2023.14 visualisation

.gitignore

@@ -254,4 +254,5 @@ dmypy.json
 
 input.txt
 credentials.json
-flamegraph.pl
+flamegraph.pl
+frames/

challenges/2023/14-parabolicReflectorDish/digits.txt

@@ -0,0 +1,78 @@
+ xxx 
+x   x
+x   x
+x   x
+x   x
+ xxx 
+
+  x  
+xxx  
+  x  
+  x  
+  x  
+  x  
+xxxxx
+
+ xxx 
+x   x
+    x
+   x
+  x
+ x
+xxxxx
+
+ xxx 
+x   x
+    x
+  xx
+    x
+x   x
+ xxx 
+
+    x
+   xx
+  x x
+ x  x
+xxxxx
+    x
+    x
+
+xxxxx
+x    
+x    
+xxxx 
+    x
+x   x
+ xxx
+
+ xxx 
+x   x
+x    
+xxxx 
+x   x
+x   x
+ xxx 
+
+xxxxx
+    x
+   x 
+  x  
+ x
+ x
+ x
+
+ xxx 
+x   x
+x   x
+ xxx 
+x   x
+x   x
+ xxx 
+
+ xxx 
+x   x
+x   x
+ xxxx
+    x
+x   x
+ xxx

challenges/2023/14-parabolicReflectorDish/vis.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python3
+
+import main
+import os
+from pathlib import Path
+from PIL import Image
+import sys
+from typing import Optional
+
+
+digits = [[] for block in open("digits.txt").split("\n\n")]
+
+
+background_colour = (0, 0, 0) # black
+stationary_colour = (190, 52, 58) # red
+falling_colour = (50, 49, 49) # grey
+scan_colour = (52, 190, 58) # green
+alt_scan_colour = (24, 77, 191) # blue
+
+frame_dir = Path("frames")
+os.mkdir(frame_dir)
+
+counter = 0
+frame_number = 0
+scale_factor = 4
+
+def draw_frame(platform: tuple[str], highlight_y: Optional[int] = None, allow_skip: bool = True):
+    global frame_number, counter
+
+    counter += 1
+    if not highlight_y and allow_skip:
+        if counter % 21 != 0:
+            return
+
+    y = len(platform)
+    x = len(platform[0])
+
+    img = Image.new("RGB", (x, y), color=background_colour)
+
+    for y, line in enumerate(platform):
+        for x, char in enumerate(line):
+            c = background_colour
+            
+            if char == "#":
+                c = stationary_colour
+            if char == "O":
+                c = falling_colour
+            if highlight_y is not None and y == highlight_y:
+                if char == "O":
+                    c = alt_scan_colour
+                else:
+                    c = scan_colour
+
+            img.putpixel((x, y), c)
+
+    img = img.resize((x*scale_factor, y*scale_factor), resample=Image.NEAREST)
+    img.save(frame_dir/f"{str(frame_number).zfill(8)}.png")
+    frame_number += 1
+
+
+def modtilt(platform: tuple[str], direction: main.TiltDirection, allow_skip = True, partial = True) -> tuple[str]:
+    needs_flip = direction == main.TiltDirection.North or direction == main.TiltDirection.South
+
+    if direction == main.TiltDirection.North or direction == main.TiltDirection.South:
+        platform = main.flip_platform(platform)
+
+    if direction == main.TiltDirection.North or direction == main.TiltDirection.West:
+        transformation = lambda x: x.replace(".O", "O.")
+    else:
+        transformation = lambda x: x.replace("O.", ".O")
+
+    res = list(platform)
+
+    changes = True
+
+    while changes:
+        changes = False
+        for i in range(len(res)):
+            after = transformation(res[i])
+            if res[i] != after:
+                changes = True
+            res[i] = after
+        if (partial and not changes) or not partial:
+            draw_frame(res if not needs_flip else main.flip_platform(res), allow_skip=allow_skip)
+
+    if direction == main.TiltDirection.North or direction == main.TiltDirection.South:
+        res = main.flip_platform(res)
+
+    return tuple(res)
+
+
+platform = main.parse(sys.stdin.read().strip())
+
+draw_frame(platform)
+platform = modtilt(platform, main.TiltDirection.North, allow_skip=False, partial=False)
+for y, line in enumerate(platform):
+    draw_frame(platform, highlight_y=y)
+platform = modtilt(platform, main.TiltDirection.West)
+platform = modtilt(platform, main.TiltDirection.North)
+platform = modtilt(platform, main.TiltDirection.East)
+
+ITERS = 1_000_000_000 - 1
+
+i = 0
+known = {}
+jumped = False
+while i < ITERS:
+    print(f"{i}/{ITERS}", end="\r")
+    for direction in [
+        main.TiltDirection.North,
+        main.TiltDirection.West,
+        main.TiltDirection.South,
+        main.TiltDirection.East,
+    ]:
+        platform = modtilt(platform, direction)
+
+    if not jumped:
+        if platform in known:
+            last_occurrence = known[platform]
+            period = i - last_occurrence
+            i += ((ITERS - i) // period) * period
+            jumped = True
+        else:
+            known[platform] = i
+    i += 1
+
+for y, line in enumerate(platform):
+        draw_frame(platform, highlight_y=y)