Fix tower building exploit

battlecode · Jan 9, 2025 · 519681b · 519681b
1 parent 7091642
commit 519681b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/battlecode25/engine/game/robot_controller.py b/battlecode25/engine/game/robot_controller.py
@@ -519,6 +519,7 @@ def remove_mark(self, loc: MapLocation) -> None:
         self.game.game_fb.add_unmark_action(loc)
 
     def assert_can_complete_tower_pattern(self, tower_type: UnitType, loc: MapLocation) -> None:
+        tower_type = tower_type.get_base_type()
         self.assert_is_robot_type(self.robot.type)
         self.assert_is_tower_type(tower_type)
         self.assert_can_act_location(loc, GameConstants.BUILD_TOWER_RADIUS_SQUARED)
@@ -546,6 +547,7 @@ def can_complete_tower_pattern(self, tower_type: UnitType, loc: MapLocation) ->
             return False
 
     def complete_tower_pattern(self, tower_type: UnitType, loc: MapLocation) -> None:
+        tower_type = tower_type.get_base_type()
         self.assert_can_complete_tower_pattern(tower_type, loc)
         robot = self.game.spawn_robot(tower_type, loc, self.robot.team)
         self.game.game_fb.add_spawn_action(robot.id, loc, robot.team, robot.type)