43a685572aa710f15c5a9da2b037ddb23309d175
[stockfish] / src / endgame.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-2009 Marco Costalba
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
21 #if !defined(ENDGAME_H_INCLUDED)
22 #define ENDGAME_H_INCLUDED
23
24 ////
25 //// Includes
26 ////
27
28 #include "position.h"
29 #include "scale.h"
30 #include "value.h"
31
32
33 ////
34 //// Types
35 ////
36
37 enum EndgameType {
38
39     // Evaluation functions
40     KXK,   // Generic "mate lone king" eval
41     KBNK,  // KBN vs K
42     KPK,   // KP vs K
43     KRKP,  // KR vs KP
44     KRKB,  // KR vs KB
45     KRKN,  // KR vs KN
46     KQKR,  // KQ vs KR
47     KBBKN, // KBB vs KN
48     KNNK,  // KNN vs K
49     KmmKm, // K and two minors vs K and one or two minors
50
51     // Scaling functions
52     KBPK,    // KBP vs K
53     KQKRP,   // KQ vs KRP
54     KRPKR,   // KRP vs KR
55     KRPPKRP, // KRPP vs KRP
56     KPsK,    // King and pawns vs king
57     KBPKB,   // KBP vs KB
58     KBPPKB,  // KBPP vs KB
59     KBPKN,   // KBP vs KN
60     KNPK,    // KNP vs K
61     KPKP     // KP vs KP
62 };
63
64 /// Template abstract base class for all special endgame functions
65
66 template<typename T>
67 class EndgameFunctionBase {
68 public:
69   EndgameFunctionBase(Color c) : strongerSide(c), weakerSide(opposite_color(c)) {}
70   virtual ~EndgameFunctionBase() {}
71   virtual T apply(const Position&) = 0;
72   Color color() const { return strongerSide; }
73
74 protected:
75   Color strongerSide, weakerSide;
76 };
77
78 typedef EndgameFunctionBase<Value> EndgameEvaluationFunctionBase;
79 typedef EndgameFunctionBase<ScaleFactor> EndgameScalingFunctionBase;
80
81
82 /// Templates subclass for various concrete endgames
83
84 template<EndgameType>
85 struct EvaluationFunction : public EndgameEvaluationFunctionBase {
86   typedef EndgameEvaluationFunctionBase Base;
87   explicit EvaluationFunction(Color c): EndgameEvaluationFunctionBase(c) {}
88   Value apply(const Position&);
89 };
90
91 template<EndgameType>
92 struct ScalingFunction : public EndgameScalingFunctionBase {
93   typedef EndgameScalingFunctionBase Base;
94   explicit ScalingFunction(Color c) : EndgameScalingFunctionBase(c) {}
95   ScaleFactor apply(const Position&);
96 };
97
98
99 ////
100 //// Prototypes
101 ////
102
103 extern void init_bitbases();
104
105
106 #endif // !defined(ENDGAME_H_INCLUDED)