qgselevationprofilelayertreeview.h

Go to the documentation of this file.

/***************************************************************************
                          qgselevationprofilelayertreeview.h
                          ---------------
    begin                : April 2022
    copyright            : (C) 2022 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 QGSELEVATIONPROFILELAYERTREEVIEW_H
#define QGSELEVATIONPROFILELAYERTREEVIEW_H
 
#include "qgsconfig.h"
 
#include "qgis_gui.h"
#include "qgslayertreemodel.h"
#include "qgslayertreeview.h"
 
#include <QSortFilterProxyModel>
#include <QTreeView>
 
#define SIP_NO_FILE
 
class QgsLayerTree;
class QgsElevationProfileLayerTreeModel;
class QgsElevationProfileLayerTreeProxyModel;
class QgsMapLayer;
 

class GUI_EXPORT QgsElevationProfileLayerTreeModel : public QgsLayerTreeModel
{
    Q_OBJECT
 
  public:
    explicit QgsElevationProfileLayerTreeModel( QgsLayerTree *rootNode, QObject *parent = nullptr );
 
    QVariant data( const QModelIndex &index, int role = Qt::DisplayRole ) const override;
    bool setData( const QModelIndex &index, const QVariant &value, int role = Qt::EditRole ) override;
    Qt::ItemFlags flags( const QModelIndex &index ) const override;
    bool canDropMimeData( const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent ) const override;
    bool dropMimeData( const QMimeData *data, Qt::DropAction action, int row, int column, const QModelIndex &parent ) override;
    QMimeData *mimeData( const QModelIndexList &indexes ) const override;

    void setAllowModifications( bool allow );

    bool allowModifications() const { return mAllowModifications; }
 
  signals:

    void addLayers( const QList<QgsMapLayer *> &layers );
 
  private:
#ifdef SIP_RUN
    QgsElevationProfileLayerTreeModel( const QgsElevationProfileLayerTreeModel &other );
#endif
 
    bool mAllowModifications = true;
};

class GUI_EXPORT QgsElevationProfileLayerTreeProxyModel : public QSortFilterProxyModel
{
    Q_OBJECT
 
  public:
    explicit QgsElevationProfileLayerTreeProxyModel( QgsElevationProfileLayerTreeModel *model, QObject *parent = nullptr );
 
  protected:
    bool filterAcceptsRow( int sourceRow, const QModelIndex &sourceParent ) const override;
 
  private:
    QgsElevationProfileLayerTreeModel *mModel = nullptr;
};

class GUI_EXPORT QgsElevationProfileLayerTreeView : public QgsLayerTreeViewBase
{
    Q_OBJECT
 
  public:
    static const QString CUSTOM_NODE_ELEVATION_PROFILE_SOURCE;

    explicit QgsElevationProfileLayerTreeView( QgsLayerTree *rootNode, QWidget *parent = nullptr );

    void setLayerTree( QgsLayerTree *rootNode );

    void populateInitialSources( QgsProject *project );

    QgsElevationProfileLayerTreeProxyModel *proxyModel();
 
  public slots:

    void addNodeForRegisteredSource( const QString &sourceId, const QString &sourceName );

    void removeNodeForUnregisteredSource( const QString &sourceId );

    void populateMissingLayers( QgsProject *project );
 
  signals:

    void addLayers( const QList<QgsMapLayer *> &layers );
 
  protected:
    void resizeEvent( QResizeEvent *event ) override;
 
  private:
    QgsElevationProfileLayerTreeModel *mModel = nullptr;
    QgsElevationProfileLayerTreeProxyModel *mProxyModel = nullptr;
    QgsLayerTree *mLayerTree = nullptr;
};
 
 
#endif // QGSELEVATIONPROFILELAYERTREEVIEW_H