qgsrasterattributetablewidget.h

Go to the documentation of this file.

/***************************************************************************
  qgsrasterattributetablewidget.h - QgsRasterAttributeTableWidget
 
 ---------------------
 begin                : 6.10.2022
 copyright            : (C) 2022 by Alessandro Pasotti
 email                : elpaso at itopen dot it
 ***************************************************************************
 *                                                                         *
 *   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 QGSRASTERATTRIBUTETABLEWIDGET_H
#define QGSRASTERATTRIBUTETABLEWIDGET_H
 
#include "qgis_gui.h"
#include "ui_qgsrasterattributetablewidgetbase.h"
#include "qgsrasterattributetablemodel.h"
#include "qgscolorrampimpl.h"
#include "qgspanelwidget.h"
#include "qgslocaleawarenumericlineeditdelegate.h"
 
#include <QWidget>
#include <QStyledItemDelegate>
 
 
#ifndef SIP_RUN
class QgsRasterLayer;
class QgsRasterAttributeTable;
class QgsMessageBar;
class QSortFilterProxyModel;
#endif
 
 
#ifndef SIP_RUN
class ColorDelegate : public QStyledItemDelegate
{
    Q_OBJECT
 
  public:
    ColorDelegate( QObject *parent = nullptr )
      : QStyledItemDelegate( parent ) {};
 
    // QAbstractItemDelegate interface
    QWidget *createEditor( QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
    void setEditorData( QWidget *editor, const QModelIndex &index ) const override;
    void setModelData( QWidget *editor, QAbstractItemModel *model, const QModelIndex &index ) const override;
};
 
class ColorAlphaDelegate : public ColorDelegate
{
    Q_OBJECT
 
  public:
    ColorAlphaDelegate( QObject *parent = nullptr )
      : ColorDelegate( parent ) {};
 
    // QAbstractItemDelegate interface
    QWidget *createEditor( QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
};
 
class ColorRampDelegate : public QStyledItemDelegate
{
    Q_OBJECT
 
  public:
    ColorRampDelegate( QObject *parent = nullptr )
      : QStyledItemDelegate( parent ) {};
 
    // QAbstractItemDelegate interface
    QWidget *createEditor( QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
    void setEditorData( QWidget *editor, const QModelIndex &index ) const override;
    void setModelData( QWidget *editor, QAbstractItemModel *model, const QModelIndex &index ) const override;
    void paint( QPainter *painter, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
 
  protected:
    QgsGradientColorRamp mRamp;
};
 
class ColorRampAlphaDelegate : public ColorRampDelegate
{
    Q_OBJECT
 
  public:
    ColorRampAlphaDelegate( QObject *parent = nullptr )
      : ColorRampDelegate( parent ) {};
 
    // QAbstractItemDelegate interface
    QWidget *createEditor( QWidget *parent, const QStyleOptionViewItem &option, const QModelIndex &index ) const override;
};
 
 
class LocalizedDoubleDelegate : public QgsLocaleAwareNumericLineEditDelegate
{
    Q_OBJECT
 
  public:
    LocalizedDoubleDelegate( QWidget *parent = nullptr )
      : QgsLocaleAwareNumericLineEditDelegate( Qgis::DataType::Float64, parent ) {};
 
    // QStyledItemDelegate interface
    QString displayText( const QVariant &value, const QLocale &locale ) const override;
};
 
 
#endif
 
class GUI_EXPORT QgsRasterAttributeTableWidget : public QgsPanelWidget, private Ui::QgsRasterAttributeTableWidgetBase
{
    Q_OBJECT
 
  public:
    explicit QgsRasterAttributeTableWidget( QWidget *parent SIP_TRANSFERTHIS = nullptr, QgsRasterLayer *rasterLayer = nullptr, const int bandNumber = 0 );
 
    void setRasterLayer( QgsRasterLayer *rasterLayer, const int bandNumber = 0 );
 
    bool isDirty() const;
 
    void setMessageBar( QgsMessageBar *bar );
 
  signals:
 
    void rendererChanged();
 
  public slots:
 
    void saveChanges();
 
    bool setEditable( bool editable, bool allowCancel = true );
 
  private:
    void classify();
    void addColumn();
    void removeColumn();
    void addRow();
    void removeRow();
    void bandChanged( const int index );
    void notify( const QString &title, const QString &message, Qgis::MessageLevel level = Qgis::MessageLevel::Info );
    void setDelegates();
 
    QgsRasterLayer *mRasterLayer = nullptr;
    std::unique_ptr<QgsRasterAttributeTable> mAttributeTableBuffer;
    // Default to invalid (bands are 1-indexed)
    int mCurrentBand = 0;
    bool mEditable = false;
 
    std::unique_ptr<QgsRasterAttributeTableModel> mModel;
 
    QAction *mActionToggleEditing = nullptr;
    QAction *mActionAddColumn = nullptr;
    QAction *mActionRemoveColumn = nullptr;
    QAction *mActionAddRow = nullptr;
    QAction *mActionRemoveRow = nullptr;
    QAction *mActionSaveChanges = nullptr;
 
    QgsMessageBar *mMessageBar = nullptr;
    QSortFilterProxyModel *mProxyModel = nullptr;
 
    void init( int bandNumber = 0 );
    void updateButtons();
 
    // QgsPanelWidget interface
  public:
    void setDockMode( bool dockMode ) override;
};
 
#endif // QGSRASTERATTRIBUTETABLEWIDGET_H