day9 animate

1 files changed, 31 insertions, 2 deletions

diff --git a/2022/day9/asd.py b/2022/day9/asd.py
index 0065441..5a1d94a 100755
--- a/2022/day9/asd.py
+++ b/2022/day9/asd.py
@@ -5,7 +5,9 @@ from functools import reduce
 from re import findall
 from copy import deepcopy
 import sys
+import time
 
+animate = False
 # filename = "ref.txt"
 # filename = "ref2.txt"
 filename = "pzl.txt"
@@ -58,6 +60,27 @@ ht = [[0,0] for i in range(10)]
 visited9 = set()
 visited9.add(tuple(ht[9]))
 
+
+def print_grid(arr, c=''):
+    lx = 100
+    ly = 40
+    grid = [['.' for _ in range(lx)] for _ in range(ly)]
+
+    for i, (x, y) in enumerate(arr):
+        try:
+            grid[-y+ly//2][x+lx//2] = c if c!='' else str(i)
+        except:
+            pass
+
+    grid[ly//2][lx//2] = "s"
+
+    print('\x1b[2J')
+    for line in grid:
+        for c in line:
+            print(c, end="")
+        print("")
+    time.sleep(1/20)
+
 for m in moves:
     ht[0] = [hi + mi for hi,mi in zip(ht[0],m)]
 
@@ -76,11 +99,17 @@ for m in moves:
     # print(ht)
     visited9.add(tuple(ht[9]))
 
+    if animate:
+        print_grid(ht)
+
 # print(visited9)
 
 res1 = len(visited)
 res2 = len(visited9)
 
-print('res1:', res1)
-print('res2:', res2)
+if animate == True:
+    print_grid(visited9, c='#')
+else:
+    print('res1:', res1)
+    print('res2:', res2)