51 lines
1.3 KiB
C++
51 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <QAbstractListModel>
|
|
|
|
class SectionsDecoratorModel : public QAbstractListModel
|
|
{
|
|
Q_OBJECT
|
|
Q_PROPERTY(QAbstractItemModel* sourceModel READ sourceModel
|
|
WRITE setSourceModel NOTIFY sourceModelChanged)
|
|
public:
|
|
enum Roles {
|
|
SectionRole = Qt::UserRole + 100,
|
|
IsSectionRole,
|
|
IsFoldedRole,
|
|
SubitemsCountRole
|
|
};
|
|
|
|
explicit SectionsDecoratorModel(QObject *parent = nullptr);
|
|
|
|
void setSourceModel(QAbstractItemModel *sourceModel);
|
|
QAbstractItemModel *sourceModel() const;
|
|
|
|
int rowCount(const QModelIndex &parent = QModelIndex()) const override;
|
|
QVariant data(const QModelIndex &index, int role) const override;
|
|
QHash<int, QByteArray> roleNames() const override;
|
|
|
|
Q_INVOKABLE void flipFolding(int index);
|
|
|
|
signals:
|
|
void sourceModelChanged();
|
|
|
|
private:
|
|
struct RowMetadata {
|
|
bool isSection = false;
|
|
QString sectionName;
|
|
bool folded = false;
|
|
int offset = 0;
|
|
int count = 0;
|
|
};
|
|
|
|
void initialize();
|
|
void calculateOffsets();
|
|
|
|
void onInserted(const QModelIndex &parent, int first, int last);
|
|
void onRemoved(const QModelIndex &parent, int first, int last);
|
|
|
|
QAbstractItemModel* m_sourceModel = nullptr;
|
|
std::vector<RowMetadata> m_rowsMetadata;
|
|
int m_sectionRole = 0;
|
|
};
|