40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
|
from graphics import Window, Point, Line
|
||
|
|
|
||
|
|
|
||
|
|
class Cell:
|
||
|
|
"""
|
||
|
|
A Cell represents a grid on the maze.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
x1: int, y1: int,
|
||
|
|
x2: int, y2: int,
|
||
|
|
window: Window
|
||
|
|
):
|
||
|
|
self.__top_wall_exists = True
|
||
|
|
self.__bottom_wall_exists = True
|
||
|
|
self.__left_wall_exists = True
|
||
|
|
self.__right_wall_exists = True
|
||
|
|
self.__top_wall = Line(Point(x1, y1), Point(x2, y1))
|
||
|
|
self.__bottom_wall = Line(Point(x1, y2), Point(x2, y2))
|
||
|
|
self.__left_wall = Line(Point(x1, y1), Point(x1, y2))
|
||
|
|
self.__right_wall = Line(Point(x2, y1), Point(x2, y2))
|
||
|
|
self.__window = window
|
||
|
|
|
||
|
|
def set_walls(self, top: bool, bottom: bool, left: bool, right: bool):
|
||
|
|
self.__top_wall_exists = top
|
||
|
|
self.__bottom_wall_exists = bottom
|
||
|
|
self.__left_wall_exists = left
|
||
|
|
self.__right_wall_exists = right
|
||
|
|
|
||
|
|
def draw(self):
|
||
|
|
if self.__top_wall_exists:
|
||
|
|
self.__window.draw_line(self.__top_wall)
|
||
|
|
if self.__bottom_wall_exists:
|
||
|
|
self.__window.draw_line(self.__bottom_wall)
|
||
|
|
if self.__left_wall_exists:
|
||
|
|
self.__window.draw_line(self.__left_wall)
|
||
|
|
if self.__right_wall_exists:
|
||
|
|
self.__window.draw_line(self.__right_wall)
|