X-Git-Url: https://git.ucc.asn.au/?p=progcomp2012.git;a=blobdiff_plain;f=progcomp%2Fagents%2Fbasic_python%2Fbasic_python.py;h=849c6f7a84da75f98412529c191cd18712458c8e;hp=cf024e8d45ae9c67256f325b4ec82432d10f228f;hb=345d8b1f49dfbc4f569c2dae992e896e6c685d55;hpb=88fc5a96c424e9a451b98f3b680bc1980345320d

diff --git a/progcomp/agents/basic_python/basic_python.py b/progcomp/agents/basic_python/basic_python.py
old mode 100644
new mode 100755
index cf024e8..849c6f7
--- a/progcomp/agents/basic_python/basic_python.py
+++ b/progcomp/agents/basic_python/basic_python.py
@@ -20,7 +20,15 @@ import random
 
 ranks = ['B','1','2','3','4','5','6','7','8','9','s','F', '?', '+']
 
-def move(x, y, direction):
+def is_integer(s):
+	""" Using exceptions for this feels... wrong..."""
+	try:
+		int(s)
+		return True
+	except ValueError:
+		return False
+
+def move(x, y, direction, multiplier):
 	""" Moves point (x,y) in direction, returns a pair """
 	if direction == "UP":
 		return (x,y-1)
@@ -203,9 +211,16 @@ class BasicAI:
 		#sys.stderr.write("	Board position " + str(x) + " " + str(y) + " is OK!\n")		
 
 		direction = result[2].strip()
+
+		multiplier = 1
 		outcome = result[3].strip()
+		outIndex = 3
+		if is_integer(outcome):
+			multiplier = int(outcome)
+			outcome = result[4].strip()
+			outIndex = 4
 		
-		p = move(x,y,direction)
+		p = move(x,y,direction, multiplier)
 
 		
 
@@ -229,7 +244,7 @@ class BasicAI:
 
 
 			self.board[p[0]][p[1]] = self.board[x][y]
-			self.board[x][y].rank = result[4].strip()
+			self.board[x][y].rank = result[outIndex+1].strip()
 
 			self.board[x][y] = None
 			
@@ -242,7 +257,7 @@ class BasicAI:
 			elif self.board[x][y].colour == oppositeColour(self.colour):
 				self.enemyUnits.remove(self.board[x][y])
 
-			self.board[p[0]][p[1]].rank = result[5].strip()
+			self.board[p[0]][p[1]].rank = result[outIndex+2].strip()
 			self.board[x][y] = None
 		elif outcome == "BOTHDIE":
 			if self.board[p[0]][p[1]] == None:
@@ -287,8 +302,9 @@ class BasicAI:
 					sys.stderr.write(str(self.board[x][y].rank));
 			sys.stderr.write("\n")
 
-#basicAI = BasicAI()
-#if basicAI.Setup():
-#	while basicAI.MoveCycle():
-#		pass
+if __name__ == "__main__":
+	basicAI = BasicAI()
+	if basicAI.Setup():
+		while basicAI.MoveCycle():
+			pass