Implement abort of search on long thinking time
[hachu.git] / hachu.c
diff --git a/hachu.c b/hachu.c
index 2f4a767..478a20b 100644 (file)
--- a/hachu.c
+++ b/hachu.c
@@ -114,7 +114,8 @@ typedef struct {
 char *array, fenArray[4000], *reason;\r
 int bWidth, bHeight, bsize, zone, currentVariant, chuFlag, tenFlag, chessFlag, repDraws;\r
 int stm, xstm, hashKeyH, hashKeyL, framePtr, msp, nonCapts, rootEval, retMSP, retFirst, retDep, pvPtr, level, cnt50, mobilityScore;\r
-int nodes, startTime, tlim1, tlim2, repCnt, comp;\r
+int nodes, startTime, tlim1, tlim2, tlim3, repCnt, comp, abortFlag;\r
+Move ponderMove;\r
 Move retMove, moveStack[10000], path[100], repStack[300], pv[1000], repeatMove[300], killer[100][2];\r
 \r
       int maxDepth;                            // used by search\r
@@ -1565,6 +1566,8 @@ FireSet (UndoInfo *tb)
     if(p[i].pos != ABSENT) tb->fireMask |= fireFlags[i-2];\r
 }\r
 \r
+void TerminationCheck();\r
+\r
 #define QSdepth 0\r
 \r
 int\r
@@ -1573,7 +1576,7 @@ Search (int alpha, int beta, int difEval, int depth, int oldPromo, int promoSupp
   int i, j, k, firstMove, oldMSP = msp, curMove, sorted, bad, phase, king, iterDep, replyDep, nextVictim, to, defer, dubious, bestMoveNr;\r
   int resDep;\r
   int myPV = pvPtr;\r
-  int score, bestScore, curEval, iterAlpha;\r
+  int score, bestScore, oldBest, curEval, iterAlpha;\r
   Move move, nullMove;\r
   UndoInfo tb;\r
 #ifdef HASH\r
@@ -1598,7 +1601,7 @@ if(PATH) /*pboard(board),pmap(attacks, BLACK),*/printf("search(%d) {%d,%d} eval=
   alpha -= (alpha < curEval);\r
   beta  -= (beta <= curEval);\r
 \r
-  nodes++;\r
+  if(!(nodes++ & 4095)) TerminationCheck();\r
   pv[pvPtr++] = 0; // start empty PV, directly behind PV of parent\r
 \r
 \r
@@ -1779,6 +1782,11 @@ level--;
     repetition:\r
       UnMake(&tb);\r
       xstm = stm; stm ^= WHITE;\r
+      if(abortFlag > 0) { // unwind search\r
+printf("# abort (%d) @ %d\n", abortFlag, level);\r
+        if(curMove == firstMove) bestScore = oldBest, bestMoveNr = firstMove; // none searched yet\r
+        goto leave;\r
+      }\r
 #if 1\r
 if(PATH) printf("%d:%2d:%d %3d %6x %-10s %6d %6d\n", level, depth, iterDep, curMove, moveStack[curMove], MoveToText(moveStack[curMove], 0), score, bestScore);\r
 \r
@@ -1842,6 +1850,7 @@ if(PATH) printf("%d:%2d:%d %3d %6x %-10s %6d %6d\n", level, depth, iterDep, curM
     } else hashTable[index].move[hit] = 0;\r
 #endif\r
   } // next depth\r
+leave:\r
   retMSP = msp;\r
   retFirst = firstMove;\r
   msp = oldMSP; // pop move list\r
@@ -1941,7 +1950,7 @@ typedef Move MOVE;
     void SetMemorySize(int n);              // if n is different from last time, resize all tables to make memory usage below n MB\r
     char *MoveToText(MOVE move, int m);     // converts the move from your internal format to text like e2e2, e1g1, a7a8q.\r
     MOVE ParseMove(char *moveText);         // converts a long-algebraic text move to your internal move format\r
-    int  SearchBestMove(int stm, int timeLeft, int mps, int timeControl, int inc, int timePerMove, MOVE *move, MOVE *ponderMove);\r
+    int  SearchBestMove(MOVE *move, MOVE *ponderMove);\r
     void PonderUntilInput(int stm);         // Search current position for stm, deepening forever until there is input.\r
 \r
 UndoInfo undoInfo;\r
@@ -2088,7 +2097,7 @@ ListMoves ()
   int i;\r
   for(i=0; i< BSIZE; i++) boardCopy[i] = !!board[i];\r
 MapFromScratch(attacks);\r
-  postThinking--; repCnt = 0; tlim1 = tlim2 = 1e8; msp = 0;\r
+  postThinking--; repCnt = 0; tlim1 = tlim2 = tlim3 = 1e8; abortFlag = msp = 0;\r
   Search(-INF-1, INF+1, 0, 1, sup1 & ~PROMOTE, sup2);\r
   postThinking++;\r
   listStart = retFirst; listEnd = msp = retMSP;\r
@@ -2200,31 +2209,43 @@ Highlight(char *coords)
   printf("highlight %s\n", buf);\r
 }\r
 \r
-int\r
-SearchBestMove (int stm, int timeLeft, int mps, int timeControl, int inc, int timePerMove, MOVE *move, MOVE *ponderMove)\r
+int timeLeft;                            // timeleft on engine's clock\r
+int mps, timeControl, inc, timePerMove;  // time-control parameters, to be used by Search\r
+char inBuf[8000], command[80], ponderMoveText[20];\r
+\r
+void\r
+SetSearchTimes (int timeLeft)\r
 {\r
-  int score, targetTime, movesLeft = BW*BH/4 + 20;\r
+  int targetTime, movesLeft = BW*BH/4 + 20;\r
   if(mps) movesLeft = mps - (moveNr>>1)%mps;\r
-  targetTime = timeLeft*10 / (movesLeft + 2) + 1000 * inc;\r
+  targetTime = (timeLeft - 1000*inc) / (movesLeft + 2) + 1000 * inc;\r
   if(moveNr < 30) targetTime *= 0.5 + moveNr/60.; // speedup in opening\r
-  if(timePerMove > 0) targetTime = timeLeft * 5;\r
-  startTime = GetTickCount();\r
+  if(timePerMove > 0) targetTime = 0.5*timeLeft, movesLeft = 1;\r
   tlim1 = 0.2*targetTime;\r
   tlim2 = 1.9*targetTime;\r
+  tlim3 = 5*timeLeft / (movesLeft + 4.1);\r
+}\r
+\r
+int\r
+SearchBestMove (MOVE *move, MOVE *ponderMove)\r
+{\r
+  int score;\r
+  startTime = GetTickCount();\r
   nodes = 0;\r
 MapFromScratch(attacks);\r
   retMove = INVALID; repCnt = 0;\r
   score = Search(-INF-1, INF+1, rootEval, maxDepth, sup1, sup2);\r
   *move = retMove;\r
-  *ponderMove = INVALID;\r
+  *ponderMove = pv[1];\r
   return score;\r
 }\r
 \r
+\r
 void\r
 PonderUntilInput (int stm)\r
 {\r
 }\r
-\r
\r
     int TakeBack(int n)\r
     { // reset the game and then replay it to the desired point\r
       int last, stm;\r
@@ -2263,14 +2284,17 @@ printf("# setup done");fflush(stdout);
       }\r
     }\r
 \r
+    void\r
+    TerminationCheck()\r
+    {\r
+        if(GetTickCount() - startTime > tlim3) abortFlag = 2;\r
+    }\r
+\r
     main()\r
     {\r
       int engineSide=NONE;                     // side played by engine\r
-      int timeLeft;                            // timeleft on engine's clock\r
-      int mps, timeControl, inc, timePerMove;  // time-control parameters, to be used by Search\r
-      MOVE move, ponderMove;\r
+      MOVE move;\r
       int i, score, curVarNr;\r
-      char inBuf[8000], command[80];\r
 \r
   Init(V_CHU); // Chu\r
 \r
@@ -2280,10 +2304,11 @@ printf("# setup done");fflush(stdout);
 \r
         if(listEnd == 0) ListMoves();   // always maintain a list of legal moves in root position\r
 \r
+        abortFlag = 0;\r
         if(stm == engineSide) {         // if it is the engine's turn to move, set it thinking, and let it move\r
      \r
 pboard(board);\r
-          score = SearchBestMove(stm, timeLeft, mps, timeControl, inc, timePerMove, &move, &ponderMove);\r
+          score = SearchBestMove(&move, &ponderMove);\r
 \r
           if(move == INVALID) {         // game apparently ended\r
             int kcapt = 0, xstm = stm ^ WHITE, king, k = p[king=royal[xstm]].pos;\r