Refactor pawns shelter and storm
[stockfish] / src / pawns.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-2012 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 #if !defined(PAWNS_H_INCLUDED)
21 #define PAWNS_H_INCLUDED
22
23 #include "position.h"
24 #include "tt.h"
25 #include "types.h"
26
27 const int PawnTableSize = 16384;
28
29 /// PawnInfo is a class which contains various information about a pawn
30 /// structure. Currently, it only includes a middle game and an end game
31 /// pawn structure evaluation, and a bitboard of passed pawns. We may want
32 /// to add further information in the future. A lookup to the pawn hash
33 /// table (performed by calling the pawn_info method in a PawnInfoTable
34 /// object) returns a pointer to a PawnInfo object.
35
36 class PawnInfo {
37
38   friend class PawnInfoTable;
39
40 public:
41   Score pawns_value() const;
42   Bitboard pawn_attacks(Color c) const;
43   Bitboard passed_pawns(Color c) const;
44   int file_is_half_open(Color c, File f) const;
45   int has_open_file_to_left(Color c, File f) const;
46   int has_open_file_to_right(Color c, File f) const;
47
48   template<Color Us>
49   Score king_safety(const Position& pos, Square ksq);
50
51 private:
52   template<Color Us>
53   Score update_safety(const Position& pos, Square ksq);
54
55   template<Color Us>
56   int shelter_storm(const Position& pos, Square ksq);
57
58   Key key;
59   Bitboard passedPawns[2];
60   Bitboard pawnAttacks[2];
61   Square kingSquares[2];
62   Score value;
63   int halfOpenFiles[2];
64   Score kingShelters[2];
65 };
66
67
68 /// The PawnInfoTable class represents a pawn hash table. The most important
69 /// method is pawn_info, which returns a pointer to a PawnInfo object.
70
71 class PawnInfoTable : public SimpleHash<PawnInfo, PawnTableSize> {
72 public:
73   PawnInfo* pawn_info(const Position& pos) const;
74
75 private:
76   template<Color Us>
77   static Score evaluate_pawns(const Position& pos, Bitboard ourPawns, Bitboard theirPawns, PawnInfo* pi);
78 };
79
80
81 inline Score PawnInfo::pawns_value() const {
82   return value;
83 }
84
85 inline Bitboard PawnInfo::pawn_attacks(Color c) const {
86   return pawnAttacks[c];
87 }
88
89 inline Bitboard PawnInfo::passed_pawns(Color c) const {
90   return passedPawns[c];
91 }
92
93 inline int PawnInfo::file_is_half_open(Color c, File f) const {
94   return halfOpenFiles[c] & (1 << int(f));
95 }
96
97 inline int PawnInfo::has_open_file_to_left(Color c, File f) const {
98   return halfOpenFiles[c] & ((1 << int(f)) - 1);
99 }
100
101 inline int PawnInfo::has_open_file_to_right(Color c, File f) const {
102   return halfOpenFiles[c] & ~((1 << int(f+1)) - 1);
103 }
104
105 template<Color Us>
106 inline Score PawnInfo::king_safety(const Position& pos, Square ksq) {
107   return kingSquares[Us] == ksq ? kingShelters[Us] : update_safety<Us>(pos, ksq);
108 }
109
110 #endif // !defined(PAWNS_H_INCLUDED)