econode.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 ECONODE_H
#define ECONODE_H

#include <QString>
#include <QMap>
#include "pgngame.h"
class QDataStream;
class PgnStream;

class LIB_EXPORT EcoNode
{
    public:
        ~EcoNode();

        bool isLeaf() const;
        EcoNode* child(const QString& sanMove) const;
        QString ecoCode() const;
        QString opening() const;
        QString variation() const;

        static void initialize();
        static void initialize(PgnStream& in);
        static const EcoNode* root();
        static const EcoNode* find(const QVector<PgnGame::MoveData>& moves);
        static void write(const QString& fileName);

    private:
        friend LIB_EXPORT QDataStream& operator<<(QDataStream& out, const EcoNode* node);
        friend LIB_EXPORT QDataStream& operator>>(QDataStream& in, EcoNode*& node);

        EcoNode();
        void addChild(const QString& sanMove, EcoNode* child);

        qint16 m_ecoCode;
        qint32 m_opening;
        QString m_variation;

        QMap<QString, EcoNode*> m_children;
};

extern LIB_EXPORT QDataStream& operator<<(QDataStream& out, const EcoNode* node);
extern LIB_EXPORT QDataStream& operator>>(QDataStream& in, EcoNode*& node);

#endif // ECONODE_H