-
Notifications
You must be signed in to change notification settings - Fork 0
/
cell.py
64 lines (56 loc) · 2.01 KB
/
cell.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
from graphics import Line, Point
class Cell:
def __init__(self,win=None):
self.has_left_wall = True
self.has_right_wall = True
self.has_top_wall = True
self.has_bottom_wall = True
self._x1 = None
self._x2 = None
self._y1 = None
self._y2 = None
self._win = win
self.visited = False
def draw(self, x1, y1, x2, y2):
if self._win is None:
return
self._x1 = x1
self._x2 = x2
self._y1 = y1
self._y2 = y2
if self.has_left_wall:
line = Line(Point(x1, y1), Point(x1, y2))
self._win.draw_line(line)
else:
line = Line(Point(x1, y1), Point(x1, y2))
self._win.draw_line(line,"white")
if self.has_top_wall:
line = Line(Point(x1, y1), Point(x2, y1))
self._win.draw_line(line)
else:
line = Line(Point(x1, y1), Point(x2, y1))
self._win.draw_line(line,"white")
if self.has_right_wall:
line = Line(Point(x2, y1), Point(x2, y2))
self._win.draw_line(line)
else:
line = Line(Point(x2, y1), Point(x2, y2))
self._win.draw_line(line,"white")
if self.has_bottom_wall:
line = Line(Point(x1, y2), Point(x2, y2))
self._win.draw_line(line)
else:
line = Line(Point(x1, y2), Point(x2, y2))
self._win.draw_line(line,"white")
def draw_move(self,to_cell,undo=False):
half_length = abs(self._x2 - self._x1) // 2#the cell is square therefore
x_center = half_length + self._x1
y_center = half_length + self._y1
half_length2 = abs(to_cell._x2 - to_cell._x1) // 2
x_center2 = half_length2 + to_cell._x1
y_center2 = half_length2 + to_cell._y1
fill_color = "red"
if undo:
fill_color = "gray"
line = Line(Point(x_center,y_center),Point(x_center2,y_center2))
self._win.draw_line(line,fill_color)