diff --git a/cockatrice/src/carddatabasemodel.cpp b/cockatrice/src/carddatabasemodel.cpp index be1311a6..9562398c 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,33 @@ 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) + if (role != Qt::DisplayRole && role != SortRole) 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 role == SortRole ? + QString("%1%2").arg(card->getCmc(), 4, QChar('0')).arg(card->getManaCost()) : + card->getManaCost(); + case CardTypeColumn: return card->getCardType(); + case PTColumn: return card->getPowTough(); default: return QVariant(); } } @@ -57,11 +61,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 +90,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) @@ -120,10 +124,10 @@ CardDatabaseDisplayModel::CardDatabaseDisplayModel(QObject *parent) bool CardDatabaseDisplayModel::lessThan(const QModelIndex &left, const QModelIndex &right) const { - QString leftString = sourceModel()->data(left).toString(); - QString rightString = sourceModel()->data(right).toString(); + QString leftString = sourceModel()->data(left, CardDatabaseModel::SortRole).toString(); + QString rightString = sourceModel()->data(right, CardDatabaseModel::SortRole).toString(); - if (!cardName.isEmpty()) + if (!cardName.isEmpty() && left.column() == CardDatabaseModel::NameColumn) { bool isLeftType = leftString.startsWith(cardName, Qt::CaseInsensitive); bool isRightType = rightString.startsWith(cardName, Qt::CaseInsensitive); diff --git a/cockatrice/src/carddatabasemodel.h b/cockatrice/src/carddatabasemodel.h index 91a01e8c..da72558b 100644 --- a/cockatrice/src/carddatabasemodel.h +++ b/cockatrice/src/carddatabasemodel.h @@ -12,6 +12,8 @@ class FilterTree; class CardDatabaseModel : public QAbstractListModel { Q_OBJECT public: + enum Columns { NameColumn, SetListColumn, ManaCostColumn, CardTypeColumn, PTColumn, CMCColumn }; + enum Role { SortRole=Qt::UserRole }; CardDatabaseModel(CardDatabase *_db, QObject *parent = 0); ~CardDatabaseModel(); int rowCount(const QModelIndex &parent = QModelIndex()) const;