some refactoring and fixes issue with rubberband and trace

qgis · Jul 29, 2020 · 932151f · 932151f
1 parent f3a8dd7
commit 932151f
Show file tree

Hide file tree

Showing 4 changed files with 134 additions and 156 deletions.
diff --git a/src/gui/qgsgeometryrubberband.cpp b/src/gui/qgsgeometryrubberband.cpp
@@ -75,6 +75,11 @@ QgsWkbTypes::GeometryType QgsGeometryRubberBand::geometryType() const
   return mGeometryType;
 }
 
+void QgsGeometryRubberBand::setGeometryType( const QgsWkbTypes::GeometryType &geometryType )
+{
+  mGeometryType = geometryType;
+}
+
 void QgsGeometryRubberBand::drawVertex( QPainter *p, double x, double y )
 {
   qreal s = ( mIconSize - 1 ) / 2.0;

diff --git a/src/gui/qgsgeometryrubberband.h b/src/gui/qgsgeometryrubberband.h
@@ -121,6 +121,9 @@ class GUI_EXPORT QgsGeometryRubberBand: public QgsMapCanvasItem
     //! Returns which geometry is handled by the rubber band, polygon or line
     QgsWkbTypes::GeometryType geometryType() const;
 
+    //! Sets which geometry is handled by the rubber band, polygon or line
+    void setGeometryType( const QgsWkbTypes::GeometryType &geometryType );
+
   private:
     std::unique_ptr<QgsAbstractGeometry> mGeometry = nullptr;
     QBrush mBrush;