# -*- coding: UTF-8 -*- from __future__ import with_statement import gtk, pango from pychess.System import uistuff from pychess.System.prefix import addDataPrefix from pychess.System.glock import * from pychess.Utils.const import * from pychess.Utils.repr import reprColor, reprPiece from pychess.Utils.lutils.lsort import staticExchangeEvaluate from pychess.Utils.lutils.lmove import FLAG, TCORD, FCORD, toSAN from pychess.Utils.lutils.lmovegen import genCaptures from pychess.Utils.lutils.leval import evalMaterial from pychess.Utils.lutils import ldata from pychess.Utils.lutils import strateval __title__ = _("Comments") __icon__ = addDataPrefix("glade/panel_comments.svg") __desc__ = _("The comments panel will try to analyze and explain the moves played") class Switch: def __init__(self): self.on = False def __enter__(self): self.on = True def __exit__(self, *a): self.on = False class Sidepanel: def __init__ (self): self.givenTips = {} def load (self, gmwidg): self.gamemodel = gmwidg.board.view.model self.gmhandlers = [ glock_connect(self.gamemodel, "game_changed", self.game_changed), glock_connect(self.gamemodel, "game_started", self.game_started), glock_connect(self.gamemodel, "moves_undoing", self.moves_undoing) ] widgets = gtk.Builder() widgets.add_from_file(addDataPrefix("sidepanel/book.glade")) self.tv = widgets.get_object("treeview") scrollwin = widgets.get_object("scrolledwindow") scrollwin.unparent() self.store = gtk.ListStore(str) self.tv.set_model(self.store) self.tv.get_selection().set_mode(gtk.SELECTION_BROWSE) #r = gtk.CellRendererText() #r.set_property("wrap-width", 177) #FIXME: Fixed width size #r.set_property("wrap-mode", pango.WRAP_WORD) #self.tv.append_column(gtk.TreeViewColumn("Comment", r, text=0)) uistuff.appendAutowrapColumn(self.tv, 200, "Comment", text=0) self.tv.get_selection().connect_after('changed', self.select_cursor_row) self.boardview = gmwidg.board.view self.boardview.connect("shown_changed", self.shown_changed) def changed (vadjust): if not hasattr(vadjust, "need_scroll") or vadjust.need_scroll: vadjust.set_value(vadjust.upper-vadjust.page_size) vadjust.need_scroll = True scrollwin.get_vadjustment().connect("changed", changed) def value_changed (vadjust): vadjust.need_scroll = abs(vadjust.value + vadjust.page_size - \ vadjust.upper) < vadjust.step_increment scrollwin.get_vadjustment().connect("value-changed", value_changed) self.frozen = Switch() return scrollwin def select_cursor_row (self, selection): iter = selection.get_selected()[1] if iter == None: return if self.frozen.on: return row = self.tv.get_model().get_path(iter)[0] board = self.gamemodel.boards[row] self.boardview.setShownBoard(board) def shown_changed (self, boardview, shown): if not boardview.shownIsMainLine(): return row = shown - self.gamemodel.lowply try: iter = self.store.get_iter(row) self.tv.get_selection().select_iter(iter) except ValueError: pass # deleted variations by moves_undoing def moves_undoing (self, game, moves): assert game.ply > 0, "Can't undo when ply <= 0" model = self.tv.get_model() for i in xrange(moves): model.remove(model.get_iter( (len(model)-1,) )) def game_started (self, model): self.game_changed(model) def game_changed (self, model): for ply in xrange(len(self.store)+model.lowply, model.ply+1): self.addComment(model, self.__chooseComment(model, ply)) def addComment (self, model, comment): self.store.append([comment]) # If latest ply is shown, we select the new latest iter = self.tv.get_selection().get_selected()[1] if iter: row = self.tv.get_model().get_path(iter)[0] if row < self.boardview.shown-1: return if self.boardview.shown >= model.ply: iter = self.store.get_iter(len(self.store)-1) with self.frozen: self.tv.get_selection().select_iter(iter) def __chooseComment(self, model, ply): if ply == model.lowply: return _("Initial position") ######################################################################## # Set up variables ######################################################################## color = model.getBoardAtPly(ply-1).board.color s, phase = evalMaterial (model.getBoardAtPly(ply).board, model.getBoardAtPly(ply-1).color) # * Final: Will be shown alone: "mates", "draws" # * Prefix: Will always be shown: "castles", "promotes" # * Attack: Will always be shown: "threaten", "preassures", "defendes" # * Moves (s): Will always be shown: "put into *" # * State: (s) Will always be shown: "new *" # * Simple: (s) Max one will be shown: "develops", "activity" # * Tip: (s) Will sometimes be shown: "pawn storm", "cramped position" ######################################################################## # Call strategic evaluation functions ######################################################################## def getMessages (prefix): messages = [] for functionName in dir(strateval): if not functionName.startswith(prefix+"_"): continue function = getattr(strateval, functionName) messages.extend(function (model, ply, phase)) return messages #move = model.moves[-1].move #print "----- %d - %s -----" % (model.ply/2, toSAN(oldboard, move)) # ---------------------------------------------------------------------- # Final # ---------------------------------------------------------------------- messages = getMessages ("final") if messages: return "%s %s" % (reprColor[color], messages[0]) # --- strings = [] # ---------------------------------------------------------------------- # Attacks # ---------------------------------------------------------------------- messages = getMessages ("attack") for message in messages: strings.append("%s %s" % (reprColor[color], message)) # ---------------------------------------------------------------------- # Check for prefixes # ---------------------------------------------------------------------- messages = getMessages ("prefix") if messages: prefix = messages[0] else: prefix = "" # ---------------------------------------------------------------------- # Check for special move stuff. All of which accept prefixes # ---------------------------------------------------------------------- for message in getMessages("offencive_moves") \ + getMessages("defencive_moves"): if prefix: strings.append("%s %s %s %s" % (reprColor[color], prefix, _("and"), message)) prefix = "" else: strings.append("%s %s" % (reprColor[color], message)) # ---------------------------------------------------------------------- # Simple # ---------------------------------------------------------------------- # We only add simples if there hasn't been too much stuff to say if not strings: messages = getMessages ("simple") if messages: messages.sort(reverse=True) score, message = messages[0] if prefix: strings.append("%s %s %s %s" % (reprColor[color], prefix, _("and"), message)) prefix = "" # ---------------------------------------------------------------------- # Prefix fallback # ---------------------------------------------------------------------- # There was nothing to apply the prefix to, so we just post it here # before the states and tips if prefix: strings.append("%s %s" % (reprColor[color], prefix)) prefix = "" # ---------------------------------------------------------------------- # State # ---------------------------------------------------------------------- messages = getMessages("state") messages.sort(reverse=True) for score, message in messages: strings.append(message) # ---------------------------------------------------------------------- # Tips # ---------------------------------------------------------------------- tips = getMessages("tip") tips.sort(reverse=True) for (score, tip) in tips: if tip in self.givenTips: oldscore, oldply = self.givenTips[tip] if score < oldscore*1.3 or model.ply < oldply+10: continue self.givenTips[tip] = (score, model.ply) strings.append(tip) break # ---------------------------------------------------------------------- # Last solution # ---------------------------------------------------------------------- if not strings: tcord = TCORD(model.getMoveAtPly(ply-1).move) piece = model.getBoardAtPly(ply).board.arBoard[tcord] strings.append( _("%(color)s moves a %(piece)s to %(cord)s") % { 'color': reprColor[color], 'piece': reprPiece[piece], 'cord': reprCord[tcord]}) return ";\n".join(strings)