110 lines
4.2 KiB
Python
110 lines
4.2 KiB
Python
|
|
|
|
class Move:
|
|
GTP_COORD = "ABCDEFGHJKLMNOPQRSTUVWYXYZ"
|
|
PLAYERS = "BW"
|
|
SGF_COORD = [chr(i) for i in range(97, 123)]
|
|
|
|
def __init__(self, player, coords=None, gtpcoords=None, sgfcoords=None, robot=False):
|
|
self.id = None
|
|
self.player = player
|
|
self.coords = coords or (gtpcoords and self.gtp2ix(gtpcoords)) or self.sgf2ix(sgfcoords)
|
|
self.children = []
|
|
self.parent = None
|
|
self.robot = robot
|
|
self.analysis = None
|
|
self.pass_analysis = None
|
|
self.outdated_evaluation = None
|
|
self.evaluation = None
|
|
self.ownership = None
|
|
self.points_lost = 0
|
|
self.previous_temperature = None
|
|
self.comment = ""
|
|
|
|
def __repr__(self):
|
|
return f"{Move.PLAYERS[self.player]}{self.gtp()}"
|
|
|
|
def __eq__(self, other):
|
|
return self.coords == other.coords and self.player == other.player
|
|
|
|
def __hash__(self):
|
|
return self.gtp().__hash__()
|
|
|
|
def play(self, move):
|
|
try:
|
|
return self.children[self.children.index(move)]
|
|
except ValueError:
|
|
move.parent = self
|
|
self.children.append(move)
|
|
return move
|
|
|
|
def temperature(self):
|
|
if self.analysis:
|
|
best_score = float(self.analysis[0]["scoreLead"])
|
|
worst_score = -float(self.pass_analysis[0]["scoreLead"])
|
|
return best_score - worst_score
|
|
else:
|
|
return 0
|
|
|
|
def evaluate(self,analysis_blob):
|
|
self.analysis = analysis_blob['moveInfos']
|
|
self.ownership = analysis_blob['ownership']
|
|
previous_move = self.parent
|
|
if not self.analysis and self.pass_analysis and previous_move.analysis:
|
|
return
|
|
# TODO: update children?
|
|
best_score = float(previous_move.analysis[0]["scoreLead"])
|
|
worst_score = -float(previous_move.pass_analysis[0]["scoreLead"])
|
|
last_move_score = -float(self.analysis[0]["scoreLead"])
|
|
self.previous_temperature = best_score - worst_score
|
|
self.points_lost = best_score - last_move_score
|
|
prev_analysis_current_move = [d for d in previous_move.analysis if d["move"] == self.gtp()]
|
|
|
|
if abs(self.previous_temperature) > 0.5:
|
|
self.evaluation = (last_move_score - worst_score) / (best_score - worst_score)
|
|
self.move_options = [previous_move.analysis[0]["scoreLead"]]
|
|
else:
|
|
self.evaluation = None
|
|
if self.evaluation:
|
|
self.comment = f"Evaluation: {100*self.evaluation:.1f}%{' (AI Move)' if self.robot else ''}\n"
|
|
if prev_analysis_current_move:
|
|
self.outdated_evaluation = (prev_analysis_current_move[0]["scoreLead"] - worst_score) / (
|
|
best_score - worst_score
|
|
)
|
|
self.comment += f"(Was considered last move as: {100 * self.outdated_evaluation:.1f}%)\n"
|
|
else:
|
|
self.comment = "Temperature too low for evaluation\n"
|
|
self.comment += f"Estimate point loss: {self.points_lost:.1f}\n"
|
|
self.comment += f"Last move score was {last_move_score:.1f}\n"
|
|
self.comment += f"Score of top move was {previous_move.analysis[0]['scoreLead']:.1f} @ {previous_move.analysis[0]['move']}\n"
|
|
self.comment += f"Pass score was {worst_score:.1f}\n"
|
|
|
|
@property
|
|
def is_pass(self):
|
|
return self.coords[0] is None
|
|
|
|
def gtp2ix(self, gtpmove):
|
|
if "pass" in gtpmove:
|
|
return (None, None)
|
|
return Move.GTP_COORD.index(gtpmove[0]), int(gtpmove[1:]) - 1
|
|
|
|
def sgf2ix(self, sgfmove_with_boardsize):
|
|
sgfmove, boardsize = sgfmove_with_boardsize
|
|
if sgfmove == "":
|
|
return (None, None)
|
|
return Move.SGF_COORD.index(sgfmove[0]), boardsize - Move.SGF_COORD.index(sgfmove[1]) - 1
|
|
|
|
def gtp(self):
|
|
if self.is_pass:
|
|
return "pass"
|
|
return Move.GTP_COORD[self.coords[0]] + str(self.coords[1] + 1)
|
|
|
|
def sgfcoords(self, boardsize):
|
|
return f"{Move.SGF_COORD[self.coords[0]]}{Move.SGF_COORD[boardsize - self.coords[1] - 1]}"
|
|
|
|
def sgf(self, boardsize):
|
|
if self.is_pass:
|
|
return f"{Move.PLAYERS[self.player]}[]"
|
|
else:
|
|
return f"{Move.PLAYERS[self.player]}[{self.sgfcoords(boardsize)}]"
|