X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fbook.h;h=33f72fb517333f0fb511d1d74c1a0fe409742bd6;hp=516ea5cc6c077627f6a564da04503e116127e4a6;hb=0eedf47661784f86c9b9c927d2d7b9ba51e551d6;hpb=afadc33fb430abd2ba2783a62180cf971b13652f diff --git a/src/book.h b/src/book.h index 516ea5cc..33f72fb5 100644 --- a/src/book.h +++ b/src/book.h @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008 Marco Costalba + Copyright (C) 2008-2009 Marco Costalba Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -52,31 +52,24 @@ struct BookEntry { uint16_t sum; }; -class Book { - +class Book : private std::ifstream { public: - // Constructors - Book(); - - // Open and close book files + ~Book(); void open(const std::string& fName); void close(); - - // Testing if a book is opened - bool is_open() const; - - // The file name of the currently active book const std::string file_name() const; - - // Get a book move for a given position - Move get_move(const Position& pos) const; + Move get_move(const Position& pos); private: - int find_key(uint64_t key) const; - void read_entry(BookEntry& entry, int n) const; + Book& operator>>(uint64_t& n) { n = read_integer(8); return *this; } + Book& operator>>(uint16_t& n) { n = (uint16_t)read_integer(2); return *this; } + void operator>>(BookEntry& e) { *this >> e.key >> e.move >> e.count >> e.n >> e.sum; } + + uint64_t read_integer(int size); + void read_entry(BookEntry& e, int n); + int find_key(uint64_t key); std::string fileName; - mutable std::ifstream bookFile; int bookSize; };