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 "qgis_gui.h"
 
#include <QObject>
#include <QTreeView>
#include <QMap>
#include <QMenu>
#include <QVector>
#include <QItemSelection>
#include <QStandardItemModel>
#include <QStyledItemDelegate>
#include <QList>
#include <QSortFilterProxyModel>
#include <memory>
#include "qgsmeshdataset.h"
 
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
{
  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
{
  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 QgsMeshDatasetGroupTreeItemDelagate: public QStyledItemDelegate
{
    Q_OBJECT
  public:
    QgsMeshDatasetGroupTreeItemDelagate( 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;
    QgsMeshDatasetGroupTreeItemDelagate mDelegate;
    QgsMeshLayer *mMeshLayer = nullptr; // not owned
};
 
class GUI_EXPORT QgsMeshDatasetGroupListModel: public QAbstractListModel
{
  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