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 probe(const Position& pos, bool findBestMove);
  48   const std::string name() const { return bookName; }
  49
  50 private:
  51   template<typename T> Book& operator>>(T& n);
  52
  53   BookEntry read_entry(int idx);
  54   int first_entry(uint64_t key);
  55
  56   RKISS RKiss;
  57   std::ifstream bookFile;
  58   std::string bookName;
  59   int bookSize;
  60 };
  61
  62 #endif // !defined(BOOK_H_INCLUDED)