#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import numpy as np
from PIL import Image, ImageDraw, ImageFont
import cv2
from time import time
import re

file = "./input.txt"
Y_TALL = 103
X_WIDE = 101
SOLUTION = 6516
#file = "./ex.txt"
#Y_TALL = 7
#X_WIDE = 11

start_time = time()
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
fps = 30
frame_size = (X_WIDE, Y_TALL)
out = cv2.VideoWriter('./solution.mp4', fourcc, fps, frame_size)


def parse_and_go(input_file:str, seconds) -> list[list[int]]:
    global Y_TALL, X_WIDE
    field = []
    for i in range(Y_TALL):
        temp = []
        for j in range(X_WIDE):
            temp.append(0)
        field.append(temp)
    with open(input_file, 'r') as d:
        datei = d.read()
    pat = r'p\=(-?\d+),(-?\d+) v\=(-?\d+),(-?\d+)'
    match = re.findall(pat, datei)
    for m in match:
        x, y, vx, vy = map(int, m)
        # move X seconds
        x = (x + (seconds * vx)) % X_WIDE
        y = (y + (seconds * vy)) % Y_TALL
        field[y][x] += 1
    return field

def array_to_image(ar):
    # 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, height))
    # Fülle das Bild mit Pixeln basierend auf den Array-Werten
    for y in range(height):
        for x in range(width):
            if np_array[y, x] == 0:
                image.putpixel((x, y), (0, 0, 0))  # Schwarz
            else:
                image.putpixel((x, y), (0, 255, 0))  # Grün
    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)

if __name__ == "__main__":

    for sec in range(SOLUTION+1): #my solution 6516
        field_s = parse_and_go(file, sec)
        img = array_to_image(field_s)
        img_with_num=add_text_to_image(img, sec)
        if sec == SOLUTION:
            for s in range(SOLUTION, SOLUTION+200):
                cv2_image = pil_to_cv2(img_with_num)
                out.write(cv2_image)
        else:
            cv2_image = pil_to_cv2(img_with_num)
            out.write(cv2_image)
    out.release()
    print(f'Runtime: {time()-start_time:.2f} s')