Added function test map and bot.

Author: McLeopold

ants/ants.py

@@ -50,14 +50,20 @@ def __init__(self, options=None):
         self.viewradius = int(options["viewradius2"])
         self.attackradius = int(options["attackradius2"])
         self.spawnradius = int(options["spawnradius2"])
+        print("Starting game with view=%s, attack=%s and spawn=%s" % (self.viewradius, self.attackradius, self.spawnradius))
         self.do_attack = self.do_attack_closest
         if 'attack' in options:
             if options['attack'] == 'occupied':
                 self.do_attack = self.do_attack_occupied
             elif options['attack'] == 'closest':
                 self.do_attack = self.do_attack_closest
-        self.render = self.render_changes
         self.do_food = self.do_food_sections
+        if 'food' in options:
+            if options['food'] == 'none':
+                self.do_food = self.do_food_none
+            elif options['food'] == 'sections':
+                self.do_food = self.do_food_sections
+        self.render = self.render_changes
 
         self.width = None   # the map
         self.height = None
@@ -154,7 +160,7 @@ def distance(self, x1, y1, x2, y2):
         d_y = min(abs(y1 - y2), self.height - abs(y1 - y2))
         return d_x + d_y
 
-    def get_vision(self, player, radius=96):
+    def get_vision(self, player):
         vision = [[False for col in range(self.width)] for row in range(self.height)]
         squaresToCheck = deque()
         for row, col in self.player_ants(player):
@@ -168,7 +174,7 @@ def get_vision(self, player, radius=96):
                 d_row = min(d_row, self.height - d_row)
                 d_col = abs(a_col - n_col)
                 d_col = min(d_col, self.width - d_col)
-                if not vision[n_row][n_col] and (d_row**2 + d_col**2) <= radius:
+                if not vision[n_row][n_col] and (d_row**2 + d_col**2) <= self.viewradius:
                     vision[n_row][n_col] = True
                     if not self.revealed[player][n_row][n_col]:
                         self.turn_reveal[player].append((n_row, n_col))
@@ -180,8 +186,8 @@ def get_vision(self, player, radius=96):
                     squaresToCheck.append(((a_row,a_col),(n_row,n_col)))
         return vision
 
-    def get_perspective(self, player, radius=96):
-        v = self.get_vision(player, radius)
+    def get_perspective(self, player):
+        v = self.get_vision(player, self.viewradius)
         #start_row = self.center[player][1] - self.height // 2
         #stop_row = start_row + self.height
         #start_col = self.center[player][0] - self.width // 2
@@ -350,7 +356,7 @@ def do_spawn(self):
         new_ants = {}
         for f_row, f_col in self.food_list[:]:
             owner = None
-            for (n_row, n_col), n_owner in self.nearby_ants(f_row, f_col, None, 1, 9):
+            for (n_row, n_col), n_owner in self.nearby_ants(f_row, f_col, None, 1, self.spawnradius):
                 if owner == None:
                     owner = n_owner
                 elif owner != n_owner:
@@ -371,7 +377,7 @@ def do_attack_occupied(self):
         score = [Fraction(0, 1) for i in range(self.num_players)]
         for (a_row, a_col), a_owner in self.ant_list.items():
             killers = []
-            enemies = self.nearby_ants(a_row, a_col, a_owner, 1, 2)
+            enemies = self.nearby_ants(a_row, a_col, a_owner, 1, self.attackradius)
             occupied = len(enemies)
             for (e_row, e_col), e_owner in enemies:
                 e_occupied = len(self.nearby_ants(e_row, e_col, e_owner, 1, 2))
@@ -396,7 +402,7 @@ def find_enemy(row, col, owner, min_d, max_d):
                 if not (n_row, n_col) in ant_group:
                     ant_group[(n_row, n_col)] = n_owner
                     find_enemy(n_row, n_col, n_owner, min_d, max_d)
-        for distance in range(1, 10):
+        for distance in range(1, self.attackradius):
             for (a_row, a_col), a_owner in self.ant_list.items():
                 if self.map[a_row][a_col] != LAND:
                     ant_group = {(a_row, a_col): a_owner}
@@ -496,6 +502,9 @@ def find_closest_land(self, coord):
 
         return None
 
+    def do_food_none(self, amount=0):
+        pass
+
     def do_food_random(self, amount=1):
         """
             Place food randomly on the map

ants/bots/python/MyBot.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+from ants import *
+
+class TestBot:
+    def do_turn(self, ants):
+        destinations = []
+        for a_row, a_col in ants.my_ants():
+            # try all directions in order
+            directions = ('n','e','s','w')
+            for direction in directions:
+                (n_row, n_col) = ants.destination(a_row, a_col, direction)
+                if (not (n_row, n_col) in destinations and
+                        ants.passable(n_row, n_col)):
+                    ants.issue_order((a_row, a_col, direction))
+                    destinations.append((n_row, n_col))
+                    break
+            else:
+                destinations.append((a_row, a_col))
+
+if __name__ == '__main__':
+    try:
+        import psyco
+        psyco.full()
+    except ImportError:
+        pass
+    try:
+        Ants.run(TestBot())
+    except KeyboardInterrupt:
+        print('ctrl-c, leaving ...')

ants/playgame.py

@@ -62,6 +62,9 @@ def main(argv):
     parser.add_option("--attack", dest="attack",
                       default="closest",
                       help="Attack method to use for engine. (closest, occupied)")
+    parser.add_option("--food", dest="food",
+                      default="sections",
+                      help="Food spawning method. (none, random, sections)")
     parser.add_option("--viewradius2", dest="viewradius2",
                       default=96, type="int",
                       help="Vision radius of ants squared")
@@ -81,6 +84,7 @@ def main(argv):
         game_options = {
             "map": opts.map,
             "attack": opts.attack,
+            "food": opts.food,
             "viewradius2": opts.viewradius2,
             "attackradius2": opts.attackradius2,
             "spawnradius2": opts.spawnradius2,

ants/test_bot.cmd

@@ -0,0 +1,4 @@
+@echo off
+playgame.py --food none -o viewer -t 30 -m submission_test/test.map %1 "python submission_test/TestBot.py"
+
+

ants/test_bot.sh

@@ -0,0 +1,2 @@
+./playgame.py --food none -o viewer -t 30 -m submission_test/test.map %1 "python submission_test/TestBot.py"
+