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 "move.h"
  27 #include "position.h"
  28 #include "rkiss.h"
  29
  30
  31 // A Polyglot book is a series of "entries" of 16 bytes. All integers are
  32 // stored highest byte first (regardless of size). The entries are ordered
  33 // according to key. Lowest key first.
  34 struct BookEntry {
  35   uint64_t key;
  36   uint16_t move;
  37   uint16_t count;
  38   uint32_t learn;
  39 };
  40
  41 class Book {
  42 public:
  43   Book();
  44   ~Book();
  45   void open(const std::string& fileName);
  46   void close();
  47   Move get_move(const Position& pos, bool findBestMove);
  48   const std::string name() const { return bookName; }
  49
  50 private:
  51   // read n chars from the file stream and converts them in an
  52   // integer number. Integers are stored with highest byte first.
  53   template<int n> uint64_t get_int();
  54
  55   template<typename T>
  56   Book& operator>>(T& n) { n = (T)get_int<sizeof(T)>(); return *this; }
  57
  58   BookEntry read_entry(int idx);
  59   int find_entry(uint64_t key);
  60
  61   std::ifstream bookFile;
  62   std::string bookName;
  63   int bookSize;
  64   RKISS RKiss;
  65 };
  66
  67 // Yes, we indulge a bit here ;-)
  68 template<int n> inline uint64_t Book::get_int() { return 256 * get_int<n-1>() + bookFile.get(); }
  69 template<> inline uint64_t Book::get_int<1>() { return bookFile.get(); }
  70
  71
  72 #endif // !defined(BOOK_H_INCLUDED)