diff --git a/cockatrice/src/carddatabasemodel.cpp b/cockatrice/src/carddatabasemodel.cpp index 2c6d658b..341542df 100644 --- a/cockatrice/src/carddatabasemodel.cpp +++ b/cockatrice/src/carddatabasemodel.cpp @@ -1,6 +1,8 @@ #include "carddatabasemodel.h" #include "filtertree.h" +#define CARDDBMODEL_COLUMNS 5 + CardDatabaseModel::CardDatabaseModel(CardDatabase *_db, QObject *parent) : QAbstractListModel(parent), db(_db) { @@ -21,31 +23,31 @@ int CardDatabaseModel::rowCount(const QModelIndex &/*parent*/) const int CardDatabaseModel::columnCount(const QModelIndex &/*parent*/) const { - return 5; + return CARDDBMODEL_COLUMNS; } QVariant CardDatabaseModel::data(const QModelIndex &index, int role) const { if (!index.isValid()) return QVariant(); - if ((index.row() >= cardList.size()) || (index.column() >= 5)) + if ((index.row() >= cardList.size()) || (index.column() >= CARDDBMODEL_COLUMNS)) return QVariant(); if (role != Qt::DisplayRole) return QVariant(); CardInfo *card = cardList.at(index.row()); switch (index.column()){ - case 0: return card->getName(); - case 1: { + case NameColumn: return card->getName(); + case SetListColumn: { QStringList setList; const QList &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(); + case ManaCostColumn: return card->getManaCost(); + case CardTypeColumn: return card->getCardType(); + case PTColumn: return card->getPowTough(); default: return QVariant(); } } @@ -57,11 +59,11 @@ QVariant CardDatabaseModel::headerData(int section, Qt::Orientation orientation, 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")); + case NameColumn: return QString(tr("Name")); + case SetListColumn: return QString(tr("Sets")); + case ManaCostColumn: return QString(tr("Mana cost")); + case CardTypeColumn: return QString(tr("Card type")); + case PTColumn: return QString(tr("P/T")); default: return QVariant(); } } @@ -86,7 +88,7 @@ void CardDatabaseModel::cardInfoChanged(CardInfo *card) if (row == -1) return; - emit dataChanged(index(row, 0), index(row, 4)); + emit dataChanged(index(row, 0), index(row, CARDDBMODEL_COLUMNS - 1)); } void CardDatabaseModel::cardAdded(CardInfo *card) diff --git a/cockatrice/src/carddatabasemodel.h b/cockatrice/src/carddatabasemodel.h index 91a01e8c..30d3b3a5 100644 --- a/cockatrice/src/carddatabasemodel.h +++ b/cockatrice/src/carddatabasemodel.h @@ -12,6 +12,7 @@ class FilterTree; class CardDatabaseModel : public QAbstractListModel { Q_OBJECT public: + enum Columns { NameColumn, SetListColumn, ManaCostColumn, CardTypeColumn, PTColumn, CMCColumn }; CardDatabaseModel(CardDatabase *_db, QObject *parent = 0); ~CardDatabaseModel(); int rowCount(const QModelIndex &parent = QModelIndex()) const;