40 lines
1.5 KiB
Python
40 lines
1.5 KiB
Python
|
from task2.mazeObjects.cell import Cell
|
|||
|
|
|
|||
|
|
class Maze:
|
|||
|
|
"""Хранит двумерный массив клеток,
|
|||
|
|
ширину, высоту, ссылки на стартовую и выходную клетку.
|
|||
|
|
Методы:
|
|||
|
|
getCell(x, y), getNeighbors(cell) – возвращает список соседних проходимых клеток
|
|||
|
|
(вверх, вниз, влево, вправо, если в пределах границ и не стена)."""
|
|||
|
|
|
|||
|
|
def __init__(self, mazeArray: list[list[Cell]], start: dict, end: dict) -> None:
|
|||
|
|
self.mazeArray = mazeArray
|
|||
|
|
self.width = len(mazeArray)
|
|||
|
|
self.height = len(mazeArray[0])
|
|||
|
|
|
|||
|
|
self.startCell = self.getCell(start['x'], start['y'])
|
|||
|
|
self.endCell = self.getCell(end['x'], end['y'])
|
|||
|
|
|
|||
|
|
def getCell(self, x: int, y: int):
|
|||
|
|
return self.mazeArray[y][x]
|
|||
|
|
|
|||
|
|
def checkCell(self, x: int, y: int):
|
|||
|
|
if not(0 <= x and x < self.width):
|
|||
|
|
return False
|
|||
|
|
if not(0 <= y and y < self.height):
|
|||
|
|
return False
|
|||
|
|
return self.getCell(x, y).isPassable()
|
|||
|
|
|
|||
|
|
def getNeighbors(self, cell: Cell):
|
|||
|
|
point = (cell.x, cell.y)
|
|||
|
|
offsets = ((0, 1),
|
|||
|
|
(0, -1),
|
|||
|
|
(-1, 0),
|
|||
|
|
(1, 0))
|
|||
|
|
passableCells = []
|
|||
|
|
for ofst in offsets:
|
|||
|
|
x = point[0]+ofst[0]
|
|||
|
|
y = point[1]+ofst[1]
|
|||
|
|
if self.checkCell(x, y):
|
|||
|
|
passableCells.append(self.getCell(x, y))
|
|||
|
|
return passableCells
|