PoeMerchant/InvestitionModel.cpp at master · Luminatione/PoeMerchant · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
#include "InvestitionModel.h"

InvestitionModel::InvestitionModel(QObject *parent) : QAbstractListModel(parent){}

void InvestitionModel::addInvestition(Investition &investition)
{
    beginInsertRows(QModelIndex(), rowCount(), rowCount());
    investitions << investition;
    endInsertRows();
}

int InvestitionModel::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent);
    return investitions.count();
}

QVariant InvestitionModel::data(const QModelIndex &index, int role) const
{
    if(index.row() < 0 || index.row() >= investitions.count())
    {
        return QVariant();
    }

    switch (role) {
        case CardAmountRole:
        {
            return investitions[index.row()].cardAmount();
        }
        case  CardNameRole:
        {
            return investitions[index.row()].cardName();
        }
        case ItemRole:
        {
            return investitions[index.row()].item();
        }
        case ItemColorRole:
        {
            return investitions[index.row()].itemColor();
        }
        case CostRole:
        {
            return investitions[index.row()].cost();
        }
        case CostCurrencyIconRole:
        {
            return investitions[index.row()].costCurrencyIcon();
        }
        case ProfitRole:
        {
            return investitions[index.row()].profit();
        }
        case ProfitCurrencyIconRole:
        {
            return investitions[index.row()].profitCurrencyIcon();
        }
        case ItemAmountRole:
        {
            return investitions[index.row()].itemAmount();
        }
        default:
            return QVariant();
    }
}

void InvestitionModel::clear()
{
    if(investitions.count() > 0)
    {
        beginRemoveRows(QModelIndex(), 0, rowCount() - 1);
        investitions.clear();
        endRemoveRows();
    }
}

QHash<int, QByteArray> InvestitionModel::roleNames() const
{
    QHash<int, QByteArray> roles;
    roles[CardAmountRole] = "cardAmount";
    roles[CardNameRole] = "cardName";
    roles[ItemRole] = "item";
    roles[ItemColorRole] = "itemColor";
    roles[CostRole] = "cost";
    roles[CostCurrencyIconRole] = "costCurrencyIcon";
    roles[ProfitRole] = "profit";
    roles[ProfitCurrencyIconRole] = "profitCurrencyIcon";
    roles[ItemAmountRole] = "itemAmount";
    return roles;
}