feature: add shortcut m for toggle the move num of nodes

Author: xiaoyifang

katrain/__main__.py

@@ -124,6 +124,8 @@ def __init__(self, **kwargs):
         self.contribute_popup = None
 
         self.pondering = False
+        self.move_num = False
+
         self.animate_contributing = False
         self.message_queue = Queue()
 
@@ -173,6 +175,10 @@ def toggle_continuous_analysis(self, quiet=False):
             self.pondering = not self.pondering
             self.update_state()
 
+    def toggle_move_num(self):
+        self.move_num = not self.move_num
+        self.update_state()
+
     def start(self):
         if self.engine:
             return
@@ -740,6 +746,8 @@ def _on_keyboard_down(self, _keyboard, keycode, _text, modifiers):
 
         if keycode[1] == Theme.KEY_TOGGLE_CONTINUOUS_ANALYSIS:
             self.toggle_continuous_analysis(quiet=shift_pressed)
+        elif keycode[1] == Theme.KEY_TOGGLE_MOVENUM:
+            self.toggle_move_num()
         elif keycode[1] == Theme.KEY_TOGGLE_COORDINATES:
             self.board_gui.toggle_coordinates()
         elif keycode[1] in Theme.KEY_PAUSE_TIMER and not ctrl_pressed:

katrain/gui/badukpan.py

@@ -227,7 +227,7 @@ def redraw(self, *_args):
         self.draw_board_contents()
 
     def draw_stone(
-        self, x, y, player, alpha=1, innercol=None, evalcol=None, evalscale=1.0, scale=1.0, ownership=None, loss=None
+        self, x, y, player, alpha=1, innercol=None, evalcol=None, evalscale=1.0, scale=1.0, ownership=None, loss=None,depth=None
     ):
         stone_size = self.stone_size * scale
         if ownership is not None:
@@ -297,6 +297,11 @@ def draw_stone(
                 texture=cached_texture(Theme.LAST_MOVE_TEXTURE),
             )
 
+        if depth:
+            text = str(depth)
+            Color(*Theme.NUMBER_COLOR)
+            draw_text(pos=self.gridpos[y][x], text=text, font_size=self.stone_size*0.9, font_name="Roboto")
+
     def eval_color(self, points_lost, show_dots_for_class: List[bool] = None) -> Optional[List[float]]:
         i = evaluation_class(points_lost, self.trainer_config["eval_thresholds"])
         colors = Theme.EVAL_COLORS[self.trainer_config["theme"]]
@@ -663,6 +668,7 @@ def draw_board_contents(self, *_args):
                             if ownership_grid and not loss_grid and not new_move
                             else None,
                             loss=loss_grid[m.coords[1]][m.coords[0]] if loss_grid else None,
+                            depth=node.depth if katrain.move_num else None,
                         )
                 realized_points_lost = node.parent_realized_points_lost
 

katrain/gui/theme.py

@@ -108,6 +108,8 @@ class Theme:
     PASS_CIRCLE_TEXT_COLOR = [0.85, 0.85, 0.85]
 
     STONE_COLORS = {"B": BLACK, "W": WHITE}
+    NUMBER_COLOR = [0.85, 0.68, 0.40, 0.8]
+
     NEXT_MOVE_DASH_CONTRAST_COLORS = {"B": LIGHTER_GREY, "W": GREY}
     OUTLINE_COLORS = {"B": [0.3, 0.3, 0.3, 0.5], "W": [0.7, 0.7, 0.7, 0.5]}
     PV_TEXT_COLORS = {"W": BLACK, "B": WHITE}  # numbers in PV
@@ -198,6 +200,7 @@ class Theme:
     KEY_SELFPLAY_TO_END = "l"
     KEY_STOP_ANALYSIS = "escape"
     KEY_TOGGLE_CONTINUOUS_ANALYSIS = "spacebar"
+    KEY_TOGGLE_MOVENUM = "m"
 
     KEY_COPY = "c"
     KEY_PASTE = "v"