class Node(Location):
def __init__(self, x, y, node_type):
Location.__init__(self, x, y)
self.important = False
if node_type == 'N' or node_type == 'M':
self.owner = PlayerColor.NEUTRAL
self.building = Building(BuildingType.EMPTY, UnitType.EMPTY)
self.important = node_type == 'M'
elif node_type == 'B':
self.owner = PlayerColor.BLUE
self.building = Building(BuildingType.HOME, UnitType.PIKEMAN)
self.important = True
elif node_type == 'R':
self.owner = PlayerColor.RED
self.building = Building(BuildingType.HOME, UnitType.PIKEMAN)
self.important = True
self.spawn_timer = 0
self.exit_direction = {PlayerColor.RED: None, PlayerColor.BLUE: None}
self.unit_in_loc = self.unit_in_loc
@DynamicClassAttribute
def name(self):
name = super().name
node_fields = [self.owner, self.building.name, self.important, self.spawn_timer, self.exit_direction]
return name + str(node_fields)
def add_neighbor(self, direction, neighbor, set_neighbor=True):
Location.add_neighbor(self, direction, neighbor, set_neighbor)
def collect_gold(self, players):
players = Location.collect_gold(self, players)
if players.keys().__contains__(self.owner):
players[self.owner].gold += self.building.generate_gold()
return players
def attempt_spawn(self, players):
players = Location.attempt_spawn(self, players)
if not self.building.can_spawn_unit():
return players
if self.spawn_timer > 0:
self.spawn_timer -= 1
elif self.unit_in_loc is None and players[self.owner].gold >= util.UNIT_COST:
self.unit_in_loc = Unit(self.building.unit_type, self.owner, self.exit_direction[self.owner])
self.spawn_timer = util.SPAWN_DELAY
players[self.owner].gold -= util.UNIT_COST
return players
def get_direction(self, owner):
if self.exit_direction.keys().__contains__(owner):
return self.exit_direction[owner]
return None
def set_exit_direction(self, owner, direction):
self.exit_direction[owner] = direction
self.diffusion.set_unit_dir_pref(direction, owner)
if self.unit_in_loc:
self.unit_in_loc.set_direction(self.exit_direction[self.owner])
def get_next_node(self, direction):
return self
def accept_unit(self):
Location.accept_unit(self)
if self.unit_in_loc is not None:
if self.owner is not self.unit_in_loc.owner:
self.owner = self.unit_in_loc.owner
self.building.type = BuildingType.EMPTY
self.unit_in_loc.set_direction(self.exit_direction[self.owner])
def check_click(self, x, y):
top = self.y * util.NODE_SIZE
left = self.x * util.NODE_SIZE
right = left + util.NODE_SIZE
bottom = top + util.NODE_SIZE
return left <= x <= right and top <= y <= bottom
def is_home_node(self, player):
return self.owner == player and self.building.is_home()
def prepare_diffusion(self):
Location.prepare_diffusion(self)
self.diffusion.set_node_value(self.building, self.important, self.owner)
def draw_node(self, win, pos):
if self.owner == PlayerColor.RED:
win.blit(RED_NODE_IMG, pos)
elif self.owner == PlayerColor.BLUE:
win.blit(BLUE_NODE_IMG, pos)
else:
win.blit(NEUTRAL_NODE_IMG, pos)
def draw_node_exit(self, win, pos):
for exit_dir in self.exit_direction:
if self.exit_direction[exit_dir] == Direction.NORTH:
win.blit(NODE_EXIT_BG_NORTH_IMG, pos)
elif self.exit_direction[exit_dir] == Direction.SOUTH:
win.blit(NODE_EXIT_BG_SOUTH_IMG, pos)
elif self.exit_direction[exit_dir] == Direction.EAST:
win.blit(NODE_EXIT_BG_EAST_IMG, pos)
elif self.exit_direction[exit_dir] == Direction.WEST:
win.blit(NODE_EXIT_BG_WEST_IMG, pos)
if self.exit_direction[PlayerColor.RED] == Direction.NORTH:
win.blit(NODE_EXIT_RED_NORTH_IMG, pos)
elif self.exit_direction[PlayerColor.RED] == Direction.SOUTH:
win.blit(NODE_EXIT_RED_SOUTH_IMG, pos)
elif self.exit_direction[PlayerColor.RED] == Direction.EAST:
win.blit(NODE_EXIT_RED_EAST_IMG, pos)
elif self.exit_direction[PlayerColor.RED] == Direction.WEST:
win.blit(NODE_EXIT_RED_WEST_IMG, pos)
if self.exit_direction[PlayerColor.BLUE] == Direction.NORTH:
win.blit(NODE_EXIT_BLUE_NORTH_IMG, pos)
elif self.exit_direction[PlayerColor.BLUE] == Direction.SOUTH:
win.blit(NODE_EXIT_BLUE_SOUTH_IMG, pos)
elif self.exit_direction[PlayerColor.BLUE] == Direction.EAST:
win.blit(NODE_EXIT_BLUE_EAST_IMG, pos)
elif self.exit_direction[PlayerColor.BLUE] == Direction.WEST:
win.blit(NODE_EXIT_BLUE_WEST_IMG, pos)
def draw(self, win):
Location.draw(self, win)
pos = self.get_pixel_pos()
self.draw_node(win, pos)
self.draw_node_exit(win, pos)
self.building.draw(win, pos)
