-
Notifications
You must be signed in to change notification settings - Fork 0
/
grid_manager.py
73 lines (59 loc) · 2.61 KB
/
grid_manager.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
64
65
66
67
68
69
70
71
72
73
from tools import *
from all_dialogs import show_info
from a_star_algorithm import algorithm
class StartNow:
def __init__(self):
pass
def start(win, width):
ROWS = 50
grid = make_grid(ROWS, width)
start = None
end = None
run = True
while run:
draw(win, grid, ROWS, width)
for event in pygame.event.get():
if event.type == pygame.QUIT:
run = False
# LEFT MOUSE BUTTON CLICKED
if pygame.mouse.get_pressed()[0]:
pos = pygame.mouse.get_pos()
row, col = get_clicked_pos(pos, ROWS, width)
node = grid[row][col]
if not start and node != end:
start = node
start.make_start()
elif not end and node != start:
end = node
end.make_end()
elif node != end and node != start:
node.make_obstacle()
# RIGHT MOUSE BUTTON CLICKED
elif pygame.mouse.get_pressed()[2]:
pos = pygame.mouse.get_pos()
row, col = get_clicked_pos(pos, ROWS, width)
node = grid[row][col]
node.reset_grid()
if node == start:
start = None
elif node == end:
end = None
if event.type == pygame.KEYDOWN:
if event.key == pygame.K_SPACE and start and end:
for row in grid:
for node in row:
node.update_neighbor_nodes(grid)
if algorithm(lambda: draw(win, grid, ROWS, width), grid, start, end):
show_info("Shortest Path Found!", "Found The Shortest Path!!\nShown By Using Green Color!!!")
else:
show_info("No Path Found!", "All The Paths From Source Are Blocked :(")
else:
if not start:
show_info("Start Not Set...", "Please click on a grid as a source.")
elif not end:
show_info("Destination Not Set...", "Please click on a grid as a destination.")
if event.key == pygame.K_BACKSPACE:
start = None
end = None
grid = make_grid(ROWS, width)
pygame.quit()