openingbook.h

/*
    This file is part of Cute Chess.

    Cute Chess is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    Cute Chess is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with Cute Chess.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef OPENING_BOOK_H
#define OPENING_BOOK_H

#include <QtGlobal>
#include <QMultiMap>
#include "board/genericmove.h"

class QString;
class QDataStream;
class PgnGame;
class PgnStream;


class LIB_EXPORT OpeningBook
{
    public:
        enum AccessMode
        {
            Ram,    
            Disk    
        };

        struct Entry
        {
            Chess::GenericMove move;
            quint16 weight;
        };

        OpeningBook(AccessMode mode = Ram);
        virtual ~OpeningBook();
        
        int import(const PgnGame& pgn, int maxMoves);
        int import(PgnStream& in, int maxMoves);
        
        Chess::GenericMove move(quint64 key) const;

        QList<Entry> entries(quint64 key) const;

        bool read(const QString& filename);

        bool write(const QString& filename) const;


    protected:
        friend LIB_EXPORT QDataStream& operator>>(QDataStream& in, OpeningBook* book);
        friend LIB_EXPORT QDataStream& operator<<(QDataStream& out, const OpeningBook* book);

        typedef QMultiMap<quint64, Entry> Map;

        virtual int entrySize() const = 0;

        void addEntry(const Entry& entry, quint64 key);
        
        virtual Entry readEntry(QDataStream& in, quint64* key) const = 0;
        
        virtual void writeEntry(const Map::const_iterator& it,
                    QDataStream& out) const = 0;

    private:
        QList<Entry> entriesFromDisk(quint64 key) const;

        AccessMode m_mode;
        QString m_filename;
        Map m_map;
};

extern LIB_EXPORT QDataStream& operator>>(QDataStream& in, OpeningBook* book);

extern LIB_EXPORT QDataStream& operator<<(QDataStream& out, const OpeningBook* book);

#endif // OPENING_BOOK_H