uct/prior.h

   1 #ifndef ZZGO_UCT_PRIOR_H
   2 #define ZZGO_UCT_PRIOR_H
   3
   4 #include "move.h"
   5 #include "uct/tree.h"
   6
   7 struct tree;
   8 struct tree_node;
   9 struct uct;
  10 struct board;
  11
  12 struct prior_map {
  13         struct board *b;
  14         enum stone to_play;
  15         int parity;
  16         /* [board_size2(b)] array, move_stats are the prior
  17          * values to be assigned to individual moves;
  18          * move_stats.value is not updated. */
  19         struct move_stats *prior;
  20         /* [board_size2(b)] array, whether to compute
  21          * prior for the given value. */
  22         bool *consider;
  23         /* [board_size2(b)] array from cfg_distances() */
  24         int *distances;
  25 };
  26
  27 /* @value is the value, @playouts is its weight. */
  28 static void add_prior_value(struct prior_map *map, coord_t c, floating_t value, int playouts);
  29
  30 void uct_prior(struct uct *u, struct tree_node *node, struct prior_map *map);
  31
  32 struct uct_prior;
  33 struct uct_prior *uct_prior_init(char *arg, struct board *b);
  34 void uct_prior_done(struct uct_prior *p);
  35
  36
  37 static inline void
  38 add_prior_value(struct prior_map *map, coord_t c, floating_t value, int playouts)
  39 {
  40         floating_t v = map->parity > 0 ? value : 1 - value;
  41         /* We don't need atomicity: */
  42         struct move_stats s = { .playouts = playouts, .value = v };
  43         stats_merge(&map->prior[c], &s);
  44 }
  45
  46 #endif