99 lines
3 KiB
C++
99 lines
3 KiB
C++
#ifndef DECKLISTMODEL_H
|
|
#define DECKLISTMODEL_H
|
|
|
|
#include "decklist.h"
|
|
|
|
#include <QAbstractItemModel>
|
|
#include <QList>
|
|
|
|
class DeckLoader;
|
|
class CardDatabase;
|
|
class QPrinter;
|
|
class QTextCursor;
|
|
|
|
class DecklistModelCardNode : public AbstractDecklistCardNode
|
|
{
|
|
private:
|
|
DecklistCardNode *dataNode;
|
|
|
|
public:
|
|
DecklistModelCardNode(DecklistCardNode *_dataNode, InnerDecklistNode *_parent)
|
|
: AbstractDecklistCardNode(_parent), dataNode(_dataNode)
|
|
{
|
|
}
|
|
int getNumber() const override
|
|
{
|
|
return dataNode->getNumber();
|
|
}
|
|
void setNumber(int _number) override
|
|
{
|
|
dataNode->setNumber(_number);
|
|
}
|
|
QString getName() const override
|
|
{
|
|
return dataNode->getName();
|
|
}
|
|
void setName(const QString &_name) override
|
|
{
|
|
dataNode->setName(_name);
|
|
}
|
|
DecklistCardNode *getDataNode() const
|
|
{
|
|
return dataNode;
|
|
}
|
|
};
|
|
|
|
class DeckListModel : public QAbstractItemModel
|
|
{
|
|
Q_OBJECT
|
|
private slots:
|
|
void rebuildTree();
|
|
public slots:
|
|
void printDeckList(QPrinter *printer);
|
|
signals:
|
|
void deckHashChanged();
|
|
|
|
public:
|
|
explicit DeckListModel(QObject *parent = nullptr);
|
|
~DeckListModel() override;
|
|
int rowCount(const QModelIndex &parent) const override;
|
|
int columnCount(const QModelIndex & /*parent*/ = QModelIndex()) const override;
|
|
QVariant data(const QModelIndex &index, int role) const override;
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role) const override;
|
|
QModelIndex index(int row, int column, const QModelIndex &parent) const override;
|
|
QModelIndex parent(const QModelIndex &index) const override;
|
|
Qt::ItemFlags flags(const QModelIndex &index) const override;
|
|
bool setData(const QModelIndex &index, const QVariant &value, int role) override;
|
|
bool removeRows(int row, int count, const QModelIndex &parent) override;
|
|
QModelIndex findCard(const QString &cardName, const QString &zoneName) const;
|
|
QModelIndex addCard(const QString &cardName, const QString &zoneName, bool abAddAnyway = false);
|
|
void sort(int column, Qt::SortOrder order) override;
|
|
void cleanList();
|
|
DeckLoader *getDeckList() const
|
|
{
|
|
return deckList;
|
|
}
|
|
void setDeckList(DeckLoader *_deck);
|
|
|
|
private:
|
|
DeckLoader *deckList;
|
|
InnerDecklistNode *root;
|
|
int lastKnownColumn;
|
|
Qt::SortOrder lastKnownOrder;
|
|
InnerDecklistNode *createNodeIfNeeded(const QString &name, InnerDecklistNode *parent);
|
|
QModelIndex nodeToIndex(AbstractDecklistNode *node) const;
|
|
DecklistModelCardNode *findCardNode(const QString &cardName, const QString &zoneName) const;
|
|
void emitRecursiveUpdates(const QModelIndex &index);
|
|
void sortHelper(InnerDecklistNode *node, Qt::SortOrder order);
|
|
|
|
void printDeckListNode(QTextCursor *cursor, InnerDecklistNode *node);
|
|
|
|
template <typename T> T getNode(const QModelIndex &index) const
|
|
{
|
|
if (!index.isValid())
|
|
return dynamic_cast<T>(root);
|
|
return dynamic_cast<T>(static_cast<AbstractDecklistNode *>(index.internalPointer()));
|
|
}
|
|
};
|
|
|
|
#endif
|