X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fsearch.h;h=e2a136062c689a79695b4c725719f7c9cb562bdc;hp=812f5eea896f7c6da88c514fe7ea3a68b1f1592d;hb=45b0aea875860e9f0fe2d0435ee6163906639194;hpb=d3600c39a745179ed6b094b305d0645e83a1ee86 diff --git a/src/search.h b/src/search.h index 812f5eea..e2a13606 100644 --- a/src/search.h +++ b/src/search.h @@ -1,91 +1,112 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008 Marco Costalba + Copyright (C) 2008-2013 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 the Free Software Foundation, either version 3 of the License, or (at your option) any later version. - + 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 +#include +#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; -//// -//// Constants -//// +namespace Search { -const int PLY_MAX = 100; -const int PLY_MAX_PLUS_2 = 102; +/// 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 Stack { + SplitPoint* splitPoint; + int ply; + Move currentMove; + Move excludedMove; + Move killers[2]; + Depth reduction; + Value staticEval; + Value evalMargin; + int skipNullMove; + int futilityMoveCount; +}; -//// -//// Types -//// -/// 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. +/// 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 { -struct SearchStack { - Move pv[PLY_MAX]; - Move currentMove; - Value currentMoveCaptureValue; - Move mateKiller, killer1, killer2; - Move threatMove; - Depth reduction; + 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 extract_pv_from_tt(Position& pos); + void insert_pv_in_tt(Position& pos); + + Value score; + Value prevScore; + std::vector pv; }; -//// -//// Global variables -//// +/// 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. -extern TranspositionTable TT; +struct LimitsType { -extern int ActiveThreads; + LimitsType() { std::memset(this, 0, sizeof(LimitsType)); } + bool use_time_management() const { return !(mate | movetime | depth | nodes | infinite); } + + int time[COLOR_NB], inc[COLOR_NB], movestogo, depth, nodes, movetime, mate, infinite, ponder; +}; -extern Lock SMPLock; -// Perhaps better to make H local, and pass as parameter to MovePicker? -extern History H; +/// The SignalsType struct stores volatile flags updated during the search +/// typically in an async fashion, for instance to stop the search by the GUI. + +struct SignalsType { + bool stopOnPonderhit, firstRootMove, stop, failedLowAtRoot; +}; +typedef std::auto_ptr > StateStackPtr; -//// -//// Prototypes -//// +extern volatile SignalsType Signals; +extern LimitsType Limits; +extern std::vector RootMoves; +extern Position RootPos; +extern Color RootColor; +extern Time::point SearchTime; +extern StateStackPtr SetupStates; -extern void init_threads(); -extern void stop_threads(); -extern void think(const Position &pos, bool infinite, bool ponder, int side_to_move, - int time[], int increment[], int movesToGo, int maxDepth, - int maxNodes, int maxTime, Move searchMoves[]); -extern int64_t nodes_searched(); +extern void init(); +extern size_t perft(Position& pos, Depth depth); +extern void think(); +} // namespace Search -#endif // !defined(SEARCH_H_INCLUDED) +#endif // #ifndef SEARCH_H_INCLUDED