qgsprovidersublayermodel.h

Go to the documentation of this file.

/***************************************************************************
                             qgsprovidersublayermodel.h
                             ----------------------
    begin                : June 2021
    copyright            : (C) 2021 by Nyall Dawson
    email                : nyall dot dawson at gmail dot com
 ***************************************************************************/
 
/***************************************************************************
 *                                                                         *
 *   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 QGSPROVIDERSUBLAYERMODEL_H
#define QGSPROVIDERSUBLAYERMODEL_H
 
#include "qgstaskmanager.h"
#include <QAbstractItemModel>
#include <QSortFilterProxyModel>
 
class QgsProviderSublayerDetails;
 
class CORE_EXPORT QgsProviderSublayerModel: public QAbstractItemModel
{
    Q_OBJECT
 
  public:
 
    enum class Role : int
    {
      ProviderKey = Qt::UserRole + 1, 
      LayerType, 
      Uri, 
      Name, 
      Description, 
      Path, 
      FeatureCount, 
      WkbType, 
      GeometryColumnName, 
      LayerNumber, 
      IsNonLayerItem, 
      NonLayerItemType, 
      Flags, 
    };
 
    enum class Column : int
    {
      Name = 0, 
      Description = 1, 
    };
 
    class CORE_EXPORT NonLayerItem
    {
      public:
 
        QString type() const;
 
        void setType( const QString &type );
 
        QString name() const;
 
        void setName( const QString &name );
 
        QString description() const;
 
        void setDescription( const QString &description );
 
        QString uri() const;
 
        void setUri( const QString &uri );
 
        QIcon icon() const;
 
        void setIcon( const QIcon &icon );
 
        bool operator==( const QgsProviderSublayerModel::NonLayerItem &other ) const;
        bool operator!=( const QgsProviderSublayerModel::NonLayerItem &other ) const;
 
#ifdef SIP_RUN
        SIP_PYOBJECT __repr__();
        % MethodCode
        QString str = QStringLiteral( "<QgsProviderSublayerModel.NonLayerItem: %1 - %2>" ).arg( sipCpp->type(), sipCpp->name() );
        sipRes = PyUnicode_FromString( str.toUtf8().constData() );
        % End
#endif
 
      private:
 
        QString mType;
        QString mName;
        QString mDescription;
        QString mUri;
        QIcon mIcon;
 
    };
 
    QgsProviderSublayerModel( QObject *parent SIP_TRANSFERTHIS = nullptr );
 
    void setSublayerDetails( const QList< QgsProviderSublayerDetails > &details );
 
    QList< QgsProviderSublayerDetails > sublayerDetails() const;
 
    QgsProviderSublayerDetails indexToSublayer( const QModelIndex &index ) const;
 
    QgsProviderSublayerModel::NonLayerItem indexToNonLayerItem( const QModelIndex &index ) const;
 
    void addNonLayerItem( const QgsProviderSublayerModel::NonLayerItem &item );
 
    QModelIndex index( int row, int column, const QModelIndex &parent = QModelIndex() ) const override;
    QModelIndex parent( const QModelIndex &index ) const override;
    int columnCount( const QModelIndex &parent = QModelIndex() ) const override;
    int rowCount( const QModelIndex &parent ) const override;
    QVariant data( const QModelIndex &index, int role ) const override;
    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const override;
 
  protected:
 
    QList<QgsProviderSublayerDetails> mSublayers;
 
    QList<NonLayerItem> mNonLayerItems;
 
};
 
 
class CORE_EXPORT QgsProviderSublayerProxyModel: public QSortFilterProxyModel
{
    Q_OBJECT
 
  public:
 
    QgsProviderSublayerProxyModel( QObject *parent SIP_TRANSFERTHIS = nullptr );
 
    QString filterString() const;
 
    void setFilterString( const QString &filter );
 
    bool includeSystemTables() const;
 
    void setIncludeSystemTables( bool include );
 
  protected:
    bool filterAcceptsRow( int source_row, const QModelIndex &source_parent ) const override;
    bool lessThan( const QModelIndex &source_left, const QModelIndex &source_right ) const override;
 
  private:
 
    QString mFilterString;
    bool mIncludeSystemTables = false;
 
};
 
 
#endif // QGSPROVIDERSUBLAYERMODEL_H