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 : private std::ifstream {
  42   Book(const Book&); // just decleared..
  43   Book& operator=(const Book&); // ..to avoid a warning
  44 public:
  45   Book();
  46   ~Book();
  47   void open(const std::string& fileName);
  48   void close();
  49   Move get_move(const Position& pos, bool findBestMove);
  50   const std::string name() const { return bookName; }
  51
  52 private:
  53   // read n chars from the file stream and converts them in an
  54   // integer number. Integers are stored with highest byte first.
  55   template<int n> uint64_t get_int();
  56
  57   template<typename T>
  58   Book& operator>>(T& n) { n = (T)get_int<sizeof(T)>(); return *this; }
  59
  60   BookEntry read_entry(int idx);
  61   int find_entry(uint64_t key);
  62
  63   std::string bookName;
  64   int bookSize;
  65   RKISS RKiss;
  66 };
  67
  68 // Yes, we indulge a bit here ;-)
  69 template<int n> inline uint64_t Book::get_int() { return 256 * get_int<n-1>() + get(); }
  70 template<> inline uint64_t Book::get_int<1>() { return get(); }
  71
  72
  73 #endif // !defined(BOOK_H_INCLUDED)