def process(self):
'''Function that does all the real work'''
title = QtCore.QCoreApplication.translate("digitizingtools", "Splitter")
splitterLayer = dtutils.dtChooseVectorLayer(self.iface, 1, msg = QtCore.QCoreApplication.translate("digitizingtools", "splitter layer"))
if splitterLayer == None:
self.iface.messageBar().pushMessage(title, QtCore.QCoreApplication.translate("digitizingtools", "Please provide a line layer to split with."))
else:
passiveLayer = self.iface.activeLayer()
msgLst = dtutils.dtGetNoSelMessage()
noSelMsg1 = msgLst[0]
if splitterLayer.selectedFeatureCount() == 0:
self.iface.messageBar().pushMessage(title, noSelMsg1 + " " + splitterLayer.name())
return None
elif splitterLayer.selectedFeatureCount() != 1:
numSelSplitMsg = dtutils.dtGetManySelMessage(splitterLayer)
self.iface.messageBar().pushMessage(title, numSelSplitMsg + \
QtCore.QCoreApplication.translate("digitizingtools", " Please select only one feature to split with."))
else:
if passiveLayer.selectedFeatureCount() == 0:
self.iface.messageBar().pushMessage(title, noSelMsg1 + " " + passiveLayer.name() + ".\n" + \
QtCore.QCoreApplication.translate("digitizingtools", " Please select the features to be splitted."))
return None
# determine srs, we work in the project's srs
splitterCRSSrsid = splitterLayer.crs().srsid()
passiveCRSSrsid = passiveLayer.crs().srsid()
mc = self.iface.mapCanvas()
renderer = mc.mapRenderer()
projectCRSSrsid = renderer.destinationCrs().srsid()
passiveLayer.beginEditCommand(QtCore.QCoreApplication.translate("editcommand", "Split features"))
featuresBeingSplit = 0
featuresToAdd = []
for feat in splitterLayer.selectedFeatures():
splitterGeom = QgsGeometry(feat.geometry())
if not splitterGeom.isGeosValid():
thisWarning = dtutils.dtGetInvalidGeomWarning(splitterLayer)
dtutils.dtShowWarning(self.iface, thisWarning)
continue
if splitterCRSSrsid != projectCRSSrsid:
splitterGeom.transform(QgsCoordinateTransform(splitterCRSSrsid, projectCRSSrsid))
for selFeat in passiveLayer.selectedFeatures():
selGeom = QgsGeometry(selFeat.geometry())
if not selGeom.isGeosValid():
thisWarning = dtutils.dtGetInvalidGeomWarning(passiveLayer)
dtutils.dtShowWarning(self.iface, thisWarning)
continue
if passiveCRSSrsid != projectCRSSrsid:
selGeom.transform(QgsCoordinateTransform(passiveCRSSrsid, projectCRSSrsid))
if splitterGeom.intersects(selGeom): # we have a candidate
splitterPList = dtutils.dtExtractPoints(splitterGeom)
try:
result, newGeometries, topoTestPoints = selGeom.splitGeometry(splitterPList, False) #QgsProject.instance().topologicalEditing())
except:
self.iface.messageBar().pushMessage(title,
dtutils.dtGetErrorMessage() + QtCore.QCoreApplication.translate("digitizingtools", "splitting of feature") + " " + str(selFeat.id()),
level=QgsMessageBar.CRITICAL)
return None
if result == 0:
selFeat.setGeometry(selGeom)
passiveLayer.updateFeature(selFeat)
if len(newGeometries) > 0:
featuresBeingSplit += 1
newFeatures = dtutils.dtMakeFeaturesFromGeometries(passiveLayer, selFeat, newGeometries)
for newFeat in newFeatures:
newGeom = QgsGeometry(newFeat.geometry())
if passiveCRSSrsid != projectCRSSrsid:
newGeom.transform(QgsCoordinateTransform( projectCRSSrsid, passiveCRSSrsid))
newFeat.setGeometry(newGeom)
featuresToAdd.append(newFeat)
if featuresBeingSplit > 0:
passiveLayer.addFeatures(featuresToAdd, False)
passiveLayer.endEditCommand()
passiveLayer.removeSelection()
else:
passiveLayer.destroyEditCommand()
def process(self):
'''Function that does all the real work'''
title = QtCore.QCoreApplication.translate("digitizingtools",
"Splitter")
splitterLayer = dtutils.dtChooseVectorLayer(
self.iface,
1,
msg=QtCore.QCoreApplication.translate("digitizingtools",
"splitter layer"))
if splitterLayer == None:
self.iface.messageBar().pushMessage(
title,
QtCore.QCoreApplication.translate(
"digitizingtools",
"Please provide a line layer to split with."))
else:
passiveLayer = self.iface.activeLayer()
msgLst = dtutils.dtGetNoSelMessage()
noSelMsg1 = msgLst[0]
if splitterLayer.selectedFeatureCount() == 0:
self.iface.messageBar().pushMessage(
title, noSelMsg1 + " " + splitterLayer.name())
return None
elif splitterLayer.selectedFeatureCount() != 1:
numSelSplitMsg = dtutils.dtGetManySelMessage(splitterLayer)
self.iface.messageBar().pushMessage(title, numSelSplitMsg + \
QtCore.QCoreApplication.translate("digitizingtools", " Please select only one feature to split with."))
else:
if passiveLayer.selectedFeatureCount() == 0:
self.iface.messageBar().pushMessage(title, noSelMsg1 + " " + passiveLayer.name() + ".\n" + \
QtCore.QCoreApplication.translate("digitizingtools", " Please select the features to be splitted."))
return None
# determine srs, we work in the project's srs
splitterCRSSrsid = splitterLayer.crs().srsid()
passiveCRSSrsid = passiveLayer.crs().srsid()
mc = self.iface.mapCanvas()
renderer = mc.mapRenderer()
projectCRSSrsid = renderer.destinationCrs().srsid()
passiveLayer.beginEditCommand(
QtCore.QCoreApplication.translate("editcommand",
"Split features"))
featuresBeingSplit = 0
featuresToAdd = []
for feat in splitterLayer.selectedFeatures():
splitterGeom = feat.geometry()
if splitterCRSSrsid != projectCRSSrsid:
splitterGeom.transform(
QgsCoordinateTransform(splitterCRSSrsid,
projectCRSSrsid))
for selFeat in passiveLayer.selectedFeatures():
selGeom = selFeat.geometry()
if passiveCRSSrsid != projectCRSSrsid:
selGeom.transform(
QgsCoordinateTransform(passiveCRSSrsid,
projectCRSSrsid))
if splitterGeom.intersects(
selGeom): # we have a candidate
splitterPList = dtutils.dtExtractPoints(
splitterGeom)
try:
result, newGeometries, topoTestPoints = selGeom.splitGeometry(
splitterPList, False
) #QgsProject.instance().topologicalEditing())
except:
self.iface.messageBar().pushMessage(
title,
dtutils.dtGetErrorMessage() +
QtCore.QCoreApplication.translate(
"digitizingtools",
"splitting of feature") + " " +
str(selFeat.id()),
level=QgsMessageBar.CRITICAL)
return None
if result == 0:
selFeat.setGeometry(selGeom)
passiveLayer.updateFeature(selFeat)
if len(newGeometries) > 0:
featuresBeingSplit += 1
newFeatures = dtutils.dtMakeFeaturesFromGeometries(
passiveLayer, selFeat, newGeometries)
for newFeat in newFeatures:
newGeom = newFeat.geometry()
if passiveCRSSrsid != projectCRSSrsid:
newGeom.transform(
QgsCoordinateTransform(
projectCRSSrsid,
passiveCRSSrsid))
newFeat.setGeometry(newGeom)
featuresToAdd.append(newFeat)
if featuresBeingSplit > 0:
passiveLayer.addFeatures(featuresToAdd, False)
passiveLayer.endEditCommand()
passiveLayer.removeSelection()
else:
passiveLayer.destroyEditCommand()
def digitizingFinished(self, splitGeom):
title = QtCore.QCoreApplication.translate("digitizingtools", "Split Features")
hlColor, hlFillColor, hlBuffer, hlMinWidth = dtutils.dtGetHighlightSettings()
selIds = self.editLayer.selectedFeatureIds()
self.editLayer.removeSelection()
if self.editLayer.crs().srsid() != QgsProject.instance().crs().srsid():
splitGeom.transform(QgsCoordinateTransform(
QgsProject.instance().crs(), self.editLayer.crs(),
QgsProject.instance()
))
splitterPList = dtutils.dtExtractPoints(splitGeom)
featuresToAdd = [] # store new features in this array
featuresToKeep = {} # store geoms that will stay with their id as key
featuresToSplit = {}
topoGeoms = [] # store all new geometries in this array
for aFeat in self.editLayer.getFeatures(QgsFeatureRequest(splitGeom.boundingBox())):
anId = aFeat.id()
# work either on selected or all features if no selection exists
if len(selIds) == 0 or selIds.count(anId) != 0:
aGeom = aFeat.geometry()
if splitGeom.intersects(aGeom):
featuresToSplit[anId] = aFeat
if len(featuresToSplit) > 0:
self.editLayer.beginEditCommand(QtCore.QCoreApplication.translate("editcommand", "Features split"))
for anId, aFeat in list(featuresToSplit.items()):
aGeom = aFeat.geometry()
wasMultipart = aGeom.isMultipart() and len(aGeom.asGeometryCollection()) > 1
splitResult = []
geomsToSplit = []
if wasMultipart:
keepGeom = None
for aPart in aGeom.asGeometryCollection():
if splitGeom.intersects(aPart):
geomsToSplit.append(aPart)
else:
if keepGeom == None:
keepGeom = aPart
else:
keepGeom = keepGeom.combine(aPart)
else:
geomsToSplit.append(aGeom)
for thisGeom in geomsToSplit:
try:
result, newGeometries, topoTestPoints = thisGeom.splitGeometry(splitterPList, False)
except:
self.iface.messageBar().pushCritical(title,
dtutils.dtGetErrorMessage() + QtCore.QCoreApplication.translate(
"digitizingtools", "splitting of feature") + " " + str(aFeat.id()))
return None
if result == 0: # success
if len(newGeometries) > 0:
splitResult = newGeometries
if wasMultipart:
splitResult.append(thisGeom)
if wasMultipart and len(splitResult) > 1:
takeThisOne = -1
while takeThisOne == -1:
for i in range(len(splitResult)):
aNewGeom = splitResult[i]
hl = QgsHighlight(self.iface.mapCanvas(), aNewGeom, self.editLayer)
hl.setColor(hlColor)
hl.setFillColor(hlFillColor)
hl.setBuffer(hlBuffer)
hl.setWidth(hlMinWidth)
answer = QtWidgets.QMessageBox.question(
None, QtCore.QCoreApplication.translate("digitizingtools", "Split Multipart Feature"),
QtCore.QCoreApplication.translate("digitizingtools", "Create new feature from this part?"),
QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No | QtWidgets.QMessageBox.Cancel | QtWidgets.QMessageBox.NoToAll)
if answer == QtWidgets.QMessageBox.Yes:
takeThisOne = i
break
elif answer == QtWidgets.QMessageBox.NoToAll:
keepGeom = aGeom
newGeoms = []
takeThisOne = -2
break
elif answer == QtWidgets.QMessageBox.Cancel:
self.editLayer.destroyEditCommand()
return None
if takeThisOne == -2:
break
elif takeThisOne >= 0:
newGeoms = [splitResult.pop(takeThisOne)]
if len(splitResult) > 0: #should be
for aNewGeom in splitResult:
if keepGeom == None:
keepGeom = aNewGeom
else:
keepGeom = keepGeom.combine(aNewGeom)
else: # singlePart
keepGeom = thisGeom
newGeoms = newGeometries
newFeatures = dtutils.dtMakeFeaturesFromGeometries(self.editLayer, aFeat, newGeoms)
featuresToAdd = featuresToAdd + newFeatures
topoGeoms = topoGeoms + newGeoms
aFeat.setGeometry(keepGeom)
featuresToKeep[anId] = aFeat
topoGeoms.append(keepGeom)
for anId, aFeat in list(featuresToKeep.items()):
aGeom = aFeat.geometry()
self.editLayer.updateFeature(aFeat)
if len(featuresToAdd) > 0:
if self.editLayer.addFeatures(featuresToAdd):
if QgsProject.instance().topologicalEditing():
for aGeom in topoGeoms:
self.editLayer.addTopologicalPoints(aGeom)
self.editLayer.endEditCommand()
else:
self.editLayer.destroyEditCommand()
else:
self.editLayer.destroyEditCommand()
if hasattr(self.editLayer, "selectByIds"): # since QGIS 2.16
self.editLayer.selectByIds(selIds)
else:
self.editLayer.setSelectedFeatures(selIds)
def run(self):
'''Function that does all the real work'''
title = QtCore.QCoreApplication.translate("digitizingtools", "Splitter")
splitterLayer = dtutils.dtChooseVectorLayer(self.iface, 1, msg = QtCore.QCoreApplication.translate("digitizingtools", "splitter layer"))
if splitterLayer == None:
QtGui.QMessageBox.information(None, title, QtCore.QCoreApplication.translate("digitizingtools", "Please provide a line layer to split with."))
else:
passiveLayer = self.iface.activeLayer()
msgLst = dtutils.dtGetNoSelMessage()
noSelMsg1 = msgLst[0]
noSelMsg2 = msgLst[1]
if splitterLayer.selectedFeatureCount() == 0:
reply = QtGui.QMessageBox.question(None, title,
noSelMsg1 + " " + splitterLayer.name() + "\n" + noSelMsg2,
QtGui.QMessageBox.Yes | QtGui.QMessageBox.Cancel )
if reply == QtGui.QMessageBox.Yes:
splitterLayer.invertSelection()
else:
return None
if splitterLayer.selectedFeatureCount() > 0:
if passiveLayer.selectedFeatureCount() == 0:
QtGui.QMessageBox.information(None, title, noSelMsg1 + " " + passiveLayer.name() + ".\n" + \
QtCore.QCoreApplication.translate("digitizingtools", "Please select the features to be splitted."))
return None
# determine srs, we work in the project's srs
splitterCRSSrsid = splitterLayer.crs().srsid()
passiveCRSSrsid = passiveLayer.crs().srsid()
mc = self.iface.mapCanvas()
renderer = mc.mapRenderer()
projectCRSSrsid = renderer.destinationCrs().srsid()
passiveLayer.beginEditCommand(QtCore.QCoreApplication.translate("digitizingtools", "Split Features"))
featuresBeingSplit = 0
featuresToAdd = []
for feat in splitterLayer.selectedFeatures():
splitterGeom = feat.geometry()
if splitterCRSSrsid != projectCRSSrsid:
splitterGeom.transform(QgsCoordinateTransform(splitterCRSSrsid, projectCRSSrsid))
for selFeat in passiveLayer.selectedFeatures():
selGeom = selFeat.geometry()
if passiveCRSSrsid != projectCRSSrsid:
selGeom.transform(QgsCoordinateTransform(passiveCRSSrsid, projectCRSSrsid))
if splitterGeom.intersects(selGeom): # we have a candidate
splitterPList = dtutils.dtExtractPoints(splitterGeom)
try:
result, newGeometries, topoTestPoints = selGeom.splitGeometry(splitterPList, False) #QgsProject.instance().topologicalEditing())
except:
QtGui.QMessageBox.warning(None, title,
dtutils.dtGetErrorMessage() + QtCore.QCoreApplication.translate("digitizingtools", "splitting of feature") + " " + str(selFeat.id()))
return None
if result == 0:
selFeat.setGeometry(selGeom)
passiveLayer.updateFeature(selFeat)
if len(newGeometries) > 0:
featuresBeingSplit += 1
newFeatures = dtutils.dtMakeFeaturesFromGeometries(passiveLayer, selFeat, newGeometries)
for newFeat in newFeatures:
newGeom = newFeat.geometry()
if passiveCRSSrsid != projectCRSSrsid:
newGeom.transform(QgsCoordinateTransform( projectCRSSrsid, passiveCRSSrsid))
newFeat.setGeometry(newGeom)
featuresToAdd.append(newFeat)
if featuresBeingSplit > 0:
passiveLayer.addFeatures(featuresToAdd, False)
passiveLayer.endEditCommand()
passiveLayer.removeSelection()
else:
passiveLayer.destroyEditCommand()
def digitizingFinished(self, splitGeom):
title = QtCore.QCoreApplication.translate("digitizingtools",
"Split Features")
hlColor, hlFillColor, hlBuffer, hlMinWidth = dtutils.dtGetHighlightSettings(
)
selIds = self.editLayer.selectedFeatureIds()
self.editLayer.removeSelection()
if self.editLayer.crs().srsid() != QgsProject.instance().crs().srsid():
splitGeom.transform(
QgsCoordinateTransform(QgsProject.instance().crs(),
self.editLayer.crs(),
QgsProject.instance()))
splitterPList = dtutils.dtExtractPoints(splitGeom)
featuresToAdd = [] # store new features in this array
featuresToKeep = {} # store geoms that will stay with their id as key
featuresToSplit = {}
topoEditEnabled = QgsProject.instance().topologicalEditing()
topoTestPointsAll = [] # store all topoTestPoints for all parts
for aFeat in self.editLayer.getFeatures(
QgsFeatureRequest(splitGeom.boundingBox())):
anId = aFeat.id()
# work either on selected or all features if no selection exists
if len(selIds) == 0 or selIds.count(anId) != 0:
aGeom = aFeat.geometry()
if splitGeom.intersects(aGeom):
featuresToSplit[anId] = aFeat
if len(featuresToSplit) > 0:
self.editLayer.beginEditCommand(
QtCore.QCoreApplication.translate("editcommand",
"Features split"))
for anId, aFeat in list(featuresToSplit.items()):
aGeom = aFeat.geometry()
wasMultipart = aGeom.isMultipart() and len(
aGeom.asGeometryCollection()) > 1
splitResult = []
geomsToSplit = []
if wasMultipart:
keepGeom = None
for aPart in aGeom.asGeometryCollection():
if splitGeom.intersects(aPart):
geomsToSplit.append(aPart)
else:
if keepGeom == None:
keepGeom = aPart
else:
keepGeom = keepGeom.combine(aPart)
else:
geomsToSplit.append(aGeom)
for thisGeom in geomsToSplit:
try:
result, newGeometries, topoTestPoints = thisGeom.splitGeometry(
splitterPList, topoEditEnabled)
except:
self.iface.messageBar().pushCritical(
title,
dtutils.dtGetErrorMessage() +
QtCore.QCoreApplication.translate(
"digitizingtools", "splitting of feature") + " " +
str(aFeat.id()))
return None
topoTestPointsAll.append(topoTestPoints)
if result == 0: # success
if len(newGeometries) > 0:
splitResult = newGeometries
if wasMultipart:
splitResult.append(thisGeom)
if wasMultipart and len(splitResult) > 1:
takeThisOne = -1
while takeThisOne == -1:
for i in range(len(splitResult)):
aNewGeom = splitResult[i]
hl = QgsHighlight(self.iface.mapCanvas(), aNewGeom,
self.editLayer)
hl.setColor(hlColor)
hl.setFillColor(hlFillColor)
hl.setBuffer(hlBuffer)
hl.setWidth(hlMinWidth)
hl.show()
answer = QtWidgets.QMessageBox.question(
None,
QtCore.QCoreApplication.translate(
"digitizingtools",
"Split Multipart Feature"),
QtCore.QCoreApplication.translate(
"digitizingtools",
"Create new feature from this part?"),
QtWidgets.QMessageBox.Yes
| QtWidgets.QMessageBox.No
| QtWidgets.QMessageBox.Cancel
| QtWidgets.QMessageBox.NoToAll)
hl.hide()
hl = None
if answer == QtWidgets.QMessageBox.Yes:
takeThisOne = i
break
elif answer == QtWidgets.QMessageBox.NoToAll:
keepGeom = aGeom
newGeoms = []
takeThisOne = -2
break
elif answer == QtWidgets.QMessageBox.Cancel:
self.editLayer.destroyEditCommand()
return None
if takeThisOne == -2:
break
elif takeThisOne >= 0:
newGeoms = [splitResult.pop(takeThisOne)]
if len(splitResult) > 0: #should be
for aNewGeom in splitResult:
if keepGeom == None:
keepGeom = aNewGeom
else:
keepGeom = keepGeom.combine(aNewGeom)
else: # singlePart
keepGeom = thisGeom
newGeoms = newGeometries
newFeatures = dtutils.dtMakeFeaturesFromGeometries(
self.editLayer, aFeat, newGeoms)
featuresToAdd = featuresToAdd + newFeatures
aFeat.setGeometry(keepGeom)
featuresToKeep[anId] = aFeat
for anId, aFeat in list(featuresToKeep.items()):
aGeom = aFeat.geometry()
self.editLayer.updateFeature(aFeat)
if len(featuresToAdd) > 0:
if self.editLayer.addFeatures(featuresToAdd):
for topoTestPoint in topoTestPointsAll:
for pt in topoTestPoint:
self.editLayer.addTopologicalPoints(pt)
self.editLayer.endEditCommand()
else:
self.editLayer.destroyEditCommand()
else:
self.editLayer.destroyEditCommand()
if hasattr(self.editLayer, "selectByIds"): # since QGIS 2.16
self.editLayer.selectByIds(selIds)
else:
self.editLayer.setSelectedFeatures(selIds)
