Rev 96 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed
| Rev 96 | Rev 169 | ||
|---|---|---|---|
| Line 4... | Line 4... | ||
| 4 | Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad |
4 | Copyright (C) 2008-2015 Marco Costalba, Joona Kiiski, Tord Romstad |
| 5 | Copyright (C) 2015- |
5 | Copyright (C) 2015-2018 Marco Costalba, Joona Kiiski, Gary Linscott, Tord Romstad |
| 6 | 6 | ||
| 7 | Stockfish is free software: you can redistribute it and/or modify |
7 | Stockfish is free software: you can redistribute it and/or modify |
| 8 | it under the terms of the GNU General Public License as published by |
8 | it under the terms of the GNU General Public License as published by |
| 9 | the Free Software Foundation, either version 3 of the License, or |
9 | the Free Software Foundation, either version 3 of the License, or |
| 10 | (at your option) any later version. |
10 | (at your option) any later version. |
| Line 29... | Line 29... | ||
| 29 | 29 | ||
| 30 | #include "position.h" |
30 | #include "position.h" |
| 31 | #include "types.h" |
31 | #include "types.h" |
| 32 | 32 | ||
| 33 | 33 | ||
| 34 | /// |
34 | /// EndgameCode lists all supported endgame functions by corresponding codes |
| 35 | 35 | ||
| 36 | enum |
36 | enum EndgameCode { |
| 37 | - | ||
| 38 | // Evaluation functions |
- | |
| 39 | 37 | ||
| - | 38 | EVALUATION_FUNCTIONS, |
|
| 40 | KNNK, // KNN vs K |
39 | KNNK, // KNN vs K |
| 41 | KXK, // Generic "mate lone king" eval |
40 | KXK, // Generic "mate lone king" eval |
| 42 | KBNK, // KBN vs K |
41 | KBNK, // KBN vs K |
| 43 | KPK, // KP vs K |
42 | KPK, // KP vs K |
| 44 | KRKP, // KR vs KP |
43 | KRKP, // KR vs KP |
| 45 | KRKB, // KR vs KB |
44 | KRKB, // KR vs KB |
| 46 | KRKN, // KR vs KN |
45 | KRKN, // KR vs KN |
| 47 | KQKP, // KQ vs KP |
46 | KQKP, // KQ vs KP |
| 48 | KQKR, // KQ vs KR |
47 | KQKR, // KQ vs KR |
| 49 | 48 | ||
| 50 | - | ||
| 51 | // Scaling functions |
- | |
| 52 | SCALING_FUNCTIONS, |
49 | SCALING_FUNCTIONS, |
| 53 | - | ||
| 54 | KBPsK, // KB and pawns vs K |
50 | KBPsK, // KB and pawns vs K |
| 55 | KQKRPs, // KQ vs KR and pawns |
51 | KQKRPs, // KQ vs KR and pawns |
| 56 | KRPKR, // KRP vs KR |
52 | KRPKR, // KRP vs KR |
| 57 | KRPKB, // KRP vs KB |
53 | KRPKB, // KRP vs KB |
| 58 | KRPPKRP, // KRPP vs KRP |
54 | KRPPKRP, // KRPP vs KRP |
| Line 66... | Line 62... | ||
| 66 | }; |
62 | }; |
| 67 | 63 | ||
| 68 | 64 | ||
| 69 | /// Endgame functions can be of two types depending on whether they return a |
65 | /// Endgame functions can be of two types depending on whether they return a |
| 70 | /// Value or a ScaleFactor. |
66 | /// Value or a ScaleFactor. |
| 71 | template< |
67 | template<EndgameCode E> using |
| 72 | eg_type = typename std::conditional<(E < SCALING_FUNCTIONS), Value, ScaleFactor>::type; |
68 | eg_type = typename std::conditional<(E < SCALING_FUNCTIONS), Value, ScaleFactor>::type; |
| 73 | 69 | ||
| 74 | 70 | ||
| 75 | /// Base and derived |
71 | /// Base and derived functors for endgame evaluation and scaling functions |
| 76 | 72 | ||
| 77 | template<typename T> |
73 | template<typename T> |
| 78 | struct EndgameBase { |
74 | struct EndgameBase { |
| 79 | 75 | ||
| - | 76 | explicit EndgameBase(Color c) : strongSide(c), weakSide(~c) {} |
|
| 80 | virtual ~EndgameBase() = default; |
77 | virtual ~EndgameBase() = default; |
| 81 | virtual Color strong_side() const = 0; |
- | |
| 82 | virtual T operator()(const Position&) const = 0; |
78 | virtual T operator()(const Position&) const = 0; |
| - | 79 | ||
| - | 80 | const Color strongSide, weakSide; |
|
| 83 | }; |
81 | }; |
| 84 | 82 | ||
| 85 | 83 | ||
| 86 | template< |
84 | template<EndgameCode E, typename T = eg_type<E>> |
| 87 | struct Endgame : public EndgameBase<T> { |
85 | struct Endgame : public EndgameBase<T> { |
| 88 | 86 | ||
| 89 | explicit Endgame(Color c) : |
87 | explicit Endgame(Color c) : EndgameBase<T>(c) {} |
| 90 | Color strong_side() const { return strongSide; } |
- | |
| 91 | T operator()(const Position&) const; |
88 | T operator()(const Position&) const override; |
| 92 | - | ||
| 93 | private: |
- | |
| 94 | Color strongSide, weakSide; |
- | |
| 95 | }; |
89 | }; |
| 96 | 90 | ||
| 97 | 91 | ||
| 98 | /// The Endgames class stores the pointers to endgame evaluation and scaling |
92 | /// The Endgames class stores the pointers to endgame evaluation and scaling |
| 99 | /// base objects in two std::map. We use polymorphism to invoke the actual |
93 | /// base objects in two std::map. We use polymorphism to invoke the actual |
| 100 | /// endgame function by calling its virtual operator(). |
94 | /// endgame function by calling its virtual operator(). |
| 101 | 95 | ||
| 102 | class Endgames { |
96 | class Endgames { |
| 103 | 97 | ||
| 104 | template<typename T> using |
98 | template<typename T> using Ptr = std::unique_ptr<EndgameBase<T>>; |
| 105 | - | ||
| 106 | template< |
99 | template<typename T> using Map = std::map<Key, Ptr<T>>; |
| 107 | void add(const std::string& code); |
- | |
| 108 | 100 | ||
| 109 | template<typename T> |
101 | template<typename T> |
| 110 | Map<T>& map() { |
102 | Map<T>& map() { |
| 111 | return std::get<std::is_same<T, ScaleFactor>::value>(maps); |
103 | return std::get<std::is_same<T, ScaleFactor>::value>(maps); |
| - | 104 | } |
|
| - | 105 | ||
| - | 106 | template<EndgameCode E, typename T = eg_type<E>, typename P = Ptr<T>> |
|
| - | 107 | void add(const std::string& code) { |
|
| - | 108 | ||
| - | 109 | StateInfo st; |
|
| - | 110 | map<T>()[Position().set(code, WHITE, &st).material_key()] = P(new Endgame<E>(WHITE)); |
|
| - | 111 | map<T>()[Position().set(code, BLACK, &st).material_key()] = P(new Endgame<E>(BLACK)); |
|
| 112 | } |
112 | } |
| 113 | 113 | ||
| 114 | std::pair<Map<Value>, Map<ScaleFactor>> maps; |
114 | std::pair<Map<Value>, Map<ScaleFactor>> maps; |
| 115 | 115 | ||
| 116 | public: |
116 | public: |