38804f9acf572b3518d105955d96539f04726c6e
[stockfish] / 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   template<typename T> Book& operator>>(T& n);
52
53   BookEntry read_entry(int idx);
54   int first_entry(uint64_t key);
55
56   std::ifstream bookFile;
57   std::string bookName;
58   int bookSize;
59   RKISS RKiss;
60 };
61
62 #endif // !defined(BOOK_H_INCLUDED)