def _validateGeometry(self):
geometryType = self.geometryType()
if (geometryType == QGis.Point or geometryType == QGis.UnknownGeometry
or geometryType == QGis.NoGeometry
or len(self._mapPointList) < 2):
return
settings = QSettings()
if (settings.value('/qgis/digitizing/validate_geometries', 1) == 0):
return
self._deleteGeometryValidation()
geometry = None # QgsGeometry()
if (geometryType == QGis.Line):
geometry = QgsGeometry.fromPolyline(self._mapPointList)
elif (geometryType == QGis.Polygon):
if (len(self._mapPointList) < 3):
return
closed = list(self._mapPointList)
closed.append(self._mapPointList[0])
geometry = QgsGeometry.fromPolygon([closed])
if (geometry is None):
return
self._validator = QgsGeometryValidator(geometry)
self._validator.errorFound.connect(self._addGeometryError)
self._validator.finished.connect(self.validationFinished)
self._validator.start()
self._iface.mainWindow().statusBar().showMessage(
self.tr('Geometry validation started.'))
def test_ring_intersections(self):
# no intersections
g = QgsGeometry.fromWkt(
"Polygon ((0 0, 10 0, 10 10, 0 10, 0 0),(1 1, 5 1, 1 9, 1 1), (6 9, 2 9, 6 1, 6 9))"
)
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
# two interior rings intersecting
g = QgsGeometry.fromWkt(
"Polygon ((0 0, 10 0, 10 10, 0 10, 0 0),(1 1, 5 1, 1 9, 1 1), (2 2, 5 2, 2 9, 2 2))"
)
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 2)
self.assertEqual(spy[0][0].where(), QgsPointXY(4.5, 2))
self.assertEqual(
spy[0][0].what(),
'segment 1 of ring 1 of polygon 0 intersects segment 0 of ring 2 of polygon 0 at 4.5, 2'
)
self.assertEqual(spy[1][0].where(), QgsPointXY(2, 7))
self.assertEqual(
spy[1][0].what(),
'segment 1 of ring 1 of polygon 0 intersects segment 2 of ring 2 of polygon 0 at 2, 7'
)
def test_linestring_duplicate_nodes(self):
g = QgsGeometry.fromWkt("LineString (1 1, 1 1, 1 1, 1 2, 1 3, 1 3, 1 3, 1 4, 1 5, 1 6, 1 6)")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 3)
self.assertEqual(spy[0][0].where(), QgsPointXY(1, 6))
self.assertEqual(spy[0][0].what(), 'line 1 contains 2 duplicate node(s) starting at vertex 10')
self.assertEqual(spy[1][0].where(), QgsPointXY(1, 3))
self.assertEqual(spy[1][0].what(), 'line 1 contains 3 duplicate node(s) starting at vertex 5')
self.assertEqual(spy[2][0].where(), QgsPointXY(1, 1))
self.assertEqual(spy[2][0].what(), 'line 1 contains 3 duplicate node(s) starting at vertex 1')
def _validateGeometry(self):
geometryType = self.geometryType()
if (geometryType == QGis.Point
or geometryType == QGis.UnknownGeometry
or geometryType == QGis.NoGeometry
or len(self._mapPointList) < 2):
return
settings = QSettings()
if (settings.value('/qgis/digitizing/validate_geometries', 1) == 0):
return
self._deleteGeometryValidation()
geometry = None # QgsGeometry()
if (geometryType == QGis.Line):
geometry = QgsGeometry.fromPolyline(self._mapPointList)
elif (geometryType == QGis.Polygon):
if (len(self._mapPointList) < 3):
return
closed = list(self._mapPointList)
closed.append(self._mapPointList[0])
geometry = QgsGeometry.fromPolygon([closed])
if (geometry is None):
return
self._validator = QgsGeometryValidator(geometry)
self._validator.errorFound.connect(self._addGeometryError)
self._validator.finished.connect(self.validationFinished)
self._validator.start()
self._iface.mainWindow().statusBar().showMessage(self.tr('Geometry validation started.'))
def test_multi_part_curve(self):
# A circle inside another one
g = QgsGeometry.fromWkt("MultiSurface (CurvePolygon (CircularString (0 5, 5 0, 0 -5, -5 0, 0 5)),CurvePolygon (CircularString (100 1, 100 0, 100 -1, 99 0, 100 1)))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
# converted as a straight polygon
g.convertToStraightSegment()
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
def testIssue15660(self):
""" Test crash when validating geometry (#15660) """
g = QgsGeometry.fromWkt(
"Polygon ((0.44256348235389709 -47.87645625696347906, -2.88231630340906797 -47.90003919913998232,"
"-2.88589842578005751 -48.91215450743293047, -2.8858984257800584 -48.91215450743293047,"
"-2.88589842578005751 -48.91215450743292337, -2.88589842632776516 -48.9121545074024624,"
"-3.24858148608664266 -48.89198543875494352, -3.27689849271356159 -49.40119850743292318,"
"-4.37689842578006072 -49.40119850743292318, -4.3768984257800625 -49.40119850743293739,"
"-6.11689842578005738 -49.40119850743293739, -6.11689842578006093 -49.40119850743292318,"
"-7.03689842578006086 -49.40119850743292318, -7.02239489567359776 -48.93302358132471852,"
"-7.02177188091450688 -48.91291272055079276, -7.02177188393206286 -48.91291272045731375,"
"-7.02141642578006309 -48.9014385074329212, -7.7002102410998674 -48.88041051649613422,"
"-7.70077301577442341 -48.89187793078160382, -7.70077301577442341 -48.89187793078160382,"
"-7.70233865095334291 -48.92378019651650334, -7.72576842578006051 -49.40119850743292318,"
"-9.26576842578005966 -49.40119850743292318, -9.26576842578006321 -49.40119850743293739,"
"-13.28076842578006023 -49.40119850743293739, -13.04700849136197149 -44.82717853953759857,"
"-12.22739168108193297 -44.85224022031006541, -12.22501286465108805 -44.774446133668377,"
"-12.22288921611744783 -44.774511069530881, -12.2155540445085915 -44.53462318893357264,"
"-13.0310217329353506 -44.50968822589503304, -12.87640859053235687 -41.29089836691012749,"
"-7.83390711693117936 -41.74840291007100745, -7.88211379129075596 -42.99075321817508666,"
"-7.43245210877673657 -43.00820115628285123, -7.50410812906098013 -44.67868742523263847,"
"-7.52086717830689011 -44.67817497540850979, -7.52820234991574644 -44.91806285600581816,"
"-7.51439432253991058 -44.91848507095944854, -7.57421591980290287 -46.3130804785479242,"
"-8.32385639731993621 -46.28985691678211367, -8.44985043007881842 -48.85718773355701217,"
"-6.26478736265246283 -48.92487774800262912, -6.18500945357052245 -46.35611749220839073,"
"-6.94163842159974198 -46.33267751510010157, -6.82382190915497944 -40.77447960826383166,"
"-5.48772831582523146 -40.77837853452808758, -5.48973219643841759 -39.92687558952010107,"
"-2.75658441771447116 -40.04490036239724304, -3.1241861109063862 -46.6551270968877958,"
"-2.78977790434136885 -46.6737244011090695, -2.78977790434136796 -46.6737244011090695,"
"-2.71083842578005996 -44.83541850743291945, -2.71083842578005729 -44.83541850743291945,"
"-0.86779302740823816 -44.89143693883772812, -0.86745855610774569 -44.87743669555854353,"
"0.29843811058281489 -44.90401226269042922, 0.20437651721061911 -46.69301920907949466,"
"0.50389019278376956 -46.71008040893148916, 0.44256348235389709 -47.87645625696347906))"
)
self.assertTrue(g)
# make sure validating this geometry doesn't crash QGIS
QgsGeometryValidator.validateGeometry(g)
def testIssue15660(self):
""" Test crash when validating geometry (#15660) """
g = QgsGeometry.fromWkt(
"Polygon ((0.44256348235389709 -47.87645625696347906, -2.88231630340906797 -47.90003919913998232,"
"-2.88589842578005751 -48.91215450743293047, -2.8858984257800584 -48.91215450743293047,"
"-2.88589842578005751 -48.91215450743292337, -2.88589842632776516 -48.9121545074024624,"
"-3.24858148608664266 -48.89198543875494352, -3.27689849271356159 -49.40119850743292318,"
"-4.37689842578006072 -49.40119850743292318, -4.3768984257800625 -49.40119850743293739,"
"-6.11689842578005738 -49.40119850743293739, -6.11689842578006093 -49.40119850743292318,"
"-7.03689842578006086 -49.40119850743292318, -7.02239489567359776 -48.93302358132471852,"
"-7.02177188091450688 -48.91291272055079276, -7.02177188393206286 -48.91291272045731375,"
"-7.02141642578006309 -48.9014385074329212, -7.7002102410998674 -48.88041051649613422,"
"-7.70077301577442341 -48.89187793078160382, -7.70077301577442341 -48.89187793078160382,"
"-7.70233865095334291 -48.92378019651650334, -7.72576842578006051 -49.40119850743292318,"
"-9.26576842578005966 -49.40119850743292318, -9.26576842578006321 -49.40119850743293739,"
"-13.28076842578006023 -49.40119850743293739, -13.04700849136197149 -44.82717853953759857,"
"-12.22739168108193297 -44.85224022031006541, -12.22501286465108805 -44.774446133668377,"
"-12.22288921611744783 -44.774511069530881, -12.2155540445085915 -44.53462318893357264,"
"-13.0310217329353506 -44.50968822589503304, -12.87640859053235687 -41.29089836691012749,"
"-7.83390711693117936 -41.74840291007100745, -7.88211379129075596 -42.99075321817508666,"
"-7.43245210877673657 -43.00820115628285123, -7.50410812906098013 -44.67868742523263847,"
"-7.52086717830689011 -44.67817497540850979, -7.52820234991574644 -44.91806285600581816,"
"-7.51439432253991058 -44.91848507095944854, -7.57421591980290287 -46.3130804785479242,"
"-8.32385639731993621 -46.28985691678211367, -8.44985043007881842 -48.85718773355701217,"
"-6.26478736265246283 -48.92487774800262912, -6.18500945357052245 -46.35611749220839073,"
"-6.94163842159974198 -46.33267751510010157, -6.82382190915497944 -40.77447960826383166,"
"-5.48772831582523146 -40.77837853452808758, -5.48973219643841759 -39.92687558952010107,"
"-2.75658441771447116 -40.04490036239724304, -3.1241861109063862 -46.6551270968877958,"
"-2.78977790434136885 -46.6737244011090695, -2.78977790434136796 -46.6737244011090695,"
"-2.71083842578005996 -44.83541850743291945, -2.71083842578005729 -44.83541850743291945,"
"-0.86779302740823816 -44.89143693883772812, -0.86745855610774569 -44.87743669555854353,"
"0.29843811058281489 -44.90401226269042922, 0.20437651721061911 -46.69301920907949466,"
"0.50389019278376956 -46.71008040893148916, 0.44256348235389709 -47.87645625696347906))"
)
self.assertTrue(g)
# make sure validating this geometry doesn't crash QGIS
QgsGeometryValidator.validateGeometry(g)
def test_multi_part_curve_nested_shell(self):
# A circle inside another one
g = QgsGeometry.fromWkt("MultiSurface (CurvePolygon (CircularString (0 5, 5 0, 0 -5, -5 0, 0 5)),CurvePolygon (CircularString (0 1, 1 0, 0 -1, -1 0, 0 1)))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'Polygon 1 lies inside polygon 0')
# converted as a straight polygon
g.convertToStraightSegment()
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'Polygon 1 lies inside polygon 0')
def test_ring_closed(self):
# valid ring
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 0))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
# not closed
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 1 1))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY(0, 0))
self.assertEqual(spy[0][0].what(), 'ring 0 not closed')
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 0),(1 1, 2 1, 2 2, 1.1 1))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY(1, 1))
self.assertEqual(spy[0][0].what(), 'ring 1 not closed')
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 10, 0 0),(1 1, 2 1, 2 2, 1 1),(3 3, 3 4, 4 4, 3.1 3))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY(3, 3))
self.assertEqual(spy[0][0].what(), 'ring 2 not closed')
# not closed but 2d closed
g = QgsGeometry.fromWkt("POLYGONZ((1 1 0, 1 2 1, 2 2 2, 2 1 3, 1 1 4))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY(1, 1))
self.assertEqual(spy[0][0].what(), 'ring 0 not closed, Z mismatch: 0 vs 4')
def test_ring_vertex_count(self):
# valid ring
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 0))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
# not enough vertices
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 0 with less than four points')
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 0 0))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 0 with less than four points')
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 0 with less than four points')
g = QgsGeometry.fromWkt("Polygon ((0 0))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 0 with less than four points')
g = QgsGeometry.fromWkt("Polygon (())")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 0),(1 1, 2 1, 2 2))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 1 with less than four points')
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 0),(1 1, 2 1, 2 2, 1 1))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
g = QgsGeometry.fromWkt("Polygon ((0 0, 10 0, 10 10, 0 10, 0 0),(1 1, 2 1, 2 2, 1 1),(3 3, 3 4, 4 4))")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'ring 2 with less than four points')
def test_line_vertices(self):
# valid line
g = QgsGeometry.fromWkt("LineString (0 0, 10 0)")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 0)
# not enough vertices
g = QgsGeometry.fromWkt("LineString (1 0)")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'line 0 with less than two points')
g = QgsGeometry.fromWkt("LineString ()")
validator = QgsGeometryValidator(g)
spy = QSignalSpy(validator.errorFound)
validator.run()
self.assertEqual(len(spy), 1)
self.assertEqual(spy[0][0].where(), QgsPointXY())
self.assertEqual(spy[0][0].what(), 'line 0 with less than two points')
class MapToolCapture(MapToolInteractive):
"""Tool to capture and show mouse clicks as geometry using map points."""
canvasClicked = pyqtSignal(QgsPointV2, Qt.MouseButton)
def __init__(self, iface, geometryType=QGis.UnknownGeometry):
super(MapToolCapture, self).__init__(iface.mapCanvas())
self._iface = iface
self._useCurrentLayerGeometry = False
self._geometryType = geometryType
self._mapPointList = [] # QList<QgsPointV2>
self._rubberBand = None # QgsRubberBand()
self._moveRubberBand = None # QgsRubberBand()
self._tip = ''
self._validator = None # QgsGeometryValidator()
self._geometryErrors = [] # QList<QgsGeometry.Error>
self._geometryErrorMarkers = [] # QList<QgsVertexMarker>
if (geometryType == QGis.UnknownGeometry):
self._useCurrentLayerGeometry = True
self.setCursor(CapturePointCursor)
def __del__(self):
self.deactivate()
super(MapToolCapture, self).__del__()
def activate(self):
super(MapToolCapture, self).activate()
geometryType = self.geometryType()
self._rubberBand = self._createRubberBand(geometryType)
self._moveRubberBand = self._createRubberBand(geometryType, True)
if (self._useCurrentLayerGeometry is True):
self._iface.currentLayerChanged.connect(self._currentLayerChanged)
def deactivate(self):
self.resetCapturing()
if (self._rubberBand is not None):
self.canvas().scene().removeItem(self._rubberBand)
self._rubberBand = None
if (self._moveRubberBand is not None):
self.canvas().scene().removeItem(self._moveRubberBand)
self._moveRubberBand = None
if (self._useCurrentLayerGeometry is True):
self._iface.currentLayerChanged.disconnect(
self._currentLayerChanged)
super(MapToolCapture, self).deactivate()
def geometryType(self):
if (self._useCurrentLayerGeometry
and self.canvas().currentLayer().type()
== QgsMapLayer.VectorLayer):
return self.canvas().currentLayer().geometryType()
else:
return self._geometryType
def _currentLayerChanged(self, layer):
if (not self._useCurrentLayerGeometry):
return
# TODO Update rubber bands
if (self._rubberBand is not None):
self._rubberBand.reset(self.geometryType())
for point in self._mapPointList[:-1]:
self._rubberBand.addPoint(point, False)
self._updateMoveRubberBand(self._mapPointList[-1])
self._validateGeometry()
self.canvas().refresh()
def canvasMoveEvent(self, e):
super(MapToolCapture, self).canvasMoveEvent(e)
if e.isAccepted():
return
if (self._moveRubberBand is not None):
# Capture mode
mapPoint, mapPointV2, snapped = self._snapCursorPoint(e.pos())
self._moveRubberBand.movePoint(mapPoint)
def canvasReleaseEvent(self, e):
super(MapToolCapture, self).canvasReleaseEvent(e)
if e.isAccepted():
return
mapPoint, mapPointV2, snapped = self._snapCursorPoint(e.pos())
if (e.button() == Qt.LeftButton):
# Capture mode
self._addVertex(mapPoint, mapPointV2)
# Emit mode
self.canvasClicked.emit(mapPointV2, e.button())
def keyPressEvent(self, e):
if (e.key() == Qt.Key_Escape):
self.resetCapturing()
e.accept()
elif (e.key() == Qt.Key_Backspace or e.key() == Qt.Key_Delete):
self._undo()
e.accept()
else:
super(MapToolCapture, self).keyPressEvent(e)
def _createRubberBand(self, geometryType, moveBand=False):
settings = QSettings()
rb = QgsRubberBand(self.canvas(), geometryType)
rb.setWidth(int(settings.value('/qgis/digitizing/line_width', 1)))
color = QColor(
int(settings.value('/qgis/digitizing/line_color_red', 255)),
int(settings.value('/qgis/digitizing/line_color_green', 0)),
int(settings.value('/qgis/digitizing/line_color_blue', 0)))
myAlpha = int(settings.value('/qgis/digitizing/line_color_alpha',
200)) / 255.0
if (moveBand):
myAlpha = myAlpha * float(
settings.value('/qgis/digitizing/line_color_alpha_scale',
0.75))
rb.setLineStyle(Qt.DotLine)
if (geometryType == QGis.Polygon):
color.setAlphaF(myAlpha)
color.setAlphaF(myAlpha)
rb.setColor(color)
rb.show()
return rb
def _addVertex(self, mapPoint, mapPointV2):
self._mapPointList.append(mapPoint)
self._rubberBand.addPoint(mapPoint)
self._updateMoveRubberBand(mapPoint)
self._validateGeometry()
return
def _updateMoveRubberBand(self, mapPoint):
if (self._moveRubberBand is None):
return
geometryType = self.geometryType()
self._moveRubberBand.reset(geometryType)
if (geometryType == QGis.Polygon):
self._moveRubberBand.addPoint(self._rubberBand.getPoint(0, 0),
False)
self._moveRubberBand.movePoint(mapPoint, False)
self._moveRubberBand.addPoint(mapPoint)
def _undo(self):
if (self._rubberBand is not None):
rubberBandSize = self._rubberBand.numberOfVertices()
moveRubberBandSize = self._moveRubberBand.numberOfVertices()
if (rubberBandSize < 1 or len(self._mapPointList) < 1):
return
self._rubberBand.removePoint(-1)
if (rubberBandSize > 1):
if (moveRubberBandSize > 1):
point = self._rubberBand.getPoint(0, rubberBandSize - 2)
self._moveRubberBand.movePoint(moveRubberBandSize - 2,
point)
else:
self._moveRubberBand.reset(self.geometryType())
self._mapPointList.pop()
self._validateGeometry()
def resetCapturing(self):
if (self._rubberBand is not None):
self._rubberBand.reset(self.geometryType())
if (self._moveRubberBand is not None):
self._moveRubberBand.reset(self.geometryType())
self._deleteGeometryValidation()
del self._mapPointList[:]
self.canvas().refresh()
def _deleteGeometryValidation(self):
if (self._validator is not None):
self._validator.errorFound.disconnect(self._addGeometryError)
self._validator.finished.disconnect(self.validationFinished)
self._validator.deleteLater()
self._validator = None
for errorMarker in self._geometryErrorMarkers:
self.canvas().scene().removeItem(errorMarker)
del self._geometryErrorMarkers[:]
self._geometryErrors[:]
self._tip = ''
def _validateGeometry(self):
geometryType = self.geometryType()
if (geometryType == QGis.Point or geometryType == QGis.UnknownGeometry
or geometryType == QGis.NoGeometry
or len(self._mapPointList) < 2):
return
settings = QSettings()
if (settings.value('/qgis/digitizing/validate_geometries', 1) == 0):
return
self._deleteGeometryValidation()
geometry = None # QgsGeometry()
if (geometryType == QGis.Line):
geometry = QgsGeometry.fromPolyline(self._mapPointList)
elif (geometryType == QGis.Polygon):
if (len(self._mapPointList) < 3):
return
closed = list(self._mapPointList)
closed.append(self._mapPointList[0])
geometry = QgsGeometry.fromPolygon([closed])
if (geometry is None):
return
self._validator = QgsGeometryValidator(geometry)
self._validator.errorFound.connect(self._addGeometryError)
self._validator.finished.connect(self.validationFinished)
self._validator.start()
self._iface.mainWindow().statusBar().showMessage(
self.tr('Geometry validation started.'))
def _addGeometryError(self, error):
self._geometryErrors.append(error)
if (self._tip != ''):
self._tip += '\n'
self._tip += error.what()
if (error.hasWhere()):
marker = QgsVertexMarker(self.canvas())
marker.setCenter(error.where())
marker.setIconType(QgsVertexMarker.ICON_X)
marker.setPenWidth(2)
marker.setToolTip(error.what())
marker.setColor(Qt.green)
marker.setZValue(marker.zValue() + 1)
self._geometryErrorMarkers.append(marker)
self._iface.mainWindow().statusBar().showMessage(error.what())
if (self._tip != ''):
self._iface.mainWindow().statusBar().setToolTip(self._tip)
def validationFinished(self):
self._iface.mainWindow().statusBar().showMessage(
self.tr('Geometry validation finished.'))
class MapToolCapture(MapToolInteractive):
"""Tool to capture and show mouse clicks as geometry using map points."""
canvasClicked = pyqtSignal(QgsPointV2, Qt.MouseButton)
def __init__(self, iface, geometryType=QGis.UnknownGeometry):
super(MapToolCapture, self).__init__(iface.mapCanvas())
self._iface = iface
self._useCurrentLayerGeometry = False
self._geometryType = geometryType
self._mapPointList = [] # QList<QgsPointV2>
self._rubberBand = None # QgsRubberBand()
self._moveRubberBand = None # QgsRubberBand()
self._tip = ''
self._validator = None # QgsGeometryValidator()
self._geometryErrors = [] # QList<QgsGeometry.Error>
self._geometryErrorMarkers = [] # QList<QgsVertexMarker>
if (geometryType == QGis.UnknownGeometry):
self._useCurrentLayerGeometry = True
self.setCursor(CapturePointCursor)
def __del__(self):
self.deactivate()
super(MapToolCapture, self).__del__()
def activate(self):
super(MapToolCapture, self).activate()
geometryType = self.geometryType()
self._rubberBand = self._createRubberBand(geometryType)
self._moveRubberBand = self._createRubberBand(geometryType, True)
if (self._useCurrentLayerGeometry is True):
self._iface.currentLayerChanged.connect(self._currentLayerChanged)
def deactivate(self):
self.resetCapturing()
if (self._rubberBand is not None):
self.canvas().scene().removeItem(self._rubberBand)
self._rubberBand = None
if (self._moveRubberBand is not None):
self.canvas().scene().removeItem(self._moveRubberBand)
self._moveRubberBand = None
if (self._useCurrentLayerGeometry is True):
self._iface.currentLayerChanged.disconnect(self._currentLayerChanged)
super(MapToolCapture, self).deactivate()
def geometryType(self):
if (self._useCurrentLayerGeometry and self.canvas().currentLayer().type() == QgsMapLayer.VectorLayer):
return self.canvas().currentLayer().geometryType()
else:
return self._geometryType
def _currentLayerChanged(self, layer):
if (not self._useCurrentLayerGeometry):
return
# TODO Update rubber bands
if (self._rubberBand is not None):
self._rubberBand.reset(self.geometryType())
for point in self._mapPointList[:-1]:
self._rubberBand.addPoint(point, False)
self._updateMoveRubberBand(self._mapPointList[-1])
self._validateGeometry()
self.canvas().refresh()
def canvasMoveEvent(self, e):
super(MapToolCapture, self).canvasMoveEvent(e)
if e.isAccepted():
return
if (self._moveRubberBand is not None):
# Capture mode
mapPoint, mapPointV2, snapped = self._snapCursorPoint(e.pos())
self._moveRubberBand.movePoint(mapPoint)
def canvasReleaseEvent(self, e):
super(MapToolCapture, self).canvasReleaseEvent(e)
if e.isAccepted():
return
mapPoint, mapPointV2, snapped = self._snapCursorPoint(e.pos())
if (e.button() == Qt.LeftButton):
# Capture mode
self._addVertex(mapPoint, mapPointV2)
# Emit mode
self.canvasClicked.emit(mapPointV2, e.button())
def keyPressEvent(self, e):
if (e.key() == Qt.Key_Escape):
self.resetCapturing()
e.accept()
elif (e.key() == Qt.Key_Backspace or e.key() == Qt.Key_Delete):
self._undo()
e.accept()
else:
super(MapToolCapture, self).keyPressEvent(e)
def _createRubberBand(self, geometryType, moveBand=False):
settings = QSettings()
rb = QgsRubberBand(self.canvas(), geometryType)
rb.setWidth(int(settings.value('/qgis/digitizing/line_width', 1)))
color = QColor(int(settings.value('/qgis/digitizing/line_color_red', 255)),
int(settings.value('/qgis/digitizing/line_color_green', 0)),
int(settings.value('/qgis/digitizing/line_color_blue', 0)))
myAlpha = int(settings.value('/qgis/digitizing/line_color_alpha', 200)) / 255.0
if (moveBand):
myAlpha = myAlpha * float(settings.value('/qgis/digitizing/line_color_alpha_scale', 0.75))
rb.setLineStyle(Qt.DotLine)
if (geometryType == QGis.Polygon):
color.setAlphaF(myAlpha)
color.setAlphaF(myAlpha)
rb.setColor(color)
rb.show()
return rb
def _addVertex(self, mapPoint, mapPointV2):
self._mapPointList.append(mapPoint)
self._rubberBand.addPoint(mapPoint)
self._updateMoveRubberBand(mapPoint)
self._validateGeometry()
return
def _updateMoveRubberBand(self, mapPoint):
if (self._moveRubberBand is None):
return
geometryType = self.geometryType()
self._moveRubberBand.reset(geometryType)
if (geometryType == QGis.Polygon):
self._moveRubberBand.addPoint(self._rubberBand.getPoint(0, 0), False)
self._moveRubberBand.movePoint(mapPoint, False)
self._moveRubberBand.addPoint(mapPoint)
def _undo(self):
if (self._rubberBand is not None):
rubberBandSize = self._rubberBand.numberOfVertices()
moveRubberBandSize = self._moveRubberBand.numberOfVertices()
if (rubberBandSize < 1 or len(self._mapPointList) < 1):
return
self._rubberBand.removePoint(-1)
if (rubberBandSize > 1):
if (moveRubberBandSize > 1):
point = self._rubberBand.getPoint(0, rubberBandSize - 2)
self._moveRubberBand.movePoint(moveRubberBandSize - 2, point)
else:
self._moveRubberBand.reset(self.geometryType())
self._mapPointList.pop()
self._validateGeometry()
def resetCapturing(self):
if (self._rubberBand is not None):
self._rubberBand.reset(self.geometryType())
if (self._moveRubberBand is not None):
self._moveRubberBand.reset(self.geometryType())
self._deleteGeometryValidation()
del self._mapPointList[:]
self.canvas().refresh()
def _deleteGeometryValidation(self):
if (self._validator is not None):
self._validator.errorFound.disconnect(self._addGeometryError)
self._validator.finished.disconnect(self.validationFinished)
self._validator.deleteLater()
self._validator = None
for errorMarker in self._geometryErrorMarkers:
self.canvas().scene().removeItem(errorMarker)
del self._geometryErrorMarkers[:]
self._geometryErrors[:]
self._tip = ''
def _validateGeometry(self):
geometryType = self.geometryType()
if (geometryType == QGis.Point
or geometryType == QGis.UnknownGeometry
or geometryType == QGis.NoGeometry
or len(self._mapPointList) < 2):
return
settings = QSettings()
if (settings.value('/qgis/digitizing/validate_geometries', 1) == 0):
return
self._deleteGeometryValidation()
geometry = None # QgsGeometry()
if (geometryType == QGis.Line):
geometry = QgsGeometry.fromPolyline(self._mapPointList)
elif (geometryType == QGis.Polygon):
if (len(self._mapPointList) < 3):
return
closed = list(self._mapPointList)
closed.append(self._mapPointList[0])
geometry = QgsGeometry.fromPolygon([closed])
if (geometry is None):
return
self._validator = QgsGeometryValidator(geometry)
self._validator.errorFound.connect(self._addGeometryError)
self._validator.finished.connect(self.validationFinished)
self._validator.start()
self._iface.mainWindow().statusBar().showMessage(self.tr('Geometry validation started.'))
def _addGeometryError(self, error):
self._geometryErrors.append(error)
if (self._tip != ''):
self._tip += '\n'
self._tip += error.what()
if (error.hasWhere()):
marker = QgsVertexMarker(self.canvas())
marker.setCenter(error.where())
marker.setIconType(QgsVertexMarker.ICON_X)
marker.setPenWidth(2)
marker.setToolTip(error.what())
marker.setColor(Qt.green)
marker.setZValue(marker.zValue() + 1)
self._geometryErrorMarkers.append(marker)
self._iface.mainWindow().statusBar().showMessage(error.what())
if (self._tip != ''):
self._iface.mainWindow().statusBar().setToolTip(self._tip)
def validationFinished(self):
self._iface.mainWindow().statusBar().showMessage(self.tr('Geometry validation finished.'))
