servatrice/cockatrice/src/carddatabase.h

#ifndef CARDDATABASE_H
#define CARDDATABASE_H

#include <QHash>
#include <QPixmap>
#include <QMap>
#include <QDataStream>
#include <QList>
#include <QXmlStreamReader>
#include <QNetworkRequest>
#include <QThread>
#include <QMutex>
#include <QWaitCondition>

class CardDatabase;
class CardInfo;
class QNetworkAccessManager;
class QNetworkReply;
class QNetworkRequest;

typedef QMap<QString, QString> QStringMap;

class CardSet : public QList<CardInfo *> {
private:
	QString shortName, longName;
	unsigned int sortKey;
public:
	CardSet(const QString &_shortName = QString(), const QString &_longName = QString());
	QString getShortName() const { return shortName; }
	QString getLongName() const { return longName; }
	int getSortKey() const { return sortKey; }
	void setSortKey(unsigned int _sortKey);
	void updateSortKey();
};

class SetList : public QList<CardSet *> {
private:
	class CompareFunctor;
public:
	void sortByKey();
};

class PictureToLoad {
private:
	CardInfo *card;
	bool stripped;
	SetList sortedSets;
	int setIndex;
	bool hq;
public:
	PictureToLoad(CardInfo *_card = 0, bool _stripped = false, bool _hq = true);
	CardInfo *getCard() const { return card; }
	bool getStripped() const { return stripped; }
	QString getSetName() const { return sortedSets[setIndex]->getShortName(); }
	bool nextSet();
		
	bool getHq() const { return hq; }
	void setHq(bool _hq) { hq = _hq; }
	
};

class PictureLoader : public QObject {
	Q_OBJECT
private:
	QString _picsPath;
	QList<PictureToLoad> loadQueue;
	QMutex mutex;
	QNetworkAccessManager *networkManager;
	QList<PictureToLoad> cardsToDownload;
	PictureToLoad cardBeingDownloaded;
	bool picDownload, downloadRunning, loadQueueRunning;
	void startNextPicDownload();
public:
	PictureLoader(QObject *parent = 0);
	void setPicsPath(const QString &path);
	void setPicDownload(bool _picDownload);
	void loadImage(CardInfo *card, bool stripped);
private slots:
	void picDownloadFinished(QNetworkReply *reply);
public slots:
	void processLoadQueue();
signals:
	void startLoadQueue();
	void imageLoaded(CardInfo *card, const QImage &image);
};

class PictureLoadingThread : public QThread {
	Q_OBJECT
private:
	QString picsPath;
	bool picDownload;
	PictureLoader *pictureLoader;
	QWaitCondition initWaitCondition;
protected:
	void run();
public:
	PictureLoadingThread(const QString &_picsPath, bool _picDownload, QObject *parent);
	~PictureLoadingThread();
	PictureLoader *getPictureLoader() const { return pictureLoader; }
	void waitForInit();
signals:
	void imageLoaded(CardInfo *card, const QImage &image);
};

class CardInfo : public QObject {
	Q_OBJECT
private:
	CardDatabase *db;

	QString name;
	SetList sets;
	QString manacost;
	QString cardtype;
	QString powtough;
	QString text;
	QStringList colors;
	int loyalty;
	QMap<QString, QString> picURLs, picURLsHq, picURLsSt;
	bool cipt;
	int tableRow;
	QPixmap *pixmap;
	QMap<int, QPixmap *> scaledPixmapCache;
public:
	CardInfo(CardDatabase *_db,
		const QString &_name = QString(),
		const QString &_manacost = QString(),
		const QString &_cardtype = QString(),
		const QString &_powtough = QString(),
		const QString &_text = QString(),
		const QStringList &_colors = QStringList(),
		int _loyalty = 0,
		bool cipt = false,
		int _tableRow = 0,
		const SetList &_sets = SetList(),
		const QStringMap &_picURLs = QStringMap(),
		const QStringMap &_picURLsHq = QStringMap(),
		const QStringMap &_picURLsSt = QStringMap());
	~CardInfo();
	const QString &getName() const { return name; }
	const SetList &getSets() const { return sets; }
	const QString &getManaCost() const { return manacost; }
	const QString &getCardType() const { return cardtype; }
	const QString &getPowTough() const { return powtough; }
	const QString &getText() const { return text; }
	const int &getLoyalty() const { return loyalty; }
	bool getCipt() const { return cipt; }
	void setText(const QString &_text) { text = _text; }
	const QStringList &getColors() const { return colors; }
	QString getPicURL(const QString &set) const { return picURLs.value(set); }
	QString getPicURLHq(const QString &set) const { return picURLsHq.value(set); }
	QString getPicURLSt(const QString &set) const { return picURLsSt.value(set); }
	QString getPicURL() const;
	const QMap<QString, QString> &getPicURLs() const { return picURLs; }
	QString getMainCardType() const;
	QString getCorrectedName() const;
	int getTableRow() const { return tableRow; }
	void setTableRow(int _tableRow) { tableRow = _tableRow; }
	void setLoyalty(int _loyalty) { loyalty = _loyalty; }
	void setPicURL(const QString &_set, const QString &_picURL) { picURLs.insert(_set, _picURL); }
	void setPicURLHq(const QString &_set, const QString &_picURL) { picURLsHq.insert(_set, _picURL); }
	void setPicURLSt(const QString &_set, const QString &_picURL) { picURLsSt.insert(_set, _picURL); }
	void addToSet(CardSet *set);
	QPixmap *loadPixmap();
	QPixmap *getPixmap(QSize size);
	void clearPixmapCache();
	void clearPixmapCacheMiss();
	void imageLoaded(const QImage &image);
public slots:
	void updatePixmapCache();
signals:
	void pixmapUpdated();
};

class CardDatabase : public QObject {
	Q_OBJECT
protected:
	QHash<QString, CardInfo *> cardHash;
	QHash<QString, CardSet *> setHash;
	bool loadSuccess;
	CardInfo *noCard;
	PictureLoadingThread *loadingThread;
private:
	static const int versionNeeded;
	void loadCardsFromXml(QXmlStreamReader &xml);
	void loadSetsFromXml(QXmlStreamReader &xml);
public:
	CardDatabase(QObject *parent = 0);
	~CardDatabase();
	void clear();
	CardInfo *getCard(const QString &cardName = QString());
	CardSet *getSet(const QString &setName);
	QList<CardInfo *> getCardList() const { return cardHash.values(); }
	SetList getSetList() const;
	bool loadFromFile(const QString &fileName);
	bool saveToFile(const QString &fileName);
	QStringList getAllColors() const;
	QStringList getAllMainCardTypes() const;
	bool getLoadSuccess() const { return loadSuccess; }
	void cacheCardPixmaps(const QStringList &cardNames);
	void loadImage(CardInfo *card);
public slots:
	void clearPixmapCache();
	bool loadCardDatabase(const QString &path);
	bool loadCardDatabase();
private slots:
	void imageLoaded(CardInfo *card, QImage image);
	void picDownloadChanged();
	void picsPathChanged();
signals:
	void cardListChanged();
};

#endif