servatrice/common/decklist.h

#ifndef DECKLIST_H
#define DECKLIST_H

#include <QList>
#include <QVector>
#include <QPair>
#include <QObject>
#include <QStringList>
#include <QSet>
#include <QMap>

// Required on Mac. Forward declaration doesn't work. Don't ask why.
#include <QtCore/QXmlStreamReader>
#include <QtCore/QXmlStreamWriter>

#include "pb/move_card_to_zone.pb.h"

class CardDatabase;
class QIODevice;
class QTextStream;

class InnerDecklistNode;

class SideboardPlan {
private:
	QString name;
	QList<MoveCard_ToZone> moveList;
public:
	SideboardPlan(const QString &_name = QString(), const QList<MoveCard_ToZone> &_moveList = QList<MoveCard_ToZone>());
	bool readElement(QXmlStreamReader *xml);
	void write(QXmlStreamWriter *xml);
	
	QString getName() const { return name; }
	const QList<MoveCard_ToZone> &getMoveList() const { return moveList; }
	void setMoveList(const QList<MoveCard_ToZone> &_moveList);
};

class AbstractDecklistNode {
protected:
	InnerDecklistNode *parent;
	int sortMethod;
public:
	AbstractDecklistNode(InnerDecklistNode *_parent = 0);
	virtual ~AbstractDecklistNode() { }
	virtual void setSortMethod(int method) { sortMethod = method; }
	virtual QString getName() const = 0;
	InnerDecklistNode *getParent() const { return parent; }
	int depth() const;
	virtual int height() const = 0;
	virtual bool compare(AbstractDecklistNode *other) const = 0;
	
	virtual bool readElement(QXmlStreamReader *xml) = 0;
	virtual void writeElement(QXmlStreamWriter *xml) = 0;
};

class InnerDecklistNode : public AbstractDecklistNode, public QList<AbstractDecklistNode *> {
private:
	QString name;
	class compareFunctor;
public:
	InnerDecklistNode(const QString &_name = QString(), InnerDecklistNode *_parent = 0) : AbstractDecklistNode(_parent), name(_name) { }
	InnerDecklistNode(InnerDecklistNode *other, InnerDecklistNode *_parent = 0);
	virtual ~InnerDecklistNode();
	void setSortMethod(int method);
	QString getName() const { return name; }
	void setName(const QString &_name) { name = _name; }
	static QString visibleNameFromName(const QString &_name);
	virtual QString getVisibleName() const;
	void clearTree();
	AbstractDecklistNode *findChild(const QString &name);
	int height() const;
	int recursiveCount(bool countTotalCards = false) const;
        float recursivePrice(bool countTotalCards = false) const;
	bool compare(AbstractDecklistNode *other) const;
	bool compareNumber(AbstractDecklistNode *other) const;
	bool compareName(AbstractDecklistNode *other) const;
	bool comparePrice(AbstractDecklistNode *other) const;
	QVector<QPair<int, int> > sort(Qt::SortOrder order = Qt::AscendingOrder);
	
	bool readElement(QXmlStreamReader *xml);
	void writeElement(QXmlStreamWriter *xml);
};

class AbstractDecklistCardNode : public AbstractDecklistNode {
public:
	AbstractDecklistCardNode(InnerDecklistNode *_parent = 0) : AbstractDecklistNode(_parent) { }
	virtual int getNumber() const = 0;
	virtual void setNumber(int _number) = 0;
	virtual QString getName() const = 0;
	virtual void setName(const QString &_name) = 0;
        virtual float getPrice() const = 0;
        virtual void setPrice(float _price) = 0;
        float getTotalPrice() const { return getNumber() * getPrice(); }
	int height() const { return 0; }
	bool compare(AbstractDecklistNode *other) const;
	bool compareNumber(AbstractDecklistNode *other) const;
	bool compareName(AbstractDecklistNode *other) const;
	bool compareTotalPrice(AbstractDecklistNode *other) const;
	
	bool readElement(QXmlStreamReader *xml);
	void writeElement(QXmlStreamWriter *xml);
};

class DecklistCardNode : public AbstractDecklistCardNode {
private:
	QString name;
	int number;
        float price;
public:
        DecklistCardNode(const QString &_name = QString(), int _number = 1, float _price = 0, InnerDecklistNode *_parent = 0) : AbstractDecklistCardNode(_parent), name(_name), number(_number), price(_price) { }
        DecklistCardNode(const QString &_name = QString(), int _number = 1, InnerDecklistNode *_parent = 0) : AbstractDecklistCardNode(_parent), name(_name), number(_number), price(0) { }
	DecklistCardNode(DecklistCardNode *other, InnerDecklistNode *_parent);
	int getNumber() const { return number; }
	void setNumber(int _number) { number = _number; }
	QString getName() const { return name; }
	void setName(const QString &_name) { name = _name; }
        float getPrice() const { return price; }
        void setPrice(const float _price) { price = _price; }
};

class DeckList : public QObject {
	Q_OBJECT
private:
	QString name, comments;
	QString deckHash;
	QMap<QString, SideboardPlan *> sideboardPlans;
	InnerDecklistNode *root;
	void getCardListHelper(InnerDecklistNode *node, QSet<QString> &result) const;
signals:
	void deckHashChanged();
public slots:
	void setName(const QString &_name = QString()) { name = _name; }
	void setComments(const QString &_comments = QString()) { comments = _comments; }
public:
	DeckList();
	DeckList(const DeckList &other);
	DeckList(const QString &nativeString);
	~DeckList();
	QString getName() const { return name; }
	QString getComments() const { return comments; }
	QList<MoveCard_ToZone> getCurrentSideboardPlan();
	void setCurrentSideboardPlan(const QList<MoveCard_ToZone> &plan);
	const QMap<QString, SideboardPlan *> &getSideboardPlans() const { return sideboardPlans; }

	bool readElement(QXmlStreamReader *xml);
	void write(QXmlStreamWriter *xml);
	bool loadFromXml(QXmlStreamReader *xml);
	bool loadFromString_Native(const QString &nativeString);
	QString writeToString_Native();
	bool loadFromFile_Native(QIODevice *device);
	bool saveToFile_Native(QIODevice *device);
	bool loadFromStream_Plain(QTextStream &stream);
	bool loadFromFile_Plain(QIODevice *device);
	bool saveToStream_Plain(QTextStream &stream);
	bool saveToFile_Plain(QIODevice *device);
	QString writeToString_Plain();

	void cleanList();
	bool isEmpty() const { return root->isEmpty() && name.isEmpty() && comments.isEmpty() && sideboardPlans.isEmpty(); }
	QStringList getCardList() const;
	
	QString getDeckHash() const { return deckHash; }
	void updateDeckHash();

	InnerDecklistNode *getRoot() const { return root; }
	DecklistCardNode *addCard(const QString &cardName, const QString &zoneName);
	bool deleteNode(AbstractDecklistNode *node, InnerDecklistNode *rootNode = 0);
};

#endif