39 lines
1.3 KiB
C++
39 lines
1.3 KiB
C++
#ifndef TIMEMODEL_H
|
|
#define TIMEMODEL_H
|
|
|
|
#include <QAbstractTableModel>
|
|
#include <qcolor.h>
|
|
#include <qvector.h>
|
|
|
|
class TimeModel : public QAbstractTableModel
|
|
{
|
|
Q_OBJECT
|
|
|
|
public:
|
|
explicit TimeModel(QVector<QVector<int>> timeData, QVector<int> minimumTimes, QObject *parent = nullptr);
|
|
explicit TimeModel(QVector<QVector<int>> timeData, QVector<int> minimumTimes, QVector<int> topTime, QObject *parent = nullptr);
|
|
|
|
// Header:
|
|
QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const override;
|
|
|
|
// Basic functionality:
|
|
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;
|
|
|
|
// Add data:
|
|
bool insertRows(int row, int count, const QModelIndex &parent = QModelIndex()) override;
|
|
bool insertColumns(int column, int count, const QModelIndex &parent = QModelIndex()) override;
|
|
|
|
private:
|
|
int getMin(const QVector<int> x) const;
|
|
|
|
QColor getColor(const QVector<QVector<int>> data, const int col, const int row) const; // row and col starts counting at 0
|
|
QVector<QVector<int>> timeData;
|
|
QVector<int> topTime;
|
|
QVector<int> minimumTimes;
|
|
};
|
|
|
|
#endif // TIMEMODEL_H
|