1 #ifndef ZZGO_UCT_INTERNAL_H
2 #define ZZGO_UCT_INTERNAL_H
4 #include <signal.h> // sig_atomic_t
18 /* Internal UCT structures */
20 /* Stats for each child of the root node. */
22 struct move_stats last_sent_own
;
23 struct move_stats added_from_others
;
24 struct tree_node
*node
;
27 /* Internal engine state. */
33 double best2_ratio
, bestr_ratio
;
35 bool territory_scoring
;
37 bool playout_amaf
, playout_amaf_nakade
;
39 int playout_amaf_cutoff
;
44 unsigned long max_tree_size
;
48 enum uct_thread_model
{
49 TM_ROOT
, /* Root parallelization. */
50 TM_TREE
, /* Tree parallelization w/o virtual loss. */
51 TM_TREEVL
, /* Tree parallelization with virtual loss. */
55 bool pondering_opt
; /* User wants pondering */
56 bool pondering
; /* Actually pondering now */
57 bool slave
; /* Act as slave in distributed engine. */
65 struct uct_dynkomi
*dynkomi
;
71 int random_policy_chance
;
74 float local_tree_aging
;
75 bool local_tree_allseq
;
76 /* Playout-localtree integration. */
77 bool local_tree_playout
; // can be true only if ELO playout
78 bool local_tree_pseqroot
;
82 struct uct_policy
*policy
;
83 struct uct_policy
*random_policy
;
84 struct playout_policy
*playout
;
85 struct uct_prior
*prior
;
87 /* Used within frame of single genmove. */
88 struct board_ownermap ownermap
;
89 struct node_stats
*stats
;
91 int played_all
; /* games played by all slaves */
93 /* Game state - maintained by setup_state(), reset_state(). */
97 #define UDEBUGL(n) DEBUGL_(u->debug_level, n)
99 extern volatile sig_atomic_t uct_halt
;
100 extern __thread
int thread_id
;
102 bool uct_pass_is_safe(struct uct
*u
, struct board
*b
, enum stone color
, bool pass_all_alive
);
105 /* This is the state used for descending the tree; we use this wrapper
106 * structure in order to be able to easily descend in multiple trees
107 * in parallel (e.g. main tree and local tree) or compute cummulative
108 * "path value" throughout the tree descent. */
110 /* Active tree nodes: */
111 struct tree_node
*node
; /* Main tree. */
112 struct tree_node
*lnode
; /* Local tree. */
113 /* Value of main tree node (with all value factors, but unbiased
114 * - without exploration factor), from black's perspective. */
115 struct move_stats value
;
119 typedef struct tree_node
*(*uctp_choose
)(struct uct_policy
*p
, struct tree_node
*node
, struct board
*b
, enum stone color
, coord_t exclude
);
120 typedef float (*uctp_evaluate
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
);
121 typedef void (*uctp_descend
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
, bool allow_pass
);
122 typedef void (*uctp_winner
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
);
123 typedef void (*uctp_prior
)(struct uct_policy
*p
, struct tree
*tree
, struct tree_node
*node
, struct board
*b
, enum stone color
, int parity
);
124 typedef void (*uctp_update
)(struct uct_policy
*p
, struct tree
*tree
, struct tree_node
*node
, enum stone node_color
, enum stone player_color
, struct playout_amafmap
*amaf
, float result
);
130 uctp_evaluate evaluate
;
131 uctp_descend descend
;