uct/tree.h

   1 #ifndef ZZGO_UCT_TREE_H
   2 #define ZZGO_UCT_TREE_H
   3
   4 #include <stdbool.h>
   5 #include "move.h"
   6
   7 struct board;
   8
   9 struct tree_node {
  10         /*
  11          *            +------+
  12          *            | node |
  13          *            +------+
  14          *          / <- parent
  15          * +------+   v- sibling +------+
  16          * | node | ------------ | node |
  17          * +------+              +------+
  18          *    | <- children          |
  19          * +------+   +------+   +------+   +------+
  20          * | node | - | node |   | node | - | node |
  21          * +------+   +------+   +------+   +------+
  22          */
  23         struct tree_node *parent, *sibling, *children;
  24
  25         coord_t coord;
  26         int playouts; // # of playouts coming through this node
  27         int wins; // # of wins coming through this node
  28         float value; // wins/playouts
  29 };
  30
  31 struct tree {
  32         struct tree_node *root;
  33         struct board *board;
  34 };
  35
  36 struct tree *tree_init(struct board *board);
  37 void tree_done(struct tree *tree);
  38 void tree_dump(struct tree *tree);
  39
  40 void tree_expand_node(struct tree *tree, struct tree_node *node, struct board *b);
  41 void tree_delete_node(struct tree_node *node);
  42 void tree_promote_node(struct tree *tree, struct tree_node *node);
  43 bool tree_leaf_node(struct tree_node *node);
  44 void tree_uct_update(struct tree_node *node, int result);
  45
  46 #endif