def test_initialise_dijk_grid_simple(self):
self.grid.width, self.grid.height, self.grid.blocks = 2, 2, [(1, 1)]
self.grid.initialise_dijk_grid()
test_grid = Graph()
for i in range(2):
test_grid.add_edge((i, 0), (i, 1), 1)
test_grid.add_edge((i, 1), (i, 0), 1)
test_grid.add_edge((0, i), (1, i), 1)
test_grid.add_edge((1, i), (0, i), 1)
test_grid.remove_node((1, 1))
self.assertEqual(self.grid.dijk_grid, test_grid)
class Grid:
def __init__(self, height, width, blocks, spawner_square, exit_square,
lives, souls):
self.height = height
self.width = width
self.blocks = blocks
self.spawner_square = spawner_square
self.exit_square = exit_square
self.lives = lives
self.souls = souls
self.dijk_grid = Graph()
self.route_dict = {}
self.num_loc_list = []
self.tower_list = []
self.square_grid = []
self.forbidden_squares = set()
def initialise_dijk_grid(self):
"""Initialises the dijk_grid attribute for routing"""
for i in range(self.height):
for j in range(self.width):
if i < self.height - 1:
self.dijk_grid.add_edge((i, j), (i + 1, j), 1)
self.dijk_grid.add_edge((i + 1, j), (i, j), 1)
if j < self.width - 1:
self.dijk_grid.add_edge((i, j), (i, j + 1), 1)
self.dijk_grid.add_edge((i, j + 1), (i, j), 1)
for node in self.blocks:
self.dijk_grid.remove_node(node)
def initialise_route_dict(self):
"""Initialises the route_dict attribute"""
self.route_dict[self.spawner_square] = find_route(
self.dijk_grid, self.spawner_square, self.exit_square)
def initialise_square_grid(self, display):
"""Initialises the square_grid attribute"""
screenWidth, screenHeight = display.get_size()
squarelen = int(screenHeight /
self.height) # Length of each square on the grid
gridStartX = int(screenWidth / 2 - screenHeight / 2)
gridStartY = 0
for i in range(self.height):
row = []
for j in range(self.width):
sqrX = gridStartX + squarelen * j
sqrY = gridStartY + squarelen * i
greySqr = pygame.Rect(sqrX, sqrY, squarelen, squarelen)
sqr = Square(greySqr)
row.append([sqr])
self.square_grid.append(row)
def initialise_exit(self):
"""Adds the Exit to its square_grid square"""
try:
sqr = self.square_grid[self.exit_square[0]][self.exit_square[1]][0]
w, h = sqr.surface.width, sqr.surface.height
rect = pygame.Rect(sqr.surface.x, sqr.surface.y, w, h)
self.square_grid[self.exit_square[0]][self.exit_square[1]].append(
Exit(rect, self.lives))
except (IndexError):
print(
"Instance variable square_grid has not been initialised properly. Please call the method initialise_square_grid to initialise it."
)
def initialise_spawner(self):
"""Adds the Spawner to its square_grid square"""
try:
sqr = self.square_grid[self.spawner_square[0]][
self.spawner_square[1]][0]
w, h = sqr.surface.width, sqr.surface.height
rect = pygame.Rect(sqr.surface.x, sqr.surface.y, w, h)
self.square_grid[self.spawner_square[0]][
self.spawner_square[1]].append(Spawner(rect))
except (IndexError):
print(
"Instance variable square_grid has not been initialised properly. Please call the method initialise_square_grid to initialise it."
)
def initialise_blocks(self):
"""Adds the Blocks to their square_grid squares"""
try:
for block in self.blocks:
sqr = self.square_grid[block[0]][block[1]][0]
w, h = sqr.surface.width, sqr.surface.height
rect = pygame.Rect(sqr.surface.x, sqr.surface.y, w, h)
self.square_grid[block[0]][block[1]].append(Block(rect))
except (IndexError):
print(
"Instance variable square_grid has not been initialised properly. Please call the method initialise_square_grid to initialise it."
)
def build_tower(self, value, operation, location):
"""Adds a new Tower to a square_grid square and blocks off its dijk_grid square"""
sqr = self.square_grid[location[0]][location[1]][0]
w, h = sqr.surface.width, sqr.surface.height
rect = pygame.Rect(sqr.surface.x, sqr.surface.y, w, h)
tower = Tower(rect, value, operation, location)
self.square_grid[tower.location[0]][tower.location[1]].append(tower)
self.dijk_grid.remove_node(tower.location)
self.tower_list.append(tower)
self.souls -= tower.cost
def sell_tower(self, tower):
"""For selling a Tower and removing from square_grid and adding its square to dijk_grid"""
pos = (tower.location[0], tower.location[1])
self.tower_list.remove(tower)
self.square_grid[pos[0]][pos[1]].remove(tower)
for node in [(pos[0], pos[1] + 1), (pos[0], pos[1] - 1),
(pos[0] + 1, pos[1]), (pos[0] - 1, pos[1])]:
if node in self.dijk_grid:
self.dijk_grid.add_edge(pos, node, 1)
self.dijk_grid.add_edge(node, pos, 1)
else:
self.dijk_grid.add_node((pos[0], pos[1]))
self.souls += max(tower.cost // 3, 1)
def spawn_numemy(self, start_val, speed, weight):
"""Adds a new Numemy object to the square_grid spawner_square"""
sqr = self.square_grid[self.spawner_square[0]][
self.spawner_square[1]][0]
w, h = sqr.surface.width, sqr.surface.height
rect = pygame.Rect(sqr.surface.x, sqr.surface.y, w, h)
numemy = Numemy(rect, start_val, speed, weight)
numemy.location = self.spawner_square
self.square_grid[numemy.location[0]][numemy.location[1]].append(numemy)
self.num_loc_list.append(numemy.location)
# Should be called after build_tower() and move_numemies()
def update_routes(self):
"""Updates all routes which pass through a new_square being built upon"""
new_route_dict = {
self.spawner_square:
find_route(self.dijk_grid, self.spawner_square, self.exit_square)
}
for num_loc in self.num_loc_list:
new_route_dict[num_loc] = find_route(self.dijk_grid, num_loc,
self.exit_square)
self.route_dict = new_route_dict
# Should be called after build_tower(), update_routes()
def update_forbidden_squares(self):
"""Updates the list of squares that cannot be built upon without isolating a Numemy"""
self.forbidden_squares = set(
self.num_loc_list) | {self.spawner_square, self.exit_square}
for num_loc in self.route_dict:
for square in self.route_dict[num_loc]:
test_grid = copy.deepcopy(self.dijk_grid)
test_grid.remove_node(square)
try:
find_route(test_grid, num_loc, self.exit_square)
except NoPathError:
self.forbidden_squares |= {square}
