qgsdatadefinedsizelegendwidget.h

Go to the documentation of this file.

/***************************************************************************
  qgsdatadefinedsizelegendwidget.h
  --------------------------------------
  Date                 : June 2017
  Copyright            : (C) 2017 by Martin Dobias
  Email                : wonder dot sk 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 QGSDATADEFINEDSIZELEGENDWIDGET_H
#define QGSDATADEFINEDSIZELEGENDWIDGET_H
 
#include "ui_qgsdatadefinedsizelegendwidget.h"
 
#include <memory>
 
#include "qgis_gui.h"
#include "qgis_sip.h"
#include "qgsdoublevalidator.h"
#include "qgspanelwidget.h"
#include "qgsproperty.h"
 
#include <QStyledItemDelegate>
 
class QStandardItemModel;
 
class QgsDataDefinedSizeLegend;
class QgsLayerTree;
class QgsLayerTreeLayer;
class QgsLayerTreeModel;
class QgsMapCanvas;
class QgsMarkerSymbol;
class QgsProperty;
class QgsVectorLayer;

class GUI_EXPORT QgsDataDefinedSizeLegendWidget : public QgsPanelWidget, private Ui::QgsDataDefinedSizeLegendWidget
{
    Q_OBJECT
  public:
    explicit QgsDataDefinedSizeLegendWidget(
      const QgsDataDefinedSizeLegend *ddsLegend, const QgsProperty &ddSize, QgsMarkerSymbol *overrideSymbol SIP_TRANSFER, QgsMapCanvas *canvas = nullptr, QWidget *parent SIP_TRANSFERTHIS = nullptr
    );
    ~QgsDataDefinedSizeLegendWidget() override;

    QgsDataDefinedSizeLegend *dataDefinedSizeLegend() const SIP_FACTORY;
 
  signals:
 
  private slots:
    void updatePreview();
    void changeSymbol();
    void addSizeClass();
    void removeSizeClass();
    void onSizeClassesChanged();
 
  private:
    std::unique_ptr<QgsMarkerSymbol> mSourceSymbol; 
    bool mOverrideSymbol = false;                   
    QgsProperty mSizeProperty;                      
    QgsLayerTreeModel *mPreviewModel = nullptr;
    std::unique_ptr<QgsLayerTree> mPreviewTree;
    QgsLayerTreeLayer *mPreviewLayerNode = nullptr;
    std::unique_ptr<QgsVectorLayer> mPreviewLayer;
    QgsMapCanvas *mMapCanvas = nullptr;
    QStandardItemModel *mSizeClassesModel = nullptr;
};
 
#ifndef SIP_RUN

class SizeClassDelegate : public QStyledItemDelegate
{
    Q_OBJECT
 
  public:
    SizeClassDelegate( QObject *parent )
      : QStyledItemDelegate( parent )
    {}
 
    QWidget *createEditor( QWidget *parent, const QStyleOptionViewItem &, const QModelIndex & ) const override
    {
      QLineEdit *lineEdit = new QLineEdit( parent );
      QgsDoubleValidator *validator = new QgsDoubleValidator( 0, 1e6, 1, lineEdit );
      lineEdit->setValidator( validator );
      return lineEdit;
    }
};

#endif
 
#endif // QGSDATADEFINEDSIZELEGENDWIDGET_H