From 2ee7eaba813ffaca6a3715d8bc7cf25d854af835 Mon Sep 17 00:00:00 2001
From: Peter Ward <pwar3236@usyd.edu.au>
Date: Tue, 5 Jan 2010 07:07:10 +1100
Subject: Gave common arg formatting.

---
 console_snake.py | 13 ++++++-------
 pygame_snake.py  | 23 +++++++++--------------
 pyglet_snake.py  | 12 ++++++------
 3 files changed, 21 insertions(+), 27 deletions(-)

diff --git a/console_snake.py b/console_snake.py
index 3c40b24..1db5e38 100755
--- a/console_snake.py
+++ b/console_snake.py
@@ -58,15 +58,14 @@ class ConsoleSnakeEngine(SnakeEngine):
             self.update_snakes()
 
 def main(*args):
-    from bots import *
+    import sys
     from processbot import BotWrapper
 
-    game = ConsoleSnakeEngine(25, 25, 50)
-    game.add_bot(right_bot)
-    game.add_bot(random_bot)
-    game.add_bot(random_bounds_bot)
-    game.add_bot(random_square_bot)
-    game.add_bot(BotWrapper('bots/peter.py'))
+    rows, columns, apples = map(int, sys.argv[1:4])
+    game = ConsoleSnakeEngine(rows, columns, apples)
+    for filename in sys.argv[4:]:
+        bot = BotWrapper(filename)
+        game.add_bot(bot)
     game.run()
 
 if __name__ == '__main__':
diff --git a/pygame_snake.py b/pygame_snake.py
index 6ee0eeb..cf07297 100755
--- a/pygame_snake.py
+++ b/pygame_snake.py
@@ -2,12 +2,14 @@
 
 from __future__ import division
 
+import os
 import time
 
 import pygame
 pygame.init()
 from pygame.locals import *
 
+from common import *
 from snake import SnakeEngine
 
 class Sprites(object):
@@ -137,22 +139,15 @@ class PygameSnakeEngine(SnakeEngine):
             time.sleep(2)
 
 if __name__ == '__main__':
-    from bots import *
+    import sys
     from processbot import BotWrapper
 
-    ROWS = 25
-    COLUMNS = 25
-    APPLES = 50
-    game = PygameSnakeEngine(ROWS, COLUMNS, APPLES, results=True)
-
-    while True:
-        game.add_bot(right_bot)
-        game.add_bot(random_bot)
-        game.add_bot(random_bounds_bot)
-        game.add_bot(random_square_bot)
-        game.add_bot(BotWrapper('bots/peter.py'))
-        game.run()
-        game.new_game(ROWS, COLUMNS, APPLES)
+    rows, columns, apples = map(int, sys.argv[1:4])
+    game = PygameSnakeEngine(rows, columns, apples)
+    for filename in sys.argv[4:]:
+        bot = BotWrapper(filename)
+        game.add_bot(bot)
+    game.run()
 
     # Early window close, late process cleanup.
     pygame.display.quit()
diff --git a/pyglet_snake.py b/pyglet_snake.py
index 2ef11a8..3d3cfc1 100755
--- a/pyglet_snake.py
+++ b/pyglet_snake.py
@@ -108,13 +108,13 @@ class PygletSnakeEngine(SnakeEngine, pyglet.window.Window):
         pyglet.app.run()
 
 if __name__ == '__main__':
-    from bots import random_bounds_bot, random_square_bot
+    import sys
     from processbot import BotWrapper
 
-    game = PygletSnakeEngine(25, 25, 50, results=True)
-    game.add_bot(random_bounds_bot)
-    game.add_bot(random_square_bot)
-    for i in xrange(2):
-        game.add_bot(BotWrapper('bots/peter.py'))
+    rows, columns, apples = map(int, sys.argv[1:4])
+    game = PygletSnakeEngine(rows, columns, apples)
+    for filename in sys.argv[4:]:
+        bot = BotWrapper(filename)
+        game.add_bot(bot)
     game.run()
 
-- 
cgit v1.2.3