mirror of
https://github.com/sockspls/badfish
synced 2025-04-29 16:23:09 +00:00
Introduce Search namespace
Move global search-related variables under "Search" namespace. As a side effect we can move uci_async_command() and wait_for_stop_or_ponderhit() away from search.cpp No functional change. Signed-off-by: Marco Costalba <mcostalba@gmail.com>
This commit is contained in:
parent
ed04c010eb
commit
c4517c013c
8 changed files with 121 additions and 116 deletions
|
@ -76,11 +76,11 @@ void benchmark(int argc, char* argv[]) {
|
||||||
|
|
||||||
// Search should be limited by nodes, time or depth ?
|
// Search should be limited by nodes, time or depth ?
|
||||||
if (valType == "nodes")
|
if (valType == "nodes")
|
||||||
Limits.maxNodes = atoi(valStr.c_str());
|
Search::Limits.maxNodes = atoi(valStr.c_str());
|
||||||
else if (valType == "time")
|
else if (valType == "time")
|
||||||
Limits.maxTime = 1000 * atoi(valStr.c_str()); // maxTime is in ms
|
Search::Limits.maxTime = 1000 * atoi(valStr.c_str()); // maxTime is in ms
|
||||||
else
|
else
|
||||||
Limits.maxDepth = atoi(valStr.c_str());
|
Search::Limits.maxDepth = atoi(valStr.c_str());
|
||||||
|
|
||||||
// Do we need to load positions from a given FEN file?
|
// Do we need to load positions from a given FEN file?
|
||||||
if (fenFile != "default")
|
if (fenFile != "default")
|
||||||
|
@ -105,22 +105,22 @@ void benchmark(int argc, char* argv[]) {
|
||||||
fenList.push_back(Defaults[i]);
|
fenList.push_back(Defaults[i]);
|
||||||
|
|
||||||
// Ok, let's start the benchmark !
|
// Ok, let's start the benchmark !
|
||||||
|
Search::RootMoves.push_back(MOVE_NONE);
|
||||||
totalNodes = 0;
|
totalNodes = 0;
|
||||||
time = get_system_time();
|
time = get_system_time();
|
||||||
SearchMoves.push_back(MOVE_NONE);
|
|
||||||
|
|
||||||
for (size_t i = 0; i < fenList.size(); i++)
|
for (size_t i = 0; i < fenList.size(); i++)
|
||||||
{
|
{
|
||||||
Position pos(fenList[i], false, 0);
|
Position pos(fenList[i], false, 0);
|
||||||
RootPosition = &pos;
|
Search::RootPosition = &pos;
|
||||||
|
|
||||||
cerr << "\nBench position: " << i + 1 << '/' << fenList.size() << endl;
|
cerr << "\nBench position: " << i + 1 << '/' << fenList.size() << endl;
|
||||||
|
|
||||||
if (valType == "perft")
|
if (valType == "perft")
|
||||||
{
|
{
|
||||||
int64_t cnt = perft(pos, Limits.maxDepth * ONE_PLY);
|
int64_t cnt = Search::perft(pos, Search::Limits.maxDepth * ONE_PLY);
|
||||||
|
|
||||||
cerr << "\nPerft " << Limits.maxDepth
|
cerr << "\nPerft " << Search::Limits.maxDepth
|
||||||
<< " nodes counted: " << cnt << endl;
|
<< " nodes counted: " << cnt << endl;
|
||||||
|
|
||||||
totalNodes += cnt;
|
totalNodes += cnt;
|
||||||
|
|
|
@ -53,7 +53,7 @@ int main(int argc, char* argv[]) {
|
||||||
init_bitboards();
|
init_bitboards();
|
||||||
Position::init();
|
Position::init();
|
||||||
kpk_bitbase_init();
|
kpk_bitbase_init();
|
||||||
init_search();
|
Search::init();
|
||||||
Threads.init();
|
Threads.init();
|
||||||
|
|
||||||
#ifdef USE_CALLGRIND
|
#ifdef USE_CALLGRIND
|
||||||
|
|
126
src/search.cpp
126
src/search.cpp
|
@ -42,10 +42,16 @@
|
||||||
using std::cout;
|
using std::cout;
|
||||||
using std::endl;
|
using std::endl;
|
||||||
using std::string;
|
using std::string;
|
||||||
|
using Search::Signals;
|
||||||
|
using Search::Limits;
|
||||||
|
|
||||||
SearchLimits Limits;
|
namespace Search {
|
||||||
std::vector<Move> SearchMoves;
|
|
||||||
Position* RootPosition;
|
volatile SignalsType Signals;
|
||||||
|
LimitsType Limits;
|
||||||
|
std::vector<Move> RootMoves;
|
||||||
|
Position* RootPosition;
|
||||||
|
}
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
|
@ -78,7 +84,7 @@ namespace {
|
||||||
// RootMoveList struct is mainly a std::vector of RootMove objects
|
// RootMoveList struct is mainly a std::vector of RootMove objects
|
||||||
struct RootMoveList : public std::vector<RootMove> {
|
struct RootMoveList : public std::vector<RootMove> {
|
||||||
|
|
||||||
void init(Position& pos, Move searchMoves[]);
|
void init(Position& pos, Move rootMoves[]);
|
||||||
RootMove* find(const Move& m, int startIndex = 0);
|
RootMove* find(const Move& m, int startIndex = 0);
|
||||||
|
|
||||||
int bestMoveChanges;
|
int bestMoveChanges;
|
||||||
|
@ -166,7 +172,6 @@ namespace {
|
||||||
int MultiPV, UCIMultiPV, MultiPVIdx;
|
int MultiPV, UCIMultiPV, MultiPVIdx;
|
||||||
|
|
||||||
// Time management variables
|
// Time management variables
|
||||||
volatile bool StopOnPonderhit, FirstRootMove, StopRequest, AspirationFailLow;
|
|
||||||
TimeManager TimeMgr;
|
TimeManager TimeMgr;
|
||||||
|
|
||||||
// Skill level adjustment
|
// Skill level adjustment
|
||||||
|
@ -179,7 +184,7 @@ namespace {
|
||||||
|
|
||||||
/// Local functions
|
/// Local functions
|
||||||
|
|
||||||
Move id_loop(Position& pos, Move searchMoves[], Move* ponderMove);
|
Move id_loop(Position& pos, Move rootMoves[], Move* ponderMove);
|
||||||
|
|
||||||
template <NodeType NT>
|
template <NodeType NT>
|
||||||
Value search(Position& pos, SearchStack* ss, Value alpha, Value beta, Depth depth);
|
Value search(Position& pos, SearchStack* ss, Value alpha, Value beta, Depth depth);
|
||||||
|
@ -295,7 +300,7 @@ namespace {
|
||||||
|
|
||||||
/// init_search() is called during startup to initialize various lookup tables
|
/// init_search() is called during startup to initialize various lookup tables
|
||||||
|
|
||||||
void init_search() {
|
void Search::init() {
|
||||||
|
|
||||||
int d; // depth (ONE_PLY == 2)
|
int d; // depth (ONE_PLY == 2)
|
||||||
int hd; // half depth (ONE_PLY == 1)
|
int hd; // half depth (ONE_PLY == 1)
|
||||||
|
@ -323,7 +328,7 @@ void init_search() {
|
||||||
/// perft() is our utility to verify move generation. All the leaf nodes up to
|
/// perft() is our utility to verify move generation. All the leaf nodes up to
|
||||||
/// the given depth are generated and counted and the sum returned.
|
/// the given depth are generated and counted and the sum returned.
|
||||||
|
|
||||||
int64_t perft(Position& pos, Depth depth) {
|
int64_t Search::perft(Position& pos, Depth depth) {
|
||||||
|
|
||||||
StateInfo st;
|
StateInfo st;
|
||||||
int64_t sum = 0;
|
int64_t sum = 0;
|
||||||
|
@ -353,7 +358,7 @@ int64_t perft(Position& pos, Depth depth) {
|
||||||
/// variables, and calls id_loop(). It returns false when a "quit" command is
|
/// variables, and calls id_loop(). It returns false when a "quit" command is
|
||||||
/// received during the search.
|
/// received during the search.
|
||||||
|
|
||||||
void think() {
|
void Search::think() {
|
||||||
|
|
||||||
static Book book; // Defined static to initialize the PRNG only once
|
static Book book; // Defined static to initialize the PRNG only once
|
||||||
|
|
||||||
|
@ -362,8 +367,8 @@ void think() {
|
||||||
// Save "search start" time and reset elapsed time to zero
|
// Save "search start" time and reset elapsed time to zero
|
||||||
elapsed_search_time(get_system_time());
|
elapsed_search_time(get_system_time());
|
||||||
|
|
||||||
// Initialize global search-related variables
|
// Reset global search signals
|
||||||
StopOnPonderhit = StopRequest = AspirationFailLow = false;
|
memset((void*)&Signals, 0, sizeof(Signals));
|
||||||
|
|
||||||
// Set output stream mode: normal or chess960. Castling notation is different
|
// Set output stream mode: normal or chess960. Castling notation is different
|
||||||
cout << set960(pos.is_chess960());
|
cout << set960(pos.is_chess960());
|
||||||
|
@ -377,7 +382,7 @@ void think() {
|
||||||
Move bookMove = book.probe(pos, Options["Best Book Move"].value<bool>());
|
Move bookMove = book.probe(pos, Options["Best Book Move"].value<bool>());
|
||||||
if (bookMove != MOVE_NONE)
|
if (bookMove != MOVE_NONE)
|
||||||
{
|
{
|
||||||
if (!StopRequest && (Limits.ponder || Limits.infinite))
|
if (!Signals.stop && (Limits.ponder || Limits.infinite))
|
||||||
Threads.wait_for_stop_or_ponderhit();
|
Threads.wait_for_stop_or_ponderhit();
|
||||||
|
|
||||||
cout << "bestmove " << bookMove << endl;
|
cout << "bestmove " << bookMove << endl;
|
||||||
|
@ -437,7 +442,7 @@ void think() {
|
||||||
|
|
||||||
// We're ready to start thinking. Call the iterative deepening loop function
|
// We're ready to start thinking. Call the iterative deepening loop function
|
||||||
Move ponderMove = MOVE_NONE;
|
Move ponderMove = MOVE_NONE;
|
||||||
Move bestMove = id_loop(pos, &SearchMoves[0], &ponderMove);
|
Move bestMove = id_loop(pos, &RootMoves[0], &ponderMove);
|
||||||
|
|
||||||
// Stop timer, no need to check for available time any more
|
// Stop timer, no need to check for available time any more
|
||||||
Threads.set_timer(0);
|
Threads.set_timer(0);
|
||||||
|
@ -464,7 +469,7 @@ void think() {
|
||||||
// When we reach max depth we arrive here even without a StopRequest, but if
|
// When we reach max depth we arrive here even without a StopRequest, but if
|
||||||
// we are pondering or in infinite search, we shouldn't print the best move
|
// we are pondering or in infinite search, we shouldn't print the best move
|
||||||
// before we are told to do so.
|
// before we are told to do so.
|
||||||
if (!StopRequest && (Limits.ponder || Limits.infinite))
|
if (!Signals.stop && (Limits.ponder || Limits.infinite))
|
||||||
Threads.wait_for_stop_or_ponderhit();
|
Threads.wait_for_stop_or_ponderhit();
|
||||||
|
|
||||||
// Could be MOVE_NONE when searching on a stalemate position
|
// Could be MOVE_NONE when searching on a stalemate position
|
||||||
|
@ -485,7 +490,7 @@ namespace {
|
||||||
// with increasing depth until the allocated thinking time has been consumed,
|
// with increasing depth until the allocated thinking time has been consumed,
|
||||||
// user stops the search, or the maximum search depth is reached.
|
// user stops the search, or the maximum search depth is reached.
|
||||||
|
|
||||||
Move id_loop(Position& pos, Move searchMoves[], Move* ponderMove) {
|
Move id_loop(Position& pos, Move rootMoves[], Move* ponderMove) {
|
||||||
|
|
||||||
SearchStack ss[PLY_MAX_PLUS_2];
|
SearchStack ss[PLY_MAX_PLUS_2];
|
||||||
Value bestValues[PLY_MAX_PLUS_2];
|
Value bestValues[PLY_MAX_PLUS_2];
|
||||||
|
@ -505,7 +510,7 @@ namespace {
|
||||||
ss->currentMove = MOVE_NULL; // Hack to skip update gains
|
ss->currentMove = MOVE_NULL; // Hack to skip update gains
|
||||||
|
|
||||||
// Moves to search are verified and copied
|
// Moves to search are verified and copied
|
||||||
Rml.init(pos, searchMoves);
|
Rml.init(pos, rootMoves);
|
||||||
|
|
||||||
// Handle special case of searching on a mate/stalemate position
|
// Handle special case of searching on a mate/stalemate position
|
||||||
if (!Rml.size())
|
if (!Rml.size())
|
||||||
|
@ -517,7 +522,7 @@ namespace {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Iterative deepening loop until requested to stop or target depth reached
|
// Iterative deepening loop until requested to stop or target depth reached
|
||||||
while (!StopRequest && ++depth <= PLY_MAX && (!Limits.maxDepth || depth <= Limits.maxDepth))
|
while (!Signals.stop && ++depth <= PLY_MAX && (!Limits.maxDepth || depth <= Limits.maxDepth))
|
||||||
{
|
{
|
||||||
// Save now last iteration's scores, before Rml moves are reordered
|
// Save now last iteration's scores, before Rml moves are reordered
|
||||||
for (size_t i = 0; i < Rml.size(); i++)
|
for (size_t i = 0; i < Rml.size(); i++)
|
||||||
|
@ -576,7 +581,7 @@ namespace {
|
||||||
// If search has been stopped exit the aspiration window loop,
|
// If search has been stopped exit the aspiration window loop,
|
||||||
// note that sorting and writing PV back to TT is safe becuase
|
// note that sorting and writing PV back to TT is safe becuase
|
||||||
// Rml is still valid, although refers to the previous iteration.
|
// Rml is still valid, although refers to the previous iteration.
|
||||||
if (StopRequest)
|
if (Signals.stop)
|
||||||
break;
|
break;
|
||||||
|
|
||||||
// Send full PV info to GUI if we are going to leave the loop or
|
// Send full PV info to GUI if we are going to leave the loop or
|
||||||
|
@ -611,8 +616,8 @@ namespace {
|
||||||
}
|
}
|
||||||
else if (bestValue <= alpha)
|
else if (bestValue <= alpha)
|
||||||
{
|
{
|
||||||
AspirationFailLow = true;
|
Signals.failedLowAtRoot = true;
|
||||||
StopOnPonderhit = false;
|
Signals.stopOnPonderhit = false;
|
||||||
|
|
||||||
alpha = std::max(alpha - aspirationDelta, -VALUE_INFINITE);
|
alpha = std::max(alpha - aspirationDelta, -VALUE_INFINITE);
|
||||||
aspirationDelta += aspirationDelta / 2;
|
aspirationDelta += aspirationDelta / 2;
|
||||||
|
@ -644,7 +649,7 @@ namespace {
|
||||||
bestMoveNeverChanged = false;
|
bestMoveNeverChanged = false;
|
||||||
|
|
||||||
// Do we have time for the next iteration? Can we stop searching now?
|
// Do we have time for the next iteration? Can we stop searching now?
|
||||||
if (!StopRequest && !StopOnPonderhit && Limits.useTimeManagement())
|
if (!Signals.stop && !Signals.stopOnPonderhit && Limits.useTimeManagement())
|
||||||
{
|
{
|
||||||
// Take in account some extra time if the best move has changed
|
// Take in account some extra time if the best move has changed
|
||||||
if (depth > 4 && depth < 50)
|
if (depth > 4 && depth < 50)
|
||||||
|
@ -653,11 +658,11 @@ namespace {
|
||||||
// Stop search if most of available time is already consumed. We probably don't
|
// Stop search if most of available time is already consumed. We probably don't
|
||||||
// have enough time to search the first move at the next iteration anyway.
|
// have enough time to search the first move at the next iteration anyway.
|
||||||
if (elapsed_search_time() > (TimeMgr.available_time() * 62) / 100)
|
if (elapsed_search_time() > (TimeMgr.available_time() * 62) / 100)
|
||||||
StopRequest = true;
|
Signals.stop = true;
|
||||||
|
|
||||||
// Stop search early if one move seems to be much better than others
|
// Stop search early if one move seems to be much better than others
|
||||||
if ( depth >= 10
|
if ( depth >= 10
|
||||||
&& !StopRequest
|
&& !Signals.stop
|
||||||
&& ( bestMoveNeverChanged
|
&& ( bestMoveNeverChanged
|
||||||
|| elapsed_search_time() > (TimeMgr.available_time() * 40) / 100))
|
|| elapsed_search_time() > (TimeMgr.available_time() * 40) / 100))
|
||||||
{
|
{
|
||||||
|
@ -669,14 +674,14 @@ namespace {
|
||||||
(ss+1)->excludedMove = MOVE_NONE;
|
(ss+1)->excludedMove = MOVE_NONE;
|
||||||
|
|
||||||
if (v < rBeta)
|
if (v < rBeta)
|
||||||
StopRequest = true;
|
Signals.stop = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// If we are allowed to ponder do not stop the search now but keep pondering
|
// If we are allowed to ponder do not stop the search now but keep pondering
|
||||||
if (StopRequest && Limits.ponder) // FIXME Limits.ponder is racy
|
if (Signals.stop && Limits.ponder) // FIXME Limits.ponder is racy
|
||||||
{
|
{
|
||||||
StopRequest = false;
|
Signals.stop = false;
|
||||||
StopOnPonderhit = true;
|
Signals.stopOnPonderhit = true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -756,7 +761,7 @@ namespace {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Step 2. Check for aborted search and immediate draw
|
// Step 2. Check for aborted search and immediate draw
|
||||||
if (( StopRequest
|
if (( Signals.stop
|
||||||
|| pos.is_draw<false>()
|
|| pos.is_draw<false>()
|
||||||
|| ss->ply > PLY_MAX) && !RootNode)
|
|| ss->ply > PLY_MAX) && !RootNode)
|
||||||
return VALUE_DRAW;
|
return VALUE_DRAW;
|
||||||
|
@ -1015,7 +1020,7 @@ split_point_start: // At split points actual search starts from here
|
||||||
if (RootNode)
|
if (RootNode)
|
||||||
{
|
{
|
||||||
// This is used by time management
|
// This is used by time management
|
||||||
FirstRootMove = (moveCount == 1);
|
Signals.firstRootMove = (moveCount == 1);
|
||||||
|
|
||||||
// Save the current node count before the move is searched
|
// Save the current node count before the move is searched
|
||||||
nodes = pos.nodes_searched();
|
nodes = pos.nodes_searched();
|
||||||
|
@ -1184,7 +1189,7 @@ split_point_start: // At split points actual search starts from here
|
||||||
// was aborted because the user interrupted the search or because we
|
// was aborted because the user interrupted the search or because we
|
||||||
// ran out of time. In this case, the return value of the search cannot
|
// ran out of time. In this case, the return value of the search cannot
|
||||||
// be trusted, and we don't update the best move and/or PV.
|
// be trusted, and we don't update the best move and/or PV.
|
||||||
if (RootNode && !StopRequest)
|
if (RootNode && !Signals.stop)
|
||||||
{
|
{
|
||||||
// Remember searched nodes counts for this move
|
// Remember searched nodes counts for this move
|
||||||
RootMove* rm = Rml.find(move);
|
RootMove* rm = Rml.find(move);
|
||||||
|
@ -1235,7 +1240,7 @@ split_point_start: // At split points actual search starts from here
|
||||||
&& depth >= Threads.min_split_depth()
|
&& depth >= Threads.min_split_depth()
|
||||||
&& bestValue < beta
|
&& bestValue < beta
|
||||||
&& Threads.available_slave_exists(pos.thread())
|
&& Threads.available_slave_exists(pos.thread())
|
||||||
&& !StopRequest
|
&& !Signals.stop
|
||||||
&& !thread.cutoff_occurred())
|
&& !thread.cutoff_occurred())
|
||||||
bestValue = Threads.split<FakeSplit>(pos, ss, alpha, beta, bestValue, depth,
|
bestValue = Threads.split<FakeSplit>(pos, ss, alpha, beta, bestValue, depth,
|
||||||
threatMove, moveCount, &mp, NT);
|
threatMove, moveCount, &mp, NT);
|
||||||
|
@ -1253,7 +1258,7 @@ split_point_start: // At split points actual search starts from here
|
||||||
// Step 21. Update tables
|
// Step 21. Update tables
|
||||||
// If the search is not aborted, update the transposition table,
|
// If the search is not aborted, update the transposition table,
|
||||||
// history counters, and killer moves.
|
// history counters, and killer moves.
|
||||||
if (!SpNode && !StopRequest && !thread.cutoff_occurred())
|
if (!SpNode && !Signals.stop && !thread.cutoff_occurred())
|
||||||
{
|
{
|
||||||
move = bestValue <= oldAlpha ? MOVE_NONE : ss->bestMove;
|
move = bestValue <= oldAlpha ? MOVE_NONE : ss->bestMove;
|
||||||
vt = bestValue <= oldAlpha ? VALUE_TYPE_UPPER
|
vt = bestValue <= oldAlpha ? VALUE_TYPE_UPPER
|
||||||
|
@ -1943,7 +1948,7 @@ split_point_start: // At split points actual search starts from here
|
||||||
|
|
||||||
/// RootMove and RootMoveList method's definitions
|
/// RootMove and RootMoveList method's definitions
|
||||||
|
|
||||||
void RootMoveList::init(Position& pos, Move searchMoves[]) {
|
void RootMoveList::init(Position& pos, Move rootMoves[]) {
|
||||||
|
|
||||||
Move* sm;
|
Move* sm;
|
||||||
bestMoveChanges = 0;
|
bestMoveChanges = 0;
|
||||||
|
@ -1952,11 +1957,11 @@ split_point_start: // At split points actual search starts from here
|
||||||
// Generate all legal moves and add them to RootMoveList
|
// Generate all legal moves and add them to RootMoveList
|
||||||
for (MoveList<MV_LEGAL> ml(pos); !ml.end(); ++ml)
|
for (MoveList<MV_LEGAL> ml(pos); !ml.end(); ++ml)
|
||||||
{
|
{
|
||||||
// If we have a searchMoves[] list then verify the move
|
// If we have a rootMoves[] list then verify the move
|
||||||
// is in the list before to add it.
|
// is in the list before to add it.
|
||||||
for (sm = searchMoves; *sm && *sm != ml.move(); sm++) {}
|
for (sm = rootMoves; *sm && *sm != ml.move(); sm++) {}
|
||||||
|
|
||||||
if (sm != searchMoves && *sm != ml.move())
|
if (sm != rootMoves && *sm != ml.move())
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
RootMove rm;
|
RootMove rm;
|
||||||
|
@ -2138,49 +2143,6 @@ void Thread::idle_loop(SplitPoint* sp) {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// ThreadsManager::wait_for_stop_or_ponderhit() is called when the maximum depth
|
|
||||||
// is reached while the program is pondering. The point is to work around a wrinkle
|
|
||||||
// in the UCI protocol: When pondering, the engine is not allowed to give a
|
|
||||||
// "bestmove" before the GUI sends it a "stop" or "ponderhit" command.
|
|
||||||
// We simply wait here until one of these commands (that raise StopRequest) is
|
|
||||||
// sent, and return, after which the bestmove and pondermove will be printed.
|
|
||||||
|
|
||||||
void ThreadsManager::wait_for_stop_or_ponderhit() {
|
|
||||||
|
|
||||||
StopOnPonderhit = true;
|
|
||||||
|
|
||||||
Thread& main = threads[0];
|
|
||||||
|
|
||||||
lock_grab(&main.sleepLock);
|
|
||||||
|
|
||||||
while (!StopRequest)
|
|
||||||
cond_wait(&main.sleepCond, &main.sleepLock);
|
|
||||||
|
|
||||||
lock_release(&main.sleepLock);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// uci_async_command() is called when a 'cmd' input line is received from the
|
|
||||||
// GUI while searching.
|
|
||||||
|
|
||||||
void uci_async_command(const std::string& cmd) {
|
|
||||||
|
|
||||||
if (cmd == "quit" || cmd == "stop")
|
|
||||||
StopRequest = true;
|
|
||||||
|
|
||||||
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.
|
|
||||||
Limits.ponder = false;
|
|
||||||
|
|
||||||
if (StopOnPonderhit)
|
|
||||||
StopRequest = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
// do_timer_event() is called by the timer thread when the timer triggers
|
// do_timer_event() is called by the timer thread when the timer triggers
|
||||||
|
|
||||||
void do_timer_event() {
|
void do_timer_event() {
|
||||||
|
@ -2201,8 +2163,8 @@ void do_timer_event() {
|
||||||
if (Limits.ponder)
|
if (Limits.ponder)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
bool stillAtFirstMove = FirstRootMove
|
bool stillAtFirstMove = Signals.firstRootMove
|
||||||
&& !AspirationFailLow
|
&& !Signals.failedLowAtRoot
|
||||||
&& e > TimeMgr.available_time();
|
&& e > TimeMgr.available_time();
|
||||||
|
|
||||||
bool noMoreTime = e > TimeMgr.maximum_time()
|
bool noMoreTime = e > TimeMgr.maximum_time()
|
||||||
|
@ -2211,5 +2173,5 @@ void do_timer_event() {
|
||||||
if ( (Limits.useTimeManagement() && noMoreTime)
|
if ( (Limits.useTimeManagement() && noMoreTime)
|
||||||
|| (Limits.maxTime && e >= Limits.maxTime)
|
|| (Limits.maxTime && e >= Limits.maxTime)
|
||||||
/* missing nodes limit */ ) // FIXME
|
/* missing nodes limit */ ) // FIXME
|
||||||
StopRequest = true;
|
Signals.stop = true;
|
||||||
}
|
}
|
||||||
|
|
18
src/search.h
18
src/search.h
|
@ -46,26 +46,34 @@ struct SearchStack {
|
||||||
int skipNullMove;
|
int skipNullMove;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
namespace Search {
|
||||||
|
|
||||||
/// The SearchLimits struct stores information sent by GUI about available time
|
/// The SearchLimits struct stores information sent by GUI about available time
|
||||||
/// to search the current move, maximum depth/time, if we are in analysis mode
|
/// to search the current move, maximum depth/time, if we are in analysis mode
|
||||||
/// or if we have to ponder while is our opponent's side to move.
|
/// or if we have to ponder while is our opponent's side to move.
|
||||||
|
|
||||||
struct SearchLimits {
|
struct LimitsType {
|
||||||
|
|
||||||
bool useTimeManagement() const { return !(maxTime | maxDepth | maxNodes | infinite); }
|
bool useTimeManagement() const { return !(maxTime | maxDepth | maxNodes | infinite); }
|
||||||
|
|
||||||
int time, increment, movesToGo, maxTime, maxDepth, maxNodes, infinite, ponder;
|
int time, increment, movesToGo, maxTime, maxDepth, maxNodes, infinite, ponder;
|
||||||
};
|
};
|
||||||
|
|
||||||
extern SearchLimits Limits;
|
struct SignalsType {
|
||||||
extern std::vector<Move> SearchMoves;
|
bool stopOnPonderhit, firstRootMove, stop, failedLowAtRoot;
|
||||||
|
};
|
||||||
|
|
||||||
|
extern volatile SignalsType Signals;
|
||||||
|
extern LimitsType Limits;
|
||||||
|
extern std::vector<Move> RootMoves;
|
||||||
extern Position* RootPosition;
|
extern Position* RootPosition;
|
||||||
|
|
||||||
extern void init_search();
|
extern void init();
|
||||||
extern int64_t perft(Position& pos, Depth depth);
|
extern int64_t perft(Position& pos, Depth depth);
|
||||||
extern void think();
|
extern void think();
|
||||||
extern void uci_async_command(const std::string& cmd);
|
|
||||||
|
}
|
||||||
|
|
||||||
extern void do_timer_event();
|
extern void do_timer_event();
|
||||||
|
|
||||||
#endif // !defined(SEARCH_H_INCLUDED)
|
#endif // !defined(SEARCH_H_INCLUDED)
|
||||||
|
|
|
@ -418,7 +418,7 @@ void Thread::main_loop() {
|
||||||
if (do_terminate)
|
if (do_terminate)
|
||||||
return;
|
return;
|
||||||
|
|
||||||
think(); // Search entry point
|
Search::think();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -446,3 +446,25 @@ void ThreadsManager::start_thinking(bool asyncMode) {
|
||||||
|
|
||||||
lock_release(&main.sleepLock);
|
lock_release(&main.sleepLock);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// ThreadsManager::wait_for_stop_or_ponderhit() is called when the maximum depth
|
||||||
|
// is reached while the program is pondering. The point is to work around a wrinkle
|
||||||
|
// in the UCI protocol: When pondering, the engine is not allowed to give a
|
||||||
|
// "bestmove" before the GUI sends it a "stop" or "ponderhit" command.
|
||||||
|
// We simply wait here until one of these commands (that raise StopRequest) is
|
||||||
|
// sent, and return, after which the bestmove and pondermove will be printed.
|
||||||
|
|
||||||
|
void ThreadsManager::wait_for_stop_or_ponderhit() {
|
||||||
|
|
||||||
|
Search::Signals.stopOnPonderhit = true;
|
||||||
|
|
||||||
|
Thread& main = threads[0];
|
||||||
|
|
||||||
|
lock_grab(&main.sleepLock);
|
||||||
|
|
||||||
|
while (!Search::Signals.stop)
|
||||||
|
cond_wait(&main.sleepCond, &main.sleepLock);
|
||||||
|
|
||||||
|
lock_release(&main.sleepLock);
|
||||||
|
}
|
||||||
|
|
|
@ -84,7 +84,7 @@ void TimeManager::pv_instability(int curChanges, int prevChanges) {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
void TimeManager::init(const SearchLimits& limits, int currentPly)
|
void TimeManager::init(const Search::LimitsType& limits, int currentPly)
|
||||||
{
|
{
|
||||||
/* We support four different kind of time controls:
|
/* We support four different kind of time controls:
|
||||||
|
|
||||||
|
|
|
@ -25,7 +25,7 @@ struct SearchLimits;
|
||||||
class TimeManager {
|
class TimeManager {
|
||||||
public:
|
public:
|
||||||
|
|
||||||
void init(const SearchLimits& limits, int currentPly);
|
void init(const Search::LimitsType& limits, int currentPly);
|
||||||
void pv_instability(int curChanges, int prevChanges);
|
void pv_instability(int curChanges, int prevChanges);
|
||||||
int available_time() const { return optimumSearchTime + unstablePVExtraTime; }
|
int available_time() const { return optimumSearchTime + unstablePVExtraTime; }
|
||||||
int maximum_time() const { return maximumSearchTime; }
|
int maximum_time() const { return maximumSearchTime; }
|
||||||
|
|
49
src/uci.cpp
49
src/uci.cpp
|
@ -67,14 +67,27 @@ void uci_loop() {
|
||||||
|
|
||||||
is >> skipws >> token;
|
is >> skipws >> token;
|
||||||
|
|
||||||
quit = (token == "quit");
|
if (cmd == "quit" || cmd == "stop")
|
||||||
|
|
||||||
if (token == "quit" || token == "stop" || token == "ponderhit")
|
|
||||||
{
|
{
|
||||||
uci_async_command(token);
|
quit = (token == "quit");
|
||||||
|
Search::Signals.stop = true;
|
||||||
Threads[0].wake_up(); // In case is waiting for stop or ponderhit
|
Threads[0].wake_up(); // In case is waiting for stop or ponderhit
|
||||||
}
|
}
|
||||||
|
|
||||||
|
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")
|
else if (token == "go")
|
||||||
go(pos, is);
|
go(pos, is);
|
||||||
|
|
||||||
|
@ -195,16 +208,16 @@ namespace {
|
||||||
string token;
|
string token;
|
||||||
int time[] = { 0, 0 }, inc[] = { 0, 0 };
|
int time[] = { 0, 0 }, inc[] = { 0, 0 };
|
||||||
|
|
||||||
memset(&Limits, 0, sizeof(SearchLimits));
|
memset(&Search::Limits, 0, sizeof(Search::Limits));
|
||||||
SearchMoves.clear();
|
Search::RootMoves.clear();
|
||||||
RootPosition = &pos;
|
Search::RootPosition = &pos;
|
||||||
|
|
||||||
while (is >> token)
|
while (is >> token)
|
||||||
{
|
{
|
||||||
if (token == "infinite")
|
if (token == "infinite")
|
||||||
Limits.infinite = true;
|
Search::Limits.infinite = true;
|
||||||
else if (token == "ponder")
|
else if (token == "ponder")
|
||||||
Limits.ponder = true;
|
Search::Limits.ponder = true;
|
||||||
else if (token == "wtime")
|
else if (token == "wtime")
|
||||||
is >> time[WHITE];
|
is >> time[WHITE];
|
||||||
else if (token == "btime")
|
else if (token == "btime")
|
||||||
|
@ -214,21 +227,21 @@ namespace {
|
||||||
else if (token == "binc")
|
else if (token == "binc")
|
||||||
is >> inc[BLACK];
|
is >> inc[BLACK];
|
||||||
else if (token == "movestogo")
|
else if (token == "movestogo")
|
||||||
is >> Limits.movesToGo;
|
is >> Search::Limits.movesToGo;
|
||||||
else if (token == "depth")
|
else if (token == "depth")
|
||||||
is >> Limits.maxDepth;
|
is >> Search::Limits.maxDepth;
|
||||||
else if (token == "nodes")
|
else if (token == "nodes")
|
||||||
is >> Limits.maxNodes;
|
is >> Search::Limits.maxNodes;
|
||||||
else if (token == "movetime")
|
else if (token == "movetime")
|
||||||
is >> Limits.maxTime;
|
is >> Search::Limits.maxTime;
|
||||||
else if (token == "searchmoves")
|
else if (token == "searchmoves")
|
||||||
while (is >> token)
|
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);
|
Search::RootMoves.push_back(MOVE_NONE);
|
||||||
Limits.time = time[pos.side_to_move()];
|
Search::Limits.time = time[pos.side_to_move()];
|
||||||
Limits.increment = inc[pos.side_to_move()];
|
Search::Limits.increment = inc[pos.side_to_move()];
|
||||||
|
|
||||||
Threads.start_thinking();
|
Threads.start_thinking();
|
||||||
}
|
}
|
||||||
|
@ -248,7 +261,7 @@ namespace {
|
||||||
|
|
||||||
time = get_system_time();
|
time = get_system_time();
|
||||||
|
|
||||||
n = perft(pos, depth * ONE_PLY);
|
n = Search::perft(pos, depth * ONE_PLY);
|
||||||
|
|
||||||
time = get_system_time() - time;
|
time = get_system_time() - time;
|
||||||
|
|
||||||
|
|
Loading…
Add table
Reference in a new issue