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-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(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
  41 class Book : private std::ifstream {
  42 public:
  43   Book();
  44   ~Book();
  45   Move probe(const Position& pos, const std::string& fName, bool pickBest);
  46
  47 private:
  48   template<typename T> Book& operator>>(T& n);
  49
  50   bool open(const char* fName);
  51   void binary_search(uint64_t key);
  52
  53   RKISS RKiss;
  54   std::string fileName;
  55   size_t size;
  56 };
  57
  58 #endif // !defined(BOOK_H_INCLUDED)