1 #ifndef PACHI_UCT_INTERNAL_H
2 #define PACHI_UCT_INTERNAL_H
4 /* Internal UCT structures */
10 #include "patternsp.h"
11 #include "patternprob.h"
24 /* How big proportion of ownermap counts must be of one color to consider
27 /* How many games to consider at minimum before judging groups. */
28 #define GJ_MINGAMES 500
30 /* TODO: Generalize this for other engines too. */
32 struct pattern_config pc
;
34 struct pattern_pdict
*pd
;
37 /* Internal engine state. */
41 floating_t resign_threshold
, sure_win_threshold
;
42 double best2_ratio
, bestr_ratio
;
43 floating_t max_maintime_ratio
;
44 bool pass_all_alive
; /* Current value */
45 bool territory_scoring
;
49 int playout_amaf_cutoff
;
54 unsigned long max_tree_size
;
55 unsigned long max_pruned_size
;
56 unsigned long pruning_threshold
;
58 int significant_threshold
;
61 enum uct_thread_model
{
62 TM_TREE
, /* Tree parallelization w/o virtual loss. */
63 TM_TREEVL
, /* Tree parallelization with virtual loss. */
66 bool pondering_opt
; /* User wants pondering */
67 bool pondering
; /* Actually pondering now */
68 bool slave
; /* Act as slave in distributed engine. */
69 int max_slaves
; /* Optional, -1 if not set */
70 int slave_index
; /* 0..max_slaves-1, or -1 if not set */
78 struct uct_dynkomi
*dynkomi
;
84 int random_policy_chance
;
87 floating_t local_tree_aging
;
88 #define LTREE_PLAYOUTS_MULTIPLIER 100
89 floating_t local_tree_depth_decay
;
90 bool local_tree_allseq
;
91 bool local_tree_neival
;
97 bool local_tree_rootchoose
;
101 struct uct_policy
*policy
;
102 struct uct_policy
*random_policy
;
103 struct playout_policy
*playout
;
104 struct uct_prior
*prior
;
105 struct uct_pluginset
*plugins
;
106 struct joseki_dict
*jdict
;
107 struct uct_pattern pat
;
109 /* Used within frame of single genmove. */
110 struct board_ownermap ownermap
;
111 /* Used for coordination among slaves of the distributed engine. */
115 double stats_delay
; /* stored in seconds */
117 int played_all
; /* games played by all slaves */
119 /* Game state - maintained by setup_state(), reset_state(). */
123 #define UDEBUGL(n) DEBUGL_(u->debug_level, n)
125 bool uct_pass_is_safe(struct uct
*u
, struct board
*b
, enum stone color
, bool pass_all_alive
);
127 void uct_prepare_move(struct uct
*u
, struct board
*b
, enum stone color
);
128 void uct_genmove_setup(struct uct
*u
, struct board
*b
, enum stone color
);
129 void uct_pondering_stop(struct uct
*u
);
132 /* This is the state used for descending the tree; we use this wrapper
133 * structure in order to be able to easily descend in multiple trees
134 * in parallel (e.g. main tree and local tree) or compute cummulative
135 * "path value" throughout the tree descent. */
137 /* Active tree nodes: */
138 struct tree_node
*node
; /* Main tree. */
139 struct tree_node
*lnode
; /* Local tree. */
140 /* Value of main tree node (with all value factors, but unbiased
141 * - without exploration factor), from black's perspective. */
142 struct move_stats value
;
146 typedef struct tree_node
*(*uctp_choose
)(struct uct_policy
*p
, struct tree_node
*node
, struct board
*b
, enum stone color
, coord_t exclude
);
147 typedef floating_t (*uctp_evaluate
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
);
148 typedef void (*uctp_descend
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
, int parity
, bool allow_pass
);
149 typedef void (*uctp_winner
)(struct uct_policy
*p
, struct tree
*tree
, struct uct_descent
*descent
);
150 typedef void (*uctp_prior
)(struct uct_policy
*p
, struct tree
*tree
, struct tree_node
*node
, struct board
*b
, enum stone color
, int parity
);
151 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
, struct board
*final_board
, floating_t result
);
157 uctp_evaluate evaluate
;
158 uctp_descend descend
;