qgssymbolbutton.h

Go to the documentation of this file.

/***************************************************************************
     qgssymbolbutton.h
     -----------------
    Date                 : July 2017
    Copyright            : (C) 2017 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 QGSSYMBOLBUTTON_H
#define QGSSYMBOLBUTTON_H
 
#include "qgis_gui.h"
#include "qgis_sip.h"
#include "qgis.h"
 
#include <QToolButton>
#include <QPointer>
#include <memory>
 
class QgsMapCanvas;
class QgsVectorLayer;
class QgsExpressionContextGenerator;
class QgsPanelWidget;
class QgsMessageBar;
class QMimeData;
class QgsSymbol;
 
class GUI_EXPORT QgsSymbolButton : public QToolButton
{
    Q_OBJECT
 
    Q_PROPERTY( QString dialogTitle READ dialogTitle WRITE setDialogTitle )
 
  public:
 
    QgsSymbolButton( QWidget *parent SIP_TRANSFERTHIS = nullptr, const QString &dialogTitle = QString() );
    ~QgsSymbolButton();
 
    QSize minimumSizeHint() const override;
    QSize sizeHint() const override;
 
    void setSymbolType( Qgis::SymbolType type );
 
    Qgis::SymbolType symbolType() const { return mType; }
 
    void setDialogTitle( const QString &title );
 
    QString dialogTitle() const;
 
    QgsSymbol *symbol();
 
    template <class SymbolType> SymbolType *clonedSymbol() SIP_SKIP
    {
      QgsSymbol *tmpSymbol = mSymbol.get();
      SymbolType *symbolCastToType = dynamic_cast<SymbolType *>( tmpSymbol );
 
      if ( symbolCastToType )
      {
        return symbolCastToType->clone();
      }
      else
      {
        //could not cast
        return nullptr;
      }
    }
 
    QgsMapCanvas *mapCanvas() const;
 
    void setMapCanvas( QgsMapCanvas *canvas );
 
    void setMessageBar( QgsMessageBar *bar );
 
    QgsMessageBar *messageBar() const;
 
    QgsVectorLayer *layer() const;
 
    void setLayer( QgsVectorLayer *layer );
 
    void registerExpressionContextGenerator( QgsExpressionContextGenerator *generator );
 
  public slots:
 
    void setSymbol( QgsSymbol *symbol SIP_TRANSFER );
 
    void setColor( const QColor &color );
 
    void copySymbol();
 
    void pasteSymbol();
 
    void copyColor();
 
    void pasteColor();
 
  signals:
 
    void changed();
 
  protected:
 
    void changeEvent( QEvent *e ) override;
    void showEvent( QShowEvent *e ) override;
    void resizeEvent( QResizeEvent *event ) override;
 
    // Reimplemented to detect right mouse button clicks on the color button and allow dragging colors
    void mousePressEvent( QMouseEvent *e ) override;
    // Reimplemented to allow dragging colors/symbols from button
    void mouseMoveEvent( QMouseEvent *e ) override;
    void mouseReleaseEvent( QMouseEvent *e ) override;
    void keyPressEvent( QKeyEvent *e ) override;
    // Reimplemented to accept dragged colors
    void dragEnterEvent( QDragEnterEvent *e ) override;
 
    // Reimplemented to reset button appearance after drag leave
    void dragLeaveEvent( QDragLeaveEvent *e ) override;
 
    // Reimplemented to accept dropped colors
    void dropEvent( QDropEvent *e ) override;
 
  private slots:
 
    void showSettingsDialog();
    void updateSymbolFromWidget();
    void cleanUpSymbolSelector( QgsPanelWidget *container );
 
    void prepareMenu();
 
    void addRecentColor( const QColor &color );
 
    void activatePicker();
 
  private:
 
    QSize mSizeHint;
 
    QString mDialogTitle;
 
    Qgis::SymbolType mType = Qgis::SymbolType::Fill;
 
    QgsMapCanvas *mMapCanvas = nullptr;
    QgsMessageBar *mMessageBar = nullptr;
 
    QPoint mDragStartPosition;
 
    QMenu *mMenu = nullptr;
 
    QPointer< QgsVectorLayer > mLayer;
 
    QSize mIconSize;
 
    std::unique_ptr< QgsSymbol > mSymbol;
 
    QgsExpressionContextGenerator *mExpressionContextGenerator = nullptr;
 
    bool mPickingColor = false;
 
    void updatePreview( const QColor &color = QColor(), QgsSymbol *tempSymbol = nullptr );
 
    bool colorFromMimeData( const QMimeData *mimeData, QColor &resultColor, bool &hasAlpha );
 
    QPixmap createColorIcon( const QColor &color ) const;
 
    void stopPicking( QPoint eventPos, bool samplingColor = true );
 
    void showColorDialog();
 
};
 
#endif // QGSSYMBOLBUTTON_H