qgslayertreefilterproxymodel.h

Go to the documentation of this file.

/***************************************************************************
  qgslayertreefilterproxymodel.h
 
 ---------------------
 begin                : 05.06.2020
 copyright            : (C) 2020 by Denis Rouzaud
 email                : [email protected]
 ***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 
#ifndef QGSLAYERTREEFILTERPROXYMODEL_H
#define QGSLAYERTREEFILTERPROXYMODEL_H
 
#include <QSortFilterProxyModel>
 
#include "qgis_core.h"
#include "qgsmaplayer.h"
#include "qgsmaplayerproxymodel.h"
 
class QgsLayerTreeModel;
class QgsLayerTreeNode;
 
class CORE_EXPORT QgsLayerTreeFilterProxyModel : public QSortFilterProxyModel
{
    Q_OBJECT
  public:
    QgsLayerTreeFilterProxyModel( QObject *parent = nullptr );
 
    void setCheckedLayers( const QList<QgsMapLayer *> layers );
 
    QList<QgsMapLayer *> checkedLayers() const {return mCheckedLayers;}
 
    QgsMapLayer *mapLayer( const QModelIndex &idx ) const;
 
    QgsLayerTreeModel *layerTreeModel() const;
    void setLayerTreeModel( QgsLayerTreeModel *layerTreeModel );
 
    void setFilters( const QgsMapLayerProxyModel::Filters &filters );
 
    virtual int columnCount( const QModelIndex &parent ) const override;
    virtual Qt::ItemFlags flags( const QModelIndex &idx ) const override;
    QModelIndex index( int row, int column, const QModelIndex &parent = QModelIndex() ) const override;
    QModelIndex parent( const QModelIndex &child ) const override;
    QModelIndex sibling( int row, int column, const QModelIndex &idx ) const override;
    virtual QVariant data( const QModelIndex &index, int role ) const override;
    virtual bool setData( const QModelIndex &index, const QVariant &value, int role ) override;
 
  public slots:
    virtual void setFilterText( const QString &filterText = QString() );
 
  protected:
    virtual bool isLayerChecked( QgsMapLayer *layer ) const;
 
    virtual void setLayerChecked( QgsMapLayer *layer, bool checked );
 
    bool filterAcceptsRow( int sourceRow, const QModelIndex &sourceParent ) const override;
 
  private:
 
    virtual bool layerShown( QgsMapLayer *layer ) const;
 
    bool nodeShown( QgsLayerTreeNode *node ) const;
 
    void setLayerCheckedPrivate( QgsMapLayer *layer, bool checked );
 
    QgsLayerTreeModel *mLayerTreeModel = nullptr;
    QList<QgsMapLayer *> mCheckedLayers;
    QString mFilterText;
    QgsMapLayerProxyModel::Filters mFilters = QgsMapLayerProxyModel::All;
};
 
#endif // QGSLAYERTREEFILTERPROXYMODEL_H