46 lines
1.3 KiB
C++
46 lines
1.3 KiB
C++
#ifndef FRUITMODEL_H
|
|
#define FRUITMODEL_H
|
|
|
|
#include <QAbstractTableModel>
|
|
#include <QList>
|
|
#include <QMap>
|
|
#include <QIcon>
|
|
|
|
struct Fruit {
|
|
QString name;
|
|
QString origin;
|
|
int largeQuantity;
|
|
int mediumQuantity;
|
|
int smallQuantity;
|
|
QIcon icon;
|
|
Fruit(const QString &name,
|
|
const QString &origin,
|
|
int largeQuantity,
|
|
int mediumQuantity,
|
|
int smallQuantity,
|
|
const QIcon &icon)
|
|
: name(name),
|
|
origin(origin),
|
|
largeQuantity(largeQuantity),
|
|
mediumQuantity(mediumQuantity),
|
|
smallQuantity(smallQuantity),
|
|
icon(icon) {}
|
|
};
|
|
|
|
class FruitModel : public QAbstractTableModel {
|
|
Q_OBJECT
|
|
|
|
public:
|
|
explicit FruitModel(QObject *parent = nullptr);
|
|
void addFruit(const Fruit &fruit);
|
|
Qt::ItemFlags flags(const QModelIndex& index) const override;
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
int columnCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override;
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
|
|
|
|
private:
|
|
QList<Fruit> fruits;
|
|
};
|
|
|
|
#endif // FRUITMODEL_H
|