X-Git-Url: http://winboard.nl/cgi-bin?a=blobdiff_plain;f=UCI2WB.c;h=c075731fd01b4cf87d80ddc08d0eee70e4cb3ff5;hb=19d15a4533382e6c7ff78165279151bef8209aee;hp=a0abcfe1d5cd3347c93a159b34e918726be54dfa;hpb=19fd1d671bb688d926f1f1c6b7ab4abc304b6db1;p=uci2wb.git

diff --git a/UCI2WB.c b/UCI2WB.c
index a0abcfe..c075731 100644
--- a/UCI2WB.c
+++ b/UCI2WB.c
@@ -1,6 +1,6 @@
 /************************* UCI2WB by H.G.Muller ****************************/
 
-#define VERSION "1.6"
+#define VERSION "1.9"
 
 #include <stdio.h>
 #include <stdlib.h>
@@ -30,9 +30,9 @@
 #define NONE  2
 #define ANALYZE 3
 
-char move[2000][10], checkOptions[8192], iniPos[256], hashOpt[20], pause, pondering, ponder, post, hasHash, c, sc='c', *suffix;
+char move[2000][10], checkOptions[8192], iniPos[256], hashOpt[20], pause, pondering, ponder, post, hasHash, c, sc='c', *suffix, *variants;
 int mps, tc, inc, sTime, depth, myTime, hisTime, stm, computer = NONE, memory, oldMem=0, cores, moveNr, lastDepth, lastScore, startTime, debug;
-int statDepth, statScore, statNodes, statTime, currNr, size; char currMove[20]; // for analyze mode
+int statDepth, statScore, statNodes, statTime, currNr, size, collect, nr, sm, inex, on[500]; char currMove[20], moveMap[500][10]; // for analyze mode
 
 FILE *toE, *fromE;
 int pid;
@@ -122,6 +122,25 @@ char *Convert(char *pv)
     return buf;
 }
 
+void
+Move4GUI(char *m)
+{
+    if(sc == 's') {
+      // convert USI move to WB format
+      m[2] = 'a'+'0'+size - m[2];
+      m[3] = 'a'+'0'+size - m[3];
+      if(m[1] == '*') { // drop
+	m[1] = '@';
+      } else {
+	m[0] = 'a'+'0'+size - m[0];
+	m[1] = 'a'+'0'+size - m[1];
+	if((stm == WHITE ? (m[1]>'0'+size-size/3 || m[3]>'0'+size-size/3)
+                                : (m[1] <= '0'+size/3 || m[3] <= '0'+size/3)) && m[4] != '+')
+	     m[4] = '=', m[5] = 0;
+      }
+    }
+}
+
 void *
 Engine2GUI()
 {
@@ -157,20 +176,7 @@ Engine2GUI()
 	      }
 	      p[-1] = '\n'; *p = 0; // strip off ponder move
 	    }
-	    if(sc == 's') {
-	      // convert USI move to WB format
-	      line[11] = 'a'+'0'+size - line[11];
-	      line[12] = 'a'+'0'+size - line[12];
-	      if(line[10] == '*') { // drop
-		line[10] = '@';
-	      } else {
-		line[9]  = 'a'+'0'+size - line[9];
-		line[10] = 'a'+'0'+size - line[10];
-		if((stm == WHITE ? (line[10]>'0'+size-size/3 || line[12]>'0'+size-size/3)
-                                 : (line[10] <= '0'+size/3 || line[12] <= '0'+size/3)) && line[13] != '+')
-		     line[13] = '=', line[14] = 0;
-	      }
-	    }
+	    Move4GUI(line+9);
 	    printf("move %s\n", line+9); // send move to GUI
 	    if(lastScore == 100001 && iniPos[0] != 'f') { printf("%s {mate}\n", stm == WHITE ? "1-0" : "0-1"); computer = NONE; }
 	    stm = WHITE+BLACK - stm;
@@ -179,6 +185,17 @@ Engine2GUI()
 	    int d=0, s=0, t=0, n=0;
 	    char *pv;
 	    if(sscanf(line+5, "string times @ %c", &dummy) == 1) { printf("# %s", line+12); continue; }
+	    if(collect && (pv = strstr(line+5, "currmove "))) {
+		if(p = strstr(line+5, "currmovenumber ")) {
+		    n = atoi(p+15);
+		    if(collect == 1 && n != 1) continue; // wait for move 1
+		    if(collect + (n == 1) > 2) { // done collecting
+			if(inex && collect == 2) printf("%d 0 0 0 OK to exclude\n", lastDepth);
+			collect = 3; continue;
+		    }
+		    collect = 2; on[nr=n] = 1; sscanf(pv+9, "%s", moveMap[n]); continue; // store move
+		}
+	    }
 	    if(!post) continue;
 	    if(sscanf(line+5, "string %c", &dummy) == 1) printf("%d 0 0 0 %s", lastDepth, line+12); else {
 		if(p = strstr(line+4, " depth "))      sscanf(p+7, "%d", &d), statDepth = d;
@@ -231,14 +248,32 @@ Engine2GUI()
 	    if(sscanf(line, "id name %[^\n]", name) == 1) printf("feature myname=\"%s (U%cI2WB)\"\n", name, sc-32);
 	}
 	else if(!strcmp(command, "readyok")) { pause = 0; Sync(WAKEUP); } // resume processing of GUI commands
-	else if(sscanf(command, "u%ciok", &c)==1 && c==sc)   printf("feature smp=1 memory=%d done=1\n", hasHash); // done with options
+	else if(sscanf(command, "u%ciok", &c)==1 && c==sc)   printf("feature smp=1 memory=%d done=1\n", hasHash), Sync(WAKEUP); // done with options
+    }
+}
+
+void
+Move4Engine(char *m)
+{
+    if(sc == 's') {
+      // convert input move to USI format
+      if(m[1] == '@') { // drop
+	m[1] = '*';
+      } else {
+	m[0] = 'a'+'0'+size - m[0];
+	m[1] = 'a'+'0'+size - m[1];
+      }
+      m[2] = 'a'+'0'+size - m[2];
+      m[3] = 'a'+'0'+size - m[3];
+      if(m[4] == '=') m[4] = 0; // no '=' in USI format!
+      else if(m[4] != '\n') m[4] = '+'; // cater to WB 4.4 bug :-(
     }
 }
 
 void
 GUI2Engine()
 {
-    char line[256], command[256], *p, *q;
+    char line[256], command[256], *p, *q, *r;
 
     while(1) {
 	int i, x;
@@ -248,8 +283,15 @@ GUI2Engine()
 	    LoadPos(moveNr); // load position
 	    // and set engine thinking (note USI swaps colors!)
 	    startTime = GetTickCount();
-	    if(computer == ANALYZE) fprintf(toE, "\ngo infinite\n"), printf("\ngo infinite\n");
-	    else StartSearch("");
+	    if(computer == ANALYZE) {
+		fprintf(toE, "\ngo infinite"); DPRINT("\n# go infinite");
+		if(sm & 1) { // some moves are disabled
+		    fprintf(toE, " searchmoves"); DPRINT(" searchmoves");
+		    for(i=1; i<nr; i++) if(on[i]) { fprintf(toE, " %s", moveMap[i]); DPRINT(" %s", moveMap[i]); }
+		}
+		fprintf(toE, "\n"); DPRINT("\n");
+	    // code for searchmoves goes here
+	    } else StartSearch("");
 	}
       nomove:
 	fflush(toE); fflush(stdout);
@@ -268,21 +310,9 @@ GUI2Engine()
 	    Sync(PAUSE); // wait for readyok
 	}
 	else if(!strcmp(command, "usermove")) {
-	    if(sc == 's') {
-	      // convert input move to USI format
-	      if(line[10] == '@') { // drop
-		line[10] = '*';
-	      } else {
-		line[9]  = 'a'+'0'+size - line[9];
-		line[10] = 'a'+'0'+size - line[10];
-	      }
-	      line[11] = 'a'+'0'+size - line[11];
-	      line[12] = 'a'+'0'+size - line[12];
-	      if(line[13] == '=') line[13] = 0; // no '=' in USI format!
-	      else if(line[13] != '\n') line[13] = '+'; // cater to WB 4.4 bug :-(
-	    }
 	    sscanf(line, "usermove %s", command); // strips off linefeed
-	    stm = WHITE+BLACK - stm;
+	    Move4Engine(command);
+	    stm = WHITE+BLACK - stm; collect = (computer == ANALYZE); sm = 0;
 	    // when pondering we either continue the ponder search as normal search, or abort it
 	    if(pondering || computer == ANALYZE) {
 		if(pondering && !strcmp(command, move[moveNr])) { // ponder hit
@@ -292,7 +322,7 @@ GUI2Engine()
 		}
 		StopPonder(1);
 	    }
-	    sscanf(line, "usermove %s", move[moveNr++]); // possibly overwrites ponder move
+	    strcpy(move[moveNr++], command); // possibly overwrites ponder move
 	}
 	else if(!strcmp(command, "level")) {
 	    int sec = 0;
@@ -310,17 +340,22 @@ GUI2Engine()
 	    } else { fprintf(toE, "setoption name %s\n", line+7); DPRINT("# setoption name %s\n", line+7); }
 	}
 	else if(!strcmp(command, "protover")) {
-	    printf("feature variants=\"%s\" setboard=1 usermove=1 debug=1 ping=1 reuse=0 done=0\n", sc=='s' ? "shogi,5x5+5_shogi" : VARIANTS);
+	    if(!variants) variants = sc=='s' ? "shogi,5x5+5_shogi" : VARIANTS;
+	    printf("feature variants=\"%s\" setboard=1 usermove=1 debug=1 ping=1 reuse=0 exclude=1 done=0\n", variants);
 	    printf("feature option=\"UCI2WB debug output -check %d\"\n", debug);
-	    fprintf(toE, "u%ci\n", sc); // this prompts UCI engine for options
+	    fprintf(toE, "u%ci\n", sc); fflush(toE); // this prompts UCI engine for options
+	    Sync(PAUSE); // wait for uciok
 	}
 	else if(!strcmp(command, "setboard")) {
 		if(strstr(line+9, " b ")) stm = BLACK;
-                if(sc == 's' && (p = strchr(line+9, '['))) { char c;
-                    *p++ = ' '; q = strchr(p, ']'); c = 'w' + 'b' - q[2]; strcpy(q+2, " 1\n"); while(*--q != ' ') q[2] = *q; *p = c; p[1] = ' '; 
-                }
-		sprintf(iniPos, "%s%sfen %s", iniPos[0]=='p' ? "position " : "", sc=='s' ? "s" : "", line+9);
-		iniPos[strlen(iniPos)-1] = 0;
+                if(p = strchr(line+9, '[')) { char c;
+                    *p++ = 0; q = strchr(p, ']'); *q = 0; r = q + 4; 
+		    if(sc == 's') q[2] = 'w' + 'b' - q[2], strcpy(r=q+3, " 1\n"); // Shogi: reverse color
+		    else r = strchr(strchr(q+4, ' ') + 1, ' '); // skip to second space (after e.p. square)
+		    *r = 0; sprintf(command, "%s%s %s %s", line+9, q+1, p, r+1);
+                } else strcpy(command, line+9);
+		sprintf(iniPos, "%s%sfen %s", iniPos[0]=='p' ? "position " : "", sc=='s' ? "s" : "", command);
+		iniPos[strlen(iniPos)-1] = sm = 0; collect = (computer == ANALYZE);
 	}
 	else if(!strcmp(command, "variant")) {
 		if(!strcmp(line+8, "shogi\n")) size = 9, strcpy(iniPos, "position startpos");
@@ -329,14 +364,21 @@ GUI2Engine()
 	}
 	else if(!strcmp(command, "undo") && (i=1) || !strcmp(command, "remove") && (i=2)) {
 	    if(pondering || computer == ANALYZE) StopPonder(1);
-	    moveNr = moveNr > i ? moveNr - i : 0;
+	    moveNr = moveNr > i ? moveNr - i : 0; collect = (computer == ANALYZE); sm = 0;
 	}
 	else if(!strcmp(command, ".")) {
 	    printf("stat01: %d %d %d %d 100 %s\n", statTime, statNodes, statDepth, 100-currNr, currMove);
 	    goto nomove;
 	}
+	else if(!strcmp(command+2, "clude") && collect > 2) { // include or exclude
+	    int all = !strcmp(line+8, "all"), in = command[1] == 'n';
+	    inex = 1; line[strlen(line)-1] = sm = 0; // strip LF and clear sm flag
+	    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
+	    if(!(sm & 2)) goto nomove; // no moves enabled; continue current search
+	    if(computer == ANALYZE) StopPonder(1); // abort old analysis
+	}
 	else if(!strcmp(command, "xboard")) ;
-	else if(!strcmp(command, "analyze"))computer = ANALYZE;
+	else if(!strcmp(command, "analyze"))computer = ANALYZE, collect = 1, sm = 0;
 	else if(!strcmp(command, "exit"))   computer = NONE, StopPonder(1);
 	else if(!strcmp(command, "force"))  computer = NONE, StopPonder(pondering);
 	else if(!strcmp(command, "go"))     computer = stm;
@@ -350,7 +392,7 @@ GUI2Engine()
 	else if(!strcmp(command, "memory")) sscanf(line, "memory %d", &memory);
 	else if(!strcmp(command, "cores"))  sscanf(line, "cores %d", &cores);
 	else if(!strcmp(command, "sd"))     sscanf(line, "sd %d", &depth);
-	else if(!strcmp(command, "st"))     sscanf(line, "st %d", &sTime), sTime *= 1000, inc = 0;
+	else if(!strcmp(command, "st"))     sscanf(line, "st %d", &sTime), sTime = 1000*sTime - 30, inc = 0;
 	else if(!strcmp(command, "quit"))   fprintf(toE, "quit\n"), fflush(toE), exit(0);
     }
 }
@@ -446,6 +488,7 @@ main(int argc, char **argv)
 
 	if(argc == 2 && !strcmp(argv[1], "-v")) { printf("UCI2WB " VERSION " by H.G.Muller\n"); exit(0); }
 	if(argc > 1 && !strcmp(argv[1], "debug")) { debug = 1; argc--; argv++; }
+	if(argc > 1 && !strcmp(argv[1], "-var")) { variants = argv[2]; argc-=2; argv+=2; }
 	if(argc > 1 && argv[1][0] == '-') { sc = argv[1][1]; argc--; argv++; }
 	if(argc < 2) { printf("usage is: U%cI2WB [debug] [-s] <engine.exe> [<engine directory>]\n", sc-32); exit(-1); }
 	if(argc > 2) dir = argv[2];