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. */
59 int gtp_id
; /* id of the last gtp command */
66 struct uct_dynkomi
*dynkomi
;
72 int random_policy_chance
;
75 float local_tree_aging
;
76 bool local_tree_allseq
;
77 /* Playout-localtree integration. */
78 bool local_tree_playout
; // can be true only if ELO playout
79 bool local_tree_pseqroot
;
83 struct uct_policy
*policy
;
84 struct uct_policy
*random_policy
;
85 struct playout_policy
*playout
;
86 struct uct_prior
*prior
;
88 /* Used within frame of single genmove. */
89 struct board_ownermap ownermap
;
90 struct node_stats
*stats
;
92 int played_all
; /* games played by all slaves */
94 /* Game state - maintained by setup_state(), reset_state(). */
98 #define UDEBUGL(n) DEBUGL_(u->debug_level, n)
100 extern volatile sig_atomic_t uct_halt
;
101 extern __thread
int thread_id
;
103 bool uct_pass_is_safe(struct uct
*u
, struct board
*b
, enum stone color
, bool pass_all_alive
);
106 /* This is the state used for descending the tree; we use this wrapper
107 * structure in order to be able to easily descend in multiple trees
108 * in parallel (e.g. main tree and local tree) or compute cummulative
109 * "path value" throughout the tree descent. */
111 /* Active tree nodes: */
112 struct tree_node
*node
; /* Main tree. */
113 struct tree_node
*lnode
; /* Local tree. */
114 /* Value of main tree node (with all value factors, but unbiased
115 * - without exploration factor), from black's perspective. */
116 struct move_stats value
;
120 typedef struct tree_node
*(*uctp_choose
)(struct uct_policy
*p
, struct tree_node
*node
, struct board
*b
, enum stone color
, coord_t exclude
);
121 typedef float (*uctp_evaluate
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
);
122 typedef void (*uctp_descend
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
, bool allow_pass
);
123 typedef void (*uctp_winner
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
);
124 typedef void (*uctp_prior
)(struct uct_policy
*p
, struct tree
*tree
, struct tree_node
*node
, struct board
*b
, enum stone color
, int parity
);
125 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
);
131 uctp_evaluate evaluate
;
132 uctp_descend descend
;