X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fsearch.h;h=afa12b3a4ee6b56104918178aeee4bdd50e6845a;hp=a2e783b110895a6fb92fa71665c4f9edc0cc9820;hb=4739037f967ac3c818907e89cc88c7b97021d027;hpb=1d368bbbdc50bbb4e10933c4986abc07a08010fd diff --git a/src/search.h b/src/search.h index a2e783b1..afa12b3a 100644 --- a/src/search.h +++ b/src/search.h @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad + Copyright (C) 2008-2014 Marco Costalba, Joona Kiiski, Tord Romstad Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -17,55 +17,111 @@ along with this program. If not, see . */ -#if !defined(SEARCH_H_INCLUDED) +#ifndef SEARCH_H_INCLUDED #define SEARCH_H_INCLUDED -#include "move.h" +#include +#include +#include + +#include "misc.h" +#include "position.h" #include "types.h" -class Position; struct SplitPoint; -/// The SearchStack struct keeps track of the information we need to remember -/// from nodes shallower and deeper in the tree during the search. Each -/// search thread has its own array of SearchStack objects, indexed by the -/// current ply. +namespace Search { + +struct PVEntry { + Move pv[MAX_PLY+1]; + + void update(Move move, PVEntry* child) { + pv[0] = move; + + int i = 1; + for (; child && i < MAX_PLY && child->pv[i - 1] != MOVE_NONE; ++i) + pv[i] = child->pv[i - 1]; + pv[i] = MOVE_NONE; + } +}; + +/// The Stack struct keeps track of the information we need to remember from +/// nodes shallower and deeper in the tree during the search. Each search thread +/// has its own array of Stack objects, indexed by the current ply. -struct SearchStack { +struct Stack { + SplitPoint* splitPoint; + PVEntry* pv; int ply; Move currentMove; - Move mateKiller; + Move ttMove; Move excludedMove; - Move bestMove; Move killers[2]; Depth reduction; - Value eval; - Value evalMargin; + Value staticEval; bool skipNullMove; - SplitPoint* sp; }; -/// The SearchLimits struct stores information sent by GUI about available time +/// RootMove struct is used for moves at the root of the tree. For each root +/// move we store a score, a node count, and a PV (really a refutation in the +/// case of moves which fail low). Score is normally set at -VALUE_INFINITE for +/// all non-pv moves. +struct RootMove { + + RootMove(Move m) : score(-VALUE_INFINITE), prevScore(-VALUE_INFINITE) { + pv.push_back(m); pv.push_back(MOVE_NONE); + } + + bool operator<(const RootMove& m) const { return score > m.score; } // Ascending sort + bool operator==(const Move& m) const { return pv[0] == m; } + + void insert_pv_in_tt(Position& pos); + + Value score; + Value prevScore; + std::vector pv; +}; + + +/// The LimitsType struct stores information sent by GUI about available time /// 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 it's our opponent's turn to move. -struct SearchLimits { +struct LimitsType { - SearchLimits() {} - SearchLimits(int t, int i, int mtg, int mt, int md, int mn, bool inf, bool pon) - : time(t), increment(i), movesToGo(mtg), maxTime(mt), maxDepth(md), - maxNodes(mn), infinite(inf), ponder(pon) {} + LimitsType() { // Using memset on a std::vector is undefined behavior + nodes = time[WHITE] = time[BLACK] = inc[WHITE] = inc[BLACK] = movestogo = + depth = movetime = mate = infinite = ponder = 0; + } + bool use_time_management() const { return !(mate | movetime | depth | nodes | infinite); } + + std::vector searchmoves; + int time[COLOR_NB], inc[COLOR_NB], movestogo, depth, movetime, mate, infinite, ponder; + int64_t nodes; +}; - bool useTimeManagement() const { return !(maxTime | maxDepth | maxNodes | int(infinite)); } - int time, increment, movesToGo, maxTime, maxDepth, maxNodes; - bool infinite, ponder; +/// The SignalsType struct stores volatile flags updated during the search +/// typically in an async fashion e.g. to stop the search by the GUI. + +struct SignalsType { + bool stop, stopOnPonderhit, firstRootMove, failedLowAtRoot; }; -extern void init_threads(); -extern void exit_threads(); -extern int64_t perft(Position& pos, Depth depth); -extern bool think(Position& pos, const SearchLimits& limits, Move searchMoves[]); +typedef std::auto_ptr > StateStackPtr; + +extern volatile SignalsType Signals; +extern LimitsType Limits; +extern std::vector RootMoves; +extern Position RootPos; +extern Time::point SearchTime; +extern StateStackPtr SetupStates; + +void init(); +void think(); +template uint64_t perft(Position& pos, Depth depth); + +} // namespace Search -#endif // !defined(SEARCH_H_INCLUDED) +#endif // #ifndef SEARCH_H_INCLUDED