time_in_byoyomi(), time_start_timer(): Swap, more logical organization now
[pachi/json.git] / gtp.c
blobe716a4582e278156cc2e2fbd0efaeb7c3c4239e0
1 #define DEBUG
2 #include <assert.h>
3 #include <ctype.h>
4 #include <stdarg.h>
5 #include <stdio.h>
6 #include <stdlib.h>
7 #include <string.h>
9 #include "board.h"
10 #include "debug.h"
11 #include "engine.h"
12 #include "gtp.h"
13 #include "mq.h"
14 #include "uct/uct.h"
15 #include "version.h"
16 #include "timeinfo.h"
18 void
19 gtp_prefix(char prefix, int id)
21 if (id >= 0)
22 printf("%c%d ", prefix, id);
23 else
24 printf("%c ", prefix);
27 void
28 gtp_flush(void)
30 putchar('\n');
31 fflush(stdout);
34 void
35 gtp_output(char prefix, int id, va_list params)
37 gtp_prefix(prefix, id);
38 char *s;
39 while ((s = va_arg(params, char *))) {
40 fputs(s, stdout);
42 putchar('\n');
43 gtp_flush();
46 void
47 gtp_reply(int id, ...)
49 va_list params;
50 va_start(params, id);
51 gtp_output('=', id, params);
52 va_end(params);
55 void
56 gtp_error(int id, ...)
58 va_list params;
59 va_start(params, id);
60 gtp_output('?', id, params);
61 va_end(params);
65 /* XXX: THIS IS TOTALLY INSECURE!!!!
66 * Even basic input checking is missing. */
68 void
69 gtp_parse(struct board *board, struct engine *engine, struct time_info *ti, char *buf)
71 #define next_tok(to_) \
72 to_ = next; \
73 next = next + strcspn(next, " \t\r\n"); \
74 if (*next) { \
75 *next = 0; next++; \
76 next += strspn(next, " \t\r\n"); \
79 if (strchr(buf, '#'))
80 *strchr(buf, '#') = 0;
82 char *cmd, *next = buf;
83 next_tok(cmd);
85 int id = -1;
86 if (isdigit(*cmd)) {
87 id = atoi(cmd);
88 next_tok(cmd);
91 if (!*cmd)
92 return;
94 if (!strcasecmp(cmd, "protocol_version")) {
95 gtp_reply(id, "2", NULL);
97 } else if (!strcasecmp(cmd, "name")) {
98 /* KGS hack */
99 gtp_reply(id, "Pachi ", engine->name, NULL);
101 } else if (!strcasecmp(cmd, "version")) {
102 gtp_reply(id, PACHI_VERSION, ": ", engine->comment, NULL);
104 /* TODO: known_command */
106 } else if (!strcasecmp(cmd, "list_commands")) {
107 gtp_reply(id, "protocol_version\n"
108 "name\n"
109 "version\n"
110 "list_commands\n"
111 "quit\n"
112 "boardsize\n"
113 "clear_board\n"
114 "komi\n"
115 "play\n"
116 "genmove\n"
117 "kgs-genmove_cleanup\n"
118 "set_free_handicap\n"
119 "place_free_handicap\n"
120 "final_status_list\n"
121 "kgs-chat\n"
122 "time_left\n"
123 "time_settings\n"
124 "kgs-time_settings", NULL);
126 } else if (!strcasecmp(cmd, "quit")) {
127 gtp_reply(id, NULL);
128 exit(0);
130 } else if (!strcasecmp(cmd, "boardsize")) {
131 char *arg;
132 next_tok(arg);
133 board_resize(board, atoi(arg));
135 gtp_reply(id, NULL);
137 } else if (!strcasecmp(cmd, "clear_board")) {
138 board_clear(board);
139 engine_reset = true;
140 if (DEBUGL(1))
141 board_print(board, stderr);
142 gtp_reply(id, NULL);
144 } else if (!strcasecmp(cmd, "komi")) {
145 char *arg;
146 next_tok(arg);
147 sscanf(arg, "%f", &board->komi);
149 if (DEBUGL(1))
150 board_print(board, stderr);
151 gtp_reply(id, NULL);
153 } else if (!strcasecmp(cmd, "play")) {
154 struct move m;
156 char *arg;
157 next_tok(arg);
158 m.color = str2stone(arg);
159 next_tok(arg);
160 coord_t *c = str2coord(arg, board_size(board));
161 m.coord = *c; coord_done(c);
162 char *reply = NULL;
164 if (DEBUGL(1))
165 fprintf(stderr, "got move %d,%d,%d\n", m.color, coord_x(m.coord, board), coord_y(m.coord, board));
167 // This is where kgs starts our timer, not at coming genmove!
168 time_start_timer(&ti[stone_other(m.color)]);
170 if (engine->notify_play)
171 reply = engine->notify_play(engine, board, &m);
172 if (board_play(board, &m) < 0) {
173 if (DEBUGL(0)) {
174 fprintf(stderr, "! ILLEGAL MOVE %d,%d,%d\n", m.color, coord_x(m.coord, board), coord_y(m.coord, board));
175 board_print(board, stderr);
177 gtp_error(id, "illegal move", NULL);
178 } else {
179 if (DEBUGL(1))
180 board_print(board, stderr);
181 gtp_reply(id, reply, NULL);
184 } else if (!strcasecmp(cmd, "genmove") || !strcasecmp(cmd, "kgs-genmove_cleanup")) {
185 char *arg;
186 next_tok(arg);
187 enum stone color = str2stone(arg);
189 coord_t *c = engine->genmove(engine, board, &ti[color], color, !strcasecmp(cmd, "kgs-genmove_cleanup"));
190 struct move m = { *c, color };
191 board_play(board, &m);
192 char *str = coord2str(*c, board);
193 if (DEBUGL(1))
194 fprintf(stderr, "playing move %s\n", str);
195 if (DEBUGL(1)) {
196 board_print_custom(board, stderr, engine->printhook);
198 gtp_reply(id, str, NULL);
199 free(str); coord_done(c);
201 } else if (!strcasecmp(cmd, "set_free_handicap")) {
202 struct move m;
203 m.color = S_BLACK;
205 char *arg;
206 next_tok(arg);
207 do {
208 coord_t *c = str2coord(arg, board_size(board));
209 m.coord = *c; coord_done(c);
210 if (DEBUGL(1))
211 fprintf(stderr, "setting handicap %d,%d\n", coord_x(m.coord, board), coord_y(m.coord, board));
213 if (board_play(board, &m) < 0) {
214 if (DEBUGL(0))
215 fprintf(stderr, "! ILLEGAL MOVE %d,%d,%d\n", m.color, coord_x(m.coord, board), coord_y(m.coord, board));
216 gtp_error(id, "illegal move", NULL);
218 board->handicap++;
219 next_tok(arg);
220 } while (*arg);
221 if (DEBUGL(1))
222 board_print(board, stderr);
223 gtp_reply(id, NULL);
225 /* TODO: Engine should choose free handicap; however, it tends to take
226 * overly long to think it all out, and unless it's clever its
227 * handicap stones won't be of much help. ;-) */
228 } else if (!strcasecmp(cmd, "place_free_handicap")
229 || !strcasecmp(cmd, "fixed_handicap")) {
230 char *arg;
231 next_tok(arg);
232 int stones = atoi(arg);
234 gtp_prefix('=', id);
235 board_handicap(board, stones, stdout);
236 if (DEBUGL(1))
237 board_print(board, stderr);
238 putchar('\n');
239 gtp_flush();
241 } else if (!strcasecmp(cmd, "final_score")) {
242 struct move_queue q = { .moves = 0 };
243 if (engine->dead_group_list)
244 engine->dead_group_list(engine, board, &q);
245 float score = board_official_score(board, &q);
246 char str[64];
247 if (DEBUGL(1))
248 fprintf(stderr, "counted score %.1f\n", score);
249 if (score == 0) {
250 gtp_reply(id, "0", NULL);
251 } else if (score > 0) {
252 snprintf(str, 64, "W+%.1f", score);
253 gtp_reply(id, str, NULL);
254 } else {
255 snprintf(str, 64, "B+%.1f", -score);
256 gtp_reply(id, str, NULL);
259 /* XXX: This is a huge hack. */
260 } else if (!strcasecmp(cmd, "final_status_list")) {
261 char *arg;
262 next_tok(arg);
263 struct move_queue q = { .moves = 0 };
264 if (engine->dead_group_list)
265 engine->dead_group_list(engine, board, &q);
266 /* else we return empty list - i.e. engine not supporting
267 * this assumes all stones alive at the game end. */
268 if (!strcasecmp(arg, "dead")) {
269 gtp_prefix('=', id);
270 for (int i = 0; i < q.moves; i++) {
271 foreach_in_group(board, q.move[i]) {
272 printf("%s ", coord2sstr(c, board));
273 } foreach_in_group_end;
274 putchar('\n');
276 if (!q.moves)
277 putchar('\n');
278 gtp_flush();
279 } else if (!strcasecmp(arg, "seki") || !strcasecmp(arg, "alive")) {
280 gtp_prefix('=', id);
281 bool printed_group = false;
282 foreach_point(board) { // foreach_group, effectively
283 group_t g = group_at(board, c);
284 if (!g || g != c) continue;
286 for (int i = 0; i < q.moves; i++) {
287 if (q.move[i] == g)
288 goto next_group;
290 foreach_in_group(board, g) {
291 printf("%s ", coord2sstr(c, board));
292 } foreach_in_group_end;
293 putchar('\n');
294 printed_group = true;
295 next_group:;
296 } foreach_point_end;
297 if (!printed_group)
298 putchar('\n');
299 gtp_flush();
300 } else {
301 gtp_error(id, "illegal status specifier", NULL);
304 /* Custom commands for handling UCT opening book */
305 } else if (!strcasecmp(cmd, "uct_genbook")) {
306 /* Board must be initialized properly, as if for genmove;
307 * makes sense only as 'uct_genbook b'. */
308 char *arg;
309 next_tok(arg);
310 enum stone color = str2stone(arg);
311 if (uct_genbook(engine, board, &ti[color], color))
312 gtp_reply(id, NULL);
313 else
314 gtp_error(id, "error generating book", NULL);
316 } else if (!strcasecmp(cmd, "uct_dumpbook")) {
317 char *arg;
318 next_tok(arg);
319 enum stone color = str2stone(arg);
320 uct_dumpbook(engine, board, color);
321 gtp_reply(id, NULL);
323 } else if (!strcasecmp(cmd, "kgs-chat")) {
324 char *loc;
325 next_tok(loc);
326 char *src;
327 next_tok(src);
328 char *msg;
329 next_tok(msg);
330 char *reply = NULL;
331 if (engine->chat)
332 reply = engine->chat(engine, board, msg);
333 if (reply)
334 gtp_reply(id, reply, NULL);
335 else
336 gtp_error(id, "unknown chat command", NULL);
338 } else if (!strcasecmp(cmd, "time_left")) {
339 char *arg;
340 next_tok(arg);
341 enum stone color = str2stone(arg);
342 next_tok(arg);
343 int time = atoi(arg);
344 next_tok(arg);
345 int stones = atoi(arg);
346 if (!ti[color].ignore_gtp) {
347 time_left(&ti[color], time, stones);
348 } else {
349 if (DEBUGL(2)) fprintf(stderr, "ignored time info\n");
352 gtp_reply(id, NULL);
354 } else if (!strcasecmp(cmd, "time_settings") || !strcasecmp(cmd, "kgs-time_settings")) {
355 char *time_system = "canadian";
356 char *arg;
357 int main_time = -1, byoyomi_time = 0, byoyomi_stones = 0, byoyomi_periods = 0;
358 if (!strcasecmp(cmd, "kgs-time_settings")) {
359 next_tok(time_system);
360 if (!strcasecmp(time_system, "none")) {
361 // time > 0, stones 0: convention for unlimited
362 byoyomi_time = 1;
363 main_time = 0;
364 } else if (!strcasecmp(time_system, "absolute")) {
365 next_tok(arg);
366 main_time = atoi(arg);
367 } else if (!strcasecmp(time_system, "byoyomi")) {
368 next_tok(arg);
369 main_time = atoi(arg);
370 next_tok(arg);
371 byoyomi_time = atoi(arg);
372 byoyomi_stones = 1;
373 next_tok(arg);
374 byoyomi_periods = atoi(arg);
377 if (main_time < 0) { // canadian time system
378 next_tok(arg);
379 main_time = atoi(arg);
380 next_tok(arg);
381 byoyomi_time = atoi(arg);
382 next_tok(arg);
383 byoyomi_stones = atoi(arg);
385 if (DEBUGL(1))
386 fprintf(stderr, "time_settings %d %d %d %d\n",
387 main_time, byoyomi_time, byoyomi_stones, byoyomi_periods);
388 if (!ti[S_BLACK].ignore_gtp) {
389 time_settings(&ti[S_BLACK], main_time, byoyomi_time, byoyomi_stones, byoyomi_periods);
390 ti[S_WHITE] = ti[S_BLACK];
391 } else {
392 if (DEBUGL(1)) fprintf(stderr, "ignored time info\n");
395 gtp_reply(id, NULL);
397 } else {
398 gtp_error(id, "unknown command", NULL);
401 #undef next_tok