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
=
91 "kgs-genmove_cleanup\n"
93 "pachi-genmoves_cleanup\n"
95 "place_free_handicap\n"
110 /* Return true if cmd is a valid gtp command. */
112 gtp_is_valid(char *cmd
)
114 if (!cmd
|| !*cmd
) return false;
115 char *s
= strcasestr(known_commands
, cmd
);
116 if (!s
) return false;
117 if (s
!= known_commands
&& s
[-1] != '\n') return false;
119 int len
= strlen(cmd
);
120 return s
[len
] == '\0' || s
[len
] == '\n';
123 /* XXX: THIS IS TOTALLY INSECURE!!!!
124 * Even basic input checking is missing. */
127 gtp_parse(struct board
*board
, struct engine
*engine
, struct time_info
*ti
, char *buf
)
129 #define next_tok(to_) \
131 next = next + strcspn(next, " \t\r\n"); \
134 next += strspn(next, " \t\r\n"); \
137 if (strchr(buf
, '#'))
138 *strchr(buf
, '#') = 0;
140 char *cmd
, *next
= buf
;
152 if (!strcasecmp(cmd
, "protocol_version")) {
153 gtp_reply(id
, "2", NULL
);
156 } else if (!strcasecmp(cmd
, "name")) {
158 gtp_reply(id
, "Pachi ", engine
->name
, NULL
);
161 } else if (!strcasecmp(cmd
, "echo")) {
162 gtp_reply(id
, next
, NULL
);
165 } else if (!strcasecmp(cmd
, "version")) {
166 gtp_reply(id
, PACHI_VERSION
, ": ", engine
->comment
, NULL
);
169 } else if (!strcasecmp(cmd
, "list_commands")) {
170 gtp_reply(id
, known_commands
, NULL
);
173 } else if (!strcasecmp(cmd
, "known_command")) {
176 if (gtp_is_valid(arg
)) {
177 gtp_reply(id
, "true", NULL
);
179 gtp_reply(id
, "false", NULL
);
184 if (engine
->notify
&& gtp_is_valid(cmd
)) {
186 enum parse_code c
= engine
->notify(engine
, board
, id
, cmd
, next
, &reply
);
187 if (c
== P_NOREPLY
) {
189 } else if (c
== P_DONE_OK
) {
190 gtp_reply(id
, reply
, NULL
);
192 } else if (c
== P_DONE_ERROR
) {
193 gtp_error(id
, reply
, NULL
);
194 /* This is an internal error for the engine, but
195 * it is still OK from main's point of view. */
197 } else if (c
!= P_OK
) {
202 if (!strcasecmp(cmd
, "quit")) {
206 } else if (!strcasecmp(cmd
, "boardsize")) {
209 int size
= atoi(arg
);
210 if (size
< 1 || size
> BOARD_MAX_SIZE
) {
211 gtp_error(id
, "illegal board size", NULL
);
214 board_resize(board
, size
);
218 } else if (!strcasecmp(cmd
, "clear_board")) {
220 if (DEBUGL(1) && debug_boardprint
)
221 board_print(board
, stderr
);
223 return P_ENGINE_RESET
;
225 } else if (!strcasecmp(cmd
, "kgs-game_over")) {
226 /* The game may not be really over, just adjourned.
227 * Do not clear the board to avoid illegal moves
228 * if the game is resumed immediately after. KGS
229 * may start directly with genmove on resumption. */
231 fprintf(stderr
, "game is over\n");
234 /* Sleep before replying, so that kgs doesn't
235 * start another game immediately. */
236 sleep(GAME_OVER_SLEEP
);
239 } else if (!strcasecmp(cmd
, "komi")) {
242 sscanf(arg
, PRIfloating
, &board
->komi
);
244 if (DEBUGL(1) && debug_boardprint
)
245 board_print(board
, stderr
);
248 } else if (!strcasecmp(cmd
, "kgs-rules")) {
251 if (!strcasecmp(arg
, "japanese")) {
252 board
->rules
= RULES_JAPANESE
;
253 } else if (!strcasecmp(arg
, "chinese")) {
254 board
->rules
= RULES_CHINESE
;
255 } else if (!strcasecmp(arg
, "aga")) {
256 board
->rules
= RULES_AGA
;
257 } else if (!strcasecmp(arg
, "new_zealand")) {
258 board
->rules
= RULES_NEW_ZEALAND
;
260 gtp_error(id
, "unknown rules", NULL
);
265 } else if (!strcasecmp(cmd
, "play")) {
270 m
.color
= str2stone(arg
);
272 coord_t
*c
= str2coord(arg
, board_size(board
));
273 m
.coord
= *c
; coord_done(c
);
277 fprintf(stderr
, "got move %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
279 // This is where kgs starts the timer, not at genmove!
280 time_start_timer(&ti
[stone_other(m
.color
)]);
282 if (engine
->notify_play
)
283 reply
= engine
->notify_play(engine
, board
, &m
);
284 if (board_play(board
, &m
) < 0) {
286 fprintf(stderr
, "! ILLEGAL MOVE %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
287 board_print(board
, stderr
);
289 gtp_error(id
, "illegal move", NULL
);
291 if (DEBUGL(1) && debug_boardprint
)
292 board_print_custom(board
, stderr
, engine
->printhook
);
293 gtp_reply(id
, reply
, NULL
);
296 } else if (!strcasecmp(cmd
, "genmove") || !strcasecmp(cmd
, "kgs-genmove_cleanup")) {
299 enum stone color
= str2stone(arg
);
302 if (!ti
[color
].len
.t
.timer_start
) {
303 /* First game move. */
304 time_start_timer(&ti
[color
]);
309 cf
= fbook_check(board
);
313 c
= engine
->genmove(engine
, board
, &ti
[color
], color
, !strcasecmp(cmd
, "kgs-genmove_cleanup"));
315 struct move m
= { *c
, color
};
316 if (board_play(board
, &m
) < 0) {
317 fprintf(stderr
, "Attempted to generate an illegal move: [%s, %s]\n", coord2sstr(m
.coord
, board
), stone2str(m
.color
));
320 char *str
= coord2str(*c
, board
);
322 fprintf(stderr
, "playing move %s\n", str
);
323 if (DEBUGL(1) && debug_boardprint
) {
324 board_print_custom(board
, stderr
, engine
->printhook
);
326 gtp_reply(id
, str
, NULL
);
327 free(str
); coord_done(c
);
329 /* Account for spent time. If our GTP peer keeps our clock, this will
330 * be overriden by next time_left GTP command properly. */
331 /* (XXX: Except if we pass to byoyomi and the peer doesn't, but that
332 * should be absolutely rare situation and we will just spend a little
333 * less time than we could on next few moves.) */
334 if (ti
[color
].period
!= TT_NULL
&& ti
[color
].dim
== TD_WALLTIME
)
335 time_sub(&ti
[color
], time_now() - ti
[color
].len
.t
.timer_start
, true);
337 } else if (!strcasecmp(cmd
, "pachi-genmoves") || !strcasecmp(cmd
, "pachi-genmoves_cleanup")) {
340 enum stone color
= str2stone(arg
);
344 char *reply
= engine
->genmoves(engine
, board
, &ti
[color
], color
, next
,
345 !strcasecmp(cmd
, "pachi-genmoves_cleanup"),
346 &stats
, &stats_size
);
348 gtp_error(id
, "genmoves error", NULL
);
352 fprintf(stderr
, "proposing moves %s\n", reply
);
353 if (DEBUGL(4) && debug_boardprint
)
354 board_print_custom(board
, stderr
, engine
->printhook
);
355 gtp_reply(id
, reply
, NULL
);
356 if (stats_size
> 0) {
357 double start
= time_now();
358 fwrite(stats
, 1, stats_size
, stdout
);
361 fprintf(stderr
, "sent reply %d bytes in %.4fms\n",
362 stats_size
, (time_now() - start
)*1000);
365 } else if (!strcasecmp(cmd
, "set_free_handicap")) {
372 coord_t
*c
= str2coord(arg
, board_size(board
));
373 m
.coord
= *c
; coord_done(c
);
375 fprintf(stderr
, "setting handicap %d,%d\n", coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
377 if (board_play(board
, &m
) < 0) {
379 fprintf(stderr
, "! ILLEGAL MOVE %d,%d,%d\n", m
.color
, coord_x(m
.coord
, board
), coord_y(m
.coord
, board
));
380 gtp_error(id
, "illegal move", NULL
);
385 if (DEBUGL(1) && debug_boardprint
)
386 board_print(board
, stderr
);
389 /* TODO: Engine should choose free handicap; however, it tends to take
390 * overly long to think it all out, and unless it's clever its
391 * handicap stones won't be of much help. ;-) */
392 } else if (!strcasecmp(cmd
, "place_free_handicap")
393 || !strcasecmp(cmd
, "fixed_handicap")) {
396 int stones
= atoi(arg
);
399 board_handicap(board
, stones
, id
== NO_REPLY
? NULL
: stdout
);
400 if (DEBUGL(1) && debug_boardprint
)
401 board_print(board
, stderr
);
402 if (id
== NO_REPLY
) return P_OK
;
406 } else if (!strcasecmp(cmd
, "final_score")) {
407 struct move_queue q
= { .moves
= 0 };
408 if (engine
->dead_group_list
)
409 engine
->dead_group_list(engine
, board
, &q
);
410 floating_t score
= board_official_score(board
, &q
);
413 fprintf(stderr
, "counted score %.1f\n", score
);
415 gtp_reply(id
, "0", NULL
);
416 } else if (score
> 0) {
417 snprintf(str
, 64, "W+%.1f", score
);
418 gtp_reply(id
, str
, NULL
);
420 snprintf(str
, 64, "B+%.1f", -score
);
421 gtp_reply(id
, str
, NULL
);
424 /* XXX: This is a huge hack. */
425 } else if (!strcasecmp(cmd
, "final_status_list")) {
426 if (id
== NO_REPLY
) return P_OK
;
429 struct move_queue q
= { .moves
= 0 };
430 if (engine
->dead_group_list
)
431 engine
->dead_group_list(engine
, board
, &q
);
432 /* else we return empty list - i.e. engine not supporting
433 * this assumes all stones alive at the game end. */
434 if (!strcasecmp(arg
, "dead")) {
436 for (unsigned int i
= 0; i
< q
.moves
; i
++) {
437 foreach_in_group(board
, q
.move
[i
]) {
438 printf("%s ", coord2sstr(c
, board
));
439 } foreach_in_group_end
;
445 } else if (!strcasecmp(arg
, "seki") || !strcasecmp(arg
, "alive")) {
447 bool printed_group
= false;
448 foreach_point(board
) { // foreach_group, effectively
449 group_t g
= group_at(board
, c
);
450 if (!g
|| g
!= c
) continue;
452 for (unsigned int i
= 0; i
< q
.moves
; i
++) {
456 foreach_in_group(board
, g
) {
457 printf("%s ", coord2sstr(c
, board
));
458 } foreach_in_group_end
;
460 printed_group
= true;
467 gtp_error(id
, "illegal status specifier", NULL
);
470 } else if (!strcasecmp(cmd
, "undo")) {
471 if (board_undo(board
) < 0) {
473 fprintf(stderr
, "undo on non-pass move %s\n", coord2sstr(board
->last_move
.coord
, board
));
474 board_print(board
, stderr
);
476 gtp_error(id
, "cannot undo", NULL
);
481 reply
= engine
->undo(engine
, board
);
482 if (DEBUGL(1) && debug_boardprint
)
483 board_print(board
, stderr
);
484 gtp_reply(id
, reply
, NULL
);
486 /* Custom commands for handling the tree opening tbook */
487 } else if (!strcasecmp(cmd
, "pachi-gentbook")) {
488 /* Board must be initialized properly, as if for genmove;
489 * makes sense only as 'uct_gentbook b'. */
492 enum stone color
= str2stone(arg
);
493 if (uct_gentbook(engine
, board
, &ti
[color
], color
))
496 gtp_error(id
, "error generating tbook", NULL
);
498 } else if (!strcasecmp(cmd
, "pachi-dumptbook")) {
501 enum stone color
= str2stone(arg
);
502 uct_dumptbook(engine
, board
, color
);
505 } else if (!strcasecmp(cmd
, "pachi-evaluate")) {
508 enum stone color
= str2stone(arg
);
510 if (!engine
->evaluate
) {
511 gtp_error(id
, "pachi-evaluate not supported by engine", NULL
);
514 floating_t vals
[board
->flen
];
515 engine
->evaluate(engine
, board
, &ti
[color
], vals
, color
);
516 for (int i
= 0; i
< board
->flen
; i
++) {
517 if (!board_coord_in_symmetry(board
, board
->f
[i
])
520 printf("%s %1.3f\n", coord2sstr(board
->f
[i
], board
), (double) vals
[i
]);
525 } else if (!strcasecmp(cmd
, "pachi-result")) {
526 /* More detailed result of the last genmove. */
527 /* For UCT, the output format is: = color move playouts winrate dynkomi */
530 reply
= engine
->result(engine
, board
);
532 gtp_reply(id
, reply
, NULL
);
534 gtp_error(id
, "unknown pachi-result command", NULL
);
536 } else if (!strcasecmp(cmd
, "kgs-chat")) {
545 reply
= engine
->chat(engine
, board
, msg
);
547 gtp_reply(id
, reply
, NULL
);
549 gtp_error(id
, "unknown kgs-chat command", NULL
);
551 } else if (!strcasecmp(cmd
, "time_left")) {
554 enum stone color
= str2stone(arg
);
556 int time
= atoi(arg
);
558 int stones
= atoi(arg
);
559 if (!ti
[color
].ignore_gtp
) {
560 time_left(&ti
[color
], time
, stones
);
562 if (DEBUGL(2)) fprintf(stderr
, "ignored time info\n");
567 } else if (!strcasecmp(cmd
, "time_settings") || !strcasecmp(cmd
, "kgs-time_settings")) {
570 if (!strcasecmp(cmd
, "kgs-time_settings")) {
571 next_tok(time_system
);
573 time_system
= "canadian";
576 int main_time
= 0, byoyomi_time
= 0, byoyomi_stones
= 0, byoyomi_periods
= 0;
577 if (!strcasecmp(time_system
, "none")) {
579 } else if (!strcasecmp(time_system
, "absolute")) {
581 main_time
= atoi(arg
);
582 } else if (!strcasecmp(time_system
, "byoyomi")) {
584 main_time
= atoi(arg
);
586 byoyomi_time
= atoi(arg
);
588 byoyomi_periods
= atoi(arg
);
589 } else if (!strcasecmp(time_system
, "canadian")) {
591 main_time
= atoi(arg
);
593 byoyomi_time
= atoi(arg
);
595 byoyomi_stones
= atoi(arg
);
599 fprintf(stderr
, "time_settings %d %d/%d*%d\n",
600 main_time
, byoyomi_time
, byoyomi_stones
, byoyomi_periods
);
601 if (!ti
[S_BLACK
].ignore_gtp
) {
602 time_settings(&ti
[S_BLACK
], main_time
, byoyomi_time
, byoyomi_stones
, byoyomi_periods
);
603 ti
[S_WHITE
] = ti
[S_BLACK
];
605 if (DEBUGL(1)) fprintf(stderr
, "ignored time info\n");
611 gtp_error(id
, "unknown command", NULL
);
612 return P_UNKNOWN_COMMAND
;