15 part1 bigger anim

15/15-1-with-bigger-anim.py

@@ -0,0 +1,157 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import numpy as np
+from PIL import Image, ImageDraw, ImageFont
+import cv2
+from time import time
+
+file = "./input.txt"
+#file = "./ex.txt"
+#file = "./ex_s.txt"
+out_file_name = "sol_bigger"
+grow_factor = 10
+
+start_time = time()
+
+def array_to_image(ar):
+    global grow_factor
+    # Konvertiere das Array in ein NumPy-Array
+    np_array = np.array(ar)
+    # Erstelle ein leeres RGB-Bild
+    height, width = np_array.shape
+    image = Image.new('RGB', (width * grow_factor, height * grow_factor))
+    # Fülle das Bild mit Pixeln basierend auf den Array-Werten
+    for y in range(height):
+        for x in range(width):
+            color = (0, 0, 0)  # Standardfarbe schwarz
+            if np_array[y, x] == "#":
+                color = (255, 0, 0)
+            elif np_array[y, x] == "O":
+                color = (0, 255, 0)
+            elif np_array[y, x] == "@":
+                color = (255, 255, 255)
+            for dy in range(grow_factor):
+                for dx in range(grow_factor):
+                    image.putpixel((x * grow_factor + dx, y * grow_factor + dy), color)
+
+    return image
+
+def add_text_to_image(image, num):
+    draw = ImageDraw.Draw(image)
+    font = ImageFont.truetype("arial.ttf", 20)  # Wählen Sie eine Schriftart und Größe
+    draw.text((5,5), f"{num:04d}", fill=(255, 255, 255), font=font)  # Weißer Text
+    return image
+
+def pil_to_cv2(pil_image):
+    return cv2.cvtColor(np.array(pil_image), cv2.COLOR_RGB2BGR)
+
+def read_input(input_file:str) -> tuple[list[list[str]], list[str]]:
+    out_field=[]
+    out_moves=[]
+    f = open(input_file, "r")
+    for line in f:
+        line = line.strip()
+        if line == "":
+            continue
+        if "#" in line:
+            temp = []
+            for b in line:
+                temp.append(b)
+            out_field.append(temp)
+        elif "<" in line or ">" in line or "^" in line or "v" in line:
+            out_moves.append(line)
+    f.close()
+    return out_field, out_moves
+
+def get_position(game_field) -> (int, int):
+    for li in range(len(game_field)):
+        for ro in range(len(game_field[li])):
+            if game_field[li][ro] == '@':
+                return li, ro
+
+def print_field(f:list[list[str]]):
+    for r in f:
+        for s in r:
+            print(s, end="")
+        print()
+
+def do_move(f:list[list[str]], m:str)-> list[list[str]]:
+    li, ro = get_position(f)
+    #print(li, ro, m)
+    if m == ">":
+        for i in range(ro+1, len(f[li])):
+            if f[li][i] == '.':
+                for ml in range(i,ro,-1):
+                    f[li][ml] = f[li][ml-1]
+                f[li][ro] = '.'
+                return f
+            elif f[li][i] == '#':
+                return f
+    elif m == "<":
+        for i in range(ro-1, 0, -1):
+            if f[li][i] == '.':
+                for ml in range(i,ro):
+                    f[li][ml] = f[li][ml+1]
+                f[li][ro] = '.'
+                return f
+            elif f[li][i] == '#':
+                return f
+    elif m == "v":
+        for i in range(li+1, len(f)):
+            if f[i][ro] == '.':
+                for mr in range(i,li,-1):
+                    f[mr][ro] = f[mr-1][ro]
+                f[li][ro] = '.'
+                return f
+            elif f[i][ro] == '#':
+                return f
+    elif m == "^":
+        for i in range(li-1, 0, -1):
+            if f[i][ro] == '.':
+                for mr in range(i,li):
+                    f[mr][ro] = f[mr+1][ro]
+                f[li][ro] = '.'
+                return f
+            elif f[i][ro] == '#':
+                return f
+    return f
+
+def sum_gps_coord(f:list[list[str]]):
+    result = 0
+    for l in range(len(f)):
+        for r in range(len(f[l])):
+            if f[l][r] == "O":
+                result += (100 * l) + r
+    return result
+
+if __name__ == "__main__":
+    field, moves_list = read_input(file)
+    print_field(field)
+
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    fps = 30
+    frame_size = (len(field[0]) * grow_factor, len(field) * grow_factor)
+    out = cv2.VideoWriter(f'./{out_file_name}.mp4', fourcc, fps, frame_size)
+    gif_anim=[]
+
+    img = array_to_image(field)
+    gif_anim.append(img)
+    cv2_image = pil_to_cv2(img)
+    out.write(cv2_image)
+
+    for moves in moves_list:
+        for move in moves:
+            field = do_move(field,move)
+            #print_field(field)
+            img = array_to_image(field)
+            gif_anim.append(img)
+            cv2_image = pil_to_cv2(img)
+            out.write(cv2_image)
+    out.release()
+    gif_anim[0].save(f'./{out_file_name}.gif', save_all=True, append_images=gif_anim[1:],
+                       optimize=False, duration=30, loop=0)
+
+    print_field(field)
+    print(f'Solution Part1: {sum_gps_coord(field)}')
+    print(f'Runtime: {time()-start_time:.2f} s')

15/ex_p2.txt

@@ -0,0 +1,9 @@
+#######
+#...#.#
+#.....#
+#..OO@#
+#..O..#
+#.....#
+#######
+
+<vv<<^^<<^^