Skip to content

/ QGIS

Permalink
Browse files
Add rubberband on mouse press, code reorganisation and cleanup
  • Loading branch information
@leyan
leyan committed May 8, 2014
1 parent 6b0796e commit 609b692f7f6801849dcbf25be013b3a91a86b8c4
Unified Split
Showing with 199 additions and 224 deletions.
  1. +107 −153 src/app/qgsmaptooldeletepart.cpp
  2. +8 −9 src/app/qgsmaptooldeletepart.h
  3. +72 −60 src/app/qgsmaptooldeletering.cpp
  4. +12 −2 src/app/qgsmaptooldeletering.h
260 src/app/qgsmaptooldeletepart.cpp
View file
@@ -15,23 +15,23 @@

#include "qgsmaptooldeletepart.h"

#include "qgisapp.h"
#include "qgsmapcanvas.h"
#include "qgsvertexmarker.h"
#include "qgsvectorlayer.h"
#include "qgsmessagebar.h"
#include "qgsgeometry.h"
#include "qgstolerance.h"

#include <QMouseEvent>
#include <QMessageBox>

QgsMapToolDeletePart::QgsMapToolDeletePart( QgsMapCanvas* canvas )
: QgsMapToolEdit( canvas )
: QgsMapToolEdit( canvas ), mRubberBand( 0 )
{
}

QgsMapToolDeletePart::~QgsMapToolDeletePart()
{
delete mRubberBand;
}

void QgsMapToolDeletePart::canvasMoveEvent( QMouseEvent *e )
@@ -42,18 +42,10 @@ void QgsMapToolDeletePart::canvasMoveEvent( QMouseEvent *e )

void QgsMapToolDeletePart::canvasPressEvent( QMouseEvent *e )
{

mRecentSnappingResults.clear();
//do snap -> new recent snapping results
if ( mSnapper.snapToCurrentLayer( e->pos(), mRecentSnappingResults, QgsSnapper::SnapToVertexAndSegment ) != 0 )
{
//error
}
}

void QgsMapToolDeletePart::canvasReleaseEvent( QMouseEvent *e )
{
Q_UNUSED( e );
mPressedFid = -1;
mPressedPartNum = -1;
delete mRubberBand;
mRubberBand = 0;

QgsMapLayer* currentLayer = mCanvas->currentLayer();
if ( !currentLayer )
@@ -71,62 +63,48 @@ void QgsMapToolDeletePart::canvasReleaseEvent( QMouseEvent *e )
notifyNotEditableLayer();
return;
}
QgsGeometry* g;
QgsFeature f;
int partNum;
switch( vlayer->geometryType() )

QgsGeometry* geomPart;

geomPart = partUnderPoint( e->pos(), mPressedFid, mPressedPartNum );

if ( mPressedFid != -1 )
{
case QGis::Point:
case QGis::Line:
{
if ( mRecentSnappingResults.size() == 0 )
{
return;
}
QgsSnappingResult sr = mRecentSnappingResults.first();
vlayer->getFeatures( QgsFeatureRequest().setFilterFid( sr.snappedAtGeometry ) ).nextFeature( f );
g = f.geometry();
if ( !g )
return;
if ( g->wkbType() == QGis::WKBPoint25D || g->wkbType() == QGis::WKBPoint ||
g->wkbType() == QGis::WKBLineString25D || g->wkbType() == QGis::WKBLineString)
{
emit messageEmitted( tr( "The Delete part tool cannot be used on single part features." ) );
return;
}
int vertex = sr.snappedVertexNr;
if ( vertex == -1 )
{
vertex = sr.beforeVertexNr;
}
partNum = partNumberOfVertex( g, vertex );
break;
}
case QGis::Polygon:
{
QgsPoint p = mCanvas->getCoordinateTransform()->toMapCoordinates( e->x(),e->y());
p = toLayerCoordinates(vlayer, p);
f = featureUnderPoint(p);
g = f.geometry();
if ( !g )
return;
mRubberBand = createRubberBand( vlayer->geometryType() );

if ( g->wkbType() == QGis::WKBPolygon25D || g->wkbType() == QGis::WKBPolygon)
{
emit messageEmitted( tr( "The Delete part tool cannot be used on single part features." ) );
return;
}
partNum = partNumberOfPoint( g, p );
if ( partNum < 0 )
return;
break;
}
default:
{
QgsDebugMsg("Unknown geometry type");
return;
}
mRubberBand->setToGeometry( geomPart, vlayer );
mRubberBand->setColor( QColor( 255, 0, 0, 65 ) );
mRubberBand->setWidth( 2 );
mRubberBand->show();
}

}

void QgsMapToolDeletePart::canvasReleaseEvent( QMouseEvent *e )
{

delete mRubberBand;
mRubberBand = 0;

if ( !vlayer || !vlayer->isEditable() )
{
return;
}

if ( mPressedFid == -1 )
return;

int fid;
int partNum;
partUnderPoint( e->pos(), fid, partNum );

if ( fid != mPressedFid || partNum != mPressedPartNum )
return;

QgsFeature f;
vlayer->getFeatures( QgsFeatureRequest().setFilterFid( fid ) ).nextFeature( f );
QgsGeometry* g = f.geometry();

if ( g->deletePart( partNum ) )
{
vlayer->beginEditCommand( tr( "Part of multipart feature deleted" ) );
@@ -141,113 +119,89 @@ void QgsMapToolDeletePart::canvasReleaseEvent( QMouseEvent *e )
return;
}

QgsFeature QgsMapToolDeletePart::featureUnderPoint(QgsPoint p)
QgsGeometry* QgsMapToolDeletePart::partUnderPoint( QPoint point, int& fid, int& partNum )
{
QgsRectangle r;
double searchRadius = mCanvas->extent().width()/100;
r.setXMinimum( p.x() - searchRadius );
r.setXMaximum( p.x() + searchRadius );
r.setYMinimum( p.y() - searchRadius );
r.setYMaximum( p.y() + searchRadius );
QgsFeatureIterator fit = vlayer->getFeatures( QgsFeatureRequest().setFilterRect( r ) );
QgsFeature f;
fit.nextFeature( f );
return f;
}


int QgsMapToolDeletePart::partNumberOfPoint(QgsGeometry *g, QgsPoint point)
{
int part;
switch ( g->wkbType() )
{
case QGis::WKBMultiPolygon25D:
case QGis::WKBMultiPolygon:
{
QgsMultiPolygon mpolygon = g->asMultiPolygon();
for ( part = 0; part < mpolygon.count(); part++ ) // go through the polygons
{
const QgsPolygon& polygon = mpolygon[part];
QgsGeometry* partGeo = QgsGeometry::fromPolygon(polygon);
if ( partGeo->contains( &point ) )
return part;
}
return -1; // not found
}
default:
return -1;
}
}

int QgsMapToolDeletePart::partNumberOfVertex( QgsGeometry* g, int beforeVertexNr )
{
int part;
QgsGeometry* geomPart = new QgsGeometry();

switch ( g->wkbType() )
switch ( vlayer->geometryType() )
{
case QGis::WKBLineString25D:
case QGis::WKBLineString:
case QGis::WKBPoint25D:
case QGis::WKBPoint:
case QGis::WKBPolygon25D:
case QGis::WKBPolygon:
return 1;

case QGis::WKBMultiPoint25D:
case QGis::WKBMultiPoint:
if ( beforeVertexNr < g->asMultiPoint().count() )
return beforeVertexNr;
else
return -1;

case QGis::WKBMultiLineString25D:
case QGis::WKBMultiLineString:
case QGis::Point:
case QGis::Line:
{
QgsMultiPolyline mline = g->asMultiPolyline();
for ( part = 0; part < mline.count(); part++ )
if ( mSnapper.snapToCurrentLayer( point, mRecentSnappingResults, QgsSnapper::SnapToVertexAndSegment ) == 0 )
{
if ( beforeVertexNr < mline[part].count() )
return part;

beforeVertexNr -= mline[part].count();
if ( mRecentSnappingResults.length() > 0 )
{
QgsSnappingResult sr = mRecentSnappingResults.first();
int snapVertex = sr.snappedVertexNr;
if ( snapVertex == -1 )
snapVertex = sr.beforeVertexNr;
vlayer->getFeatures( QgsFeatureRequest().setFilterFid( sr.snappedAtGeometry ) ).nextFeature( f );
QgsGeometry* g = f.geometry();
if ( !g->isMultipart() )
return geomPart;
if ( g->wkbType() == QGis::WKBMultiPoint || g->wkbType() == QGis::WKBMultiPoint25D )
{
fid = sr.snappedAtGeometry;
partNum = snapVertex;
return QgsGeometry::fromPoint( sr.snappedVertex );
}
if ( g->wkbType() == QGis::WKBMultiLineString || g->wkbType() == QGis::WKBMultiLineString25D )
{
QgsMultiPolyline mline = g->asMultiPolyline();
for ( int part = 0; part < mline.count(); part++ )
{
if ( snapVertex < mline[part].count() )
{
fid = sr.snappedAtGeometry;
partNum = part;
return QgsGeometry::fromPolyline( mline[part] );
}
snapVertex -= mline[part].count();
}
}
}
}
return -1; // not found
break;
}

case QGis::WKBMultiPolygon25D:
case QGis::WKBMultiPolygon:
case QGis::Polygon:
{
QgsPoint layerCoords = toLayerCoordinates( vlayer, point );
double searchRadius = QgsTolerance::vertexSearchRadius( mCanvas->currentLayer(), mCanvas->mapSettings() );
QgsRectangle selectRect( layerCoords.x() - searchRadius, layerCoords.y() - searchRadius,
layerCoords.x() + searchRadius, layerCoords.y() + searchRadius );
QgsFeatureIterator fit = vlayer->getFeatures( QgsFeatureRequest().setFilterRect( selectRect ) );
fit.nextFeature( f );
QgsGeometry* g = f.geometry();
if ( !g )
return geomPart;
if ( !g->isMultipart() )
return geomPart;
QgsMultiPolygon mpolygon = g->asMultiPolygon();
for ( part = 0; part < mpolygon.count(); part++ ) // go through the polygons
for ( int part = 0; part < mpolygon.count(); part++ ) // go through the polygons
{
const QgsPolygon& polygon = mpolygon[part];
for ( int ring = 0; ring < polygon.count(); ring++ ) // go through the rings
QgsGeometry* partGeo = QgsGeometry::fromPolygon( polygon );
if ( partGeo->contains( &layerCoords ) )
{
if ( beforeVertexNr < polygon[ring].count() )
return part;

beforeVertexNr -= polygon[ring].count();
fid = f.id();
partNum = part;
return partGeo;
}
}
return -1; // not found
break;
}

default:
return -1;
{
break;
}
}
return geomPart;
}

void QgsMapToolDeletePart::deactivate()
{
QgsMapTool::deactivate();
}

void QgsMapToolDeletePart::notifySinglePart()
{
QgisApp::instance()->messageBar()->pushMessage(
tr( "Cannot use delete part" ),
tr( "The Delete part tool cannot be used on single part features." ),
QgsMessageBar::INFO,
QgisApp::instance()->messageTimeout() );
return;
}
17 src/app/qgsmaptooldeletepart.h
View file
@@ -17,6 +17,7 @@
#define QGSMAPTOOLDELETEPART_H

#include "qgsmaptooledit.h"
#include "qgsrubberband.h"

class QgsVertexMarker;

@@ -42,17 +43,15 @@ class APP_EXPORT QgsMapToolDeletePart: public QgsMapToolEdit
QgsVectorLayer* vlayer;
QList<QgsSnappingResult> mRecentSnappingResults;

//! find out the part number of geometry given the vertex number
int partNumberOfVertex( QgsGeometry* g, int beforeVertexNr );
QgsGeometry* partUnderPoint( QPoint p, int &fid, int &partNum );

//! find out the part number of geometry including the point
int partNumberOfPoint( QgsGeometry* g, QgsPoint point );

//! find which feature is under the point position (different from snapping as we allow the whole polygon surface)
QgsFeature featureUnderPoint(QgsPoint p);

void notifySinglePart();
/* Rubberband that shows the part being deleted*/
QgsRubberBand* mRubberBand;

//The feature and part where the mouse cursor was pressed
//This is used to check whether we are still in the same part at cursor release
int mPressedFid;
int mPressedPartNum;
};

#endif

0 comments on commit 609b692

Please sign in to comment.