X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fendgame.h;h=a5e4efecafea8831c0bc4ee032ddc5c5aa5af6c8;hp=dabf233e7058f2f367f429badf25bcd00f4fd40e;hb=4554d8b2ac1ad260c3f7123b2a7d3a3385fa8306;hpb=5dc23121215039938a9ef4e59ae934312774571c diff --git a/src/endgame.h b/src/endgame.h index dabf233e..a5e4efec 100644 --- a/src/endgame.h +++ b/src/endgame.h @@ -1,243 +1,120 @@ /* 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-2010 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(ENDGAME_H_INCLUDED) #define ENDGAME_H_INCLUDED -//// -//// Includes -//// +#include +#include #include "position.h" -#include "scale.h" -#include "value.h" - +#include "types.h" -//// -//// Types -//// -/// Abstract base class for all special endgame evaluation functions: +/// EndgameType lists all supported endgames -class EndgameEvaluationFunction { -public: - EndgameEvaluationFunction(Color c); - virtual ~EndgameEvaluationFunction() { } +enum EndgameType { - virtual Value apply(const Position &pos) =0; + // Evaluation functions -protected: - Color strongerSide, weakerSide; -}; + KXK, // Generic "mate lone king" eval + KBNK, // KBN vs K + KPK, // KP vs K + KRKP, // KR vs KP + KRKB, // KR vs KB + KRKN, // KR vs KN + KQKR, // KQ vs KR + KBBKN, // KBB vs KN + KNNK, // KNN vs K + KmmKm, // K and two minors vs K and one or two minors -/// Subclasses for various concrete endgames: + // Scaling functions + SCALE_FUNS, -// Generic "mate lone king" eval: -class KXKEvaluationFunction : public EndgameEvaluationFunction { -public: - KXKEvaluationFunction(Color c); - Value apply(const Position &pos); + KBPsK, // KB+pawns vs K + KQKRPs, // KQ vs KR+pawns + KRPKR, // KRP vs KR + KRPPKRP, // KRPP vs KRP + KPsK, // King and pawns vs king + KBPKB, // KBP vs KB + KBPPKB, // KBPP vs KB + KBPKN, // KBP vs KN + KNPK, // KNP vs K + KPKP // KP vs KP }; -// KBN vs K: -class KBNKEvaluationFunction : public EndgameEvaluationFunction { -public: - KBNKEvaluationFunction(Color c); - Value apply(const Position &pos); -}; -// KP vs K: -class KPKEvaluationFunction : public EndgameEvaluationFunction { -public: - KPKEvaluationFunction(Color c); - Value apply(const Position &pos); -}; +/// Some magic to detect family type of endgame from its enum value -// KR vs KP: -class KRKPEvaluationFunction : public EndgameEvaluationFunction { -public: - KRKPEvaluationFunction(Color c); - Value apply(const Position &pos); -}; +template struct bool_to_type { typedef Value type; }; +template<> struct bool_to_type { typedef ScaleFactor type; }; +template struct eg_family : public bool_to_type<(E > SCALE_FUNS)> {}; -// KR vs KB: -class KRKBEvaluationFunction : public EndgameEvaluationFunction { -public: - KRKBEvaluationFunction(Color c); - Value apply(const Position &pos); -}; -// KR vs KN: -class KRKNEvaluationFunction : public EndgameEvaluationFunction { -public: - KRKNEvaluationFunction(Color c); - Value apply(const Position &pos); -}; +/// Base and derived templates for endgame evaluation and scaling functions -// KQ vs KR: -class KQKREvaluationFunction : public EndgameEvaluationFunction { -public: - KQKREvaluationFunction(Color c); - Value apply(const Position &pos); -}; +template +struct EndgameBase { + virtual ~EndgameBase() {} + virtual Color color() const = 0; + virtual T operator()(const Position&) const = 0; +}; -/// Abstract base class for all evaluation scaling functions: -class ScalingFunction { -public: - ScalingFunction(Color c); - virtual ~ScalingFunction() { } +template::type> +struct Endgame : public EndgameBase { - virtual ScaleFactor apply(const Position &pos) =0; + explicit Endgame(Color c) : strongerSide(c), weakerSide(flip(c)) {} + Color color() const { return strongerSide; } + T operator()(const Position&) const; -protected: +private: Color strongerSide, weakerSide; }; -/// Subclasses for various concrete endgames: +/// Endgames class stores in two std::map the pointers to endgame evaluation +/// and scaling base objects. Then we use polymorphism to invoke the actual +/// endgame function calling its operator() method that is virtual. -// KBP vs K: -class KBPKScalingFunction : public ScalingFunction { -public: - KBPKScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; +class Endgames { -// KQ vs KRP: -class KQKRPScalingFunction: public ScalingFunction { -public: - KQKRPScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; - -// KRP vs KR: -class KRPKRScalingFunction : public ScalingFunction { -public: - KRPKRScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; - -// KRPP vs KRP: -class KRPPKRPScalingFunction : public ScalingFunction { -public: - KRPPKRPScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; + typedef std::map*> M1; + typedef std::map*> M2; -// King and pawns vs king: -class KPsKScalingFunction : public ScalingFunction { -public: - KPsKScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; + M1 m1; + M2 m2; -// KBP vs KB: -class KBPKBScalingFunction : public ScalingFunction { -public: - KBPKBScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; + M1& map(Value*) { return m1; } + M2& map(ScaleFactor*) { return m2; } -// KBP vs KN: -class KBPKNScalingFunction : public ScalingFunction { -public: - KBPKNScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; + template void add(const std::string& code); -// KNP vs K: -class KNPKScalingFunction : public ScalingFunction { public: - KNPKScalingFunction(Color c); - ScaleFactor apply(const Position &pos); -}; + Endgames(); + ~Endgames(); -// KP vs KP: -class KPKPScalingFunction : public ScalingFunction { -public: - KPKPScalingFunction(Color c); - ScaleFactor apply(const Position &pos); + template EndgameBase* get(Key key) { + return map((T*)0).count(key) ? map((T*)0)[key] : NULL; + } }; - -//// -//// Constants and variables -//// - -// Generic "mate lone king" eval: -extern KXKEvaluationFunction EvaluateKXK, EvaluateKKX; - -// KBN vs K: -extern KBNKEvaluationFunction EvaluateKBNK, EvaluateKKBN; - -// KP vs K: -extern KPKEvaluationFunction EvaluateKPK, EvaluateKKP; - -// KR vs KP: -extern KRKPEvaluationFunction EvaluateKRKP, EvaluateKPKR; - -// KR vs KB: -extern KRKBEvaluationFunction EvaluateKRKB, EvaluateKBKR; - -// KR vs KN: -extern KRKNEvaluationFunction EvaluateKRKN, EvaluateKNKR; - -// KQ vs KR: -extern KQKREvaluationFunction EvaluateKQKR, EvaluateKRKQ; - -// KBP vs K: -extern KBPKScalingFunction ScaleKBPK, ScaleKKBP; - -// KQ vs KRP: -extern KQKRPScalingFunction ScaleKQKRP, ScaleKRPKQ; - -// KRP vs KR: -extern KRPKRScalingFunction ScaleKRPKR, ScaleKRKRP; - -// KRPP vs KRP: -extern KRPPKRPScalingFunction ScaleKRPPKRP, ScaleKRPKRPP; - -// King and pawns vs king: -extern KPsKScalingFunction ScaleKPsK, ScaleKKPs; - -// KBP vs KB: -extern KBPKBScalingFunction ScaleKBPKB, ScaleKBKBP; - -// KBP vs KN: -extern KBPKNScalingFunction ScaleKBPKN, ScaleKNKBP; - -// KNP vs K: -extern KNPKScalingFunction ScaleKNPK, ScaleKKNP; - -// KP vs KP: -extern KPKPScalingFunction ScaleKPKPw, ScaleKPKPb; - - -//// -//// Prototypes -//// - -extern void init_bitbases(); - - #endif // !defined(ENDGAME_H_INCLUDED)