54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
|
from source.classes.maze import Maze, Cell
|
|||
|
|
from .builder import MazeBuilder
|
|||
|
|
|
|||
|
|
|
|||
|
|
class TextFileMazeBuilder(MazeBuilder):
|
|||
|
|
def buildFromFile(self, filename: str) -> Maze:
|
|||
|
|
"""Получает лабиринт из текстового файла"""
|
|||
|
|
with open(filename) as f:
|
|||
|
|
data = f.read().splitlines()
|
|||
|
|
x, y = 0, 0
|
|||
|
|
width = len(data[0])
|
|||
|
|
height = len(data)
|
|||
|
|
|
|||
|
|
cells = [[None] * width for _ in range(height)]
|
|||
|
|
|
|||
|
|
start, c_exit = None, None
|
|||
|
|
|
|||
|
|
for line in data:
|
|||
|
|
x = 0
|
|||
|
|
for c in line.strip():
|
|||
|
|
if c == 'S':
|
|||
|
|
cells[y][x] = Cell(x, y, isStart=True)
|
|||
|
|
start = cells[y][x]
|
|||
|
|
x += 1
|
|||
|
|
elif c == 'E':
|
|||
|
|
cells[y][x] = Cell(x, y, isExit=True)
|
|||
|
|
c_exit = cells[y][x]
|
|||
|
|
x += 1
|
|||
|
|
elif c == '#':
|
|||
|
|
cells[y][x] = Cell(x, y, isWall=True)
|
|||
|
|
x += 1
|
|||
|
|
elif c == ' ':
|
|||
|
|
cells[y][x] = Cell(x, y)
|
|||
|
|
x += 1
|
|||
|
|
else:
|
|||
|
|
print(f'Обнаружен неизвестный символ({c}) в файле лабиринта\nfilename: {filename}\nОн заменён на стену')
|
|||
|
|
cells[y][x] = Cell(x, y, isWall=True)
|
|||
|
|
x += 1
|
|||
|
|
|
|||
|
|
y += 1
|
|||
|
|
|
|||
|
|
if start == None:
|
|||
|
|
raise ValueError(f'В файле лабиринта не обнаружен вход!\nfilename: {filename}')
|
|||
|
|
|
|||
|
|
if c_exit == None:
|
|||
|
|
raise ValueError(f'В файле лабиринта не обнаружен выход!\nfilename: {filename}')
|
|||
|
|
|
|||
|
|
return Maze(
|
|||
|
|
cells=cells,
|
|||
|
|
width=width,
|
|||
|
|
height=height,
|
|||
|
|
start=start,
|
|||
|
|
exit_cell=c_exit
|
|||
|
|
)
|