# Update the board when a piece has been selected
# "type" is apparently reserved, so I'll use "state"
- def update_select(self, x, y, type_index, state):
+ def update_select(self, x, y, type_index, state, sanity=True):
piece = self.grid[x][y]
if piece.types[type_index] == "unknown":
- if not state in self.unrevealed_types[piece.colour].keys():
+ if not state in self.unrevealed_types[piece.colour].keys() and sanity == True:
raise Exception("SANITY: Too many " + piece.colour + " " + state + "s")
self.unrevealed_types[piece.colour][state] -= 1
if self.unrevealed_types[piece.colour][state] <= 0:
piece.possible_moves = None
# Update the board when a piece has been moved
- def update_move(self, x, y, x2, y2):
+ def update_move(self, x, y, x2, y2, sanity=True):
piece = self.grid[x][y]
#print "Moving " + str(x) + "," + str(y) + " to " + str(x2) + "," + str(y2) + "; possible_moves are " + str(self.possible_moves(piece))
- if not [x2,y2] in self.possible_moves(piece):
- raise Exception("ILLEGAL move")
+ if not [x2,y2] in self.possible_moves(piece) and sanity == True:
+ raise Exception("ILLEGAL move " + str(x2)+","+str(y2))
self.grid[x][y] = None
taken = self.grid[x2][y2]
# Update the board from a string
# Guesses what to do based on the format of the string
- def update(self, result):
+ def update(self, result, sanity=True):
#print "Update called with \"" + str(result) + "\""
# String always starts with 'x y'
try:
raise Exception("GIBBERISH \""+ str(result) + "\"") # Raise expectations
piece = self.grid[x][y]
- if piece == None:
+ if piece == None and sanity == True:
raise Exception("EMPTY")
# If a piece is being moved, the third token is '->'
raise Exception("GIBBERISH \"" + str(result) + "\"") # Raise the alarm
# Move the piece (take opponent if possible)
- self.update_move(x, y, x2, y2)
+ self.update_move(x, y, x2, y2, sanity)
else:
# Otherwise we will just assume a piece has been selected
except:
raise Exception("GIBBERISH \"" + result + "\"") # Throw a hissy fit
+
# Select the piece
- self.update_select(x, y, type_index, state)
+ self.update_select(x, y, type_index, state, sanity)
return result
for i in range(len(p.types)):
t = p.types[i]
- prob = 0.5
+ prob = 1.0 / float(len(p.types))
if t == "unknown" or p.types[i][0] == '?':
total_types = 0
for t2 in self.unrevealed_types[p.colour].keys():
for t2 in self.unrevealed_types[p.colour].keys():
prob2 = float(self.unrevealed_types[p.colour][t2]) / float(total_types)
- p.current_type = t2
- for point in self.possible_moves(p, reject_allied):
+ #p.current_type = t2
+ for point in self.possible_moves(p, reject_allied, state=t2):
result[point[0]][point[1]] += prob2 * prob
else:
- p.current_type = t
- for point in self.possible_moves(p, reject_allied):
- result[point[0]][point[1]] += prob
+ #p.current_type = t
+ for point in self.possible_moves(p, reject_allied, state=t):
+ result[point[0]][point[1]] += prob
#self.verify()
- p.current_type = "unknown"
+ #p.current_type = "unknown"
return result
def prob_is_type(self, p, state):
p.current_type = old_type
return result
- if p.possible_moves != None:
- return p.possible_moves
+
result = []