13 #include "patternsp.h"
22 bool random_pass
= false;
26 #define profiling_noinline __attribute__((noinline))
28 #define profiling_noinline
31 #define gi_granularity 4
32 #define gi_allocsize(gids) ((1 << gi_granularity) + ((gids) >> gi_granularity) * (1 << gi_granularity))
36 board_setup(struct board
*b
)
38 memset(b
, 0, sizeof(*b
));
40 struct move m
= { pass
, S_NONE
};
41 b
->last_move
= b
->last_move2
= b
->last_ko
= b
->ko
= m
;
47 struct board
*b
= malloc(sizeof(struct board
));
58 board_copy(struct board
*b2
, struct board
*b1
)
60 memcpy(b2
, b1
, sizeof(struct board
));
62 int bsize
= board_size2(b2
) * sizeof(*b2
->b
);
63 int gsize
= board_size2(b2
) * sizeof(*b2
->g
);
64 int fsize
= board_size2(b2
) * sizeof(*b2
->f
);
65 int nsize
= board_size2(b2
) * sizeof(*b2
->n
);
66 int psize
= board_size2(b2
) * sizeof(*b2
->p
);
67 int hsize
= board_size2(b2
) * 2 * sizeof(*b2
->h
);
68 int gisize
= board_size2(b2
) * sizeof(*b2
->gi
);
70 int csize
= board_size2(b2
) * sizeof(*b2
->c
);
75 int ssize
= board_size2(b2
) * sizeof(*b2
->spathash
);
80 int p3size
= board_size2(b2
) * sizeof(*b2
->pat3
);
85 int tsize
= board_size2(b2
) * sizeof(*b2
->t
);
90 int pbsize
= board_size2(b2
) * sizeof(*b2
->prob
[0].items
);
94 void *x
= malloc(bsize
+ gsize
+ fsize
+ psize
+ nsize
+ hsize
+ gisize
+ csize
+ ssize
+ p3size
+ tsize
+ pbsize
* 2);
95 memcpy(x
, b1
->b
, bsize
+ gsize
+ fsize
+ psize
+ nsize
+ hsize
+ gisize
+ csize
+ ssize
+ p3size
+ tsize
+ pbsize
* 2);
96 b2
->b
= x
; x
+= bsize
;
97 b2
->g
= x
; x
+= gsize
;
98 b2
->f
= x
; x
+= fsize
;
99 b2
->p
= x
; x
+= psize
;
100 b2
->n
= x
; x
+= nsize
;
101 b2
->h
= x
; x
+= hsize
;
102 b2
->gi
= x
; x
+= gisize
;
104 b2
->c
= x
; x
+= csize
;
106 #ifdef BOARD_SPATHASH
107 b2
->spathash
= x
; x
+= ssize
;
110 b2
->pat3
= x
; x
+= p3size
;
113 b2
->t
= x
; x
+= tsize
;
116 b2
->prob
[0].items
= x
; x
+= pbsize
;
117 b2
->prob
[1].items
= x
; x
+= pbsize
;
124 board_done_noalloc(struct board
*board
)
126 if (board
->b
) free(board
->b
);
130 board_done(struct board
*board
)
132 board_done_noalloc(board
);
137 board_resize(struct board
*board
, int size
)
140 assert(board_size(board
) == size
+ 2);
142 board_size(board
) = size
+ 2 /* S_OFFBOARD margin */;
143 board_size2(board
) = board_size(board
) * board_size(board
);
148 int bsize
= board_size2(board
) * sizeof(*board
->b
);
149 int gsize
= board_size2(board
) * sizeof(*board
->g
);
150 int fsize
= board_size2(board
) * sizeof(*board
->f
);
151 int nsize
= board_size2(board
) * sizeof(*board
->n
);
152 int psize
= board_size2(board
) * sizeof(*board
->p
);
153 int hsize
= board_size2(board
) * 2 * sizeof(*board
->h
);
154 int gisize
= board_size2(board
) * sizeof(*board
->gi
);
156 int csize
= board_size2(board
) * sizeof(*board
->c
);
160 #ifdef BOARD_SPATHASH
161 int ssize
= board_size2(board
) * sizeof(*board
->spathash
);
166 int p3size
= board_size2(board
) * sizeof(*board
->pat3
);
171 int tsize
= board_size2(board
) * sizeof(*board
->t
);
176 int pbsize
= board_size2(board
) * sizeof(*board
->prob
[0].items
);
180 void *x
= malloc(bsize
+ gsize
+ fsize
+ psize
+ nsize
+ hsize
+ gisize
+ csize
+ ssize
+ p3size
+ tsize
+ pbsize
* 2);
181 memset(x
, 0, bsize
+ gsize
+ fsize
+ psize
+ nsize
+ hsize
+ gisize
+ csize
+ ssize
+ p3size
+ tsize
+ pbsize
* 2);
182 board
->b
= x
; x
+= bsize
;
183 board
->g
= x
; x
+= gsize
;
184 board
->f
= x
; x
+= fsize
;
185 board
->p
= x
; x
+= psize
;
186 board
->n
= x
; x
+= nsize
;
187 board
->h
= x
; x
+= hsize
;
188 board
->gi
= x
; x
+= gisize
;
190 board
->c
= x
; x
+= csize
;
192 #ifdef BOARD_SPATHASH
193 board
->spathash
= x
; x
+= ssize
;
196 board
->pat3
= x
; x
+= p3size
;
199 board
->t
= x
; x
+= tsize
;
202 board
->prob
[0].items
= x
; x
+= pbsize
;
203 board
->prob
[1].items
= x
; x
+= pbsize
;
208 board_clear(struct board
*board
)
210 int size
= board_size(board
);
211 float komi
= board
->komi
;
213 board_done_noalloc(board
);
215 board_resize(board
, size
- 2 /* S_OFFBOARD margin */);
219 /* Setup neighborhood iterators */
220 board
->nei8
[0] = -size
- 1; // (-1,-1)
223 board
->nei8
[3] = size
- 2; // (-1,0)
225 board
->nei8
[5] = size
- 2; // (-1,1)
228 board
->dnei
[0] = -size
- 1;
230 board
->dnei
[2] = size
*2 - 2;
233 /* Setup initial symmetry */
234 board
->symmetry
.d
= 1;
235 board
->symmetry
.x1
= board
->symmetry
.y1
= board_size(board
) / 2;
236 board
->symmetry
.x2
= board
->symmetry
.y2
= board_size(board
) - 1;
237 board
->symmetry
.type
= SYM_FULL
;
239 /* Draw the offboard margin */
240 int top_row
= board_size2(board
) - board_size(board
);
242 for (i
= 0; i
< board_size(board
); i
++)
243 board
->b
[i
] = board
->b
[top_row
+ i
] = S_OFFBOARD
;
244 for (i
= 0; i
<= top_row
; i
+= board_size(board
))
245 board
->b
[i
] = board
->b
[board_size(board
) - 1 + i
] = S_OFFBOARD
;
247 foreach_point(board
) {
249 if (board_at(board
, coord
) == S_OFFBOARD
)
251 foreach_neighbor(board
, c
, {
252 inc_neighbor_count_at(board
, coord
, board_at(board
, c
));
256 /* First, pass is always a free position. */
257 board
->f
[board
->flen
++] = coord_raw(pass
);
258 /* All positions are free! Except the margin. */
259 for (i
= board_size(board
); i
< (board_size(board
) - 1) * board_size(board
); i
++)
260 if (i
% board_size(board
) != 0 && i
% board_size(board
) != board_size(board
) - 1)
261 board
->f
[board
->flen
++] = i
;
263 /* Initialize zobrist hashtable. */
264 foreach_point(board
) {
265 int max
= (sizeof(hash_t
) << history_hash_bits
);
266 /* fast_random() is 16-bit only */
267 board
->h
[coord_raw(c
) * 2] = ((hash_t
) fast_random(max
))
268 | ((hash_t
) fast_random(max
) << 16)
269 | ((hash_t
) fast_random(max
) << 32)
270 | ((hash_t
) fast_random(max
) << 48);
271 if (!board
->h
[coord_raw(c
) * 2])
272 /* Would be kinda "oops". */
273 board
->h
[coord_raw(c
) * 2] = 1;
274 /* And once again for white */
275 board
->h
[coord_raw(c
) * 2 + 1] = ((hash_t
) fast_random(max
))
276 | ((hash_t
) fast_random(max
) << 16)
277 | ((hash_t
) fast_random(max
) << 32)
278 | ((hash_t
) fast_random(max
) << 48);
279 if (!board
->h
[coord_raw(c
) * 2 + 1])
280 board
->h
[coord_raw(c
) * 2 + 1] = 1;
283 #ifdef BOARD_SPATHASH
284 /* Initialize spatial hashes. */
285 foreach_point(board
) {
286 for (int d
= 1; d
<= BOARD_SPATHASH_MAXD
; d
++) {
287 for (int j
= ptind
[d
]; j
< ptind
[d
+ 1]; j
++) {
288 ptcoords_at(x
, y
, c
, board
, j
);
289 board
->spathash
[coord_xy(board
, x
, y
)][d
- 1][0] ^=
290 pthashes
[0][j
][board_at(board
, c
)];
291 board
->spathash
[coord_xy(board
, x
, y
)][d
- 1][1] ^=
292 pthashes
[0][j
][stone_other(board_at(board
, c
))];
298 /* Initialize 3x3 pattern codes. */
299 foreach_point(board
) {
300 if (board_at(board
, c
) == S_NONE
)
301 board
->pat3
[c
] = pattern3_hash(board
, c
);
305 /* Initialize traits. */
306 foreach_point(board
) {
307 trait_at(board
, c
, S_BLACK
).cap
= 0;
308 trait_at(board
, c
, S_BLACK
).safe
= true;
309 trait_at(board
, c
, S_WHITE
).cap
= 0;
310 trait_at(board
, c
, S_WHITE
).safe
= true;
314 board
->prob
[0].n
= board
->prob
[1].n
= board_size2(board
);
320 board_print_top(struct board
*board
, FILE *f
, int c
)
322 for (int i
= 0; i
< c
; i
++) {
323 char asdf
[] = "ABCDEFGHJKLMNOPQRSTUVWXYZ";
325 for (int x
= 1; x
< board_size(board
) - 1; x
++)
326 fprintf(f
, "%c ", asdf
[x
- 1]);
330 for (int i
= 0; i
< c
; i
++) {
332 for (int x
= 1; x
< board_size(board
) - 1; x
++)
340 board_print_bottom(struct board
*board
, FILE *f
, int c
)
342 for (int i
= 0; i
< c
; i
++) {
344 for (int x
= 1; x
< board_size(board
) - 1; x
++)
352 board_print_row(struct board
*board
, int y
, FILE *f
, board_cprint cprint
)
354 fprintf(f
, " %2d | ", y
);
355 for (int x
= 1; x
< board_size(board
) - 1; x
++) {
356 if (coord_x(board
->last_move
.coord
, board
) == x
&& coord_y(board
->last_move
.coord
, board
) == y
)
357 fprintf(f
, "%c)", stone2char(board_atxy(board
, x
, y
)));
359 fprintf(f
, "%c ", stone2char(board_atxy(board
, x
, y
)));
363 fprintf(f
, " %2d | ", y
);
364 for (int x
= 1; x
< board_size(board
) - 1; x
++) {
365 cprint(board
, coord_xy(board
, x
, y
), f
);
373 board_print_custom(struct board
*board
, FILE *f
, board_cprint cprint
)
375 fprintf(f
, "Move: % 3d Komi: %2.1f Handicap: %d Captures B: %d W: %d\n",
376 board
->moves
, board
->komi
, board
->handicap
,
377 board
->captures
[S_BLACK
], board
->captures
[S_WHITE
]);
378 board_print_top(board
, f
, 1 + !!cprint
);
379 for (int y
= board_size(board
) - 2; y
>= 1; y
--)
380 board_print_row(board
, y
, f
, cprint
);
381 board_print_bottom(board
, f
, 1 + !!cprint
);
386 cprint_group(struct board
*board
, coord_t c
, FILE *f
)
388 fprintf(f
, "%d ", group_base(group_at(board
, c
)));
392 board_print(struct board
*board
, FILE *f
)
394 board_print_custom(board
, f
, DEBUGL(6) ? cprint_group
: NULL
);
398 /* Update the probability distribution we maintain incrementally. */
400 board_gamma_update(struct board
*board
, coord_t coord
, enum stone color
)
403 assert(board
->gamma
);
404 /* Punch out invalid moves and moves filling our own eyes. */
405 if (board_at(board
, coord
) != S_NONE
406 || (board_is_eyelike(board
, &coord
, stone_other(color
))
407 && !trait_at(board
, coord
, color
).cap
)
408 || (board_is_one_point_eye(board
, &coord
, color
))) {
409 probdist_set(&board
->prob
[color
- 1], coord
, 0);
413 /* We just quickly replicate the general pattern matcher stuff
414 * here in the most bare-bone way. */
416 if (trait_at(board
, coord
, color
).cap
)
417 value
*= board
->gamma
->gamma
[FEAT_CAPTURE
][0];
418 if (trait_at(board
, coord
, stone_other(color
)).cap
419 && trait_at(board
, coord
, color
).safe
)
420 value
*= board
->gamma
->gamma
[FEAT_AESCAPE
][0];
421 if (!trait_at(board
, coord
, color
).safe
)
422 value
*= board
->gamma
->gamma
[FEAT_SELFATARI
][0];
423 probdist_set(&board
->prob
[color
- 1], coord
, value
);
427 /* Recompute some of the traits for given point from scratch. Note that
428 * some traits are updated incrementally elsewhere. */
430 board_trait_recompute(struct board
*board
, coord_t coord
)
433 trait_at(board
, coord
, S_BLACK
).safe
= board_safe_to_play(board
, coord
, S_BLACK
);
434 trait_at(board
, coord
, S_WHITE
).safe
= board_safe_to_play(board
, coord
, S_WHITE
);
436 fprintf(stderr
, "traits[%s:%s lib=%d] (black cap=%d safe=%d) (white cap=%d safe=%d)\n",
437 coord2sstr(coord
, board
), stone2str(board_at(board
, coord
)), immediate_liberty_count(board
, coord
),
438 trait_at(board
, coord
, S_BLACK
).cap
, trait_at(board
, coord
, S_BLACK
).safe
,
439 trait_at(board
, coord
, S_WHITE
).cap
, trait_at(board
, coord
, S_WHITE
).safe
);
442 board_gamma_update(board
, coord
, S_BLACK
);
443 board_gamma_update(board
, coord
, S_WHITE
);
446 /* Update board hash with given coordinate. */
447 static void profiling_noinline
448 board_hash_update(struct board
*board
, coord_t coord
, enum stone color
)
450 board
->hash
^= hash_at(board
, coord
, color
);
452 fprintf(stderr
, "board_hash_update(%d,%d,%d) ^ %"PRIhash
" -> %"PRIhash
"\n", color
, coord_x(coord
, board
), coord_y(coord
, board
), hash_at(board
, coord
, color
), board
->hash
);
454 #ifdef BOARD_SPATHASH
455 /* Gridcular metric is reflective, so we update all hashes
456 * of appropriate ditance in OUR circle. */
457 for (int d
= 1; d
<= BOARD_SPATHASH_MAXD
; d
++) {
458 for (int j
= ptind
[d
]; j
< ptind
[d
+ 1]; j
++) {
459 ptcoords_at(x
, y
, coord
, board
, j
);
460 /* We either changed from S_NONE to color
461 * or vice versa; doesn't matter. */
462 board
->spathash
[coord_xy(board
, x
, y
)][d
- 1][0] ^=
463 pthashes
[0][j
][color
] ^ pthashes
[0][j
][S_NONE
];
464 board
->spathash
[coord_xy(board
, x
, y
)][d
- 1][1] ^=
465 pthashes
[0][j
][stone_other(color
)] ^ pthashes
[0][j
][S_NONE
];
470 #if defined(BOARD_PAT3)
471 /* @color is not what we need in case of capture. */
472 enum stone new_color
= board_at(board
, coord
);
473 if (new_color
== S_NONE
)
474 board
->pat3
[coord
] = pattern3_hash(board
, coord
);
475 foreach_8neighbor(board
, coord
) { // internally, the loop uses fn__i=[0..7]
476 if (board_at(board
, c
) != S_NONE
)
478 board
->pat3
[c
] &= ~(3 << (fn__i
*2));
479 board
->pat3
[c
] |= new_color
<< (fn__i
*2);
481 if (board_at(board
, c
) != S_OFFBOARD
&& pattern3_hash(board
, c
) != board
->pat3
[c
]) {
482 board_print(board
, stderr
);
483 fprintf(stderr
, "%s->%s %x != %x (%d-%d:%d)\n", coord2sstr(coord
, board
), coord2sstr(c
, board
), pattern3_hash(board
, c
), board
->pat3
[c
], coord
, c
, fn__i
);
487 board_gamma_update(board
, c
, S_BLACK
);
488 board_gamma_update(board
, c
, S_WHITE
);
489 } foreach_8neighbor_end
;
493 /* Commit current board hash to history. */
494 static void profiling_noinline
495 board_hash_commit(struct board
*board
)
498 fprintf(stderr
, "board_hash_commit %"PRIhash
"\n", board
->hash
);
499 if (likely(board
->history_hash
[board
->hash
& history_hash_mask
]) == 0) {
500 board
->history_hash
[board
->hash
& history_hash_mask
] = board
->hash
;
502 hash_t i
= board
->hash
;
503 while (board
->history_hash
[i
& history_hash_mask
]) {
504 if (board
->history_hash
[i
& history_hash_mask
] == board
->hash
) {
506 fprintf(stderr
, "SUPERKO VIOLATION noted at %d,%d\n",
507 coord_x(board
->last_move
.coord
, board
), coord_y(board
->last_move
.coord
, board
));
508 board
->superko_violation
= true;
511 i
= history_hash_next(i
);
513 board
->history_hash
[i
& history_hash_mask
] = board
->hash
;
519 board_symmetry_update(struct board
*b
, struct board_symmetry
*symmetry
, coord_t c
)
521 if (likely(symmetry
->type
== SYM_NONE
)) {
522 /* Fully degenerated already. We do not support detection
523 * of restoring of symmetry, assuming that this is too rare
524 * a case to handle. */
528 int x
= coord_x(c
, b
), y
= coord_y(c
, b
), t
= board_size(b
) / 2;
529 int dx
= board_size(b
) - 1 - x
; /* for SYM_DOWN */
531 fprintf(stderr
, "SYMMETRY [%d,%d,%d,%d|%d=%d] update for %d,%d\n",
532 symmetry
->x1
, symmetry
->y1
, symmetry
->x2
, symmetry
->y2
,
533 symmetry
->d
, symmetry
->type
, x
, y
);
536 switch (symmetry
->type
) {
538 if (x
== t
&& y
== t
) {
539 /* Tengen keeps full symmetry. */
542 /* New symmetry now? */
544 symmetry
->type
= SYM_DIAG_UP
;
545 symmetry
->x1
= symmetry
->y1
= 1;
546 symmetry
->x2
= symmetry
->y2
= board_size(b
) - 1;
548 } else if (dx
== y
) {
549 symmetry
->type
= SYM_DIAG_DOWN
;
550 symmetry
->x1
= symmetry
->y1
= 1;
551 symmetry
->x2
= symmetry
->y2
= board_size(b
) - 1;
554 symmetry
->type
= SYM_HORIZ
;
556 symmetry
->y2
= board_size(b
) - 1;
559 symmetry
->type
= SYM_VERT
;
561 symmetry
->x2
= board_size(b
) - 1;
565 symmetry
->type
= SYM_NONE
;
566 symmetry
->x1
= symmetry
->y1
= 1;
567 symmetry
->x2
= symmetry
->y2
= board_size(b
) - 1;
593 fprintf(stderr
, "NEW SYMMETRY [%d,%d,%d,%d|%d=%d]\n",
594 symmetry
->x1
, symmetry
->y1
, symmetry
->x2
, symmetry
->y2
,
595 symmetry
->d
, symmetry
->type
);
602 board_handicap_stone(struct board
*board
, int x
, int y
, FILE *f
)
605 m
.color
= S_BLACK
; m
.coord
= coord_xy(board
, x
, y
);
607 board_play(board
, &m
);
608 /* Simulate white passing; otherwise, UCT search can get confused since
609 * tree depth parity won't match the color to move. */
612 char *str
= coord2str(m
.coord
, board
);
614 fprintf(stderr
, "choosing handicap %s (%d,%d)\n", str
, x
, y
);
615 fprintf(f
, "%s ", str
);
620 board_handicap(struct board
*board
, int stones
, FILE *f
)
622 int margin
= 3 + (board_size(board
) >= 13);
624 int mid
= board_size(board
) / 2;
625 int max
= board_size(board
) - 1 - margin
;
626 const int places
[][2] = {
627 { min
, min
}, { max
, max
}, { max
, min
}, { min
, max
},
628 { min
, mid
}, { max
, mid
},
629 { mid
, min
}, { mid
, max
},
633 board
->handicap
= stones
;
635 if (stones
== 5 || stones
== 7) {
636 board_handicap_stone(board
, mid
, mid
, f
);
641 for (i
= 0; i
< stones
; i
++)
642 board_handicap_stone(board
, places
[i
][0], places
[i
][1], f
);
646 static void __attribute__((noinline
))
647 check_libs_consistency(struct board
*board
, group_t g
)
651 struct group
*gi
= &board_group_info(board
, g
);
652 for (int i
= 0; i
< GROUP_KEEP_LIBS
; i
++)
653 if (gi
->lib
[i
] && board_at(board
, gi
->lib
[i
]) != S_NONE
) {
654 fprintf(stderr
, "BOGUS LIBERTY %s of group %d[%s]\n", coord2sstr(gi
->lib
[i
], board
), g
, coord2sstr(group_base(g
), board
));
661 board_capturable_add(struct board
*board
, group_t group
, coord_t lib
)
663 //fprintf(stderr, "group %s cap %s\n", coord2sstr(group, board), coord2sstr(lib, boarD));
665 /* Increase capturable count trait of my last lib. */
666 enum stone capturing_color
= stone_other(board_at(board
, group
));
667 assert(capturing_color
== S_BLACK
|| capturing_color
== S_WHITE
);
668 foreach_neighbor(board
, lib
, {
669 if (DEBUGL(8) && group_at(board
, c
) == group
)
670 fprintf(stderr
, "%s[%d] %s cap bump bc of %s(%d) member %s\n", coord2sstr(lib
, board
), trait_at(board
, lib
, capturing_color
).cap
, stone2str(capturing_color
), coord2sstr(group
, board
), board_group_info(board
, group
).libs
, coord2sstr(c
, board
));
671 trait_at(board
, lib
, capturing_color
).cap
+= (group_at(board
, c
) == group
);
673 board_trait_recompute(board
, lib
);
677 /* Update the list of capturable groups. */
679 assert(board
->clen
< board_size2(board
));
680 board
->c
[board
->clen
++] = group
;
684 board_capturable_rm(struct board
*board
, group_t group
, coord_t lib
)
686 //fprintf(stderr, "group %s nocap %s\n", coord2sstr(group, board), coord2sstr(lib, board));
688 /* Decrease capturable count trait of my previously-last lib. */
689 enum stone capturing_color
= stone_other(board_at(board
, group
));
690 assert(capturing_color
== S_BLACK
|| capturing_color
== S_WHITE
);
691 foreach_neighbor(board
, lib
, {
692 if (DEBUGL(8) && group_at(board
, c
) == group
)
693 fprintf(stderr
, "%s[%d] cap dump bc of %s(%d) member %s\n", coord2sstr(lib
, board
), trait_at(board
, lib
, capturing_color
).cap
, coord2sstr(group
, board
), board_group_info(board
, group
).libs
, coord2sstr(c
, board
));
694 trait_at(board
, lib
, capturing_color
).cap
-= (group_at(board
, c
) == group
);
696 board_trait_recompute(board
, lib
);
700 /* Update the list of capturable groups. */
701 for (int i
= 0; i
< board
->clen
; i
++) {
702 if (unlikely(board
->c
[i
] == group
)) {
703 board
->c
[i
] = board
->c
[--board
->clen
];
707 fprintf(stderr
, "rm of bad group %d\n", group_base(group
));
713 board_group_addlib(struct board
*board
, group_t group
, coord_t coord
)
716 fprintf(stderr
, "Group %d[%s] %d: Adding liberty %s\n",
717 group_base(group
), coord2sstr(group_base(group
), board
),
718 board_group_info(board
, group
).libs
, coord2sstr(coord
, board
));
721 check_libs_consistency(board
, group
);
723 struct group
*gi
= &board_group_info(board
, group
);
724 if (gi
->libs
< GROUP_KEEP_LIBS
) {
725 for (int i
= 0; i
< GROUP_KEEP_LIBS
; i
++) {
727 /* Seems extra branch just slows it down */
731 if (unlikely(gi
->lib
[i
] == coord
))
735 board_capturable_add(board
, group
, coord
);
736 else if (gi
->libs
== 1)
737 board_capturable_rm(board
, group
, gi
->lib
[0]);
738 gi
->lib
[gi
->libs
++] = coord
;
741 check_libs_consistency(board
, group
);
745 board_group_find_extra_libs(struct board
*board
, group_t group
, struct group
*gi
, coord_t avoid
)
747 /* Add extra liberty from the board to our liberty list. */
748 unsigned char watermark
[board_size2(board
) / 8];
749 memset(watermark
, 0, sizeof(watermark
));
750 #define watermark_get(c) (watermark[coord_raw(c) >> 3] & (1 << (coord_raw(c) & 7)))
751 #define watermark_set(c) watermark[coord_raw(c) >> 3] |= (1 << (coord_raw(c) & 7))
753 for (int i
= 0; i
< GROUP_KEEP_LIBS
- 1; i
++)
754 watermark_set(gi
->lib
[i
]);
755 watermark_set(avoid
);
757 foreach_in_group(board
, group
) {
759 foreach_neighbor(board
, coord2
, {
760 if (board_at(board
, c
) + watermark_get(c
) != S_NONE
)
763 gi
->lib
[gi
->libs
++] = c
;
764 if (unlikely(gi
->libs
>= GROUP_KEEP_LIBS
))
767 } foreach_in_group_end
;
773 board_group_rmlib(struct board
*board
, group_t group
, coord_t coord
)
776 fprintf(stderr
, "Group %d[%s] %d: Removing liberty %s\n",
777 group_base(group
), coord2sstr(group_base(group
), board
),
778 board_group_info(board
, group
).libs
, coord2sstr(coord
, board
));
781 struct group
*gi
= &board_group_info(board
, group
);
782 for (int i
= 0; i
< GROUP_KEEP_LIBS
; i
++) {
784 /* Seems extra branch just slows it down */
788 if (likely(gi
->lib
[i
] != coord
))
791 coord_t lib
= gi
->lib
[i
] = gi
->lib
[--gi
->libs
];
792 gi
->lib
[gi
->libs
] = 0;
794 check_libs_consistency(board
, group
);
796 /* Postpone refilling lib[] until we need to. */
797 assert(GROUP_REFILL_LIBS
> 1);
798 if (gi
->libs
> GROUP_REFILL_LIBS
)
800 if (gi
->libs
== GROUP_REFILL_LIBS
)
801 board_group_find_extra_libs(board
, group
, gi
, coord
);
804 board_capturable_add(board
, group
, gi
->lib
[0]);
805 else if (gi
->libs
== 0)
806 board_capturable_rm(board
, group
, lib
);
810 /* This is ok even if gi->libs < GROUP_KEEP_LIBS since we
811 * can call this multiple times per coord. */
812 check_libs_consistency(board
, group
);
817 /* This is a low-level routine that doesn't maintain consistency
818 * of all the board data structures. */
820 board_remove_stone(struct board
*board
, group_t group
, coord_t c
)
822 enum stone color
= board_at(board
, c
);
823 board_at(board
, c
) = S_NONE
;
824 group_at(board
, c
) = 0;
825 board_hash_update(board
, c
, color
);
827 /* We mark as cannot-capture now. If this is a ko/snapback,
828 * we will get incremented later in board_group_addlib(). */
829 trait_at(board
, c
, S_BLACK
).cap
= 0;
830 trait_at(board
, c
, S_WHITE
).cap
= 0;
831 /* However, we do decide safety statically; we might get
832 * over-paranoid, but in that case the neighbor loop for
833 * stones removed next will repair the flag. */
834 /* We must do this update after the loop when our neighbor count is correct. */
835 board_trait_recompute(board
, c
);
838 /* Increase liberties of surrounding groups */
840 foreach_neighbor(board
, coord
, {
841 dec_neighbor_count_at(board
, c
, color
);
842 board_trait_recompute(board
, c
);
843 group_t g
= group_at(board
, c
);
845 board_group_addlib(board
, g
, coord
);
849 fprintf(stderr
, "pushing free move [%d]: %d,%d\n", board
->flen
, coord_x(c
, board
), coord_y(c
, board
));
850 board
->f
[board
->flen
++] = coord_raw(c
);
853 static int profiling_noinline
854 board_group_capture(struct board
*board
, group_t group
)
858 foreach_in_group(board
, group
) {
859 board
->captures
[stone_other(board_at(board
, c
))]++;
860 board_remove_stone(board
, group
, c
);
862 } foreach_in_group_end
;
864 if (board_group_info(board
, group
).libs
== 1)
865 board_capturable_rm(board
, group
, board_group_info(board
, group
).lib
[0]);
866 memset(&board_group_info(board
, group
), 0, sizeof(struct group
));
872 static void profiling_noinline
873 add_to_group(struct board
*board
, group_t group
, coord_t prevstone
, coord_t coord
)
875 group_at(board
, coord
) = group
;
876 groupnext_at(board
, coord
) = groupnext_at(board
, prevstone
);
877 groupnext_at(board
, prevstone
) = coord_raw(coord
);
880 if (board_group_info(board
, group
).libs
== 1) {
881 /* Our group is temporarily in atari; make sure the capturable
882 * counts also correspond to the newly added stone before we
883 * start adding liberties again so bump-dump ops match. */
884 enum stone capturing_color
= stone_other(board_at(board
, group
));
885 assert(capturing_color
== S_BLACK
|| capturing_color
== S_WHITE
);
886 coord_t lib
= board_group_info(board
, group
).lib
[0];
887 if (coord_is_adjecent(lib
, coord
, board
)) {
888 if (DEBUGL(8)) fprintf(stderr
, "add_to_group %s: %s[%d] bump\n", coord2sstr(group
, board
), coord2sstr(lib
, board
), trait_at(board
, lib
, capturing_color
).cap
);
889 trait_at(board
, lib
, capturing_color
).cap
++;
890 board_trait_recompute(board
, lib
);
895 foreach_neighbor(board
, coord
, {
896 if (board_at(board
, c
) == S_NONE
)
897 board_group_addlib(board
, group
, c
);
901 fprintf(stderr
, "add_to_group: added (%d,%d ->) %d,%d (-> %d,%d) to group %d\n",
902 coord_x(prevstone
, board
), coord_y(prevstone
, board
),
903 coord_x(coord
, board
), coord_y(coord
, board
),
904 groupnext_at(board
, coord
) % board_size(board
), groupnext_at(board
, coord
) / board_size(board
),
908 static void profiling_noinline
909 merge_groups(struct board
*board
, group_t group_to
, group_t group_from
)
912 fprintf(stderr
, "board_play_raw: merging groups %d -> %d\n",
913 group_base(group_from
), group_base(group_to
));
914 struct group
*gi_from
= &board_group_info(board
, group_from
);
915 struct group
*gi_to
= &board_group_info(board
, group_to
);
917 /* We do this early before the group info is rewritten. */
918 if (gi_from
->libs
== 1)
919 board_capturable_rm(board
, group_from
, gi_from
->lib
[0]);
922 fprintf(stderr
,"---- (froml %d, tol %d)\n", gi_from
->libs
, gi_to
->libs
);
924 if (gi_to
->libs
< GROUP_KEEP_LIBS
) {
925 for (int i
= 0; i
< gi_from
->libs
; i
++) {
926 for (int j
= 0; j
< gi_to
->libs
; j
++)
927 if (gi_to
->lib
[j
] == gi_from
->lib
[i
])
929 if (gi_to
->libs
== 0)
930 board_capturable_add(board
, group_to
, gi_from
->lib
[i
]);
931 else if (gi_to
->libs
== 1)
932 board_capturable_rm(board
, group_to
, gi_to
->lib
[0]);
933 gi_to
->lib
[gi_to
->libs
++] = gi_from
->lib
[i
];
934 if (gi_to
->libs
>= GROUP_KEEP_LIBS
)
941 if (board_group_info(board
, group_to
).libs
== 1) {
942 /* Our group is currently in atari; make sure we properly
943 * count in even the neighbors from the other group in the
944 * capturable counter. */
945 enum stone capturing_color
= stone_other(board_at(board
, group_to
));
946 assert(capturing_color
== S_BLACK
|| capturing_color
== S_WHITE
);
947 coord_t lib
= board_group_info(board
, group_to
).lib
[0];
948 foreach_neighbor(board
, lib
, {
949 if (DEBUGL(8) && group_at(board
, c
) == group_from
) fprintf(stderr
, "%s[%d] cap bump\n", coord2sstr(lib
, board
), trait_at(board
, lib
, capturing_color
).cap
);
950 trait_at(board
, lib
, capturing_color
).cap
+= (group_at(board
, c
) == group_from
);
952 board_trait_recompute(board
, lib
);
956 coord_t last_in_group
;
957 foreach_in_group(board
, group_from
) {
959 group_at(board
, c
) = group_to
;
960 } foreach_in_group_end
;
961 groupnext_at(board
, last_in_group
) = groupnext_at(board
, group_base(group_to
));
962 groupnext_at(board
, group_base(group_to
)) = group_base(group_from
);
963 memset(gi_from
, 0, sizeof(struct group
));
966 fprintf(stderr
, "board_play_raw: merged group: %d\n",
967 group_base(group_to
));
970 static group_t profiling_noinline
971 new_group(struct board
*board
, coord_t coord
)
973 group_t group
= coord_raw(coord
);
974 struct group
*gi
= &board_group_info(board
, group
);
975 foreach_neighbor(board
, coord
, {
976 if (board_at(board
, c
) == S_NONE
)
977 /* board_group_addlib is ridiculously expensive for us */
978 #if GROUP_KEEP_LIBS < 4
979 if (gi
->libs
< GROUP_KEEP_LIBS
)
981 gi
->lib
[gi
->libs
++] = c
;
984 group_at(board
, coord
) = group
;
985 groupnext_at(board
, coord
) = 0;
988 board_capturable_add(board
, group
, gi
->lib
[0]);
989 check_libs_consistency(board
, group
);
992 fprintf(stderr
, "new_group: added %d,%d to group %d\n",
993 coord_x(coord
, board
), coord_y(coord
, board
),
999 static inline group_t
1000 play_one_neighbor(struct board
*board
,
1001 coord_t coord
, enum stone color
, enum stone other_color
,
1002 coord_t c
, group_t group
)
1004 enum stone ncolor
= board_at(board
, c
);
1005 group_t ngroup
= group_at(board
, c
);
1007 inc_neighbor_count_at(board
, c
, color
);
1008 /* We can be S_NONE, in that case we need to update the safety
1009 * trait since we might be left with only one liberty. */
1010 board_trait_recompute(board
, c
);
1015 board_group_rmlib(board
, ngroup
, coord
);
1017 fprintf(stderr
, "board_play_raw: reducing libs for group %d (%d:%d,%d)\n",
1018 group_base(ngroup
), ncolor
, color
, other_color
);
1020 if (ncolor
== color
&& ngroup
!= group
) {
1023 add_to_group(board
, group
, c
, coord
);
1025 merge_groups(board
, group
, ngroup
);
1027 } else if (ncolor
== other_color
) {
1029 struct group
*gi
= &board_group_info(board
, ngroup
);
1030 fprintf(stderr
, "testing captured group %d[%s]: ", group_base(ngroup
), coord2sstr(group_base(ngroup
), board
));
1031 for (int i
= 0; i
< GROUP_KEEP_LIBS
; i
++)
1032 fprintf(stderr
, "%s ", coord2sstr(gi
->lib
[i
], board
));
1033 fprintf(stderr
, "\n");
1035 if (unlikely(board_group_captured(board
, ngroup
)))
1036 board_group_capture(board
, ngroup
);
1041 /* We played on a place with at least one liberty. We will become a member of
1042 * some group for sure. */
1043 static group_t profiling_noinline
1044 board_play_outside(struct board
*board
, struct move
*m
, int f
)
1046 coord_t coord
= m
->coord
;
1047 enum stone color
= m
->color
;
1048 enum stone other_color
= stone_other(color
);
1051 board
->f
[f
] = board
->f
[--board
->flen
];
1053 fprintf(stderr
, "popping free move [%d->%d]: %d\n", board
->flen
, f
, board
->f
[f
]);
1055 #if defined(BOARD_TRAITS) && !defined(NDEBUG)
1056 /* Sanity check that cap matches reality. */
1059 foreach_neighbor(board
, coord
, {
1060 group_t g
= group_at(board
, c
);
1061 a
+= g
&& (board_at(board
, c
) == other_color
&& board_group_info(board
, g
).libs
== 1);
1063 assert(a
== trait_at(board
, coord
, color
).cap
);
1064 assert(board_safe_to_play(board
, coord
, color
) == trait_at(board
, coord
, color
).safe
);
1067 foreach_neighbor(board
, coord
, {
1068 group
= play_one_neighbor(board
, coord
, color
, other_color
, c
, group
);
1071 board_at(board
, coord
) = color
;
1072 if (unlikely(!group
))
1073 group
= new_group(board
, coord
);
1074 board_gamma_update(board
, coord
, S_BLACK
);
1075 board_gamma_update(board
, coord
, S_WHITE
);
1077 board
->last_move2
= board
->last_move
;
1078 board
->last_move
= *m
;
1080 board_hash_update(board
, coord
, color
);
1081 board_symmetry_update(board
, &board
->symmetry
, coord
);
1082 struct move ko
= { pass
, S_NONE
};
1088 /* We played in an eye-like shape. Either we capture at least one of the eye
1089 * sides in the process of playing, or return -1. */
1090 static int profiling_noinline
1091 board_play_in_eye(struct board
*board
, struct move
*m
, int f
)
1093 coord_t coord
= m
->coord
;
1094 enum stone color
= m
->color
;
1095 /* Check ko: Capture at a position of ko capture one move ago */
1096 if (unlikely(color
== board
->ko
.color
&& coord_eq(coord
, board
->ko
.coord
))) {
1098 fprintf(stderr
, "board_check: ko at %d,%d color %d\n", coord_x(coord
, board
), coord_y(coord
, board
), color
);
1100 } else if (DEBUGL(6)) {
1101 fprintf(stderr
, "board_check: no ko at %d,%d,%d - ko is %d,%d,%d\n",
1102 color
, coord_x(coord
, board
), coord_y(coord
, board
),
1103 board
->ko
.color
, coord_x(board
->ko
.coord
, board
), coord_y(board
->ko
.coord
, board
));
1106 struct move ko
= { pass
, S_NONE
};
1108 int captured_groups
= 0;
1110 foreach_neighbor(board
, coord
, {
1111 group_t g
= group_at(board
, c
);
1113 fprintf(stderr
, "board_check: group %d has %d libs\n",
1114 g
, board_group_info(board
, g
).libs
);
1115 captured_groups
+= (board_group_info(board
, g
).libs
== 1);
1118 if (likely(captured_groups
== 0)) {
1121 board_print(board
, stderr
);
1122 fprintf(stderr
, "board_check: one-stone suicide\n");
1128 /* We _will_ for sure capture something. */
1129 assert(trait_at(board
, coord
, color
).cap
> 0);
1130 assert(trait_at(board
, coord
, color
).safe
== board_safe_to_play(board
, coord
, color
));
1133 board
->f
[f
] = board
->f
[--board
->flen
];
1135 fprintf(stderr
, "popping free move [%d->%d]: %d\n", board
->flen
, f
, board
->f
[f
]);
1137 foreach_neighbor(board
, coord
, {
1138 inc_neighbor_count_at(board
, c
, color
);
1139 /* Originally, this could not have changed any trait
1140 * since no neighbors were S_NONE, however by now some
1141 * of them might be removed from the board. */
1142 board_trait_recompute(board
, c
);
1144 group_t group
= group_at(board
, c
);
1148 board_group_rmlib(board
, group
, coord
);
1150 fprintf(stderr
, "board_play_raw: reducing libs for group %d\n",
1153 if (board_group_captured(board
, group
)) {
1154 if (board_group_capture(board
, group
) == 1) {
1155 /* If we captured multiple groups at once,
1156 * we can't be fighting ko so we don't need
1157 * to check for that. */
1158 ko
.color
= stone_other(color
);
1160 board
->last_ko
= ko
;
1161 board
->last_ko_age
= board
->moves
;
1163 fprintf(stderr
, "guarding ko at %d,%s\n", ko
.color
, coord2sstr(ko
.coord
, board
));
1168 board_at(board
, coord
) = color
;
1169 group_t group
= new_group(board
, coord
);
1170 board_gamma_update(board
, coord
, S_BLACK
);
1171 board_gamma_update(board
, coord
, S_WHITE
);
1173 board
->last_move2
= board
->last_move
;
1174 board
->last_move
= *m
;
1176 board_hash_update(board
, coord
, color
);
1177 board_hash_commit(board
);
1178 board_symmetry_update(board
, &board
->symmetry
, coord
);
1184 static int __attribute__((flatten
))
1185 board_play_f(struct board
*board
, struct move
*m
, int f
)
1188 fprintf(stderr
, "board_play(): ---- Playing %d,%d\n", coord_x(m
->coord
, board
), coord_y(m
->coord
, board
));
1190 if (likely(!board_is_eyelike(board
, &m
->coord
, stone_other(m
->color
)))) {
1191 /* NOT playing in an eye. Thus this move has to succeed. (This
1192 * is thanks to New Zealand rules. Otherwise, multi-stone
1193 * suicide might fail.) */
1194 group_t group
= board_play_outside(board
, m
, f
);
1195 if (unlikely(board_group_captured(board
, group
))) {
1196 board_group_capture(board
, group
);
1198 board_hash_commit(board
);
1201 return board_play_in_eye(board
, m
, f
);
1206 board_play(struct board
*board
, struct move
*m
)
1208 if (unlikely(is_pass(m
->coord
) || is_resign(m
->coord
))) {
1209 struct move nomove
= { pass
, S_NONE
};
1211 board
->last_move2
= board
->last_move
;
1212 board
->last_move
= *m
;
1217 for (f
= 0; f
< board
->flen
; f
++)
1218 if (board
->f
[f
] == coord_raw(m
->coord
))
1219 return board_play_f(board
, m
, f
);
1222 fprintf(stderr
, "board_check: stone exists\n");
1228 board_try_random_move(struct board
*b
, enum stone color
, coord_t
*coord
, int f
, ppr_permit permit
, void *permit_data
)
1230 coord_raw(*coord
) = b
->f
[f
];
1231 if (unlikely(is_pass(*coord
)))
1233 struct move m
= { *coord
, color
};
1235 fprintf(stderr
, "trying random move %d: %d,%d\n", f
, coord_x(*coord
, b
), coord_y(*coord
, b
));
1236 return (likely(!board_is_one_point_eye(b
, coord
, color
)) /* bad idea to play into one, usually */
1237 && board_is_valid_move(b
, &m
)
1238 && (!permit
|| permit(permit_data
, b
, &m
))
1239 && likely(board_play_f(b
, &m
, f
) >= 0));
1243 board_play_random(struct board
*b
, enum stone color
, coord_t
*coord
, ppr_permit permit
, void *permit_data
)
1245 int base
= fast_random(b
->flen
);
1246 coord_pos(*coord
, base
, b
);
1247 if (likely(board_try_random_move(b
, color
, coord
, base
, permit
, permit_data
)))
1251 for (f
= base
+ 1; f
< b
->flen
; f
++)
1252 if (board_try_random_move(b
, color
, coord
, f
, permit
, permit_data
))
1254 for (f
= 0; f
< base
; f
++)
1255 if (board_try_random_move(b
, color
, coord
, f
, permit
, permit_data
))
1259 struct move m
= { pass
, color
};
1265 board_is_false_eyelike(struct board
*board
, coord_t
*coord
, enum stone eye_color
)
1267 enum stone color_diag_libs
[S_MAX
] = {0, 0, 0, 0};
1269 /* XXX: We attempt false eye detection but we will yield false
1270 * positives in case of http://senseis.xmp.net/?TwoHeadedDragon :-( */
1272 foreach_diag_neighbor(board
, *coord
) {
1273 color_diag_libs
[(enum stone
) board_at(board
, c
)]++;
1274 } foreach_diag_neighbor_end
;
1275 /* For false eye, we need two enemy stones diagonally in the
1276 * middle of the board, or just one enemy stone at the edge
1277 * or in the corner. */
1278 color_diag_libs
[stone_other(eye_color
)] += !!color_diag_libs
[S_OFFBOARD
];
1279 return color_diag_libs
[stone_other(eye_color
)] >= 2;
1283 board_is_one_point_eye(struct board
*board
, coord_t
*coord
, enum stone eye_color
)
1285 return board_is_eyelike(board
, coord
, eye_color
)
1286 && !board_is_false_eyelike(board
, coord
, eye_color
);
1290 board_get_one_point_eye(struct board
*board
, coord_t
*coord
)
1292 if (board_is_one_point_eye(board
, coord
, S_WHITE
))
1294 else if (board_is_one_point_eye(board
, coord
, S_BLACK
))
1302 board_fast_score(struct board
*board
)
1305 memset(scores
, 0, sizeof(scores
));
1307 foreach_point(board
) {
1308 enum stone color
= board_at(board
, c
);
1309 if (color
== S_NONE
)
1310 color
= board_get_one_point_eye(board
, &c
);
1312 // fprintf(stderr, "%d, %d ++%d = %d\n", coord_x(c, board), coord_y(c, board), color, scores[color]);
1313 } foreach_point_end
;
1315 return board
->komi
+ board
->handicap
+ scores
[S_WHITE
] - scores
[S_BLACK
];
1318 /* Owner map: 0: undecided; 1: black; 2: white; 3: dame */
1320 /* One flood-fill iteration; returns true if next iteration
1323 board_tromp_taylor_iter(struct board
*board
, int *ownermap
)
1325 bool needs_update
= false;
1326 foreach_point(board
) {
1327 /* Ignore occupied and already-dame positions. */
1328 if (board_at(board
, c
) != S_NONE
|| ownermap
[c
] == 3)
1330 /* Count neighbors. */
1332 foreach_neighbor(board
, c
, {
1335 /* If we have neighbors of both colors, or dame,
1336 * we are dame too. */
1337 if ((nei
[1] && nei
[2]) || nei
[3]) {
1339 /* Speed up the propagation. */
1340 foreach_neighbor(board
, c
, {
1341 if (board_at(board
, c
) == S_NONE
)
1344 needs_update
= true;
1347 /* If we have neighbors of one color, we are owned
1348 * by that color, too. */
1349 if (!ownermap
[c
] && (nei
[1] || nei
[2])) {
1350 int newowner
= nei
[1] ? 1 : 2;
1351 ownermap
[c
] = newowner
;
1352 /* Speed up the propagation. */
1353 foreach_neighbor(board
, c
, {
1354 if (board_at(board
, c
) == S_NONE
&& !ownermap
[c
])
1355 ownermap
[c
] = newowner
;
1357 needs_update
= true;
1360 } foreach_point_end
;
1361 return needs_update
;
1364 /* Tromp-Taylor Counting */
1366 board_official_score(struct board
*board
, struct move_queue
*q
)
1369 /* A point P, not colored C, is said to reach C, if there is a path of
1370 * (vertically or horizontally) adjacent points of P's color from P to
1371 * a point of color C.
1373 * A player's score is the number of points of her color, plus the
1374 * number of empty points that reach only her color. */
1376 int ownermap
[board_size2(board
)];
1378 const int o
[4] = {0, 1, 2, 0};
1379 foreach_point(board
) {
1380 ownermap
[c
] = o
[board_at(board
, c
)];
1381 s
[board_at(board
, c
)]++;
1382 } foreach_point_end
;
1385 /* Process dead groups. */
1386 for (int i
= 0; i
< q
->moves
; i
++) {
1387 foreach_in_group(board
, q
->move
[i
]) {
1388 enum stone color
= board_at(board
, c
);
1389 ownermap
[c
] = o
[stone_other(color
)];
1390 s
[color
]--; s
[stone_other(color
)]++;
1391 } foreach_in_group_end
;
1395 /* We need to special-case empty board. */
1396 if (!s
[S_BLACK
] && !s
[S_WHITE
])
1397 return board
->komi
+ board
->handicap
;
1399 while (board_tromp_taylor_iter(board
, ownermap
))
1400 /* Flood-fill... */;
1403 memset(scores
, 0, sizeof(scores
));
1405 foreach_point(board
) {
1406 assert(board_at(board
, c
) == S_OFFBOARD
|| ownermap
[c
] != 0);
1407 if (ownermap
[c
] == 3)
1409 scores
[ownermap
[c
]]++;
1410 } foreach_point_end
;
1412 return board
->komi
+ board
->handicap
+ scores
[S_WHITE
] - scores
[S_BLACK
];