88 lines
2.2 KiB
C++
88 lines
2.2 KiB
C++
#include "carddatabasemodel.h"
|
|
|
|
CardDatabaseModel::CardDatabaseModel(CardDatabase *_db, QObject *parent)
|
|
: QAbstractListModel(parent), db(_db)
|
|
{
|
|
cardList = db->getCardList();
|
|
}
|
|
|
|
CardDatabaseModel::~CardDatabaseModel()
|
|
{
|
|
|
|
}
|
|
|
|
int CardDatabaseModel::rowCount(const QModelIndex &/*parent*/) const
|
|
{
|
|
return cardList.size();
|
|
}
|
|
|
|
int CardDatabaseModel::columnCount(const QModelIndex &/*parent*/) const
|
|
{
|
|
return 5;
|
|
}
|
|
|
|
QVariant CardDatabaseModel::data(const QModelIndex &index, int role) const
|
|
{
|
|
if (!index.isValid())
|
|
return QVariant();
|
|
if ((index.row() >= cardList.size()) || (index.column() >= 5))
|
|
return QVariant();
|
|
if (role != Qt::DisplayRole)
|
|
return QVariant();
|
|
|
|
CardInfo *card = cardList.at(index.row());
|
|
switch (index.column()){
|
|
case 0: return card->getName();
|
|
case 1: {
|
|
QStringList setList;
|
|
const QList<CardSet *> &sets = card->getSets();
|
|
for (int i = 0; i < sets.size(); i++)
|
|
setList << sets[i]->getShortName();
|
|
return setList.join(", ");
|
|
}
|
|
case 2: return card->getManaCost();
|
|
case 3: return card->getCardType();
|
|
case 4: return card->getPowTough();
|
|
default: return QVariant();
|
|
}
|
|
}
|
|
|
|
QVariant CardDatabaseModel::headerData(int section, Qt::Orientation orientation, int role) const
|
|
{
|
|
if (role != Qt::DisplayRole)
|
|
return QVariant();
|
|
if (orientation != Qt::Horizontal)
|
|
return QVariant();
|
|
switch (section) {
|
|
case 0: return QString(tr("Name"));
|
|
case 1: return QString(tr("Sets"));
|
|
case 2: return QString(tr("Mana cost"));
|
|
case 3: return QString(tr("Card type"));
|
|
case 4: return QString(tr("P/T"));
|
|
default: return QVariant();
|
|
}
|
|
}
|
|
|
|
CardDatabaseDisplayModel::CardDatabaseDisplayModel(QObject *parent)
|
|
: QSortFilterProxyModel(parent)
|
|
{
|
|
setFilterCaseSensitivity(Qt::CaseInsensitive);
|
|
setSortCaseSensitivity(Qt::CaseInsensitive);
|
|
}
|
|
|
|
bool CardDatabaseDisplayModel::filterAcceptsRow(int sourceRow, const QModelIndex & /*sourceParent*/) const
|
|
{
|
|
CardInfo *info = static_cast<CardDatabaseModel *>(sourceModel())->getCard(sourceRow);
|
|
|
|
if (!cardNameBeginning.isEmpty())
|
|
if (!info->getName().startsWith(cardNameBeginning, Qt::CaseInsensitive))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
void CardDatabaseDisplayModel::setCardNameBeginning(const QString &_beginning)
|
|
{
|
|
cardNameBeginning = _beginning;
|
|
invalidateFilter();
|
|
}
|