bac4bcb6496573ff9f8c77171026eba60ac7631d
[uci2wb.git] / UCI2WB.c
1 /****************************************************************************/\r
2 /*                           UCI2WB by H.G.Muller                           */\r
3 /*                                                                          */\r
4 /* UCI2WB is an adapter to run engines that communicate in various dialects */\r
5 /* of the Universal Chess Interface in a GUI that supports XBoard protocol  */\r
6 /* (CECP). It supports UCI (when used for Xiangqi: the 'Cyclone dialect'),  */\r
7 /* as well as USI and UCCI when used with the flags -s or -x, respectively. */\r
8 /* This version of UCI2WB is released under the GNU General Public License, */\r
9 /* of which you should have received a copy together with this file.        */\r
10 /****************************************************************************/\r
11 \r
12 #define VERSION "3.0"\r
13 \r
14 #include <stdio.h>\r
15 #include <stdlib.h>\r
16 #ifdef WIN32\r
17 #  include <windows.h>\r
18 #  include <io.h>\r
19    HANDLE process;\r
20    DWORD thread_id;\r
21 #else\r
22 #  include <pthread.h>\r
23 #  include <signal.h>\r
24 #  define NO_ERROR 0\r
25 #  include <sys/time.h>\r
26    int GetTickCount() // with thanks to Tord\r
27    { struct timeval t; gettimeofday(&t, NULL); return t.tv_sec*1000 + t.tv_usec/1000; }\r
28 #endif\r
29 #include <fcntl.h>\r
30 #include <string.h>\r
31 \r
32 // Set VARIANTS for in WinBoard variant feature. (With -s option this will always be reset to use "shogi".)\r
33 #define VARIANTS ",normal,xiangqi"\r
34 #define STDVARS "chess,chess960,crazyhouse,3check,giveaway,suicide,losers,atomic,seirawan,shogi,xiangqi"\r
35 #define EGT ",gaviotaTbPath,syzygyPath,nalimovPath,robbotripleBaseDirectory,robbototalBaseDirectory,bitbases path,"\r
36 \r
37 #define DPRINT if(debug) printf\r
38 #define EPRINT(X) { char f[999]; sprintf X; DPRINT("%s", f); fprintf(toE, "%s", f + 2*(*f == '#')); /* strip optional # prefix */ }\r
39 \r
40 #define WHITE 0\r
41 #define BLACK 1\r
42 #define NONE  2\r
43 #define ANALYZE 3\r
44 \r
45 char move[2000][10], iniPos[256], hashOpt[20], suspended, ponder, post, hasHash, c, sc='c', suffix[81], varOpt, searching, *binary;\r
46 int mps, tc, inc, sTime, depth, myTime, hisTime, stm, computer = NONE, memory, oldMem=0, cores, moveNr, lastDepth, lastScore, startTime, debug, flob;\r
47 int statDepth, statScore, statNodes, statTime, currNr, size, collect, nr, sm, inex, on[500], frc, byo = -1, namOpt, comp;\r
48 char currMove[20], moveMap[500][10], /* for analyze mode */ canPonder[20], threadOpt[20], varList[8000], anaOpt[20], checkOptions[8192] = "Ponder";\r
49 char pvs[99][999], board[100];  // XQ board for UCCI\r
50 char *nameWord = "name ", *valueWord = "value ", *wTime = "w", *bTime = "b", *wInc = "winc", *bInc = "binc", newGame; // keywords that differ in UCCI\r
51 int unit = 1, drawOffer, scores[99], mpvSP, maxDepth, ponderAlways;\r
52 \r
53 FILE *toE, *fromE, *fromF;\r
54 int pid;\r
55 \r
56 char *strcasestr (char *p, char *q) { while(*p) { char *r=p++, *s=q; while(tolower(*r++) == tolower(*s) && *s) s++; if(!*s) return p-1; } return NULL; }\r
57 \r
58 #ifdef WIN32\r
59 WinPipe(HANDLE *hRd, HANDLE *hWr)\r
60 {\r
61   SECURITY_ATTRIBUTES saAttr;\r
62 \r
63   /* Set the bInheritHandle flag so pipe handles are inherited. */\r
64   saAttr.nLength = sizeof(SECURITY_ATTRIBUTES);\r
65   saAttr.bInheritHandle = TRUE;\r
66   saAttr.lpSecurityDescriptor = NULL;\r
67 \r
68   /* Create a pipe */\r
69   return CreatePipe(hRd, hWr, &saAttr, 0);\r
70 }\r
71 #endif\r
72 \r
73 #define INIT 0\r
74 #define WAKEUP 1\r
75 #define PAUSE 2\r
76 \r
77 void\r
78 Sync (int action)\r
79 {\r
80 #ifdef WIN32\r
81         static HANDLE hWr, hRd; DWORD d; char c;\r
82         switch(action) {\r
83             case INIT:   WinPipe(&hRd, &hWr); break;\r
84             case WAKEUP: WriteFile(hWr, "\n", 1, &d, NULL); break;\r
85             case PAUSE:  ReadFile(hRd, &c, 1, &d, NULL);\r
86         }\r
87 #else\r
88         static int syncPipe[2]; char c;\r
89         switch(action) {\r
90             case INIT:   pipe(syncPipe); break;\r
91             case WAKEUP: write(syncPipe[1], "\n", 1); break;\r
92             case PAUSE:  read(syncPipe[0], &c, 1);\r
93         }\r
94 #endif\r
95 }\r
96 \r
97 void\r
98 FromFEN(char *fen)\r
99 {       int i=0;\r
100         while(*fen) {\r
101             char c = *fen++;\r
102             if(c >= 'A') board[i++] = c; else\r
103             if(c == '/') i++; else\r
104             if(c == ' ') break; else\r
105             while(c-- > '0' && i < 99) board[i++] = 0;\r
106             if(i >= 99) break;\r
107         }\r
108 }\r
109 \r
110 char *\r
111 ToFEN(int stm)\r
112 {\r
113         int i, n=0; static char fen[200]; char *p = fen;\r
114         for(i=0; i<99; i++) {\r
115             char c = board[i];\r
116             if(c >= 'A')  { if(n) *p++ = '0' + n; n = 0;  *p++ = c; } else n ++;\r
117             if(i%10 == 8) { if(n) *p++ = '0' + n; n = -1; *p++ = '/'; }\r
118         }\r
119         sprintf(p-1, " %c - - 0 1", stm);\r
120         return fen;\r
121 }\r
122 \r
123 int\r
124 Sqr(char *m, int j)\r
125 {\r
126         int n = m[j] - 'a' + 10*('9' - m[j+1]);\r
127         if(n < 0) n = 0; else if(n > 99) n = 99; return n;\r
128 }\r
129 \r
130 int\r
131 Play(int nr)\r
132 {\r
133         int i, last = -1;\r
134         FromFEN(iniPos + 4); // in XQ iniPos always has just "fen " prefix\r
135         for(i=0; i<nr; i++) {\r
136             int from=Sqr(move[i], 0), to=Sqr(move[i], 2);\r
137             if(board[to] || (board[from]|32)  == 'p' && move[i][1] != move[i][3]) last = i;\r
138             board[to] = board[from]; board[from] = 0;\r
139         }\r
140         return last;\r
141 }\r
142 \r
143 void\r
144 StartSearch(char *ponder)\r
145 {       // send the 'go' command to engine. Suffix by ponder.\r
146         int x = (ponder[0] != 0);                   // during ponder stm is the opponent\r
147         int black = (stm == BLACK ^ x ^ sc == 's'); // set if our color is what the engine calls black\r
148         int nr = moveNr + x;                        // we ponder for one move ahead!\r
149         int t = (flob ? inc + myTime/40 : 1000*byo*(byo>0)); // byoyomi time \r
150         if(sc == 'x') black = 1; else drawOffer = 0;// in UCCI 'black' refers to us and 'white' to opponent\r
151         if(!x && drawOffer) ponder = " draw", drawOffer = 0; //pass draw offer only when not pondering\r
152         EPRINT((f, "# go%s %stime %d %stime %d", ponder, bTime, (black ? myTime : hisTime) - t, wTime, (!black ? myTime : hisTime) - t))\r
153         if(sTime > 0) EPRINT((f, " movetime %d", sTime)) else\r
154         if(mps) EPRINT((f, " movestogo %d", mps*(nr/(2*mps)+1)-nr/2))\r
155         if(flob || byo >= 0) sprintf(suffix, " byoyomi %d", t); // for engines running purely on byoyomi\r
156         if(inc && !*suffix) EPRINT((f, " %s %d %s %d", wInc, inc, bInc, inc))\r
157         if(depth > 0) EPRINT((f, " depth %d", depth))\r
158         if(*suffix) EPRINT((f, suffix, inc))\r
159         EPRINT((f, "\n")); maxDepth = mpvSP = 0;\r
160 }\r
161 \r
162 void\r
163 StopSearch(int discard)\r
164 {\r
165         if(!searching) return;\r
166         if(discard) searching = 0; // this causes bestmove to be ignored\r
167         EPRINT((f, "# stop\n")) fflush(toE); // note: 'pondering' remains set until engine acknowledges 'stop' with 'bestmove'\r
168 }\r
169 \r
170 void\r
171 LoadPos(int moveNr)\r
172 {\r
173         int j, lastCapt = 0; char *pos = iniPos, buf[200], stm;\r
174         if(sc == 'x') { // UCCI: send only reversible moves\r
175             lastCapt = Play(moveNr); // find last capture (returns -1 if none!)\r
176             Play(++lastCapt);        // reconstruct board after last capture\r
177             stm = (!strstr(iniPos+4, " b ") ^ lastCapt & 1 ? 'w' :  'b');\r
178             sprintf(buf, "position fen %s", ToFEN(stm)); pos = buf; // send it as FEN (with "position" in UCCI!)\r
179         }\r
180         EPRINT((f, "# %s moves", pos))\r
181         for(j=lastCapt; j<moveNr; j++) EPRINT((f, " %s", move[j]))\r
182         EPRINT((f, "\n"))\r
183 }\r
184 \r
185 void\r
186 StartPonder(int moveNr)\r
187 {\r
188         if(!move[moveNr][0]) return; // no ponder move\r
189         LoadPos(moveNr+1);\r
190         searching = 1; lastDepth = 1;\r
191         StartSearch(" ponder");\r
192 }\r
193 \r
194 void\r
195 Analyze(char *val)\r
196 {\r
197     if(*anaOpt) EPRINT((f, "# setoption %s%s %s%s\n", nameWord, anaOpt, valueWord, val));\r
198 }\r
199 \r
200 char *Convert(char *pv)\r
201 {   // convert Shogi coordinates to WB\r
202     char *p, *q, c;\r
203     static char buf[10000];\r
204     if(sc != 's') return pv;\r
205     p = pv; q = buf;\r
206     while(c = *p++) {\r
207         if(c >= '0' && c <= '9' || c >= 'a' && c <= 'z') *q++ = 'a'+'0'+size - c; else *q++ = c;\r
208     }\r
209     *q++ = 0;\r
210     return buf;\r
211 }\r
212 \r
213 void\r
214 Move4GUI(char *m)\r
215 {\r
216     if(sc == 's') {\r
217       // convert USI move to WB format\r
218       m[2] = 'a'+'0'+size - m[2];\r
219       m[3] = 'a'+'0'+size - m[3];\r
220       if(m[1] == '*') { // drop\r
221         m[1] = '@';\r
222       } else {\r
223         m[0] = 'a'+'0'+size - m[0];\r
224         m[1] = 'a'+'0'+size - m[1];\r
225         if((stm == WHITE ? (m[1]>'0'+size-size/3 || m[3]>'0'+size-size/3)\r
226                                 : (m[1] <= '0'+size/3 || m[3] <= '0'+size/3)) && m[4] != '+')\r
227              m[4] = '=', m[5] = 0;\r
228       }\r
229     }\r
230 }\r
231 \r
232 int\r
233 ReadLine (FILE *f, char *line)\r
234 {\r
235     int x, i = 0;\r
236     while((x = fgetc(f)) != EOF && (line[i] = x) != '\n') i++; line[++i] = 0;\r
237     return (x != EOF);\r
238 }\r
239 \r
240 void\r
241 HandleEngineOutput()\r
242 {\r
243     char line[1024], command[256]; static char egts[999];\r
244 \r
245     while(1) {\r
246         int i=0, x; char *p, dummy, len;\r
247 \r
248         fflush(stdout); fflush(toE);\r
249         if(fromF && !ReadLine(fromF, line))  fromF = 0, printf("# end fake\n");\r
250         if(!fromF && !ReadLine(fromE, line)) printf("tellusererror UCI2WB: %s died on me\n", binary), exit(0);\r
251         DPRINT("# engine said: %s", line), fflush(stdout);\r
252         if(sscanf(line, "%s", command) != 1) continue;\r
253         if(!strcmp(command, "bestmove")) {\r
254             if(searching == 1) { searching = 0; printf("%d 0 0 0 UCI violation! Engine moves during ponder\n", lastDepth+1); return; } // ignore ponder search\r
255             else if(searching != 3) { searching = 0; return; } // ponder miss or analysis result; ignore.\r
256             // move was a move to be played\r
257             if(p = strstr(line+8, " draw")) *p = 0, printf("offer draw\n"); // UCCI\r
258             if(strstr(line+9, "resign")) { printf("resign\n"); computer = NONE; }\r
259             if(strstr(line+9, "win")) { printf("%s {claim}\n", stm== WHITE ? "1-0" :"0-1"); computer = NONE; } // USI\r
260             if(strstr(line+9, "(none)") || strstr(line+9, "null") ||\r
261                strstr(line+9, "0000")) { printf("%s\n", lastScore < -99999 ? "resign" : "1/2-1/2 {stalemate}"); computer = NONE; }\r
262             sscanf(line, "bestmove %s", move[moveNr++]);\r
263             myTime -= (GetTickCount() - startTime)*1.02 - inc; // update own clock, so we can give correct wtime, btime with ponder\r
264             if(mps && ((moveNr+1)/2) % mps == 0) myTime += tc; if(sTime) myTime = sTime; // new session or move starts\r
265             stm = WHITE+BLACK - stm; searching = 0;\r
266             // first start a new ponder search, if pondering is on and we have a move to ponder on\r
267             if(p = strstr(line+9, "ponder")) {\r
268               sscanf(p+7, "%s", move[moveNr]);\r
269               if(computer != NONE && ponder) {\r
270                 DPRINT("# ponder on %s\n", move[moveNr]);\r
271                 StartPonder(moveNr);\r
272               }\r
273               p[-1] = '\n'; *p = 0; // strip off ponder move\r
274             } else move[moveNr][0] = 0;\r
275             Move4GUI(line+9);\r
276             printf("move %s\n", line+9); // send move to GUI\r
277             if(move[moveNr][0]) printf("Hint: %s\n", move[moveNr]);\r
278             if(lastScore == 100001 && iniPos[0] != 'f') { printf("%s {mate}\n", stm == BLACK ? "1-0" : "0-1"); computer = NONE; }\r
279             fflush(stdout); return;\r
280         }\r
281         else if(!strcmp(command, "info")) {\r
282             int d=0, s=0, t=(GetTickCount() - startTime)/10, n=1;\r
283             char *pv, varName[80];\r
284             if(sscanf(line+5, "string times @ %c", &dummy) == 1) { printf("# %s", line+12); continue; }\r
285             if(sscanf(line+5, "string variant %s", varName) == 1) {\r
286                 if(!strstr(STDVARS, varName)) {\r
287                     int files = 8, ranks = 8, hand = 0; char parent[80];\r
288                     if(p = strstr(line+18, " files ")) sscanf(p+7, "%d", &files);\r
289                     if(p = strstr(line+18, " ranks ")) sscanf(p+7, "%d", &ranks);\r
290                     if(p = strstr(line+18, " pocket ")) sscanf(p+8, "%d", &hand);\r
291                     if(p = strstr(line+18, " template ")) sscanf(p+10, "%s", parent); else strcpy(parent, "fairy");\r
292                     if(p = strstr(line+18, " startpos "))\r
293                         printf("setup (-) %dx%d+%d_%s %s", files, ranks, hand, parent, p+10);\r
294                 }\r
295                 continue;\r
296             }\r
297             if(collect && (pv = strstr(line+5, "currmove "))) {\r
298                 if(p = strstr(line+5, "currmovenumber ")) {\r
299                     n = atoi(p+15);\r
300                     if(collect == 1 && n != 1) continue; // wait for move 1\r
301                     if(collect + (n == 1) > 2) { // done collecting\r
302                         if(inex && collect == 2) printf("%d 0 0 0 OK to exclude\n", lastDepth);\r
303                         collect = 3; continue;\r
304                     }\r
305                     collect = 2; on[nr=n] = 1; sscanf(pv+9, "%s", moveMap[n]); continue; // store move\r
306                 }\r
307             }\r
308             if(!post) continue;\r
309             if(sscanf(line+5, "string %c", &dummy) == 1) printf("%d 0 0 0 %s", lastDepth, line+12); else {\r
310                 if(p = strstr(line+4, " depth "))      sscanf(p+7, "%d", &d), statDepth = d;\r
311                 if(p = strstr(line+4, " score cp "))   sscanf(p+10, "%d", &s), statScore = s; else\r
312                 if(p = strstr(line+4, " score mate ")) sscanf(p+12, "%d", &s), s += s>0 ? 100000 : -100000, statScore = s; else\r
313                 if(p = strstr(line+4, " score "))      sscanf(p+7, "%d", &s), statScore = s;\r
314                 if(p = strstr(line+4, " nodes "))      sscanf(p+7, "%d", &n), statNodes = n;\r
315                 if(p = strstr(line+4, " time "))       sscanf(p+6, "%d", &t), t /= 10, statTime = t;\r
316                 if(p = strstr(line+4, " currmove "))   sscanf(p+10,"%s", currMove);\r
317                 if(p = strstr(line+4, " currmovenumber ")) sscanf(p+16,"%d", &currNr);\r
318                 if(pv = strstr(line+4, " pv ")) { // convert PV info to WB thinking output\r
319                   if(d > maxDepth) maxDepth = d, mpvSP = 0; else if(d < maxDepth) continue; // ignore depth regressions\r
320                   if(p = strstr(line+4, " upperbound ")) strcat(p, "?\n"); else\r
321                   if(p = strstr(line+4, " lowerbound ")) strcat(p, "!\n");\r
322                   for(i=0; i<mpvSP; i++) if(s == scores[i] && !strcmp(pvs[i], pv+4)) break; // check if duplicat\r
323                   if(i >= mpvSP) strncpy(pvs[mpvSP], pv+4, 998), scores[mpvSP++] = s,       // emit as thinking output if not\r
324                     printf("%3d  %6d %6d %10d %s", lastDepth=d, lastScore=s, t, n, Convert(pv+4));\r
325                 } else if(s == -100000) lastScore = s; // when checkmated score is valid even without PV (which might not come)\r
326             }\r
327         }\r
328         else if(!strcmp(command, "option")) { // USI option: extract data fields\r
329             char name[80], type[80], buf[1024], val[256], *q;\r
330             int min=0, max=1e9; *val = 0;\r
331             if(p = strstr(line+6, " type ")) sscanf(p+1, "type %s", type), *p = '\n';\r
332             if(p = strstr(line+6, " min "))  sscanf(p+1, "min %d", &min), *p = '\n';\r
333             if(p = strstr(line+6, " max "))  sscanf(p+1, "max %d", &max), *p = '\n';\r
334             if(p = strstr(line+6, " default "))  sscanf(p+1, "default %[^\n]*", val), *p = '\n';\r
335             if(!(p = strstr(line+6, " name "))) p = line+1; sscanf(p+6, "%[^\n]", name); // 'name' is omitted in UCCI\r
336             if(!strcasecmp(name, "UCI_Chess960")) { frc=2; continue; }\r
337             if(!strcasecmp(name, "UCI_Variant")) { if(p = strstr(line+6, " var ")) strcpy(varList, p); varOpt = 1; continue; }\r
338             if(!strcasecmp(name, "UCI_Opponent")) { namOpt = 1; continue; }\r
339             if(!strcasecmp(name+2, "I_AnalyseMode")) { strcpy(anaOpt, name); continue; }\r
340             if(frc< 0 && (strstr(name, "960") || strcasestr(name, "frc")) && !strcmp(type, "check")) {\r
341                 EPRINT((f, "# setoption name %s value true\n", name)) strcpy(val, "true"); // set non-standard suspected FRC options\r
342             }\r
343             if(!strcasecmp(name, "Threads")) { strcpy(threadOpt, name); continue; }\r
344             if(!strcasecmp(name, "Ponder") || !strcasecmp(name, "USI_Ponder")) { strcpy(canPonder, name); continue; }\r
345             if(!strcasecmp(name, "Hash") || !strcasecmp(name, "USI_Hash") || !strcasecmp(name, "hashsize")) {\r
346                 memory = oldMem = atoi(val); hasHash = 1; \r
347                 strcpy(hashOpt, name);\r
348                 continue;\r