From f72070254816c26ab9e52862e6191b2a59029e4d Mon Sep 17 00:00:00 2001
From: H.G. Muller <h.g.muller@hccnet.nl>
Date: Thu, 19 Apr 2012 21:18:28 +0200
Subject: [PATCH] Implement WB exclude feature

UCI2WB collects moves sent by the engne in analysis mode from the currmove,
currmovenumber info commands. After the full set for the current position
has been completed, UCI2Wb becomes sensitive for WB include and exclude
commands, to switch the moves in the list on or off. If not all moves
are on, the 'go infinite' will be accompanied by 'searchmoves'. If all
moves are off, the current search is allowed to continue.
---
 UCI2WB.c |   38 ++++++++++++++++++++++++++++++--------
 1 files changed, 30 insertions(+), 8 deletions(-)

diff --git a/UCI2WB.c b/UCI2WB.c
index acb34d9..71dfd4b 100644
--- a/UCI2WB.c
+++ b/UCI2WB.c
@@ -32,7 +32,7 @@
 
 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, on[500]; char currMove[20], moveMap[500][10]; // for analyze mode
 
 FILE *toE, *fromE;
 int pid;
@@ -187,6 +187,14 @@ 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) { collect = 3; continue; } // done collecting
+		    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;
@@ -279,8 +287,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);
@@ -301,7 +316,7 @@ GUI2Engine()
 	else if(!strcmp(command, "usermove")) {
 	    sscanf(line, "usermove %s", command); // strips off linefeed
 	    Move4Engine(command);
-	    stm = WHITE+BLACK - stm;
+	    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
@@ -330,7 +345,7 @@ GUI2Engine()
 	}
 	else if(!strcmp(command, "protover")) {
 	    if(!variants) variants = sc=='s' ? "shogi,5x5+5_shogi" : VARIANTS;
-	    printf("feature variants=\"%s\" setboard=1 usermove=1 debug=1 ping=1 reuse=0 done=0\n", 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); fflush(toE); // this prompts UCI engine for options
 	    Sync(PAUSE); // wait for uciok
@@ -344,7 +359,7 @@ GUI2Engine()
 		    *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] = 0;
+		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");
@@ -353,14 +368,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) {
+	    int all = !strcmp(line+8, "all"), in = command[1] == 'n';
+	    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;
-- 
1.7.0.4