qgsmeshdatasetgrouptreeview.h

Go to the documentation of this file.

/***************************************************************************
    qgsmeshdatasetgrouptreeview.h
    -----------------------------
    begin                : June 2018
    copyright            : (C) 2018 by Peter Petrik
    email                : zilolv 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 QGSMESHDATASETGROUPTREE_H
#define QGSMESHDATASETGROUPTREE_H
 
#include <memory>
 
#include "qgis_gui.h"
#include "qgsmeshdataset.h"
 
#include <QItemSelection>
#include <QList>
#include <QMap>
#include <QMenu>
#include <QObject>
#include <QSortFilterProxyModel>
#include <QStandardItemModel>
#include <QStyledItemDelegate>
#include <QTreeView>
#include <QVector>
 
SIP_NO_FILE
 
class QgsMeshLayer;

class QgsMeshDatasetGroupSaveMenu : public QObject
{
    Q_OBJECT
  public:
    QgsMeshDatasetGroupSaveMenu( QObject *parent = nullptr )
      : QObject( parent ) {}
    QMenu *createSaveMenu( int groupIndex, QMenu *parentMenu = nullptr );
 
    void setMeshLayer( QgsMeshLayer *meshLayer );
 
  signals:
    void datasetGroupSaved( const QString &uri );
 
  private:
    QgsMeshLayer *mMeshLayer = nullptr;
 
    void saveDatasetGroup( int datasetGroup, const QString &driver, const QString &fileSuffix );
};

class QgsMeshDatasetGroupTreeModel : public QAbstractItemModel
{
    Q_OBJECT
  public:
    enum Roles
    {
      Name = Qt::UserRole,
      IsVector,
      IsActiveScalarDatasetGroup,
      IsActiveVectorDatasetGroup,
      DatasetGroupIndex,
    };
 
    explicit QgsMeshDatasetGroupTreeModel( QObject *parent = nullptr );
 
    QVariant data( const QModelIndex &index, int role ) const override;
    Qt::ItemFlags flags( const QModelIndex &index ) const override;
    QVariant headerData( int section, Qt::Orientation orientation, int role = Qt::DisplayRole ) const override;
    QModelIndex index( int row, int column, const QModelIndex &parent = QModelIndex() ) const override;
    QModelIndex parent( const QModelIndex &index ) const override;
    int rowCount( const QModelIndex &parent = QModelIndex() ) const override;
    int columnCount( const QModelIndex &parent = QModelIndex() ) const override;

    void syncToLayer( QgsMeshLayer *layer );

    QgsMeshDatasetGroupTreeItem *datasetGroupTreeRootItem();

    QgsMeshDatasetGroupTreeItem *datasetGroupTreeItem( int groupIndex );

    QgsMeshDatasetGroupTreeItem *datasetGroupTreeItem( QModelIndex index );

    bool isEnabled( const QModelIndex &index ) const;

    void resetDefault( QgsMeshLayer *meshLayer );

    void setAllGroupsAsEnabled( bool isEnabled );

    void removeItem( const QModelIndex &index );

    void setPersistentDatasetGroup( const QModelIndex &index, const QString &uri );
 
  private:
    std::unique_ptr<QgsMeshDatasetGroupTreeItem> mRootItem;
};

class QgsMeshAvailableDatasetGroupTreeModel : public QgsMeshDatasetGroupTreeModel
{
    Q_OBJECT
  public:
    QgsMeshAvailableDatasetGroupTreeModel( QObject *parent = nullptr );
 
    QVariant data( const QModelIndex &index, int role ) const override;
    bool setData( const QModelIndex &index, const QVariant &value, int role ) override;
    Qt::ItemFlags flags( const QModelIndex &index ) const override;
    QVariant headerData( int section, Qt::Orientation orientation, int role ) const override;
    int columnCount( const QModelIndex &parent = QModelIndex() ) const override;
 
  private:
    QString textDisplayed( const QModelIndex &index ) const;
    QColor backGroundColor( const QModelIndex &index ) const;
};

class QgsMeshDatasetGroupProxyModel : public QSortFilterProxyModel
{
    Q_OBJECT
  public:
    QgsMeshDatasetGroupProxyModel( QAbstractItemModel *sourceModel );
 
    Qt::ItemFlags flags( const QModelIndex &index ) const override;
    QVariant data( const QModelIndex &index, int role ) const override;

    int activeScalarGroup() const;

    void setActiveScalarGroup( int group );

    int activeVectorGroup() const;

    void setActiveVectorGroup( int group );

    void syncToLayer( QgsMeshLayer *layer );
 
  protected:
    bool filterAcceptsRow( int source_row, const QModelIndex &source_parent ) const override;
 
  private:
    int mActiveScalarGroupIndex = -1;
    int mActiveVectorGroupIndex = -1;
};

class QgsMeshDatasetGroupTreeItemDelegate : public QStyledItemDelegate
{
    Q_OBJECT
  public:
    QgsMeshDatasetGroupTreeItemDelegate( QObject *parent = nullptr );
 
    void paint( QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;

    QRect iconRect( const QRect &rect, bool isVector ) const;
 
    QSize sizeHint( const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
 
  private:
    const QPixmap mScalarSelectedPixmap;
    const QPixmap mScalarDeselectedPixmap;
    const QPixmap mVectorSelectedPixmap;
    const QPixmap mVectorDeselectedPixmap;
 
    QRect iconRect( const QRect &rect, int pos ) const;
};

class GUI_EXPORT QgsMeshDatasetGroupTreeView : public QTreeView
{
    Q_OBJECT
  public:
    QgsMeshDatasetGroupTreeView( QWidget *parent = nullptr );
 
    void syncToLayer( QgsMeshLayer *layer );
    void resetDefault( QgsMeshLayer *meshLayer );
 
    QgsMeshDatasetGroupTreeItem *datasetGroupTreeRootItem();
 
  signals:
    void apply();
 
  public slots:
    void selectAllGroups();
    void deselectAllGroups();
 
  protected:
    void contextMenuEvent( QContextMenuEvent *event ) override;
 
  private slots:
    void removeCurrentItem();
    void onDatasetGroupSaved( const QString &uri );
 
  private:
    QgsMeshAvailableDatasetGroupTreeModel *mModel;
    QgsMeshDatasetGroupSaveMenu *mSaveMenu;
 
    void selectAllItem( bool isChecked );
    QMenu *createContextMenu();
};

class GUI_EXPORT QgsMeshActiveDatasetGroupTreeView : public QTreeView
{
    Q_OBJECT
 
  public:
    QgsMeshActiveDatasetGroupTreeView( QWidget *parent = nullptr );

    void setLayer( QgsMeshLayer *layer );

    int activeScalarGroup() const;

    int activeVectorGroup() const;

    void syncToLayer();
 
    void mousePressEvent( QMouseEvent *event ) override;
 
  public slots:
    void setActiveScalarGroup( int group );

    void setActiveVectorGroup( int group );
 
  signals:
    void activeScalarGroupChanged( int groupIndex );

    void activeVectorGroupChanged( int groupIndex );
 
  private:
    void setActiveGroup();
 
    QgsMeshDatasetGroupProxyModel *mProxyModel;
    QgsMeshDatasetGroupTreeItemDelegate mDelegate;
    QgsMeshLayer *mMeshLayer = nullptr; // not owned
};

class GUI_EXPORT QgsMeshDatasetGroupListModel : public QAbstractListModel
{
    Q_OBJECT
  public:
    explicit QgsMeshDatasetGroupListModel( QObject *parent );

    void syncToLayer( QgsMeshLayer *layer );
 
    int rowCount( const QModelIndex &parent ) const override;
    QVariant data( const QModelIndex &index, int role ) const override;
 
    void setDisplayProviderName( bool displayProviderName );
 
    QStringList variableNames() const;
 
  private:
    QgsMeshDatasetGroupTreeItem *mRootItem = nullptr;
    bool mDisplayProviderName = false;
};
 
#endif // QGSMESHDATASETGROUPTREE_H