afa12b3a4ee6b56104918178aeee4bdd50e6845a
[stockfish] / src / search.h
1 /*
2   Stockfish, a UCI chess playing engine derived from Glaurung 2.1
3   Copyright (C) 2004-2008 Tord Romstad (Glaurung author)
4   Copyright (C) 2008-2014 Marco Costalba, Joona Kiiski, Tord Romstad
5
6   Stockfish is free software: you can redistribute it and/or modify
7   it under the terms of the GNU General Public License as published by
8   the Free Software Foundation, either version 3 of the License, or
9   (at your option) any later version.
10
11   Stockfish is distributed in the hope that it will be useful,
12   but WITHOUT ANY WARRANTY; without even the implied warranty of
13   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14   GNU General Public License for more details.
15
16   You should have received a copy of the GNU General Public License
17   along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #ifndef SEARCH_H_INCLUDED
21 #define SEARCH_H_INCLUDED
22
23 #include <memory>
24 #include <stack>
25 #include <vector>
26
27 #include "misc.h"
28 #include "position.h"
29 #include "types.h"
30
31 struct SplitPoint;
32
33 namespace Search {
34
35 struct PVEntry {
36   Move pv[MAX_PLY+1];
37
38   void update(Move move, PVEntry* child) {
39       pv[0] = move;
40
41       int i = 1;
42       for (; child && i < MAX_PLY && child->pv[i - 1] != MOVE_NONE; ++i)
43           pv[i] = child->pv[i - 1];
44       pv[i] = MOVE_NONE;
45   } 
46 };
47
48 /// The Stack struct keeps track of the information we need to remember from
49 /// nodes shallower and deeper in the tree during the search. Each search thread
50 /// has its own array of Stack objects, indexed by the current ply.
51
52 struct Stack {
53   SplitPoint* splitPoint;
54   PVEntry* pv;
55   int ply;
56   Move currentMove;
57   Move ttMove;
58   Move excludedMove;
59   Move killers[2];
60   Depth reduction;
61   Value staticEval;
62   bool skipNullMove;
63 };
64
65
66 /// RootMove struct is used for moves at the root of the tree. For each root
67 /// move we store a score, a node count, and a PV (really a refutation in the
68 /// case of moves which fail low). Score is normally set at -VALUE_INFINITE for
69 /// all non-pv moves.
70 struct RootMove {
71
72   RootMove(Move m) : score(-VALUE_INFINITE), prevScore(-VALUE_INFINITE) {
73     pv.push_back(m); pv.push_back(MOVE_NONE);
74   }
75
76   bool operator<(const RootMove& m) const { return score > m.score; } // Ascending sort
77   bool operator==(const Move& m) const { return pv[0] == m; }
78
79   void insert_pv_in_tt(Position& pos);
80
81   Value score;
82   Value prevScore;
83   std::vector<Move> pv;
84 };
85
86
87 /// The LimitsType struct stores information sent by GUI about available time
88 /// to search the current move, maximum depth/time, if we are in analysis mode
89 /// or if we have to ponder while it's our opponent's turn to move.
90
91 struct LimitsType {
92
93   LimitsType() { // Using memset on a std::vector is undefined behavior
94     nodes = time[WHITE] = time[BLACK] = inc[WHITE] = inc[BLACK] = movestogo =
95     depth = movetime = mate = infinite = ponder = 0;
96   }
97   bool use_time_management() const { return !(mate | movetime | depth | nodes | infinite); }
98
99   std::vector<Move> searchmoves;
100   int time[COLOR_NB], inc[COLOR_NB], movestogo, depth, movetime, mate, infinite, ponder;
101   int64_t nodes;
102 };
103
104
105 /// The SignalsType struct stores volatile flags updated during the search
106 /// typically in an async fashion e.g. to stop the search by the GUI.
107
108 struct SignalsType {
109   bool stop, stopOnPonderhit, firstRootMove, failedLowAtRoot;
110 };
111
112 typedef std::auto_ptr<std::stack<StateInfo> > StateStackPtr;
113
114 extern volatile SignalsType Signals;
115 extern LimitsType Limits;
116 extern std::vector<RootMove> RootMoves;
117 extern Position RootPos;
118 extern Time::point SearchTime;
119 extern StateStackPtr SetupStates;
120
121 void init();
122 void think();
123 template<bool Root> uint64_t perft(Position& pos, Depth depth);
124
125 } // namespace Search
126
127 #endif // #ifndef SEARCH_H_INCLUDED