git.sesse.net Git - stockfish/blob - src/book.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 #if !defined(BOOK_H_INCLUDED)
  21 #define BOOK_H_INCLUDED
  22
  23 #include <fstream>
  24 #include <string>
  25
  26 #include "position.h"
  27 #include "rkiss.h"
  28
  29
  30 /// A Polyglot book is a series of "entries" of 16 bytes. All integers are
  31 /// stored highest byte first (regardless of size). The entries are ordered
  32 /// according to key. Lowest key first.
  33 struct BookEntry {
  34   uint64_t key;
  35   uint16_t move;
  36   uint16_t count;
  37   uint32_t learn;
  38 };
  39
  40 class Book {
  41 public:
  42   Book();
  43   ~Book();
  44   void open(const std::string& fileName);
  45   void close();
  46   Move probe(const Position& pos, bool findBestMove);
  47   const std::string name() const { return bookName; }
  48
  49 private:
  50   template<typename T> Book& operator>>(T& n);
  51
  52   BookEntry read_entry(int idx);
  53   int first_entry(uint64_t key);
  54
  55   RKISS RKiss;
  56   std::ifstream bookFile;
  57   std::string bookName;
  58   int bookSize;
  59 };
  60
  61 #endif // !defined(BOOK_H_INCLUDED)