After a "stop" do not read new input until search finishes

[stockfish] / src / uci.cpp

diff --git a/src/uci.cpp b/src/uci.cpp
index b7f72d9d4c8cb655df23ef9822f40724502e9d7b..57a31ad64f11f4bb0ef788a374ba67a3bd76e150 100644 (file)
--- a/src/uci.cpp
+++ b/src/uci.cpp
@@ -45,7 +45,7 @@ namespace {
 
   void set_option(istringstream& up);
   void set_position(Position& pos, istringstream& up);
-  bool go(Position& pos, istringstream& up);
+  void go(Position& pos, istringstream& up);
   void perft(Position& pos, istringstream& up);
 }
 
@@ -61,19 +61,35 @@ void uci_loop() {
   string cmd, token;
   bool quit = false;
 
-  while (!quit)
+  while (!quit && getline(cin, cmd))
   {
-      Threads.getline(cmd);
-
       istringstream is(cmd);
 
       is >> skipws >> token;
 
-      if (token == "quit")
-          quit = true;
+      if (cmd == "quit" || cmd == "stop")
+      {
+          quit = (token == "quit");
+          Search::Signals.stop = true;
+          Threads[0].wake_up(); // In case is waiting for stop or ponderhit
+          Threads.wait_end_of_search(); // Block here until search finishes
+      }
+
+      else if (cmd == "ponderhit")
+      {
+          // The opponent has played the expected move. GUI sends "ponderhit" if
+          // we were told to ponder on the same move the opponent has played. We
+          // should continue searching but switching from pondering to normal search.
+          Search::Limits.ponder = false; // FIXME racing
+
+          if (Search::Signals.stopOnPonderhit)
+              Search::Signals.stop = true;
+
+          Threads[0].wake_up(); // In case is waiting for stop or ponderhit
+      }
 
       else if (token == "go")
-          quit = !go(pos, is);
+          go(pos, is);
 
       else if (token == "ucinewgame")
           pos.from_fen(StarFEN, false);
@@ -187,19 +203,21 @@ namespace {
   // string, and then calls think(). Returns false if a quit command
   // is received while thinking, true otherwise.
 
-  bool go(Position& pos, istringstream& is) {
+  void go(Position& pos, istringstream& is) {
 
     string token;
-    SearchLimits limits;
-    std::vector<Move> searchMoves;
     int time[] = { 0, 0 }, inc[] = { 0, 0 };
 
+    memset(&Search::Limits, 0, sizeof(Search::Limits));
+    Search::RootMoves.clear();
+    Search::RootPosition = &pos;
+
     while (is >> token)
     {
         if (token == "infinite")
-            limits.infinite = true;
+            Search::Limits.infinite = true;
         else if (token == "ponder")
-            limits.ponder = true;
+            Search::Limits.ponder = true;
         else if (token == "wtime")
             is >> time[WHITE];
         else if (token == "btime")
@@ -209,23 +227,23 @@ namespace {
         else if (token == "binc")
             is >> inc[BLACK];
         else if (token == "movestogo")
-            is >> limits.movesToGo;
+            is >> Search::Limits.movesToGo;
         else if (token == "depth")
-            is >> limits.maxDepth;
+            is >> Search::Limits.maxDepth;
         else if (token == "nodes")
-            is >> limits.maxNodes;
+            is >> Search::Limits.maxNodes;
         else if (token == "movetime")
-            is >> limits.maxTime;
+            is >> Search::Limits.maxTime;
         else if (token == "searchmoves")
             while (is >> token)
-                searchMoves.push_back(move_from_uci(pos, token));
+                Search::RootMoves.push_back(move_from_uci(pos, token));
     }
 
-    searchMoves.push_back(MOVE_NONE);
-    limits.time = time[pos.side_to_move()];
-    limits.increment = inc[pos.side_to_move()];
+    Search::RootMoves.push_back(MOVE_NONE);
+    Search::Limits.time = time[pos.side_to_move()];
+    Search::Limits.increment = inc[pos.side_to_move()];
 
-    return think(pos, limits, &searchMoves[0]);
+    Threads.start_thinking();
   }
 
 
@@ -243,7 +261,7 @@ namespace {
 
     time = get_system_time();
 
-    n = perft(pos, depth * ONE_PLY);
+    n = Search::perft(pos, depth * ONE_PLY);
 
     time = get_system_time() - time;