qgsrubberband.h

Go to the documentation of this file.

/***************************************************************************
    qgsrubberband.h - Rubberband widget for drawing multilines and polygons
     --------------------------------------
    Date                 : 07-Jan-2006
    Copyright            : (C) 2006 by Tom Elwertowski
    Email                : telwertowski at users dot sourceforge dot net
 ***************************************************************************
 *                                                                         *
 *   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 QGSRUBBERBAND_H
#define QGSRUBBERBAND_H

#include "qgsmapcanvasitem.h"
#include "qgis.h"
#include "qgsgeometry.h"

#include <QBrush>
#include <QList>
#include <QPen>
#include <QPolygon>
#include <QObject>

#include "qgis_gui.h"

class QgsVectorLayer;
class QPaintEvent;

#ifdef SIP_RUN
% ModuleHeaderCode
// For ConvertToSubClassCode.
#include <qgsrubberband.h>
% End
#endif

class GUI_EXPORT QgsRubberBand : public QObject, public QgsMapCanvasItem
{
    Q_OBJECT

#ifdef SIP_RUN
    SIP_CONVERT_TO_SUBCLASS_CODE
    if ( dynamic_cast<QgsRubberBand *>( sipCpp ) )
      sipType = sipType_QgsRubberBand;
    else
      sipType = nullptr;
    SIP_END
#endif
  public:

    Q_PROPERTY( QColor fillColor READ fillColor WRITE setFillColor )
    Q_PROPERTY( QColor strokeColor READ strokeColor WRITE setStrokeColor )
    Q_PROPERTY( int iconSize READ iconSize WRITE setIconSize )
    Q_PROPERTY( QColor secondaryStrokeColor READ secondaryStrokeColor WRITE setSecondaryStrokeColor )
    Q_PROPERTY( int width READ width WRITE setWidth )

    enum IconType
    {

      ICON_NONE,

      ICON_CROSS,

      ICON_X,

      ICON_BOX,

      ICON_CIRCLE,

      ICON_FULL_BOX,

      ICON_DIAMOND,

      ICON_FULL_DIAMOND,
    };

    QgsRubberBand( QgsMapCanvas *mapCanvas SIP_TRANSFERTHIS, QgsWkbTypes::GeometryType geometryType = QgsWkbTypes::LineGeometry );

    void setColor( const QColor &color );

    void setFillColor( const QColor &color );

    QColor fillColor() const { return mBrush.color(); }

    void setStrokeColor( const QColor &color );

    QColor strokeColor() const { return mPen.color(); }

    void setSecondaryStrokeColor( const QColor &color );

    QColor secondaryStrokeColor() const { return mSecondaryPen.color(); }

    void setWidth( int width );

    int width() const { return mPen.width(); }

    void setIcon( IconType icon );


    IconType icon() const { return mIconType; }

    void setIconSize( int iconSize );

    int iconSize() const { return mIconSize; }

    void setLineStyle( Qt::PenStyle penStyle );

    void setBrushStyle( Qt::BrushStyle brushStyle );

    void reset( QgsWkbTypes::GeometryType geometryType = QgsWkbTypes::LineGeometry );

    void addPoint( const QgsPointXY &p, bool doUpdate = true, int geometryIndex = 0 );

    void closePoints( bool doUpdate = true, int geometryIndex = 0 );

    void removePoint( int index = 0, bool doUpdate = true, int geometryIndex = 0 );

    void removeLastPoint( int geometryIndex = 0, bool doUpdate = true );

    void movePoint( const QgsPointXY &p, int geometryIndex = 0 );

    void movePoint( int index, const QgsPointXY &p, int geometryIndex = 0 );

    int partSize( int geometryIndex ) const;

    void setToGeometry( const QgsGeometry &geom, QgsVectorLayer *layer );

    void setToGeometry( const QgsGeometry &geometry, const QgsCoordinateReferenceSystem &crs = QgsCoordinateReferenceSystem() );

    void setToCanvasRectangle( QRect rect );

    void addGeometry( const QgsGeometry &geometry, QgsVectorLayer *layer );

    void addGeometry( const QgsGeometry &geometry, const QgsCoordinateReferenceSystem &crs = QgsCoordinateReferenceSystem() );

    void setTranslationOffset( double dx, double dy );

    int size() const;

    int numberOfVertices() const;

    const QgsPointXY *getPoint( int i, int j = 0 ) const;

    QgsGeometry asGeometry() const;

    void updatePosition() override;

  protected:

    void paint( QPainter *p ) override;

    void drawShape( QPainter *p, const QVector<QPointF> &pts );

    void updateRect();

  private:
    QBrush mBrush;
    QPen mPen;
    QPen mSecondaryPen;

    int mIconSize = 5;

    IconType mIconType = ICON_CIRCLE;

    QList< QList <QgsPointXY> > mPoints;
    QgsWkbTypes::GeometryType mGeometryType = QgsWkbTypes::PolygonGeometry;
    double mTranslationOffsetX = 0.0;
    double mTranslationOffsetY = 0.0;

    QgsRubberBand();

    static QgsPolylineXY getPolyline( const QList<QgsPointXY> &points );

};

#endif