mirror of
https://github.com/maoakeEnterprise/amazing.git
synced 2026-04-28 16:04:35 +02:00
Compare commits
28 Commits
parsing
...
b317f7a3a0
| Author | SHA1 | Date | |
|---|---|---|---|
| b317f7a3a0 | |||
| 2fc67683d8 | |||
| cb19cf1413 | |||
| 6ec617848f | |||
| 349e58ce41 | |||
| b078241359 | |||
| 3c072de0f4 | |||
| a3bbce861d | |||
| ca9444778e | |||
| 135e13deff | |||
| 2828e37853 | |||
| ef030f70a7 | |||
| 170de8813a | |||
| 5aec319f7b | |||
| 24748c47ad | |||
| e4c91dc4a1 | |||
| 6b3fd9a6b7 | |||
| e33d0a8e29 | |||
| a408004bd7 | |||
| b8631d5b89 | |||
| 5f1ffcc01c | |||
| e717bf52e9 | |||
| 3fa0d3204e | |||
| cc6f2eb147 | |||
| c6242eeec0 | |||
| 4055a8a7a2 | |||
| a39f348b1e | |||
| 03c4d206d6 |
@@ -214,4 +214,5 @@ __marimo__/
|
||||
|
||||
# Streamlit
|
||||
.streamlit/secrets.toml
|
||||
test.txt
|
||||
|
||||
|
||||
@@ -1,9 +1,13 @@
|
||||
install:
|
||||
uv sync
|
||||
uv pip install mlx-2.2-py3-none-any.whl
|
||||
|
||||
run: install
|
||||
uv run python3 a_maze_ing.py config.txt
|
||||
|
||||
run_windows:
|
||||
.venv\Scripts\python -m a_maze_ing config.txt
|
||||
|
||||
debug:
|
||||
uv pdb python3 a_maze_ing.py config.txt
|
||||
|
||||
@@ -28,3 +32,5 @@ run_test_maze_gen:
|
||||
PYTHONPATH=src uv run pytest tests/test_MazeGenerator.py
|
||||
run_test:
|
||||
uv run pytest
|
||||
mlx:
|
||||
uv run python3 test.py
|
||||
|
||||
+215
-16
@@ -1,23 +1,222 @@
|
||||
import os
|
||||
from src.amaz_lib import Maze
|
||||
from src.amaz_lib import MazeGenerator
|
||||
import src.amaz_lib as g
|
||||
from typing import Any, Generator
|
||||
from src.AMazeIng import AMazeIng
|
||||
from src.parsing import Parsing
|
||||
from mlx import Mlx
|
||||
import numpy as np
|
||||
import math
|
||||
import time
|
||||
|
||||
|
||||
def main(maze_gen: MazeGenerator) -> None:
|
||||
# try:
|
||||
maze = Maze(maze=None)
|
||||
for alg in maze_gen.generator(30, 10):
|
||||
maze.set_maze(alg)
|
||||
os.system("clear")
|
||||
maze.ascii_print()
|
||||
# solver = AStar((1, 1), (14, 18))
|
||||
# print(solver.solve(maze))
|
||||
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.win_ptr = self.mlx.mlx_new_window(
|
||||
self.mlx_ptr, width, height + 200, "A-Maze-Ing"
|
||||
)
|
||||
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)
|
||||
)
|
||||
self.path_printer = None
|
||||
self.generator = None
|
||||
|
||||
def close(self) -> None:
|
||||
self.mlx.mlx_destroy_image(self.mlx_ptr, self.img_ptr)
|
||||
|
||||
def redraw_image(self) -> None:
|
||||
self.mlx.mlx_clear_window(self.mlx_ptr, self.win_ptr)
|
||||
self.mlx.mlx_put_image_to_window(
|
||||
self.mlx_ptr, self.win_ptr, self.img_ptr, 0, 0
|
||||
)
|
||||
self.mlx.mlx_string_put(
|
||||
self.mlx_ptr,
|
||||
self.win_ptr,
|
||||
self.width // 3,
|
||||
self.height + 100,
|
||||
0xFFFFFF,
|
||||
"1: regen; 2: path; 3: color; 4: quit;",
|
||||
)
|
||||
|
||||
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)
|
||||
|
||||
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.redraw_image()
|
||||
|
||||
def put_block(self, ul: tuple[int, int], dr: tuple[int, int]) -> None:
|
||||
for y in range(min(ul[1], dr[1]), max(dr[1], ul[1])):
|
||||
self.put_line((min(ul[0], dr[0]), y), (max(ul[0], dr[0]), y))
|
||||
|
||||
def put_path(self, amazing: AMazeIng):
|
||||
path = amazing.solve_path()
|
||||
print(path)
|
||||
actual = amazing.entry
|
||||
actual = (actual[0] - 1, actual[1] - 1)
|
||||
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])
|
||||
)
|
||||
)
|
||||
self.update_maze(maze)
|
||||
for i in range(len(path)):
|
||||
ul = (
|
||||
(actual[0]) * cell_size + margin + 12,
|
||||
(actual[1]) * cell_size + 12 + margin,
|
||||
)
|
||||
dr = (
|
||||
(actual[0]) * cell_size + cell_size + margin - 12,
|
||||
(actual[1]) * cell_size + cell_size - 12 + margin,
|
||||
)
|
||||
self.put_block(ul, dr)
|
||||
self.redraw_image()
|
||||
x0 = actual[0] * cell_size + margin + 12
|
||||
y0 = actual[1] * cell_size + margin + 12
|
||||
x1 = actual[0] * cell_size + cell_size + margin - 12
|
||||
y1 = actual[1] * cell_size + cell_size + margin - 12
|
||||
yield
|
||||
match path[i]:
|
||||
case "N":
|
||||
self.put_block((x0, y0), (x1, y0 - 24))
|
||||
actual = (actual[0], actual[1] - 1)
|
||||
case "E":
|
||||
self.put_block((x1, y0), (x1 + 24, y1))
|
||||
actual = (actual[0] + 1, actual[1])
|
||||
case "S":
|
||||
self.put_block((x0, y1), (x1, y1 + 24))
|
||||
actual = (actual[0], actual[1] + 1)
|
||||
case "W":
|
||||
self.put_block((x0, y0), (x0 - 24, y1))
|
||||
actual = (actual[0] - 1, actual[1])
|
||||
ul = (
|
||||
(actual[0]) * cell_size + margin + 12,
|
||||
(actual[1]) * cell_size + 12 + margin,
|
||||
)
|
||||
dr = (
|
||||
(actual[0]) * cell_size + cell_size + margin - 12,
|
||||
(actual[1]) * cell_size + cell_size - 12 + margin,
|
||||
)
|
||||
self.put_block(ul, dr)
|
||||
self.redraw_image()
|
||||
return
|
||||
|
||||
def close_loop(self, _: Any):
|
||||
self.mlx.mlx_loop_exit(self.mlx_ptr)
|
||||
|
||||
def handle_key_press(self, keycode: int, amazing: AMazeIng) -> None:
|
||||
if keycode == 49:
|
||||
self.restart_maze(amazing)
|
||||
if keycode == 50:
|
||||
self.restart_path(amazing)
|
||||
if keycode == 51:
|
||||
pass
|
||||
if keycode == 52:
|
||||
self.close_loop(None)
|
||||
|
||||
def start(self, amazing: AMazeIng) -> None:
|
||||
self.restart_maze(amazing)
|
||||
self.mlx.mlx_loop_hook(self.mlx_ptr, self.render_maze, amazing)
|
||||
self.mlx.mlx_hook(self.win_ptr, 33, 0, self.close_loop, None)
|
||||
self.mlx.mlx_hook(
|
||||
self.win_ptr, 2, 1 << 0, self.handle_key_press, amazing
|
||||
)
|
||||
self.mlx.mlx_loop(self.mlx_ptr)
|
||||
|
||||
def restart_maze(self, amazing: AMazeIng) -> None:
|
||||
self.generator = amazing.generate()
|
||||
|
||||
def restart_path(self, amazing: AMazeIng) -> None:
|
||||
self.path_printer = self.put_path(amazing)
|
||||
|
||||
def render_path(self):
|
||||
try:
|
||||
next(self.path_printer)
|
||||
time.sleep(0.03)
|
||||
except StopIteration:
|
||||
pass
|
||||
|
||||
def render_maze(self, amazing: AMazeIng):
|
||||
try:
|
||||
next(self.generator)
|
||||
self.update_maze(amazing.maze.get_maze())
|
||||
# time.sleep(0.01)
|
||||
except StopIteration:
|
||||
if self.path_printer is not None:
|
||||
self.render_path()
|
||||
|
||||
|
||||
# except Exception as err:
|
||||
# print(err)
|
||||
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(g.DepthFirstSearch())
|
||||
main()
|
||||
|
||||
+6
-4
@@ -1,6 +1,8 @@
|
||||
WIDTH=200
|
||||
HEIGHT=100
|
||||
ENTRY=0,0
|
||||
EXIT=19,14
|
||||
WIDTH=11
|
||||
HEIGHT=11
|
||||
ENTRY=1,1
|
||||
EXIT=11,11
|
||||
OUTPUT_FILE=maze.txt
|
||||
PERFECT=True
|
||||
GENERATOR=Kruskal
|
||||
SOLVER=AStar
|
||||
|
||||
Binary file not shown.
+11
-12
@@ -1,30 +1,28 @@
|
||||
from dataclasses import field
|
||||
from os import eventfd_read
|
||||
from typing import Generator
|
||||
import numpy
|
||||
from typing_extensions import Self
|
||||
from pydantic import AfterValidator, BaseModel, Field, model_validator
|
||||
from pydantic import BaseModel, Field, model_validator, ConfigDict
|
||||
|
||||
from amaz_lib import Maze, MazeGenerator, MazeSolver
|
||||
from amaz_lib.Cell import Cell
|
||||
from src.amaz_lib import Maze, MazeGenerator, MazeSolver
|
||||
|
||||
|
||||
class AMazeIng(BaseModel):
|
||||
width: int = Field(ge=3)
|
||||
height: int = Field(ge=3)
|
||||
model_config = ConfigDict(arbitrary_types_allowed=True)
|
||||
|
||||
width: int = Field(ge=4)
|
||||
height: int = Field(ge=4)
|
||||
entry: tuple[int, int]
|
||||
exit: tuple[int, int]
|
||||
output_file: str = Field(min_length=3)
|
||||
perfect: bool = Field(default=True)
|
||||
maze: Maze = Field(default=Maze(maze=numpy.array([])))
|
||||
maze: Maze = Field(default=Maze(None))
|
||||
generator: MazeGenerator
|
||||
solver: MazeSolver
|
||||
|
||||
@model_validator(mode="after")
|
||||
def check_entry_exit(self) -> Self:
|
||||
if self.entry[0] >= self.width or self.entry[1] >= self.height:
|
||||
if self.entry[0] > self.width or self.entry[1] > self.height:
|
||||
raise ValueError("Entry coordinates exceed the maze size")
|
||||
if self.exit[0] >= self.width or self.exit[1] >= self.height:
|
||||
if self.exit[0] > self.width or self.exit[1] > self.height:
|
||||
raise ValueError("Exit coordinates exceed the maze size")
|
||||
return self
|
||||
|
||||
@@ -32,9 +30,10 @@ class AMazeIng(BaseModel):
|
||||
for array in self.generator.generator(self.height, self.width):
|
||||
self.maze.set_maze(array)
|
||||
yield self.maze
|
||||
return
|
||||
|
||||
def solve_path(self) -> str:
|
||||
return self.solver.solve(self.maze)
|
||||
return self.solver.solve(self.maze, self.height, self.width)
|
||||
|
||||
def __str__(self) -> str:
|
||||
res = self.maze.__str__()
|
||||
|
||||
@@ -1,8 +1,6 @@
|
||||
from dataclasses import dataclass
|
||||
|
||||
import numpy
|
||||
from .Cell import Cell
|
||||
from .MazeGenerator import MazeGenerator
|
||||
|
||||
|
||||
@dataclass
|
||||
|
||||
+178
-38
@@ -6,13 +6,94 @@ import math
|
||||
|
||||
|
||||
class MazeGenerator(ABC):
|
||||
def __init__(self, start: tuple, end: tuple, perfect: bool) -> None:
|
||||
self.start = (start[0] - 1, start[1] - 1)
|
||||
self.end = (end[0] - 1, end[1] - 1)
|
||||
self.perfect = perfect
|
||||
|
||||
@abstractmethod
|
||||
def generator(
|
||||
self, height: int, width: int
|
||||
self, height: int, width: int, seed: int | None = None
|
||||
) -> Generator[np.ndarray, None, np.ndarray]: ...
|
||||
|
||||
@staticmethod
|
||||
def get_cell_ft(width: int, height: int) -> set:
|
||||
forty_two = set()
|
||||
y, x = (int(height / 2), int(width / 2))
|
||||
forty_two.add((y, x - 1))
|
||||
forty_two.add((y, x - 2))
|
||||
forty_two.add((y, x - 3))
|
||||
forty_two.add((y - 1, x - 3))
|
||||
forty_two.add((y - 2, x - 3))
|
||||
forty_two.add((y + 1, x - 1))
|
||||
forty_two.add((y + 2, x - 1))
|
||||
forty_two.add((y, x + 1))
|
||||
forty_two.add((y, x + 2))
|
||||
forty_two.add((y, x + 3))
|
||||
forty_two.add((y - 1, x + 3))
|
||||
forty_two.add((y - 2, x + 3))
|
||||
forty_two.add((y - 2, x + 2))
|
||||
forty_two.add((y - 2, x + 1))
|
||||
forty_two.add((y + 1, x + 1))
|
||||
forty_two.add((y + 2, x + 1))
|
||||
forty_two.add((y + 2, x + 2))
|
||||
forty_two.add((y + 2, x + 3))
|
||||
return forty_two
|
||||
|
||||
@staticmethod
|
||||
def unperfect_maze(width: int, height: int,
|
||||
maze: np.ndarray, forty_two: set | None,
|
||||
prob: float = 0.1
|
||||
) -> Generator[np.ndarray, None, np.ndarray]:
|
||||
directions = {
|
||||
"N": (0, -1),
|
||||
"S": (0, 1),
|
||||
"W": (-1, 0),
|
||||
"E": (1, 0)
|
||||
}
|
||||
|
||||
reverse = {
|
||||
"N": "S",
|
||||
"S": "N",
|
||||
"W": "E",
|
||||
"E": "W"
|
||||
}
|
||||
min_break = 2
|
||||
while True:
|
||||
count = 0
|
||||
for y in range(height):
|
||||
for x in range(width):
|
||||
if forty_two and (x, y) in forty_two:
|
||||
continue
|
||||
for direc, (dx, dy) in directions.items():
|
||||
nx, ny = x + dx, y + dy
|
||||
if forty_two and (
|
||||
(y, x) in forty_two
|
||||
or (ny, nx) in forty_two
|
||||
):
|
||||
continue
|
||||
if not (0 <= nx < width and 0 < ny < height):
|
||||
continue
|
||||
if direc in ["S", "E"]:
|
||||
continue
|
||||
if np.random.random() < prob:
|
||||
count += 1
|
||||
cell = maze[y][x]
|
||||
cell_n = maze[ny][nx]
|
||||
cell = DepthFirstSearch.broken_wall(cell, direc)
|
||||
cell_n = DepthFirstSearch.broken_wall(cell_n,
|
||||
reverse[
|
||||
direc])
|
||||
maze[y][x] = cell
|
||||
maze[ny][nx] = cell_n
|
||||
yield maze
|
||||
if count > min_break:
|
||||
break
|
||||
return maze
|
||||
|
||||
|
||||
class Kruskal(MazeGenerator):
|
||||
|
||||
class Set:
|
||||
def __init__(self, cells: list[int]) -> None:
|
||||
self.cells: list[int] = cells
|
||||
@@ -76,9 +157,29 @@ class Kruskal(MazeGenerator):
|
||||
return
|
||||
raise Exception("two sets not found")
|
||||
|
||||
@staticmethod
|
||||
def touch_ft(
|
||||
width: int,
|
||||
wall: tuple[int, int],
|
||||
cells_ft: None | set[tuple[int, int]],
|
||||
) -> bool:
|
||||
if cells_ft is None:
|
||||
return False
|
||||
s1 = (math.trunc(wall[0] / width), wall[0] % width)
|
||||
s2 = (math.trunc(wall[1] / width), wall[1] % width)
|
||||
return s1 in cells_ft or s2 in cells_ft
|
||||
|
||||
def generator(
|
||||
self, height: int, width: int
|
||||
self, height: int, width: int, seed: int | None = None
|
||||
) -> Generator[np.ndarray, None, np.ndarray]:
|
||||
cells_ft = None
|
||||
if height > 10 and width > 10:
|
||||
cells_ft = self.get_cell_ft(width, height)
|
||||
if cells_ft and (self.start in cells_ft or self.end in cells_ft):
|
||||
cells_ft = None
|
||||
|
||||
if seed is not None:
|
||||
np.random.seed(seed)
|
||||
sets = self.Sets([self.Set([i]) for i in range(height * width)])
|
||||
walls = []
|
||||
for h in range(height):
|
||||
@@ -87,56 +188,91 @@ class Kruskal(MazeGenerator):
|
||||
for h in range(height - 1):
|
||||
for w in range(width):
|
||||
walls += [(w + (width * h), w + (width * (h + 1)))]
|
||||
print(walls)
|
||||
np.random.shuffle(walls)
|
||||
|
||||
yield self.walls_to_maze(walls, height, width)
|
||||
while len(sets.sets) > 1:
|
||||
while (len(sets.sets) != 1 and cells_ft is None) or (
|
||||
len(sets.sets) != 19 and cells_ft is not None
|
||||
):
|
||||
for wall in walls:
|
||||
if not self.is_in_same_set(sets, wall):
|
||||
if not self.is_in_same_set(sets, wall) and not self.touch_ft(
|
||||
width, wall, cells_ft
|
||||
):
|
||||
self.merge_sets(sets, wall)
|
||||
walls.remove(wall)
|
||||
yield self.walls_to_maze(walls, height, width)
|
||||
if len(sets.sets) == 1:
|
||||
if (len(sets.sets) == 1 and cells_ft is None) or (
|
||||
len(sets.sets) == 19 and cells_ft is not None
|
||||
):
|
||||
break
|
||||
print(f"nb sets: {len(sets.sets)}")
|
||||
return self.walls_to_maze(walls, height, width)
|
||||
maze = self.walls_to_maze(walls, height, width)
|
||||
if self.perfect is False:
|
||||
gen = Kruskal.unperfect_maze(width, height, maze,
|
||||
cells_ft)
|
||||
for res in gen:
|
||||
maze = res
|
||||
yield maze
|
||||
return maze
|
||||
|
||||
|
||||
class DepthFirstSearch(MazeGenerator):
|
||||
def __init__(self, start: bool, end: bool, perfect: bool) -> None:
|
||||
self.start = (start[0] - 1, start[1] - 1)
|
||||
self.end = (end[0] - 1, end[1] - 1)
|
||||
self.perfect = perfect
|
||||
self.forty_two: set | None = None
|
||||
|
||||
def generator(
|
||||
self, height: int, width: int
|
||||
self, height: int, width: int, seed: int = None
|
||||
) -> Generator[np.ndarray, None, np.ndarray]:
|
||||
maze = DepthFirstSearch.init_maze(width, height)
|
||||
if seed is not None:
|
||||
np.random.seed(seed)
|
||||
maze = self.init_maze(width, height)
|
||||
if width > 9 and height > 9:
|
||||
self.forty_two = self.get_cell_ft(width, height)
|
||||
visited = np.zeros((height, width), dtype=bool)
|
||||
if (
|
||||
self.forty_two
|
||||
and self.start not in self.forty_two
|
||||
and self.end not in self.forty_two
|
||||
):
|
||||
visited = self.lock_cell_ft(visited, self.forty_two)
|
||||
path = list()
|
||||
w_h = (width, height)
|
||||
coord = (0, 0)
|
||||
x, y = coord
|
||||
first = True
|
||||
first_iteration = True
|
||||
|
||||
while path or first_iteration:
|
||||
first_iteration = False
|
||||
|
||||
while path or first:
|
||||
first = False
|
||||
visited[y, x] = True
|
||||
path = DepthFirstSearch.add_cell_visited(coord, path)
|
||||
random_c = DepthFirstSearch.random_cells(visited, coord, w_h)
|
||||
if len(random_c) == 0:
|
||||
path = DepthFirstSearch.back_on_step(path, w_h, visited)
|
||||
if path:
|
||||
coord = path[-1]
|
||||
random_c = DepthFirstSearch.random_cells(visited, coord, w_h)
|
||||
x, y = coord
|
||||
path = self.add_cell_visited(coord, path)
|
||||
|
||||
random_c = self.random_cells(visited, coord, w_h)
|
||||
|
||||
if not random_c:
|
||||
path = self.back_on_step(path, w_h, visited)
|
||||
if not path:
|
||||
break
|
||||
|
||||
wall = DepthFirstSearch.next_step(random_c)
|
||||
maze[y][x] = DepthFirstSearch.broken_wall(maze[y][x], wall)
|
||||
|
||||
coord = DepthFirstSearch.next_cell(x, y, wall)
|
||||
wall_r = DepthFirstSearch.reverse_path(wall)
|
||||
coord = path[-1]
|
||||
random_c = self.random_cells(visited, coord, w_h)
|
||||
x, y = coord
|
||||
maze[y][x] = DepthFirstSearch.broken_wall(maze[y][x], wall_r)
|
||||
|
||||
wall = self.next_step(random_c)
|
||||
maze[y][x] = self.broken_wall(maze[y][x], wall)
|
||||
|
||||
coord = self.next_cell(x, y, wall)
|
||||
wall_r = self.reverse_path(wall)
|
||||
x, y = coord
|
||||
maze[y][x] = self.broken_wall(maze[y][x], wall_r)
|
||||
yield maze
|
||||
if self.perfect is False:
|
||||
gen = DepthFirstSearch.unperfect_maze(width, height, maze,
|
||||
self.forty_two)
|
||||
for res in gen:
|
||||
maze = res
|
||||
yield maze
|
||||
return maze
|
||||
|
||||
@@ -194,19 +330,23 @@ class DepthFirstSearch(MazeGenerator):
|
||||
return (x + add_x, y + add_y)
|
||||
|
||||
@staticmethod
|
||||
def reverse_path(next: str) -> str:
|
||||
reverse = {"N": "S", "S": "N", "W": "E", "E": "W"}
|
||||
return reverse[next]
|
||||
def reverse_path(direction: str) -> str:
|
||||
return {"N": "S", "S": "N", "W": "E", "E": "W"}[direction]
|
||||
|
||||
@staticmethod
|
||||
def back_on_step(path: list, w_h: tuple, visited: np.array) -> list:
|
||||
def back_on_step(path: list, w_h: tuple, visited: np.ndarray) -> list:
|
||||
while path:
|
||||
last = path[-1]
|
||||
r_cells = DepthFirstSearch.random_cells(visited, last, w_h)
|
||||
while len(path) > 0:
|
||||
path.pop()
|
||||
if path:
|
||||
last = path[-1]
|
||||
r_cells = DepthFirstSearch.random_cells(visited, last, w_h)
|
||||
if r_cells:
|
||||
if DepthFirstSearch.random_cells(visited, last, w_h):
|
||||
break
|
||||
path.pop()
|
||||
return path
|
||||
|
||||
@staticmethod
|
||||
def lock_cell_ft(
|
||||
visited: np.ndarray, forty_two: set[tuple[int]]
|
||||
) -> np.ndarray:
|
||||
tab = [cell for cell in forty_two]
|
||||
for cell in tab:
|
||||
visited[cell] = True
|
||||
return visited
|
||||
|
||||
+136
-20
@@ -5,11 +5,12 @@ import numpy as np
|
||||
|
||||
class MazeSolver(ABC):
|
||||
def __init__(self, start: tuple[int, int], end: tuple[int, int]) -> None:
|
||||
self.start = (start[0] - 1, start[1] - 1)
|
||||
self.end = (end[0] - 1, end[1] - 1)
|
||||
self.start = (start[1] - 1, start[0] - 1)
|
||||
self.end = (end[1] - 1, end[0] - 1)
|
||||
|
||||
@abstractmethod
|
||||
def solve(self, maze: Maze) -> str: ...
|
||||
def solve(self, maze: Maze, height: int = None,
|
||||
width: int = None) -> str: ...
|
||||
|
||||
|
||||
class AStar(MazeSolver):
|
||||
@@ -48,35 +49,39 @@ class AStar(MazeSolver):
|
||||
return 1000
|
||||
|
||||
def best_path(
|
||||
self, maze: np.ndarray, actual: tuple[int, int]
|
||||
) -> dict[str, int | None]:
|
||||
print(actual)
|
||||
self,
|
||||
maze: np.ndarray,
|
||||
actual: tuple[int, int],
|
||||
last: str | None,
|
||||
) -> dict[str, int]:
|
||||
path = {
|
||||
"N": (
|
||||
self.f((actual[1] - 1, actual[0]))
|
||||
if not maze[actual[1]][actual[0]].get_north() and actual[0] > 0
|
||||
self.f((actual[0], actual[1] - 1))
|
||||
if not maze[actual[1]][actual[0]].get_north() and actual[1] > 0
|
||||
else None
|
||||
),
|
||||
"E": (
|
||||
self.f((actual[1], actual[0] + 1))
|
||||
self.f((actual[0] + 1, actual[1]))
|
||||
if not maze[actual[1]][actual[0]].get_est()
|
||||
and actual[1] < len(maze) - 1
|
||||
and actual[0] < len(maze[0]) - 1
|
||||
else None
|
||||
),
|
||||
"S": (
|
||||
self.f((actual[1] + 1, actual[0]))
|
||||
self.f((actual[0], actual[1] + 1))
|
||||
if not maze[actual[1]][actual[0]].get_south()
|
||||
and actual[0] < len(maze) - 1
|
||||
and actual[1] < len(maze) - 1
|
||||
else None
|
||||
),
|
||||
"W": (
|
||||
self.f((actual[1], actual[0] - 1))
|
||||
if not maze[actual[1]][actual[0]].get_west() and actual[1] > 0
|
||||
self.f((actual[0] - 1, actual[1]))
|
||||
if not maze[actual[1]][actual[0]].get_west() and actual[0] > 0
|
||||
else None
|
||||
),
|
||||
}
|
||||
return {
|
||||
k: v for k, v in sorted(path.items(), key=lambda item: item[0])
|
||||
k: v
|
||||
for k, v in sorted(path.items(), key=lambda item: item[0])
|
||||
if v is not None and k != last
|
||||
}
|
||||
|
||||
def get_opposit(self, dir: str) -> str:
|
||||
@@ -108,14 +113,125 @@ class AStar(MazeSolver):
|
||||
return actual
|
||||
|
||||
def get_path(self, maze: np.ndarray) -> str | None:
|
||||
actual = self.start
|
||||
path = ""
|
||||
path = [(self.start, self.best_path(maze, self.start, None))]
|
||||
visited = [self.start]
|
||||
while len(path) > 0 and path[-1][0] != self.end:
|
||||
if len(path[-1][1]) == 0:
|
||||
path.pop(-1)
|
||||
if len(path) == 0:
|
||||
break
|
||||
k = next(iter(path[-1][1]))
|
||||
path[-1][1].pop(k)
|
||||
continue
|
||||
|
||||
while len(path[-1][1]) > 0:
|
||||
next_pos = self.get_next_pos(
|
||||
list(path[-1][1].keys())[0], path[-1][0]
|
||||
)
|
||||
if next_pos in visited:
|
||||
k = next(iter(path[-1][1]))
|
||||
path[-1][1].pop(k)
|
||||
else:
|
||||
break
|
||||
if len(path[-1][1]) == 0:
|
||||
path.pop(-1)
|
||||
continue
|
||||
|
||||
pre = self.get_opposit(list(path[-1][1].keys())[0])
|
||||
path.append(
|
||||
(
|
||||
next_pos,
|
||||
self.best_path(maze, next_pos, pre),
|
||||
)
|
||||
)
|
||||
visited += [next_pos]
|
||||
if len(path) == 0:
|
||||
return None
|
||||
path[-1] = (self.end, {})
|
||||
return "".join(
|
||||
str(list(c[1].keys())[0]) for c in path if len(c[1]) > 0
|
||||
)
|
||||
|
||||
def solve(self, maze: Maze) -> str:
|
||||
print(maze)
|
||||
res = self.get_path(self.start, maze.get_maze(), None)
|
||||
def solve(self, maze: Maze, height: int = None,
|
||||
width: int = None) -> str:
|
||||
res = self.get_path(maze.get_maze())
|
||||
if res is None:
|
||||
raise Exception("Path not found")
|
||||
return res
|
||||
|
||||
|
||||
class DepthFirstSearchSolver(MazeSolver):
|
||||
def __init__(self, start, end):
|
||||
super().__init__(start, end)
|
||||
|
||||
def solve(self, maze: Maze, height: int = None,
|
||||
width: int = None) -> str:
|
||||
path_str = ""
|
||||
visited = np.zeros((height, width), dtype=bool)
|
||||
path = list()
|
||||
move = list()
|
||||
maze_s = maze.get_maze()
|
||||
coord = self.start
|
||||
h_w = (height, width)
|
||||
while coord != self.end:
|
||||
visited[coord] = True
|
||||
path.append(coord)
|
||||
rand_p = self.random_path(visited, coord, maze_s, h_w)
|
||||
|
||||
if not rand_p:
|
||||
path, move = self.back_on_step(path, visited, maze_s, h_w,
|
||||
move)
|
||||
if not path:
|
||||
break
|
||||
coord = path[-1]
|
||||
rand_p = self.random_path(visited, coord, maze_s, h_w)
|
||||
next = self.next_path(rand_p)
|
||||
move.append(next)
|
||||
coord = self.next_cell(coord, next)
|
||||
for m in move:
|
||||
path_str += m
|
||||
if not path:
|
||||
raise Exception("Path not found")
|
||||
return path_str
|
||||
|
||||
@staticmethod
|
||||
def random_path(visited: np.ndarray, coord: tuple,
|
||||
maze: np.ndarray, h_w: tuple) -> list:
|
||||
random_p = []
|
||||
h, w = h_w
|
||||
y, x = coord
|
||||
|
||||
if y - 1 >= 0 and not maze[y][x].get_north() and not visited[y - 1][x]:
|
||||
random_p.append("N")
|
||||
|
||||
if y + 1 < h and not maze[y][x].get_south() and not visited[y + 1][x]:
|
||||
random_p.append("S")
|
||||
|
||||
if x - 1 >= 0 and not maze[y][x].get_west() and not visited[y][x - 1]:
|
||||
random_p.append("W")
|
||||
|
||||
if x + 1 < w and not maze[y][x].get_est() and not visited[y][x + 1]:
|
||||
random_p.append("E")
|
||||
return random_p
|
||||
|
||||
@staticmethod
|
||||
def next_path(rand_path: list) -> str:
|
||||
return np.random.choice(rand_path)
|
||||
|
||||
@staticmethod
|
||||
def back_on_step(path: list, visited: np.ndarray,
|
||||
maze: np.ndarray, h_w: tuple, move: list) -> list:
|
||||
while path:
|
||||
last = path[-1]
|
||||
if DepthFirstSearchSolver.random_path(visited, last, maze, h_w):
|
||||
break
|
||||
path.pop()
|
||||
move.pop()
|
||||
return path, move
|
||||
|
||||
@staticmethod
|
||||
def next_cell(coord: tuple, next: str) -> tuple:
|
||||
y, x = coord
|
||||
next_step = {"N": (-1, 0), "S": (1, 0), "W": (0, -1), "E": (0, 1)}
|
||||
add_y, add_x = next_step[next]
|
||||
return (y + add_y, x + add_x)
|
||||
|
||||
@@ -2,9 +2,9 @@ from .Cell import Cell
|
||||
from .Maze import Maze
|
||||
from .MazeGenerator import MazeGenerator, DepthFirstSearch
|
||||
from .MazeGenerator import Kruskal
|
||||
from .MazeSolver import MazeSolver, AStar
|
||||
from .MazeSolver import MazeSolver, AStar, DepthFirstSearchSolver
|
||||
|
||||
__version__ = "1.0.0"
|
||||
__author__ = "us"
|
||||
__all__ = ["Cell", "Maze", "MazeGenerator",
|
||||
__all__ = ["Cell", "Maze", "MazeGenerator", "DepthFirstSearchSolver",
|
||||
"MazeSolver", "AStar", "Kruskal", "DepthFirstSearch"]
|
||||
|
||||
+64
-35
@@ -1,3 +1,7 @@
|
||||
from src.amaz_lib.MazeGenerator import DepthFirstSearch, Kruskal
|
||||
from src.amaz_lib.MazeSolver import AStar, DepthFirstSearchSolver
|
||||
|
||||
|
||||
class DataMaze:
|
||||
|
||||
@staticmethod
|
||||
@@ -11,28 +15,30 @@ class DataMaze:
|
||||
@staticmethod
|
||||
def transform_data(data: str) -> dict:
|
||||
tmp = data.split("\n")
|
||||
tmp2 = [
|
||||
value.split("=", 1) for value in tmp
|
||||
]
|
||||
data_t = {
|
||||
value[0]: value[1] for value in tmp2
|
||||
}
|
||||
tmp2 = [value.split("=", 1) for value in tmp if "=" in value]
|
||||
data_t = {value[0]: value[1] for value in tmp2}
|
||||
return data_t
|
||||
|
||||
@staticmethod
|
||||
def verif_key_data(data: dict) -> None:
|
||||
key_test = {
|
||||
"WIDTH", "HEIGHT", "ENTRY", "EXIT", "OUTPUT_FILE", "PERFECT"
|
||||
}
|
||||
set_key = {
|
||||
key for key in data.keys()
|
||||
"WIDTH",
|
||||
"HEIGHT",
|
||||
"ENTRY",
|
||||
"EXIT",
|
||||
"OUTPUT_FILE",
|
||||
"PERFECT",
|
||||
"GENERATOR",
|
||||
"SOLVER",
|
||||
}
|
||||
set_key = {key for key in data.keys()}
|
||||
if len(set_key) != len(key_test):
|
||||
raise KeyError("Missing some data the len do not correspond")
|
||||
res_key = {key for key in set_key if key not in key_test}
|
||||
if len(res_key) != 0:
|
||||
raise KeyError("Some Key "
|
||||
f"do not correspond the keys: {res_key}")
|
||||
raise KeyError(
|
||||
"Some Key " f"do not correspond the keys: {res_key}"
|
||||
)
|
||||
|
||||
@staticmethod
|
||||
def convert_values(data: dict):
|
||||
@@ -47,8 +53,48 @@ class DataMaze:
|
||||
for key in key_bool:
|
||||
res.update({key: DataMaze.convert_bool(data[key])})
|
||||
res.update({"OUTPUT_FILE": data["OUTPUT_FILE"]})
|
||||
res.update(
|
||||
DataMaze.get_solver_generator(data, res["ENTRY"], res["EXIT"],
|
||||
res["PERFECT"])
|
||||
)
|
||||
return res
|
||||
|
||||
@staticmethod
|
||||
def get_solver_generator(data: dict, entry: tuple, exit: tuple,
|
||||
perfect: bool) -> dict:
|
||||
available_generator = {
|
||||
"Kruskal": Kruskal,
|
||||
"DFS": DepthFirstSearch,
|
||||
}
|
||||
available_solver = {
|
||||
"AStar": AStar,
|
||||
"DFS": DepthFirstSearchSolver
|
||||
}
|
||||
res = {}
|
||||
res["GENERATOR"] = available_generator[data["GENERATOR"]](entry, exit,
|
||||
perfect)
|
||||
res["SOLVER"] = available_solver[data["SOLVER"]](entry, exit)
|
||||
return res
|
||||
|
||||
@staticmethod
|
||||
def convert_tuple(data: str) -> tuple:
|
||||
data_t = data.split(",")
|
||||
if len(data_t) != 2:
|
||||
raise ValueError(
|
||||
"There is too much " "argument in the coordinate given"
|
||||
)
|
||||
x, y = data_t
|
||||
tup = (int(x), int(y))
|
||||
return tup
|
||||
|
||||
@staticmethod
|
||||
def convert_bool(data: str) -> bool:
|
||||
if data != "True" and data != "False":
|
||||
raise ValueError("This is not True or False")
|
||||
if data == "True":
|
||||
return True
|
||||
return False
|
||||
|
||||
@staticmethod
|
||||
def get_data_maze(name_file: str) -> dict:
|
||||
try:
|
||||
@@ -56,7 +102,7 @@ class DataMaze:
|
||||
data_dict = DataMaze.transform_data(data_str)
|
||||
DataMaze.verif_key_data(data_dict)
|
||||
data_maze = DataMaze.convert_values(data_dict)
|
||||
return data_maze
|
||||
return {k.lower(): v for k, v in data_maze.items()}
|
||||
except FileNotFoundError:
|
||||
print("The file do not exist")
|
||||
exit()
|
||||
@@ -70,28 +116,11 @@ class DataMaze:
|
||||
print(f"Error on the key in the file: {e}")
|
||||
exit()
|
||||
except IndexError as e:
|
||||
print("In the function transform Data some data cannot "
|
||||
f"be splited by '=' because '=' was not present: {e}")
|
||||
print(
|
||||
"In the function transform Data some data cannot "
|
||||
f"be splited by '=' because '=' was not present: {e}"
|
||||
)
|
||||
exit()
|
||||
except AttributeError as e:
|
||||
print("Error on the "
|
||||
f"funciton get_data_maze : {e}")
|
||||
print("Error on the " f"funciton get_data_maze : {e}")
|
||||
exit()
|
||||
|
||||
@staticmethod
|
||||
def convert_tuple(data: str) -> tuple:
|
||||
data_t = data.split(",")
|
||||
if len(data_t) != 2:
|
||||
raise ValueError("There is too much "
|
||||
"argument in the coordinate given")
|
||||
x, y = data_t
|
||||
tup = (int(x), int(y))
|
||||
return tup
|
||||
|
||||
@staticmethod
|
||||
def convert_bool(data: str) -> bool:
|
||||
if data != "True" and data != "False":
|
||||
raise ValueError("This is not True or False")
|
||||
if data == "True":
|
||||
return True
|
||||
return False
|
||||
|
||||
@@ -1,23 +0,0 @@
|
||||
7D53BFD3D57951517D1D
|
||||
3D12C3903BD03AD4178D
|
||||
2BAEBEEEAA92EED547C9
|
||||
2287ED17AAAC5393FFF0
|
||||
6C6951292A87D2AEBD30
|
||||
37D43E8686E93AABAB8C
|
||||
21516D2D47FEE8284049
|
||||
6C7857C3FB9116C696D8
|
||||
751453D6D2AAC57BE970
|
||||
3BA952D17EA83BD05470
|
||||
22AAD2907BAE86967B74
|
||||
2AA83C2EFC69696FBC35
|
||||
686EE96FD7D4783FAD21
|
||||
7ED17ED3D57D3EC52FA0
|
||||
7B943D16FB7BABD3AFC8
|
||||
7407C5297EB82EB84174
|
||||
392D53C6912EE9447E9D
|
||||
62A952BBAAC13EFD7B89
|
||||
3AAC3EC6EABAAD557824
|
||||
66C7C7D7D6C6C7D556CD
|
||||
|
||||
1,1
|
||||
16,15
|
||||
@@ -1,14 +1,18 @@
|
||||
import numpy
|
||||
from amaz_lib.MazeGenerator import DepthFirstSearch
|
||||
from amaz_lib.MazeGenerator import DepthFirstSearch, MazeGenerator
|
||||
|
||||
|
||||
class TestMazeGenerator:
|
||||
|
||||
def test_generator(self) -> None:
|
||||
w_h = (300, 300)
|
||||
w_h = (10, 10)
|
||||
maze = numpy.array([])
|
||||
generator = DepthFirstSearch().generator(*w_h)
|
||||
generator = DepthFirstSearch((1, 1), (2, 2), True).generator(*w_h)
|
||||
for output in generator:
|
||||
maze = output
|
||||
|
||||
assert maze.shape == w_h
|
||||
|
||||
def test_gen_broken(self) -> None:
|
||||
test = MazeGenerator.gen_broken_set(50, 50)
|
||||
assert len(test) > 0
|
||||
|
||||
Reference in New Issue
Block a user