from typing import Any
from src.AMazeIng import AMazeIng
from src.parsing import Parsing
from mlx.mlx import Mlx
import numpy as np
import math


class MazeMLX:
    def __init__(self, height: int, width: int) -> None:
        self.mlx = Mlx()
        self.height = height
        self.width = width
        self.mlx_ptr = self.mlx.mlx_init()
        self.generator = None
        self.win_ptr = self.mlx.mlx_new_window(
            self.mlx_ptr, width, height, "amazing"
        )
        self.img_ptr = self.mlx.mlx_new_image(self.mlx_ptr, width, height)
        self.buf, self.bpp, self.size_line, self.format = (
            self.mlx.mlx_get_data_addr(self.img_ptr)
        )

    def close(self) -> None:
        self.mlx.mlx_destroy_image(self.mlx_ptr, self.img_ptr)

    def put_pixel(self, x, y) -> None:
        offset = y * self.size_line + x * (self.bpp // 8)

        self.buf[offset + 0] = 0xFF
        self.buf[offset + 1] = 0xFF
        self.buf[offset + 2] = 0xFF
        if self.bpp >= 32:
            self.buf[offset + 3] = 0xFF

    def clear_image(self) -> None:
        self.buf[:] = b"\x00" * len(self.buf)
        self.mlx.mlx_clear_window(self.mlx_ptr, self.win_ptr)

    def put_line(self, start: tuple[int, int], end: tuple[int, int]) -> None:
        sx, sy = start
        ex, ey = end
        if sy == ey:
            for x in range(min(sx, ex), max(sx, ex) + 1):
                self.put_pixel(x, sy)
        if sx == ex:
            for y in range(min(sy, ey), max(sy, ey) + 1):
                self.put_pixel(sx, y)

    def update_maze(self, maze: np.ndarray) -> None:
        self.clear_image()
        margin = math.trunc(
            math.sqrt(self.width if self.width > self.height else self.height)
            // 2
        )
        line_len = math.trunc(
            (
                (self.height - margin) // len(maze)
                if self.height > self.width
                else (self.width - margin) // len(maze[0])
            )
        )
        for y in range(len(maze)):
            for x in range(len(maze[0])):
                x0 = x * line_len + margin
                y0 = y * line_len + margin
                x1 = x * line_len + line_len + margin
                y1 = y * line_len + line_len + margin

                if maze[y][x].get_north():
                    self.put_line((x0, y0), (x1, y0))
                if maze[y][x].get_est():
                    self.put_line((x1, y0), (x1, y1))
                if maze[y][x].get_south():
                    self.put_line((x0, y1), (x1, y1))
                if maze[y][x].get_west():
                    self.put_line((x0, y0), (x0, y1))
        self.mlx.mlx_put_image_to_window(
            self.mlx_ptr, self.win_ptr, self.img_ptr, 0, 0
        )

    def put_block(self, ul: tuple[int, int], dr: tuple[int, int]) -> None:
        for y in range(ul[1], dr[1]):
            self.put_line((min(ul[0], dr[0]), y), (max(ul[0], dr[0]), y))

    def put_path(self, amazing: AMazeIng) -> None:
        path = amazing.solve_path()
        print(path)
        actual = amazing.entry
        maze = amazing.maze.get_maze()
        if maze is None:
            return
        margin = math.trunc(
            math.sqrt(self.width if self.width > self.height else self.height)
            // 2
        )
        cell_size = math.trunc(
            (
                (self.height - margin) // len(maze)
                if self.height > self.width
                else (self.width - margin) // len(maze[0])
            )
        )

        for i in range(len(path)):
            ul = (
                (actual[0]) * cell_size + margin + 6,
                (actual[1]) * cell_size + 6 + margin,
            )
            dr = (
                (actual[0]) * cell_size + cell_size + margin - 12,
                (actual[1]) * cell_size + cell_size - 12 + margin,
            )
            self.put_block(ul, dr)
            x0 = actual[0] * cell_size + margin + 3
            y0 = actual[1] * cell_size + margin + 3
            x1 = actual[0] * cell_size + cell_size + margin - 6
            y1 = actual[1] * cell_size + cell_size + margin - 6
            if i == len(path) - 1:
                continue
            match path[i]:
                case "N":
                    self.put_line((x0, y0), (x1, y0))
                    actual = (actual[0], actual[1] - 1)
                case "E":
                    self.put_line((x1, y0), (x1, y1))
                    actual = (actual[0] + 1, actual[1])
                case "S":
                    self.put_line((x0, y1), (x1, y1))
                    actual = (actual[0], actual[1] + 1)
                case "W":
                    self.put_line((x0, y0), (x0, y1))
                    actual = (actual[0] - 1, actual[1])
        self.mlx.mlx_put_image_to_window(
            self.mlx_ptr, self.win_ptr, self.img_ptr, 0, 0
        )

    def close_loop(self, _: Any):
        self.mlx.mlx_loop_exit(self.mlx_ptr)

    def gen_maze(self, amazing: AMazeIng) -> None:
        self.generator = amazing.generate()

    def start(self, amazing: AMazeIng) -> None:
        self.gen_maze(amazing)
        self.mlx.mlx_loop_hook(self.mlx_ptr, self.render, amazing)
        self.mlx.mlx_hook(self.win_ptr, 33, 0, self.close_loop, None)
        self.mlx.mlx_loop(self.mlx_ptr)

    def render(self, amazing: AMazeIng):
        try:
            next(self.generator)
            self.update_maze(amazing.maze.get_maze())
            # time.sleep(0.01)
        except StopIteration:
            self.put_path(amazing)


def main() -> None:
    mlx = None
    try:
        mlx = MazeMLX(1000, 1000)
        config = Parsing.DataMaze.get_data_maze("config.txt")
        amazing = AMazeIng(**config)
        mlx.start(amazing)
        with open("test.txt", "w") as output:
            output.write(amazing.__str__())
    except Exception as err:
        print(err)
    finally:
        if mlx is not None:
            mlx.close()


if __name__ == "__main__":
    main()