Improve efficiency
[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 "2.3"\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,threecheck,giveaway,atomic,seirawan,shogi,xiangqi"\r
35 \r
36 #define DPRINT if(debug) printf\r
37 #define EPRINT(X) { char f[999]; sprintf X; DPRINT("%s", f); fprintf(toE, "%s", f + 2*(*f == '#')); /* strip optional # prefix */ }\r
38 \r
39 #define WHITE 0\r
40 #define BLACK 1\r
41 #define NONE  2\r
42 #define ANALYZE 3\r
43 \r
44 char move[2000][10], iniPos[256], hashOpt[20], pause, pondering, suspended, ponder, post, hasHash, c, sc='c', suffix[81], *variants, varOpt, searching;\r
45 int mps, tc, inc, sTime, depth, myTime, hisTime, stm, computer = NONE, memory, oldMem=0, cores, moveNr, lastDepth, lastScore, startTime, debug, flob;\r
46 int statDepth, statScore, statNodes, statTime, currNr, size, collect, nr, sm, inex, on[500], frc, byo = -1, namOpt, comp;\r
47 char currMove[20], moveMap[500][10], /* for analyze mode */ canPonder[20], threadOpt[20], varList[8000], anaOpt[20], backLog[10000], checkOptions[8192] = "Ponder";\r
48 char board[100];  // XQ board for UCCI\r
49 char *nameWord = "name ", *valueWord = "value ", *wTime = "w", *bTime = "b", *wInc = "winc", *bInc = "binc", newGame; // keywords that differ in UCCI\r
50 int unit = 1, drawOffer;\r
51 volatile int logLen, sentLen;\r
52 \r
53 FILE *toE, *fromE, *fromF;\r
54 int pid;\r
55 \r
56 #ifdef WIN32\r
57 char *strcasestr (char *p, char *q) { char *r = p; while(*r) *r = tolower(*r), r++; return strstr(p, q); }\r
58 \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"))\r
160 }\r
161 \r
162 void\r
163 StopPonder(int pondering)\r
164 {\r
165         if(!pondering) return;\r
166         pause = 1;\r
167         EPRINT((f, "# stop\n")) fflush(toE); // note: 'pondering' remains set until engine acknowledges 'stop' with 'bestmove'\r
168         Sync(PAUSE); // wait for engine to acknowledge 'stop' with 'bestmove'.\r
169 }\r
170 \r
171 void\r
172 LoadPos(int moveNr)\r
173 {\r
174         int j, lastCapt = 0; char *pos = iniPos, buf[200], stm;\r
175         if(sc == 'x') { // UCCI: send only reversible moves\r
176             lastCapt = Play(moveNr); // find last capture (returns -1 if none!)\r
177             Play(++lastCapt);        // reconstruct board after last capture\r
178             stm = (!strstr(iniPos+4, " b ") ^ lastCapt & 1 ? 'w' :  'b');\r
179             sprintf(buf, "position fen %s", ToFEN(stm)); pos = buf; // send it as FEN (with "position" in UCCI!)\r
180         }\r
181         EPRINT((f, "# %s moves", pos))\r
182         for(j=lastCapt; j<moveNr; j++) EPRINT((f, " %s", move[j]))\r
183         EPRINT((f, "\n"))\r
184 }\r
185 \r
186 void\r
187 StartPonder()\r
188 {\r
189         if(!move[moveNr][0]) return; // no ponder move\r
190         LoadPos(moveNr+1);\r
191         pondering = 1; lastDepth = 1;\r
192         StartSearch(" ponder");\r
193 }\r
194 \r
195 void\r
196 Analyze(char *val)\r
197 {\r
198     if(*anaOpt) EPRINT((f, "# setoption %s%s %s%s\n", nameWord, anaOpt, valueWord, val));\r
199 }\r
200 \r
201 int\r
202 Release()\r
203 {   // send setoption commands backlogged during thinking to engine, aborting ponder or analysis search if necessary\r
204     int len = logLen - sentLen, analyse = searching;\r
205     if(len <= 0) return 0;\r
206     StopPonder(pondering | searching); pondering = searching = 0; // force new search if settings change during analysis (multi-PV!)\r
207     fwrite(backLog + sentLen, 1, len, toE); sentLen += len; DPRINT("# release %d\n", len);\r
208     if(ponder && computer == 1 - stm) StartPonder(); // (re)start ponder search\r
209     return analyse; // return 1 if analysis search should be restarted\r
210 }\r
211 \r
212 char *Convert(char *pv)\r
213 {   // convert Shogi coordinates to WB\r
214     char *p, *q, c;\r
215     static char buf[10000];\r
216     if(sc != 's') return pv;\r
217     p = pv; q = buf;\r
218     while(c = *p++) {\r
219         if(c >= '0' && c <= '9' || c >= 'a' && c <= 'z') *q++ = 'a'+'0'+size - c; else *q++ = c;\r
220     }\r
221     *q++ = 0;\r
222     return buf;\r
223 }\r
224 \r
225 void\r
226 Move4GUI(char *m)\r
227 {\r
228     if(sc == 's') {\r
229       // convert USI move to WB format\r
230       m[2] = 'a'+'0'+size - m[2];\r
231       m[3] = 'a'+'0'+size - m[3];\r
232       if(m[1] == '*') { // drop\r
233         m[1] = '@';\r
234       } else {\r
235         m[0] = 'a'+'0'+size - m[0];\r
236         m[1] = 'a'+'0'+size - m[1];\r
237         if((stm == WHITE ? (m[1]>'0'+size-size/3 || m[3]>'0'+size-size/3)\r
238                                 : (m[1] <= '0'+size/3 || m[3] <= '0'+size/3)) && m[4] != '+')\r
239              m[4] = '=', m[5] = 0;\r
240       }\r
241     }\r
242 }\r
243 \r
244 int\r
245 GetChar()\r
246 {\r
247     int c;\r
248     if(fromF) {\r
249         if((c = fgetc(fromF)) != EOF) return c;\r
250         fclose(fromF); fromF = 0; printf("# end fake\n");\r
251     }\r
252     return fgetc(fromE);\r
253 }\r
254 \r
255 void *\r
256 Engine2GUI()\r
257 {\r
258     char line[1024], command[256];\r
259 \r
260     if(fromF = fopen("DefectiveEngineOptions.ini", "r")) printf("# fake engine input\n");\r
261     while(1) {\r
262         int i=0, x; char *p, dummy, len;\r
263 \r
264         fflush(stdout); fflush(toE);\r
265         while((line[i] = x = GetChar()) != EOF && line[i] != '\n') i++;\r
266         line[++i] = 0;\r
267         if(x == EOF) exit(0);\r
268         DPRINT("# engine said: %s", line), fflush(stdout);\r
269         if(sscanf(line, "%s", command) != 1) continue;\r
270         if(!strcmp(command, "bestmove")) {\r
271             if(pause == 1) { pondering = pause = 0; Sync(WAKEUP); continue; } // bestmove was reply to ponder miss or analysis result; ignore.\r
272             else if(pondering) { pondering = 0; printf("%d 0 0 0 UCI violation! Engine moves during ponder\n", lastDepth+1); continue; } // ignore ponder search\r
273             // move was a move to be played\r
274             if(p = strstr(line+8, " draw")) *p = 0, printf("offer draw\n"); // UCCI\r
275             if(strstr(line+9, "resign")) { printf("resign\n"); computer = NONE; }\r
276             if(strstr(line+9, "win")) { printf("%s {claim}\n", stm== WHITE ? "1-0" :"0-1"); computer = NONE; } // USI\r
277             if(strstr(line+9, "(none)") || strstr(line+9, "null") ||\r
278                strstr(line+9, "0000")) { printf("%s\n", lastScore < -99999 ? "resign" : "1/2-1/2 {stalemate}"); computer = NONE; }\r
279             sscanf(line, "bestmove %s", move[moveNr++]);\r
280             Release(); // send setoption commands that arrived during search\r
281             myTime -= (GetTickCount() - startTime)*1.02 + inc; // update own clock, so we can give correct wtime, btime with ponder\r
282             if(mps && ((moveNr+1)/2) % mps == 0) myTime += tc; if(sTime) myTime = sTime; // new session or move starts\r
283             stm = WHITE+BLACK - stm;\r
284             // first start a new ponder search, if pondering is on and we have a move to ponder on\r
285             if(p = strstr(line+9, "ponder")) {\r
286               sscanf(p+7, "%s", move[moveNr]);\r
287               if(computer != NONE && ponder) {\r
288                 DPRINT("# ponder on %s\n", move[moveNr]);\r
289                 StartPonder();\r
290               }\r
291               p[-1] = '\n'; *p = 0; // strip off ponder move\r
292             } else move[moveNr][0] = 0;\r
293             Move4GUI(line+9);\r
294             printf("move %s\n", line+9); // send move to GUI\r
295             if(move[moveNr][0]) printf("Hint: %s\n", move[moveNr]);\r
296             if(pause) { pause = 0; Sync(WAKEUP); } // release commands that came in during think\r
297             if(lastScore == 100001 && iniPos[0] != 'f') { printf("%s {mate}\n", stm == BLACK ? "1-0" : "0-1"); computer = NONE; }\r
298         }\r
299         else if(!strcmp(command, "info")) {\r
300             int d=0, s=0, t=(GetTickCount() - startTime)/10, n=1;\r
301             char *pv, varName[80];\r
302             if(sscanf(line+5, "string times @ %c", &dummy) == 1) { printf("# %s", line+12); continue; }\r
303             if(sscanf(line+5, "string variant %s", varName) == 1) {\r
304                 if(!strstr(STDVARS, varName) && (p = strstr(line+18, " startpos "))) printf("setup (-) 8x8+0_fairy %s", p+10);\r
305                 continue;\r
306             }\r
307             if(collect && (pv = strstr(line+5, "currmove "))) {\r
308                 if(p = strstr(line+5, "currmovenumber ")) {\r
309                     n = atoi(p+15);\r
310                     if(collect == 1 && n != 1) continue; // wait for move 1\r
311                     if(collect + (n == 1) > 2) { // done collecting\r
312                         if(inex && collect == 2) printf("%d 0 0 0 OK to exclude\n", lastDepth);\r
313                         collect = 3; continue;\r
314                     }\r
315                     collect = 2; on[nr=n] = 1; sscanf(pv+9, "%s", moveMap[n]); continue; // store move\r
316                 }\r
317             }\r
318             if(!post) continue;\r
319             if(sscanf(line+5, "string %c", &dummy) == 1) printf("%d 0 0 0 %s", lastDepth, line+12); else {\r
320                 if(p = strstr(line+4, " depth "))      sscanf(p+7, "%d", &d), statDepth = d;\r
321                 if(p = strstr(line+4, " score cp "))   sscanf(p+10, "%d", &s), statScore = s; else\r
322                 if(p = strstr(line+4, " score mate ")) sscanf(p+12, "%d", &s), s += s>0 ? 100000 : -100000, statScore = s; else\r
323                 if(p = strstr(line+4, " score "))      sscanf(p+7, "%d", &s), statScore = s;\r
324                 if(p = strstr(line+4, " nodes "))      sscanf(p+7, "%d", &n), statNodes = n;\r
325                 if(p = strstr(line+4, " time "))       sscanf(p+6, "%d", &t), t /= 10, statTime = t;\r
326                 if(p = strstr(line+4, " currmove "))   sscanf(p+10,"%s", currMove);\r
327                 if(p = strstr(line+4, " currmovenumber ")) sscanf(p+16,"%d", &currNr);\r
328                 if(pv = strstr(line+4, " pv ")) // convert PV info to WB thinking output\r
329                     printf("%3d  %6d %6d %10d %s", lastDepth=d, lastScore=s, t, n, Convert(pv+4));\r
330             }\r
331         }\r
332         else if(!strcmp(command, "option")) { // USI option: extract data fields\r
333             char name[80], type[80], buf[1024], val[256], *q;\r
334             int min=0, max=1e9;\r
335             if(p = strstr(line+6, " type ")) sscanf(p+1, "type %s", type), *p = '\n';\r
336             if(p = strstr(line+6, " min "))  sscanf(p+1, "min %d", &min), *p = '\n';\r
337             if(p = strstr(line+6, " max "))  sscanf(p+1, "max %d", &max), *p = '\n';\r
338             if(p = strstr(line+6, " default "))  sscanf(p+1, "default %[^\n]*", val), *p = '\n';\r
339             if(!(p = strstr(line+6, " name "))) p = line+1; sscanf(p+6, "%[^\n]", name); // 'name' is omitted in UCCI\r
340             if(!strcasecmp(name, "UCI_Chess960")) { frc=2; continue; }\r
341             if(!strcasecmp(name, "UCI_Variant")) { if(p = strstr(line+6, " var ")) strcpy(varList, p); varOpt = 1; continue; }\r
342             if(!strcasecmp(name, "UCI_Opponent")) { namOpt = 1; continue; }\r
343             if(!strcasecmp(name+2, "I_AnalyseMode")) { strcpy(anaOpt, name); continue; }\r
344             if(frc< 0 && (strstr(name, "960") || strcasestr(name, "frc")) && !strcmp(type, "check")) {\r
345                 EPRINT((f, "# setoption name %s value true\n", name)) strcpy(val, "true"); // set non-standard suspected FRC options\r
346             }\r
347             if(!strcasecmp(name, "Threads")) { strcpy(threadOpt, name); continue; }\r
348             if(!strcasecmp(name, "Ponder") || !strcasecmp(name, "USI_Ponder")) { strcpy(canPonder, name); continue; }\r
349             if(!strcasecmp(name, "Hash") || !strcasecmp(name, "USI_Hash") || !strcasecmp(name, "hashsize")) {\r
350                 memory = oldMem = atoi(val); hasHash = 1; \r
351                 strcpy(hashOpt, name);\r
352                 continue;\r
353             }\r
354             if(!strcasecmp(name, "newgame") && !strcmp(type, "button")) { newGame++; continue; }\r
355             if(!strcasecmp(name, "usemillisec")) { unit = (!strcmp(val, "false") ? 2 : 1); continue; }\r
356             // pass on engine-defined option as WB option feature\r
357             if(!strcmp(type, "filename")) type[4] = 0;\r
358             sprintf(buf, "feature option=\"%s -%s", name, type); q = buf + strlen(buf);\r
359             if(     !strcmp(type, "file")\r
360                  || !strcmp(type, "string")) sprintf(q, " %s\"\n", val);\r
361             else if(!strcmp(type, "spin"))   sprintf(q, " %d %d %d\"\n", atoi(val), min, max);\r
362             else if(!strcmp(type, "check"))  sprintf(q, " %d\"\n", strcmp(val, "true") ? 0 : 1), strcat(checkOptions, name);\r
363             else if(!strcmp(type, "button")) sprintf(q, "\"\n");\r
364             else if(!strcmp(type, "combo")) {\r
365                 if(p = strstr(line+6, " default "))  sscanf(p+1, "default %s", type); // current setting\r
366                 min = 0; p = line+6;\r
367                 while(p = strstr(p, " var ")) {\r
368                     sscanf(p += 5, "%s", val); // next choice\r
369                     sprintf(buf + strlen(buf), "%s%s%s", min++ ? " /// " : " ", strcmp(type, val) ? "" : "*", val);\r
370                 }\r
371                 strcat(q, "\"\n");\r
372             }\r
373             else buf[0] = 0; // ignore unrecognized option types\r
374             if(buf[0]) printf("%s", buf);\r
375         }\r
376         else if(!strcmp(command, "id")) {\r
377             static char name[256], version[256];\r
378             if(sscanf(line, "id name %[^\n]", name) == 1) printf("feature myname=\"%s (U%cI2WB)\"\n", name, sc-32);\r
379             if(sscanf(line, "id version %[^\n]", version) == 1 && *name) printf("feature myname=\"%s %s (U%cI2WB)\"\n", name, version, sc-32);\r
380         }\r
381         else if(!strcmp(command, "readyok")) { pause = 0; Sync(WAKEUP); } // resume processing of GUI commands\r
382         else if(sc == 'x'&& !strcmp(command, "ucciok") || sscanf(command, "u%ciok", &c)==1 && c==sc) {\r
383             char *p = varList, *q = varList;\r
384             while(*q && *q != '\n')  if(!strncmp(q, " var ", 5)) *p++ = ',', q +=5; // replace var keywords by commas\r
385                                 else if(!strncmp(q-1, " chess ", 7)) strcpy(p, "normal"), p += 6, q += 5; // 'chess' is called 'normal' in CECP\r
386                                 else if(!strncmp(q-1, " threecheck", 11)) *p++ = '3', q += 5; // 'threecheck' is written '3check' in CECP\r
387                                 else *p++ = *q++; // copy other variant names unmodified\r
388             if(frc) sprintf(p, ",normal,fischerandom"), printf("feature oocastle=%d\n", frc<0); // unannounced FRC uses O-O castling\r
389             if(*varList) printf("feature variants=\"%s\"\n", varList+1); // from UCI_Variant combo and/or UCI_Chess960 check options\r
390             printf("feature smp=1 memory=%d done=1\n", hasHash);\r
391             if(unit == 2) { unit = 1; EPRINT((f, "# setoption usemillisec true\n")) }\r
392             Sync(WAKEUP); // done with options\r
393         }\r
394     }\r
395 }\r
396 \r
397 void\r
398 Move4Engine(char *m)\r
399 {\r
400     if(sc == 's') {\r
401       // convert input move to USI format\r
402       if(m[1] == '@') { // drop\r
403         m[1] = '*';\r
404       } else {\r
405         m[0] = 'a'+'0'+size - m[0];\r
406         m[1] = 'a'+'0'+size - m[1];\r
407       }\r
408       m[2] = 'a'+'0'+size - m[2];\r
409       m[3] = 'a'+'0'+size - m[3];\r
410       if(m[4] == '=') m[4] = 0; // no '=' in USI format!\r
411       else if(m[4]) m[4] = '+'; // cater to WB 4.4 bug :-(\r
412     }\r
413 }\r
414 \r
415 void\r
416 GUI2Engine()\r
417 {\r
418     char line[256], command[256], *p, *q, *r, mySide;\r
419 \r
420     while(1) {\r
421         int i, x, difficult, think=0;\r
422 \r
423         if((computer == stm || computer == ANALYZE && !searching) && !suspended) {\r
424             DPRINT("# start search\n");\r
425             LoadPos(moveNr); fflush(stdout); // load position\r
426             // and set engine thinking (note USI swaps colors!)\r
427             startTime = GetTickCount(); mySide = stm; // remember side we last played for\r
428             if(computer == ANALYZE) {\r
429                 EPRINT((f, "# go infinite"))\r
430                 if(sm & 1) { // some moves are disabled\r
431                     EPRINT((f, " searchmoves"))\r
432                     for(i=1; i<nr; i++) if(on[i]) EPRINT((f, " %s", moveMap[i]))\r
433                 }\r
434                 EPRINT((f, "\n")) searching = 1; // suppresses spurious commands during analysis starting new searches\r
435             } else pause = think = 2, StartSearch(""); // request suspending of input processing while thinking\r
436         }\r
437       nomove:\r
438        for(difficult=0; !difficult; ) { // read and handle commands that can (or must) be handled during thinking\r
439         fflush(toE); fflush(stdout);\r
440         i = 0; while((x = getchar()) != EOF && (line[i] = x) != '\n') i++;\r
441         line[++i] = 0; if(x == EOF) { printf("# EOF\n"); sprintf(line, "quit -1\n"); }\r
442         if(think && !pause) Sync(PAUSE), think = 0, Release(); // if no longer thinking, take dummy pause\r
443         sscanf(line, "%s", command); DPRINT("# '%s' think=%d pause=%d log=%d sent=%d\n", command, think, pause, logLen, sentLen);\r
444         if(!strcmp(command, "usermove")) { difficult--; break; } // for efficiency during game play, moves, time & otim are tried first\r
445         else if(!strcmp(command, "time"))   sscanf(line+4, "%d", &myTime),  myTime  = (10*myTime)/unit;\r
446         else if(!strcmp(command, "otim"))   sscanf(line+4, "%d", &hisTime), hisTime = (10*hisTime)/unit;\r
447         else if(!strcmp(command, "offer")) drawOffer = 1; // backlogged anyway, so this can be done instantly\r
448         else if(!strcmp(command, "post"))  post = 1;\r
449         else if(!strcmp(command, "nopost"))post = 0;\r
450         else if(!strcmp(command, "pause")) {\r
451             if(computer == stm) myTime -= GetTickCount() - startTime;\r
452             suspended = 1 + pondering; // remember if we were pondering, and stop search ignoring bestmove\r
453             StopPonder(pondering || computer == stm);\r
454         }\r
455         else { //convert easy & hard to "option" after treating their effect on the adapter\r
456           if(!strcmp(command, "easy")) {\r
457             if(*canPonder) ponder = 0, sprintf(command, "option"), sprintf(line, "option %s=0\n", canPonder); else continue;\r
458           }\r
459           else if(!strcmp(command, "hard")) {\r
460             if(*canPonder) ponder = 1, sprintf(command, "option"), sprintf(line, "option %s=1\n", canPonder); else continue;\r
461           }\r
462           if(!strcmp(command, "option")) {\r
463             char *p;\r
464             if(logLen == sentLen) logLen = 0, sentLen = 0; // engine is up to date; reset buffer\r
465             if(sscanf(line+7, "UCI2WB debug output=%d", &debug) == 1) ; else\r
466             if(sscanf(line+7, "Floating Byoyomi=%d", &flob) == 1) ; else\r
467             if(sscanf(line+7, "Byoyomi=%d", &byo) == 1) ; else\r
468             if(p = strchr(line, '=')) {\r
469                 *p++ = 0;\r
470                 if(strstr(checkOptions, line+7)) sprintf(p, "%s\n", atoi(p) ? "true" : "false");\r
471                 snprintf(backLog+logLen, 9999-logLen, "setoption %s%s %s%s", nameWord, line+7, valueWord, p);\r
472             } else snprintf(backLog+logLen, 9999-logLen, "setoption %s%s\n", nameWord, line+7);\r
473             DPRINT("# backlog: %s", backLog+logLen); logLen += strlen(backLog+logLen);\r
474             if(!think && Release()) break; // break will restart analysis; pondering is restarted by Release itself\r
475           }\r
476           else difficult = 1; // difficult command; terminate loop for easy ones\r
477         }\r
478        } // next command\r
479 \r
480         // some commands that should never come during thinking can be safely processed here\r
481         if(difficult < 0) { // used as kludge to signal "usermove" was already matched\r
482             sscanf(line, "usermove %s", command); // strips off linefeed\r
483             Move4Engine(command);\r
484             stm = WHITE+BLACK - stm; collect = (computer == ANALYZE); sm = 0;\r
485             // when pondering we either continue the ponder search as normal search, or abort it\r
486             if(pondering || computer == ANALYZE) {\r
487                 if(pondering && !strcmp(command, move[moveNr])) { // ponder hit\r
488                     char *draw = drawOffer ? " draw" : ""; drawOffer = 0;\r
489                     pondering = 0; pause = 2; moveNr++; startTime = GetTickCount(); // clock starts running now\r
490                     EPRINT((f, "# ponderhit%s\n", draw)) fflush(toE); fflush(stdout);\r
491                     think = 2; // request blocking input during thinking\r
492                     goto nomove;\r
493                 }\r
494                 StopPonder(1); searching = 0;\r
495             }\r
496             strcpy(move[moveNr++], command); // possibly overwrites ponder move\r
497             continue;\r
498         }\r
499         if(!strcmp(command, "resume")) {\r
500             if(suspended == 2) StartPonder(); // restart interrupted ponder search\r
501             suspended = think = 0; continue;  // causes thinking to start in normal way if on move or analyzing\r
502         }\r
503         if(think) { // command arrived during thinking; order abort for 'instant commands'\r
504             if(!strcmp(command, "?") || !strcmp(command, "quit") ||\r
505                !strcmp(command, "force") || !strcmp(command, "result")) { EPRINT((f, "# stop\n")); fflush(toE); }\r
506             Sync(PAUSE); Release(); // block processing of difficult commands during thinking; send backlog left because of race\r
507         }\r
508         if(!strcmp(command, "new")) {\r
509             computer = BLACK; moveNr = 0; depth = -1; move[0][0] = 0;\r
510             stm = WHITE; strcpy(iniPos, "position startpos"); frc &= ~1;\r
511             if(memory != oldMem && hasHash) EPRINT((f, "# setoption %s%s %s%d\n", nameWord, hashOpt, valueWord, memory))\r
512             oldMem = memory;\r
513             // we can set other options here\r
514             if(sc == 'x') { if(newGame) EPRINT((f, "# setoption newgame\n")) } else // optional in UCCI\r
515             if(varOpt) EPRINT((f, "# setoption name UCI_Variant value chess\n"))\r
516             pause = 1; // wait for option settings to take effect\r
517             EPRINT((f, "# isready\n")) fflush(toE);\r
518             Sync(PAUSE); // wait for readyok\r
519             EPRINT((f, "# u%cinewgame\n", sc)) fflush(toE);\r
520         }\r
521         else if(!strcmp(command, "level")) {\r
522             int sec = 0;\r
523             sscanf(line, "level %d %d:%d %d", &mps, &tc, &sec, &inc) == 4 ||\r
524             sscanf(line, "level %d %d %d", &mps, &tc, &inc);\r
525             tc = (60*tc + sec)*1000; inc *= 1000; sTime = 0; tc /= unit; inc /= unit;\r
526         }\r
527         else if(!strcmp(command, "protover")) {\r
528             if(!variants) variants = sc=='s' ? "shogi,5x5+5_shogi" : VARIANTS;\r
529             printf("feature variants=\"%s\" setboard=1 usermove=1 debug=1 ping=1 name=1 reuse=0 exclude=1 pause=1 sigint=0 sigterm=0 done=0\n", variants);\r
530             printf("feature option=\"UCI2WB debug output -check %d\"\n", debug);\r
531             if(sc == 's') printf("feature option=\"Floating Byoyomi -check %d\"\nfeature option=\"Byoyomi -spin %d -1 1000\"\n", flob, byo);\r
532             EPRINT((f, sc == 'x' ? "# ucci\n" : "u%ci\n", sc)) fflush(toE); // prompt UCI engine for options\r
533             Sync(PAUSE); // wait for uciok\r
534         }\r
535         else if(!strcmp(command, "setboard")) {\r
536                 stm = (strstr(line+9, " b ") ? BLACK : WHITE);\r
537                 if((p = strchr(line+9, '[')) && !varOpt) { char c;\r
538                     *p++ = 0; q = strchr(p, ']'); *q = 0; r = q + 4; \r
539                     if(sc == 's') q[2] = 'w' + 'b' - q[2], strcpy(r=q+3, " 1\n"); // Shogi: reverse color\r
540                     else r = strchr(strchr(q+4, ' ') + 1, ' '); // skip to second space (after e.p. square)\r
541                     *r = 0; sprintf(command, "%s%s %s %s", line+9, q+1, p, r+1);\r
542                 } else strcpy(command, line+9);\r
543                 if(frc == -1 && (p = strchr(command, ' '))) strncpy(p+3, "KQkq", 4); // unannounced FRC\r
544                 sprintf(iniPos, "%s%sfen %s", iniPos[0]=='p' ? "position " : "", sc=='s' ? "s" : "", command);\r
545                 iniPos[strlen(iniPos)-1] = sm = 0; collect = (computer == ANALYZE);\r
546         }\r
547         else if(!strcmp(command, "variant")) {\r
548                 if(varOpt) {\r
549                     EPRINT((f, "# setoption name UCI_Variant value %sucinewgame\nisready\n", strcmp(line+8, "3check\n") ? line+8 : "threecheck\n"))\r
550                     fflush(toE); Sync(PAUSE);\r
551                 }\r
552                 if(!strcmp(line+8, "shogi\n")) size = 9, strcpy(iniPos, "position startpos");\r
553                 if(!strcmp(line+8, "5x5+5_shogi\n")) size = 5, strcpy(iniPos, "position startpos");\r
554                 if(!strcmp(line+8, "xiangqi\n")) strcpy(iniPos, "fen rnbakabnr/9/1c5c1/p1p1p1p1p/9/9/P1P1P1P1P/1C5C1/9/RNBAKABNR r");\r
555                 if(!strcmp(line+8, "fischerandom\n")) { frc |= 1; if(frc > 0) EPRINT((f, "# setoption name UCI_Chess960 value true\n")) }\r
556         }\r
557         else if(!strcmp(command, "undo") && (i=1) || !strcmp(command, "remove") && (i=2)) {\r
558             if(pondering || computer == ANALYZE) StopPonder(1), searching = 0;\r
559             moveNr = moveNr > i ? moveNr - i : 0; collect = (computer == ANALYZE); sm = 0;\r
560         }\r
561         else if(!strcmp(command, ".")) {\r
562             printf("stat01: %d %d %d %d 100 %s\n", statTime, statNodes, statDepth, 100-currNr, currMove);\r
563             goto nomove;\r
564         }\r
565         else if(!strcmp(command+2, "clude") && collect > 2) { // include or exclude\r
566             int all = !strcmp(line+8, "all"), in = command[1] == 'n';\r
567             inex = 1; line[strlen(line)-1] = sm = 0; // strip LF and clear sm flag\r
568             for(i=1; i<nr; i++) { if(!strcmp(line+8, moveMap[i]) || all) on[i] = in; sm |= on[i]+1; } // sm: 2 = enabled, 1 = disabled\r
569             if(!(sm & 2)) goto nomove; // no moves enabled; continue current search\r
570             if(computer == ANALYZE) StopPonder(1), searching = 0; // abort old analysis\r
571         }\r
572         else if(!strcmp(command, "xboard")) ;\r
573         else if(!strcmp(command, "analyze"))computer = ANALYZE, collect = 1, sm = 0, Analyze("true");\r
574         else if(!strcmp(command, "exit"))   computer = NONE, StopPonder(1), searching = 0, Analyze("false");\r
575         else if(!strcmp(command, "force"))  computer = NONE, StopPonder(pondering);\r
576         else if(!strcmp(command, "go"))     computer = stm;\r
577         else if(!strcmp(command, "ping"))   { /* static int done; if(!done) pause = 1, fprintf(toE, "isready\n"), fflush(toE), printf("# send isready\n"), fflush(stdout), Sync(PAUSE); done = 1;*/ printf("po%s", line+2); }\r
578         else if(!strcmp(command, "memory")) sscanf(line, "memory %d", &memory);\r
579         else if(!strcmp(command, "cores")&& !!*threadOpt) { sscanf(line, "cores %d", &cores); EPRINT((f, "# setoption %s%s %s%d\n", nameWord, threadOpt, valueWord, cores)) }\r
580         else if(!strcmp(command, "sd"))     sscanf(line, "sd %d", &depth);\r
581         else if(!strcmp(command, "st"))     sscanf(line, "st %d", &sTime), sTime = 1000*sTime - 30, inc = 0, sTime /= unit;\r
582         else if(!strcmp(command, "name"))   { if(namOpt) EPRINT((f, "# setoption name UCI_Opponent value none none %s %s", comp ? "computer" : "human", line+5)) }\r
583         else if(!strcmp(command, "computer")) comp = 1;\r
584         else if(!strcmp(command, "result")) {\r
585             if(sc == 's') EPRINT((f, "# gameover %s\n", line[8] == '/' ? "draw" : (line[7] == '0') == mySide ? "win" : "lose"))\r
586             computer = NONE;\r
587         }\r
588         else if(!strcmp(command, "quit"))   { EPRINT((f, "# quit\n")) fflush(toE), exit(atoi(line+4)); }\r
589     }\r
590 }\r
591 \r
592 int\r
593 StartEngine(char *cmdLine, char *dir)\r
594 {\r
595 #ifdef WIN32\r
596   HANDLE hChildStdinRd, hChildStdinWr,\r
597     hChildStdoutRd, hChildStdoutWr;\r
598   BOOL fSuccess;\r
599   PROCESS_INFORMATION piProcInfo;\r
600   STARTUPINFO siStartInfo;\r
601   DWORD err;\r
602 \r
603   /* Create a pipe for the child's STDOUT. */\r
604   if (! WinPipe(&hChildStdoutRd, &hChildStdoutWr)) return GetLastError();\r
605 \r
606   /* Create a pipe for the child's STDIN. */\r
607   if (! WinPipe(&hChildStdinRd, &hChildStdinWr)) return GetLastError();\r
608 \r
609   SetCurrentDirectory(dir); // go to engine directory\r
610 \r
611   /* Now create the child process. */\r
612   siStartInfo.cb = sizeof(STARTUPINFO);\r
613   siStartInfo.lpReserved = NULL;\r
614   siStartInfo.lpDesktop = NULL;\r
615   siStartInfo.lpTitle = NULL;\r
616   siStartInfo.dwFlags = STARTF_USESTDHANDLES;\r
617   siStartInfo.cbReserved2 = 0;\r
618   siStartInfo.lpReserved2 = NULL;\r
619   siStartInfo.hStdInput = hChildStdinRd;\r
620   siStartInfo.hStdOutput = hChildStdoutWr;\r
621   siStartInfo.hStdError = hChildStdoutWr;\r
622 \r
623   fSuccess = CreateProcess(NULL,\r
624                            cmdLine,        /* command line */\r
625                            NULL,           /* process security attributes */\r
626                            NULL,           /* primary thread security attrs */\r
627                            TRUE,           /* handles are inherited */\r
628                            DETACHED_PROCESS|CREATE_NEW_PROCESS_GROUP,\r
629                            NULL,           /* use parent's environment */\r
630                            NULL,\r
631                            &siStartInfo, /* STARTUPINFO pointer */\r
632                            &piProcInfo); /* receives PROCESS_INFORMATION */\r
633 \r
634   if (! fSuccess) return GetLastError();\r
635 \r
636 //  if (0) { // in the future we could trigger this by an argument\r
637 //    SetPriorityClass(piProcInfo.hProcess, GetWin32Priority(appData.niceEngines));\r
638 //  }\r
639 \r
640   /* Close the handles we don't need in the parent */\r
641   CloseHandle(piProcInfo.hThread);\r
642   CloseHandle(hChildStdinRd);\r
643   CloseHandle(hChildStdoutWr);\r
644 \r
645   process = piProcInfo.hProcess;\r
646   pid = piProcInfo.dwProcessId;\r
647   fromE = (FILE*) _fdopen( _open_osfhandle((long)hChildStdoutRd, _O_TEXT|_O_RDONLY), "r");\r
648   toE   = (FILE*) _fdopen( _open_osfhandle((long)hChildStdinWr, _O_WRONLY), "w");\r
649 #else\r
650     char *argv[10], *p, buf[200];\r
651     int i, toEngine[2], fromEngine[2];\r
652 \r
653     if (dir && dir[0] && chdir(dir)) { perror(dir); exit(1); }\r
654     pipe(toEngine); pipe(fromEngine); // create two pipes\r
655 \r
656     if ((pid = fork()) == 0) { // Child\r
657         dup2(toEngine[0], 0);   close(toEngine[0]);   close(toEngine[1]);   // stdin from toE pipe\r
658         dup2(fromEngine[1], 1); close(fromEngine[0]); close(fromEngine[1]); // stdout into fromE pipe\r
659         dup2(1, fileno(stderr)); // stderr into frome pipe\r
660 \r
661         strcpy(buf, cmdLine); p = buf;\r
662         for (i=0;;) { argv[i++] = p; p = strchr(p, ' '); if (p == NULL) break; *p++ = 0; }\r
663         argv[i] = NULL;\r
664         execvp(argv[0], argv); // startup engine\r
665         \r
666         perror(argv[0]); exit(1); // could not start engine; quit.\r
667     }\r
668     signal(SIGPIPE, SIG_IGN);\r
669     close(toEngine[0]); close(fromEngine[1]); // close engine ends of pipes in adapter\r
670     \r
671     fromE = (FILE*) fdopen(fromEngine[0], "r"); // make into high-level I/O\r
672     toE   = (FILE*) fdopen(toEngine[1], "w");\r
673 #endif\r
674   return NO_ERROR;\r
675 }\r
676 \r
677 main(int argc, char **argv)\r
678 {\r
679         char *dir = NULL, *p, *q; int e;\r
680 \r
681         if(argc == 2 && !strcmp(argv[1], "-v")) { printf("UCI2WB " VERSION " by H.G.Muller\n"); exit(0); }\r
682         if(argc > 1 && !strcmp(argv[1], "debug")) { debug = 1; argc--; argv++; }\r
683         if(argc > 1 && !strcmp(argv[1], "-var")) { variants = argv[2]; argc-=2; argv+=2; }\r
684         if(argc > 1 && argv[1][0] == '-') { sc = argv[1][1]; argc--; argv++; }\r
685         if(argc < 2) { printf("usage is: U%cI2WB [debug] [-s] <engine.exe> [<engine directory>]\n", sc-32); exit(-1); }\r
686         if(argc > 2) dir = argv[2];\r
687         if(argc > 3) strncpy(suffix, argv[3], 80);\r
688 \r
689         if(sc == 'x') nameWord = valueWord = bTime = "", wTime = "opp", bInc = "increment", wInc = "oppincrement", unit = 1000; // switch to UCCI keywords\r
690         else if(sc == 'f' ) frc = -1, sc = 'c';   // UCI for unannounced Chess960\r
691         else if(sc == 'n') sc = 'c'; // UCI for normal Chess\r
692 \r
693         // spawn engine proc\r
694         if(StartEngine(argv[1], dir) != NO_ERROR) { perror(argv[1]), exit(-1); }\r
695 \r
696         Sync(INIT);\r
697 \r
698         // create separate thread to handle engine->GUI traffic\r
699 #ifdef WIN32\r
700         CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE) Engine2GUI, (LPVOID) NULL, 0, &thread_id);\r
701 #else\r
702         { pthread_t t; signal(SIGINT, SIG_IGN); signal(SIGTERM, SIG_IGN); pthread_create(&t, NULL, Engine2GUI, NULL); }\r
703 #endif\r
704 \r
705         // handle GUI->engine traffic in original thread\r
706         GUI2Engine();\r
707 }\r