23 /* Sleep 5 seconds after a game ends to give time to kill the program. */
24 #define GAME_OVER_SLEEP 5
27 gtp_prefix(char prefix
, int id
)
29 if (id
== NO_REPLY
) return;
31 printf("%c%d ", prefix
, id
);
33 printf("%c ", prefix
);
44 gtp_output(char prefix
, int id
, va_list params
)
46 if (id
== NO_REPLY
) return;
47 gtp_prefix(prefix
, id
);
49 while ((s
= va_arg(params
, char *))) {
57 gtp_reply(int id
, ...)
61 gtp_output('=', id
, params
);
66 gtp_error(int id
, ...)
70 gtp_output('?', id
, params
);
74 /* List of known gtp commands. The internal command pachi-genmoves is not exported,
75 * it should only be used between master and slaves of the distributed engine. */
76 static char *known_commands
=
90 "kgs-genmove_cleanup\n"
92 "place_free_handicap\n"
101 /* Return true if cmd is a valid gtp command. */
103 gtp_is_valid(char *cmd
)
105 if (!cmd
|| !*cmd
) return false;
106 char *s
= strcasestr(known_commands
, cmd
);
107 if (!s
) return false;
109 int len
= strlen(cmd
);
110 return s
[len
] == '\0' || s
[len
] == '\n';
113 /* XXX: THIS IS TOTALLY INSECURE!!!!
114 * Even basic input checking is missing. */
117 gtp_parse(struct board
*board
, struct engine
*engine
, struct time_info
*ti
, char *buf
)
119 #define next_tok(to_) \
121 next = next + strcspn(next, " \t\r\n"); \
124 next += strspn(next, " \t\r\n"); \
127 if (strchr(buf
, '#'))
128 *strchr(buf
, '#') = 0;
130 char *cmd
, *next
= buf
;
142 if (!strcasecmp(cmd
, "protocol_version")) {
143 gtp_reply(id
, "2", NULL
);
146 } else if (!strcasecmp(cmd
, "name")) {
148 gtp_reply(id
, "Pachi ", engine
->name
, NULL
);
151 } else if (!strcasecmp(cmd
, "echo")) {
152 gtp_reply(id
, next
, NULL
);
155 } else if (!strcasecmp(cmd
, "version")) {
156 gtp_reply(id
, PACHI_VERSION
, ": ", engine
->comment
, NULL
);
159 } else if (!strcasecmp(cmd
, "list_commands")) {
160 gtp_reply(id
, known_commands
, NULL
);
164 if (engine
->notify
) {
166 enum parse_code c
= engine
->notify(engine
, board
, id
, cmd
, next
, &reply
);
167 if (c
== P_NOREPLY
) {
169 } else if (c
== P_DONE_OK
) {
170 gtp_reply(id
, reply
, NULL
);
172 } else if (c
== P_DONE_ERROR
) {
173 gtp_error(id
, reply
, NULL
);
174 /* This is an internal error for the engine, but
175 * it is still OK from main's point of view. */
177 } else if (c
!= P_OK
) {
182 if (!strcasecmp(cmd
, "quit")) {
186 } else if (!strcasecmp(cmd
, "boardsize")) {
189 int size
= atoi(arg
);
190 if (size
< 1 || size
> BOARD_MAX_SIZE
) {
191 gtp_error(id
, "illegal board size", NULL
);
194 board_resize(board
, size
);
197 } else if (!strcasecmp(cmd
, "clear_board") || !strcasecmp(cmd
, "kgs-game_over")) {
200 board_print(board
, stderr
);
201 if (!strcasecmp(cmd
, "kgs-game_over")) {
203 fprintf(stderr
, "game is over\n");
204 /* Sleep before replying, so that kgs doesn't
205 * start another game immediately. */
206 sleep(GAME_OVER_SLEEP
);
209 return P_ENGINE_RESET
;
211 } else if (!strcasecmp(cmd
, "komi")) {
214 sscanf(arg
, "%f", &board
->komi
);
217 board_print(board
, stderr
);
220 } else if (!strcasecmp(cmd
, "kgs-rules")) {
223 if (!strcasecmp(arg
, "japanese")) {
224 board
->rules
= RULES_JAPANESE
;
225 } else if (!strcasecmp(arg
, "chinese")) {
226 board
->rules
= RULES_CHINESE
;
227 } else if (!strcasecmp(arg
, "aga")) {
228 board
->rules
= RULES_AGA
;
229 } else if (!strcasecmp(arg
, "new_zealand")) {
230 board
->rules
= RULES_NEW_ZEALAND
;
232 gtp_error(id
, "unknown rules", NULL
);
237 } else if (!strcasecmp(cmd
, "play")) {
242 m
.color
= str2stone(arg
);
244 coord_t
*c
= str2coord(arg
, board_size(board
));
245 m
.coord
= *c
; coord_done(c
);
249 fprintf(stderr
, "got move %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
251 // This is where kgs starts the timer, not at genmove!
252 time_start_timer(&ti
[stone_other(m
.color
)]);
254 if (engine
->notify_play
)
255 reply
= engine
->notify_play(engine
, board
, &m
);
256 if (board_play(board
, &m
) < 0) {
258 fprintf(stderr
, "! ILLEGAL MOVE %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
259 board_print(board
, stderr
);
261 gtp_error(id
, "illegal move", NULL
);
264 board_print_custom(board
, stderr
, engine
->printhook
);
265 gtp_reply(id
, reply
, NULL
);
268 } else if (!strcasecmp(cmd
, "genmove") || !strcasecmp(cmd
, "kgs-genmove_cleanup")) {
271 enum stone color
= str2stone(arg
);
274 if (!ti
[color
].len
.t
.timer_start
) {
275 /* First game move. */
276 time_start_timer(&ti
[color
]);
280 /* We have an fbook, check if we cannot make
281 * a move along it right away. */
282 hash_t hi
= board
->hash
;
284 while (!is_pass(board
->fbook
->moves
[hi
& fbook_hash_mask
])) {
285 if (board
->fbook
->hashes
[hi
& fbook_hash_mask
] == board
->hash
) {
286 cf
= board
->fbook
->moves
[hi
& fbook_hash_mask
];
293 fprintf(stderr
, "fbook match %"PRIhash
":%"PRIhash
"\n", board
->hash
, board
->hash
& fbook_hash_mask
);
296 /* No match, also prevent further fbook usage
297 * until the next clear_board. */
298 fbook_done(board
->fbook
);
303 if (!c
) c
= engine
->genmove(engine
, board
, &ti
[color
], color
, !strcasecmp(cmd
, "kgs-genmove_cleanup"));
305 struct move m
= { *c
, color
};
306 if (board_play(board
, &m
) < 0) {
307 fprintf(stderr
, "Attempted to generate an illegal move: [%s, %s]\n", coord2sstr(m
.coord
, board
), stone2str(m
.color
));
310 char *str
= coord2str(*c
, board
);
312 fprintf(stderr
, "playing move %s\n", str
);
314 board_print_custom(board
, stderr
, engine
->printhook
);
316 gtp_reply(id
, str
, NULL
);
317 free(str
); coord_done(c
);
319 /* Account for spent time. If our GTP peer keeps our clock, this will
320 * be overriden by next time_left GTP command properly. */
321 /* (XXX: Except if we pass to byoyomi and the peer doesn't, but that
322 * should be absolutely rare situation and we will just spend a little
323 * less time than we could on next few moves.) */
324 if (ti
[color
].period
!= TT_NULL
&& ti
[color
].dim
== TD_WALLTIME
)
325 time_sub(&ti
[color
], time_now() - ti
[color
].len
.t
.timer_start
, true);
327 } else if (!strcasecmp(cmd
, "pachi-genmoves") || !strcasecmp(cmd
, "pachi-genmoves_cleanup")) {
330 enum stone color
= str2stone(arg
);
334 char *reply
= engine
->genmoves(engine
, board
, &ti
[color
], color
, next
,
335 !strcasecmp(cmd
, "pachi-genmoves_cleanup"),
336 &stats
, &stats_size
);
338 gtp_error(id
, "genmoves error", NULL
);
342 fprintf(stderr
, "proposing moves %s\n", reply
);
344 board_print_custom(board
, stderr
, engine
->printhook
);
345 gtp_reply(id
, reply
, NULL
);
346 if (stats_size
> 0) {
347 double start
= time_now();
348 fwrite(stats
, 1, stats_size
, stdout
);
351 fprintf(stderr
, "sent reply %d bytes in %.4fms\n",
352 stats_size
, (time_now() - start
)*1000);
355 } else if (!strcasecmp(cmd
, "set_free_handicap")) {
362 coord_t
*c
= str2coord(arg
, board_size(board
));
363 m
.coord
= *c
; coord_done(c
);
365 fprintf(stderr
, "setting handicap %d,%d\n", coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
367 if (board_play(board
, &m
) < 0) {
369 fprintf(stderr
, "! ILLEGAL MOVE %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
370 gtp_error(id
, "illegal move", NULL
);
376 board_print(board
, stderr
);
379 /* TODO: Engine should choose free handicap; however, it tends to take
380 * overly long to think it all out, and unless it's clever its
381 * handicap stones won't be of much help. ;-) */
382 } else if (!strcasecmp(cmd
, "place_free_handicap")
383 || !strcasecmp(cmd
, "fixed_handicap")) {
386 int stones
= atoi(arg
);
389 board_handicap(board
, stones
, id
== NO_REPLY
? NULL
: stdout
);
391 board_print(board
, stderr
);
392 if (id
== NO_REPLY
) return P_OK
;
396 } else if (!strcasecmp(cmd
, "final_score")) {
397 struct move_queue q
= { .moves
= 0 };
398 if (engine
->dead_group_list
)
399 engine
->dead_group_list(engine
, board
, &q
);
400 float score
= board_official_score(board
, &q
);
403 fprintf(stderr
, "counted score %.1f\n", score
);
405 gtp_reply(id
, "0", NULL
);
406 } else if (score
> 0) {
407 snprintf(str
, 64, "W+%.1f", score
);
408 gtp_reply(id
, str
, NULL
);
410 snprintf(str
, 64, "B+%.1f", -score
);
411 gtp_reply(id
, str
, NULL
);
414 /* XXX: This is a huge hack. */
415 } else if (!strcasecmp(cmd
, "final_status_list")) {
416 if (id
== NO_REPLY
) return P_OK
;
419 struct move_queue q
= { .moves
= 0 };
420 if (engine
->dead_group_list
)
421 engine
->dead_group_list(engine
, board
, &q
);
422 /* else we return empty list - i.e. engine not supporting
423 * this assumes all stones alive at the game end. */
424 if (!strcasecmp(arg
, "dead")) {
426 for (unsigned int i
= 0; i
< q
.moves
; i
++) {
427 foreach_in_group(board
, q
.move
[i
]) {
428 printf("%s ", coord2sstr(c
, board
));
429 } foreach_in_group_end
;
435 } else if (!strcasecmp(arg
, "seki") || !strcasecmp(arg
, "alive")) {
437 bool printed_group
= false;
438 foreach_point(board
) { // foreach_group, effectively
439 group_t g
= group_at(board
, c
);
440 if (!g
|| g
!= c
) continue;
442 for (unsigned int i
= 0; i
< q
.moves
; i
++) {
446 foreach_in_group(board
, g
) {
447 printf("%s ", coord2sstr(c
, board
));
448 } foreach_in_group_end
;
450 printed_group
= true;
457 gtp_error(id
, "illegal status specifier", NULL
);
460 /* Custom commands for handling UCT opening tbook */
461 } else if (!strcasecmp(cmd
, "uct_gentbook")) {
462 /* Board must be initialized properly, as if for genmove;
463 * makes sense only as 'uct_gentbook b'. */
466 enum stone color
= str2stone(arg
);
467 if (uct_gentbook(engine
, board
, &ti
[color
], color
))
470 gtp_error(id
, "error generating tbook", NULL
);
472 } else if (!strcasecmp(cmd
, "uct_dumptbook")) {
475 enum stone color
= str2stone(arg
);
476 uct_dumptbook(engine
, board
, color
);
479 } else if (!strcasecmp(cmd
, "uct_evaluate")) {
482 enum stone color
= str2stone(arg
);
485 /* Iterate through the list of all free coordinates
486 * and call uct_evaluate() for each. uct_evaluate()
487 * will throw NAN in case of invalid moves and such. */
488 for (int i
= 0; i
< board
->flen
; i
++) {
489 float val
= uct_evaluate(engine
, board
, &ti
[color
], board
->f
[i
], color
);
492 printf("%s %1.3f\n", coord2sstr(board
->f
[i
], board
), (double) val
);
496 } else if (!strcasecmp(cmd
, "pachi-result")) {
497 /* More detailed result of the last genmove. */
498 /* For UCT, the output format is: = color move playouts winrate dynkomi */
501 reply
= engine
->result(engine
, board
);
503 gtp_reply(id
, reply
, NULL
);
505 gtp_error(id
, "unknown pachi-result command", NULL
);
507 } else if (!strcasecmp(cmd
, "kgs-chat")) {
516 reply
= engine
->chat(engine
, board
, msg
);
518 gtp_reply(id
, reply
, NULL
);
520 gtp_error(id
, "unknown kgs-chat command", NULL
);
522 } else if (!strcasecmp(cmd
, "time_left")) {
525 enum stone color
= str2stone(arg
);
527 int time
= atoi(arg
);
529 int stones
= atoi(arg
);
530 if (!ti
[color
].ignore_gtp
) {
531 time_left(&ti
[color
], time
, stones
);
533 if (DEBUGL(2)) fprintf(stderr
, "ignored time info\n");
538 } else if (!strcasecmp(cmd
, "time_settings") || !strcasecmp(cmd
, "kgs-time_settings")) {
541 if (!strcasecmp(cmd
, "kgs-time_settings")) {
542 next_tok(time_system
);
544 time_system
= "canadian";
547 int main_time
= 0, byoyomi_time
= 0, byoyomi_stones
= 0, byoyomi_periods
= 0;
548 if (!strcasecmp(time_system
, "none")) {
550 } else if (!strcasecmp(time_system
, "absolute")) {
552 main_time
= atoi(arg
);
553 } else if (!strcasecmp(time_system
, "byoyomi")) {
555 main_time
= atoi(arg
);
557 byoyomi_time
= atoi(arg
);
559 byoyomi_periods
= atoi(arg
);
560 } else if (!strcasecmp(time_system
, "canadian")) {
562 main_time
= atoi(arg
);
564 byoyomi_time
= atoi(arg
);
566 byoyomi_stones
= atoi(arg
);
570 fprintf(stderr
, "time_settings %d %d/%d*%d\n",
571 main_time
, byoyomi_time
, byoyomi_stones
, byoyomi_periods
);
572 if (!ti
[S_BLACK
].ignore_gtp
) {
573 time_settings(&ti
[S_BLACK
], main_time
, byoyomi_time
, byoyomi_stones
, byoyomi_periods
);
574 ti
[S_WHITE
] = ti
[S_BLACK
];
576 if (DEBUGL(1)) fprintf(stderr
, "ignored time info\n");
582 gtp_error(id
, "unknown command", NULL
);
583 return P_UNKNOWN_COMMAND
;