Add routine to check for input
[hachu.git] / hachu.c
diff --git a/hachu.c b/hachu.c
index 2b89d21..ff5cc8f 100644 (file)
--- a/hachu.c
+++ b/hachu.c
 \r
 #ifdef WIN32 \r
 #    include <windows.h>\r
+     int InputWaiting()\r
+     {  // checks for waiting input in pipe\r
+       static int pipe, init;\r
+       static HANDLE inp;\r
+       DWORD cnt;\r
+\r
+       if(!init) inp = GetStdHandle(STD_INPUT_HANDLE);\r
+       if(!PeekNamedPipe(inp, NULL, 0, NULL, &cnt, NULL)) return 1;\r
+       return cnt;\r
+    }\r
 #else\r
 #    include <sys/time.h>\r
+#    include <sys/ioctl.h>\r
+     int InputWaiting()\r
+     {\r
+       int cnt;\r
+       if(ioctl(0, FIONREAD, &cnt)) return 1;\r
+       return cnt;\r
+     }\r
      int GetTickCount() // with thanks to Tord\r
      { struct timeval t;\r
        gettimeofday(&t, NULL);\r
@@ -114,7 +131,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 +1583,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 +1593,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 +1618,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 +1799,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 +1867,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 +1967,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 +2114,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 +2226,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
@@ -2244,14 +2282,36 @@ printf("# setup done");fflush(stdout);
       else printf("0-1%s\n", tail);\r
     }\r
 \r
+    void GetLine(int root)\r
+    {\r
+      int i, c;\r
+      while(1) {\r
+        // wait for input, and read it until we have collected a complete line\r
+        for(i = 0; (inBuf[i] = c = getchar()) != '\n'; i++) if(c == EOF || i>7997) exit(0);\r
+        inBuf[i+1] = 0;\r
+\r
+        // extract the first word\r
+        sscanf(inBuf, "%s", command);\r
+        if(!strcmp(command, "otim"))    { continue; } // do not start pondering after receiving time commands, as move will follow immediately\r
+        if(!strcmp(command, "time"))    { sscanf(inBuf, "time %d", &timeLeft); continue; }\r
+        if(!strcmp(command, "put"))     { ReadSquare(inBuf+4, &lastPut); continue; }  // ditto\r
+        if(!strcmp(command, "."))       { inBuf[0] = 0; return; } // ignore for now\r
+        if(!strcmp(command, "lift"))    { inBuf[0] = 0; Highlight(inBuf+5); return; } // treat here\r
+        return;\r
+      }\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
@@ -2261,10 +2321,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
@@ -2307,22 +2368,14 @@ pboard(board);
           }\r
         }\r
 \r
-      noPonder:\r
-        // wait for input, and read it until we have collected a complete line\r
-        for(i = 0; (inBuf[i] = getchar()) != '\n'; i++);\r
-        inBuf[i+1] = 0;\r
-\r
-        // extract the first word\r
-        sscanf(inBuf, "%s", command);\r
-printf("in: %s\n", command);\r
+        fflush(stdout);         // make sure everything is printed before we do something that might take time\r
+        if(!*inBuf) GetLine(1); // takes care of time and otim commands\r
 \r
         // recognize the command,and execute it\r
         if(!strcmp(command, "quit"))    { break; } // breaks out of infinite loop\r
         if(!strcmp(command, "force"))   { engineSide = NONE;    continue; }\r
         if(!strcmp(command, "analyze")) { engineSide = ANALYZE; continue; }\r
         if(!strcmp(command, "exit"))    { engineSide = NONE;    continue; }\r
-        if(!strcmp(command, "otim"))    { goto noPonder; } // do not start pondering after receiving time commands, as move will follow immediately\r
-        if(!strcmp(command, "time"))    { sscanf(inBuf, "time %d", &timeLeft); goto noPonder; }\r
         if(!strcmp(command, "level"))   {\r
           int min, sec=0;\r
           sscanf(inBuf, "level %d %d %d", &mps, &min, &inc) == 3 ||  // if this does not work, it must be min:sec format\r
@@ -2356,8 +2409,6 @@ printf("in: %s\n", command);
         if(!strcmp(command, "nopost"))  { postThinking = OFF;continue; }\r
         if(!strcmp(command, "random"))  { randomize = ON;    continue; }\r
         if(!strcmp(command, "hint"))    { if(ponderMove != INVALID) printf("Hint: %s\n", MoveToText(ponderMove, 0)); continue; }\r
-        if(!strcmp(command, "lift"))    { Highlight(inBuf+5); continue; }\r
-        if(!strcmp(command, "put"))     { ReadSquare(inBuf+4, &lastPut); continue; }\r
         if(!strcmp(command, "book"))    {  continue; }\r
        // non-standard commands\r
         if(!strcmp(command, "p"))       { pboard(board); continue; }\r