X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=src%2Fsearch.h;h=e233c1c8577059dd4e416a46d829eb4621793592;hb=966bc477af9eae9be63b08111a76e8306263e4a3;hp=831248bacc8ee05cb197bd5841d45b3ebe89c660;hpb=bb751d6c890f5c50c642366d601740366cfae8d0;p=stockfish diff --git a/src/search.h b/src/search.h index 831248ba..e233c1c8 100644 --- a/src/search.h +++ b/src/search.h @@ -1,90 +1,114 @@ /* - Glaurung, a UCI chess playing engine. - Copyright (C) 2004-2008 Tord Romstad + Stockfish, a UCI chess playing engine derived from Glaurung 2.1 + Copyright (C) 2004-2008 Tord Romstad (Glaurung author) + Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad - Glaurung is free software: you can redistribute it and/or modify + Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version. - - Glaurung is distributed in the hope that it will be useful, + + Stockfish is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. - + You should have received a copy of the GNU General Public License along with this program. If not, see . */ - -#if !defined(SEARCH_H_INCLUDED) +#ifndef SEARCH_H_INCLUDED #define SEARCH_H_INCLUDED -//// -//// Includes -//// +#include // For std::auto_ptr +#include +#include -#include "types.h" -#include "depth.h" -#include "history.h" -#include "lock.h" -#include "movegen.h" +#include "misc.h" #include "position.h" -#include "tt.h" -#include "value.h" +#include "types.h" + +struct SplitPoint; +namespace Search { -//// -//// Constants -//// +/// 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 Stack { + SplitPoint* splitPoint; + Move* pv; + int ply; + Move currentMove; + Move ttMove; + Move excludedMove; + Move killers[2]; + Depth reduction; + Value staticEval; + bool skipEarlyPruning; +}; -const int PLY_MAX = 100; -const int PLY_MAX_PLUS_2 = 102; +/// RootMove struct is used for moves at the root of the tree. For each root move +/// we store a score 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 { -//// -//// Types -//// + explicit RootMove(Move m) : pv(1, m) {} -/// 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. + 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); + bool extract_ponder_from_tt(Position& pos); -struct SearchStack { - Move pv[PLY_MAX]; - Move currentMove; - Value currentMoveCaptureValue; - Move mateKiller, killer1, killer2; - Move threatMove; - Depth reduction; + Value score = -VALUE_INFINITE; + Value previousScore = -VALUE_INFINITE; + std::vector pv; }; +typedef std::vector RootMoveVector; -//// -//// Global variables -//// +/// 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 it's our opponent's turn to move. -extern TranspositionTable TT; +struct LimitsType { -extern int ActiveThreads; + LimitsType() { // Init explicitly due to broken value-initialization of non POD in MSVC + nodes = time[WHITE] = time[BLACK] = inc[WHITE] = inc[BLACK] = movestogo = + depth = movetime = mate = infinite = ponder = 0; + } -extern Lock SMPLock; + bool use_time_management() const { + return !(mate | movetime | depth | nodes | infinite); + } -// Perhaps better to make H local, and pass as parameter to MovePicker? -extern History H; + std::vector searchmoves; + int time[COLOR_NB], inc[COLOR_NB], movestogo, depth, movetime, mate, infinite, ponder; + int64_t nodes; +}; + +/// 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; +}; +typedef std::unique_ptr> StateStackPtr; -//// -//// Prototypes -//// +extern volatile SignalsType Signals; +extern LimitsType Limits; +extern RootMoveVector RootMoves; +extern Position RootPos; +extern TimePoint SearchTime; +extern StateStackPtr SetupStates; -extern void init_threads(); -extern void stop_threads(); -extern void think(const Position &pos, bool infinite, bool ponder, int time, - int increment, int movesToGo, int maxDepth, int maxNodes, - int maxTime, Move searchMoves[]); -extern int64_t nodes_searched(); +void init(); +void think(); +template uint64_t perft(Position& pos, Depth depth); +} // namespace Search -#endif // !defined(SEARCH_H_INCLUDED) +#endif // #ifndef SEARCH_H_INCLUDED