git.sesse.net Git - stockfish/blob - 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-2010 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
  21 #if !defined(PAWNS_H_INCLUDED)
  22 #define PAWNS_H_INCLUDED
  23
  24 ////
  25 //// Includes
  26 ////
  27
  28 #include "bitboard.h"
  29 #include "value.h"
  30
  31 ////
  32 //// Types
  33 ////
  34
  35 const int PawnTableSize = 16384;
  36
  37 /// PawnInfo is a class which contains various information about a pawn
  38 /// structure. Currently, it only includes a middle game and an end game
  39 /// pawn structure evaluation, and a bitboard of passed pawns. We may want
  40 /// to add further information in the future. A lookup to the pawn hash table
  41 /// (performed by calling the get_pawn_info method in a PawnInfoTable object)
  42 /// returns a pointer to a PawnInfo object.
  43 class Position;
  44
  45 class PawnInfo {
  46
  47   friend class PawnInfoTable;
  48
  49 public:
  50   Score pawns_value() const;
  51   Value kingside_storm_value(Color c) const;
  52   Value queenside_storm_value(Color c) const;
  53   Bitboard pawn_attacks(Color c) const;
  54   Bitboard passed_pawns(Color c) const;
  55   int file_is_half_open(Color c, File f) const;
  56   int has_open_file_to_left(Color c, File f) const;
  57   int has_open_file_to_right(Color c, File f) const;
  58   Score king_shelter(const Position& pos, Color c, Square ksq);
  59
  60 private:
  61   Score updateShelter(const Position& pos, Color c, Square ksq);
  62
  63   Key key;
  64   Bitboard passedPawns[2];
  65   Bitboard pawnAttacks[2];
  66   Square kingSquares[2];
  67   Score value;
  68   int ksStormValue[2];
  69   int qsStormValue[2];
  70   int halfOpenFiles[2];
  71   Score kingShelters[2];
  72 };
  73
  74 /// The PawnInfoTable class represents a pawn hash table.  It is basically
  75 /// just an array of PawnInfo objects and a few methods for accessing these
  76 /// objects.  The most important method is get_pawn_info, which looks up a
  77 /// position in the table and returns a pointer to a PawnInfo object.
  78
  79 class PawnInfoTable {
  80
  81   enum SideType { KingSide, QueenSide };
  82
  83 public:
  84   PawnInfoTable();
  85   ~PawnInfoTable();
  86   PawnInfo* get_pawn_info(const Position& pos) const;
  87   void prefetch(Key key) const;
  88
  89 private:
  90   template<Color Us>
  91   Score evaluate_pawns(const Position& pos, Bitboard ourPawns, Bitboard theirPawns, PawnInfo* pi) const;
  92
  93   template<Color Us, SideType Side>
  94   int evaluate_pawn_storm(Square s, Rank r, File f, Bitboard theirPawns) const;
  95
  96   PawnInfo* entries;
  97 };
  98
  99
 100 ////
 101 //// Inline functions
 102 ////
 103
 104 inline void PawnInfoTable::prefetch(Key key) const {
 105
 106     unsigned index = unsigned(key & (PawnTableSize - 1));
 107     PawnInfo* pi = entries + index;
 108     ::prefetch((char*) pi);
 109 }
 110
 111 inline Score PawnInfo::pawns_value() const {
 112   return value;
 113 }
 114
 115 inline Bitboard PawnInfo::pawn_attacks(Color c) const {
 116   return pawnAttacks[c];
 117 }
 118
 119 inline Value PawnInfo::kingside_storm_value(Color c) const {
 120   return Value(ksStormValue[c]);
 121 }
 122
 123 inline Value PawnInfo::queenside_storm_value(Color c) const {
 124   return Value(qsStormValue[c]);
 125 }
 126
 127 inline Bitboard PawnInfo::passed_pawns(Color c) const {
 128   return passedPawns[c];
 129 }
 130
 131 inline int PawnInfo::file_is_half_open(Color c, File f) const {
 132   return (halfOpenFiles[c] & (1 << int(f)));
 133 }
 134
 135 inline int PawnInfo::has_open_file_to_left(Color c, File f) const {
 136   return halfOpenFiles[c] & ((1 << int(f)) - 1);
 137 }
 138
 139 inline int PawnInfo::has_open_file_to_right(Color c, File f) const {
 140   return halfOpenFiles[c] & ~((1 << int(f+1)) - 1);
 141 }
 142
 143 inline Score PawnInfo::king_shelter(const Position& pos, Color c, Square ksq) {
 144   return kingSquares[c] == ksq ? kingShelters[c] : updateShelter(pos, c, ksq);
 145 }
 146
 147 #endif // !defined(PAWNS_H_INCLUDED)