started reworking image generation

draw/code_to_image.py

@@ -16,15 +16,26 @@ def NSD_init(x: float, y: float):
     img = Image.new("RGB", (x, y), "white")
     output_img = ImageDraw.Draw(img)
     #font = ImageFont.load_default()
-    font = ImageFont.truetype("res/fonts/SpaceGrotesk-Light.ttf", 12)
+    font = ImageFont.truetype("res/fonts/SpaceGrotesk-Light.ttf", 14)
+
+def get_text_size(text: str):
+    if not font:
+        raise Exception("Output image was not initialized! Make sure to call NSD_init first")
+    return font.getsize(text, direction="ltr")
+
+def draw_debug_tile(x, y, x_sz, y_sz):
+    from random import randint
+
+    output_img.rectangle((x, y) + (x + x_sz, y + y_sz), fill=(randint(0, 255)))
+    return x, y + y_sz
 
 def draw_generic_instruction(instruction: str, x, y, xsize, ysize) -> Iterable[float]:
     if not output_img:
         raise Exception("Output image was not initialized! Make sure to call NSD_init first")
     
     #size shit
-    #text_y_size = font.getsize(instruction, direction="ltr")[1]
-    #ysize = max(text_y_size, ysize) # ensure it is alway at least big enought to fit the text
+    text_y_size = font.getsize(instruction, direction="ltr")[1]
+    ysize = max(text_y_size, ysize) # ensure the bounding box is alway at least big enought to fit the text
 
     #draw shit
     output_img.rectangle((x,y) + (x + xsize, y + ysize), outline=(0), width=1)
@@ -42,42 +53,44 @@ def draw_if_statement(condition: str, x: int, y: int, xsize: int, ysize: int):
     if not output_img:
         raise Exception("Output image was not initialized! Make sure to call NSD_init first")
 
-    output_img.line((x,y) + (x + xsize / 2, y + ysize / 4), fill=(0))
-    output_img.line((x + xsize, y) + (x + xsize / 2, y + ysize / 4), fill=(0))
-    output_img.rectangle((x, y + ysize / 4) + (x + xsize, y + ysize), outline=(0), width=1)
-    output_img.rectangle((x, y) + (x + xsize, y + ysize / 4), outline=(0), width=1)
-    output_img.line((x + xsize / 2, y + ysize / 4) + (x + xsize / 2, y + ysize), fill=(0))
+    text_y_size = font.getsize(condition)[1]
+
+    output_img.line((x,y) + (x + xsize / 2, y + text_y_size), fill=(0))
+    output_img.line((x + xsize, y) + (x + xsize / 2, y + text_y_size), fill=(0))
+    output_img.rectangle((x, y + text_y_size) + (x + xsize, y + ysize), outline=(0), width=1)
+    output_img.rectangle((x, y) + (x + xsize, y + text_y_size), outline=(0), width=1)
+    output_img.line((x + xsize / 2, y + text_y_size) + (x + xsize / 2, y + ysize), fill=(0))
 
     # condition text
-    output_img.multiline_text((x + xsize / 2, y + ysize * .05 ), condition, fill=(0), font=font, anchor="mm", spacing=4, align='right')
+    output_img.multiline_text((x + xsize / 2, y + text_y_size), condition, fill=(0), font=font, anchor="md", spacing=4, align='right')
 
     # true / false
-    output_img.text((x + 5, y + ysize * .1875), "true", font = font, fill = (0), anchor="lm")
-    output_img.text((x + xsize - 5, y + ysize * .1875), "false", font = font, fill = (0), anchor="rm")
+    output_img.text((x + 5, y + text_y_size), "true", font = font, fill = (0), anchor="ld")
+    output_img.text((x + xsize - 5, y + text_y_size), "false", font = font, fill = (0), anchor="rd")
 
     #first x,y,xsize,ysize of first box then of second first true and then false
-    return x, y + ysize / 4, xsize / 2, ysize * .75, x + xsize / 2, y + ysize / 4, xsize / 2, ysize * .75
+    return x, y + text_y_size, xsize / 2, ysize - text_y_size, x + xsize / 2, y + text_y_size, xsize / 2, ysize - text_y_size
 
 def draw_while_loop_front(condition: str, x: int, y: int, xsize: int, ysize: int):
     
     if not output_img:
         raise Exception("Output image was not initialized! Make sure to call NSD_init first")
 
-    #ole #TODO
+    text_y_sz = font.getsize(condition)[1]
 
     #the box
     output_img.line((x,y) + (x + xsize, y), fill=(0))
     output_img.line((x,y) + (x, y + ysize), fill=(0))
-    output_img.line((x + xsize * .1, y + ysize * .1) + (x + xsize, y + ysize * .1), fill=(0))
-    output_img.line((x + xsize, y) + (x + xsize, y + ysize * .1), fill=(0))
+    output_img.line((x + xsize * .1, y + text_y_sz) + (x + xsize, y + text_y_sz), fill=(0))
+    output_img.line((x + xsize, y) + (x + xsize, y + text_y_sz), fill=(0))
     output_img.line((x, y + ysize) + (x + xsize * .1, y + ysize ), fill=(0))
-    output_img.line((x + xsize * .1, y + ysize) + (x + xsize * .1, y + ysize * .1), fill=(0))
+    output_img.line((x + xsize * .1, y + ysize) + (x + xsize * .1, y + text_y_sz), fill=(0))
 
     #the text
-    output_img.text((x + xsize * .1, y + ysize * .025), condition, font = font, fill = (0), anchor="lm")
+    output_img.text((x + xsize * .1, y + text_y_sz * 0.5), condition, font = font, fill = (0), anchor="lm")
 
     #the x, y offset then the x,y draw size (the canvas)
-    return x + xsize * .1, y + ysize * .1, xsize * .9, ysize * .9
+    return x + xsize * .1, y + text_y_sz, xsize * .9
 
 def draw_while_loop_back(condition: str, x: int, y: int, xsize: int, ysize: int):