TreeViewDecoratorModel.h

// SPDX-License-Identifier: LGPL-2.1-or-later
//
// SPDX-FileCopyrightText: 2014 Levente Kurusa <levex@linux.com>
 
#ifndef TREEVIEWDECORATORMODEL_H
#define TREEVIEWDECORATORMODEL_H
 
#include <QSortFilterProxyModel>
 
namespace Marble
{
 
 
class TreeViewDecoratorModel : public QSortFilterProxyModel
{
    Q_OBJECT
 
public Q_SLOTS:
    void trackExpandedState( const QModelIndex &index );
    void trackCollapsedState( const QModelIndex &index );
 
public:
    explicit TreeViewDecoratorModel( QObject *parent = nullptr );
    QVariant data( const QModelIndex & proxyIndex, int role = Qt::DisplayRole ) const override;
 
protected:
    bool filterAcceptsRow( int sourceRow, const QModelIndex &sourceParent ) const override;
 
private:
    QList<QPersistentModelIndex> m_expandedRows;
 
};
 
}
 
#endif // TREEVIEWDECORATORMODEL_H