mirror of
https://github.com/qgis/QGIS.git
synced 2025-04-24 00:04:42 -04:00
261 lines
11 KiB
Python
261 lines
11 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
"""
|
||
***************************************************************************
|
||
Eliminate.py
|
||
---------------------
|
||
Date : August 2012
|
||
Copyright : (C) 2013 by Bernhard Str<74>bl
|
||
Email : bernhard.stroebl@jena.de
|
||
***************************************************************************
|
||
* *
|
||
* 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. *
|
||
* *
|
||
***************************************************************************
|
||
"""
|
||
from processing.parameters.ParameterSelection import ParameterSelection
|
||
from processing.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException
|
||
|
||
__author__ = 'Bernhard Ströbl'
|
||
__date__ = 'September 2013'
|
||
__copyright__ = '(C) 2013, Bernhard Ströbl'
|
||
# This will get replaced with a git SHA1 when you do a git archive
|
||
__revision__ = '$Format:%H$'
|
||
|
||
from PyQt4.QtCore import *
|
||
from qgis.core import *
|
||
from processing.core.GeoAlgorithm import GeoAlgorithm
|
||
from processing.tools import dataobjects
|
||
from processing.parameters.ParameterVector import ParameterVector
|
||
from processing.parameters.ParameterBoolean import ParameterBoolean
|
||
from processing.parameters.ParameterTableField import ParameterTableField
|
||
from processing.parameters.ParameterString import ParameterString
|
||
from processing.outputs.OutputVector import OutputVector
|
||
from processing.core.ProcessingLog import ProcessingLog
|
||
|
||
class Eliminate(GeoAlgorithm):
|
||
|
||
INPUT = "INPUT"
|
||
OUTPUT = "OUTPUT"
|
||
MODE = "MODE"
|
||
KEEPSELECTION = "KEEPSELECTION"
|
||
ATTRIBUTE = "ATTRIBUTE"
|
||
COMPARISONVALUE = "COMPARISONVALUE"
|
||
COMPARISON = "COMPARISON"
|
||
|
||
MODES = ["Area", "Common boundary"]
|
||
MODE_AREA = 0
|
||
MODE_BOUNDARY = 1
|
||
|
||
def defineCharacteristics(self):
|
||
self.name = "Eliminate sliver polygons"
|
||
self.group = "Vector geometry tools"
|
||
self.addParameter(ParameterVector(self.INPUT, "Input layer", [ParameterVector.VECTOR_TYPE_POLYGON]))
|
||
self.addParameter(ParameterBoolean(self.KEEPSELECTION, "Use current selection in input layer (works only if called from toolbox)", False))
|
||
self.addParameter(ParameterTableField(self.ATTRIBUTE, "Selection attribute", self.INPUT))
|
||
self.comparisons = ['==', '!=', '>', '>=', '<', '<=', 'begins with', 'contains']
|
||
self.addParameter(ParameterSelection(self.COMPARISON, "Comparison", self.comparisons, default = 0))
|
||
self.addParameter(ParameterString(self.COMPARISONVALUE, "Value", default = "0"))
|
||
self.addParameter(ParameterSelection(self.MODE, "Merge selection with the neighbouring polygon with the largest", self.MODES))
|
||
self.addOutput(OutputVector(self.OUTPUT, "Cleaned layer"))
|
||
|
||
def processAlgorithm(self, progress):
|
||
inLayer = dataobjects.getObjectFromUri(self.getParameterValue(self.INPUT))
|
||
boundary = self.getParameterValue(self.MODE) == self.MODE_BOUNDARY
|
||
keepSelection = self.getParameterValue(self.KEEPSELECTION)
|
||
|
||
if not keepSelection:
|
||
# make a selection with the values provided
|
||
attribute = self.getParameterValue(self.ATTRIBUTE)
|
||
comparison = self.comparisons [self.getParameterValue(self.COMPARISON)]
|
||
comparisonvalue = self.getParameterValue(self.COMPARISONVALUE)
|
||
|
||
selectindex = inLayer.dataProvider().fieldNameIndex(attribute)
|
||
selectType = inLayer.dataProvider().fields()[selectindex].type()
|
||
selectionError = False
|
||
|
||
if selectType == 2:
|
||
try:
|
||
y = int(comparisonvalue)
|
||
except ValueError:
|
||
selectionError = True
|
||
msg = "Cannot convert \"" + unicode(comparisonvalue) + "\" to integer"
|
||
elif selectType == 6:
|
||
try:
|
||
y = float(comparisonvalue)
|
||
except ValueError:
|
||
selectionError = True
|
||
msg = "Cannot convert \"" + unicode(comparisonvalue) + "\" to float"
|
||
elif selectType == 10: # 10: string, boolean
|
||
try:
|
||
y = unicode(comparisonvalue)
|
||
except ValueError:
|
||
selectionError = True
|
||
msg = "Cannot convert \"" + unicode(comparisonvalue) + "\" to unicode"
|
||
elif selectType == 14: # date
|
||
dateAndFormat = comparisonvalue.split(" ")
|
||
|
||
if len(dateAndFormat) == 1:
|
||
y = QLocale.system().toDate(dateAndFormat[0]) # QtCore.QDate object
|
||
|
||
if y.isNull():
|
||
msg = "Cannot convert \"" + unicode(dateAndFormat) + "\" to date with system date format " + QLocale.system().dateFormat()
|
||
elif len(dateAndFormat) == 2:
|
||
y = QDate.fromString(dateAndFormat[0], dateAndFormat[1])
|
||
|
||
if y.isNull():
|
||
msg = "Cannot convert \"" + unicode(dateAndFormat[0]) + "\" to date with format string \"" + unicode(dateAndFormat[1] + "\". ")
|
||
else:
|
||
y = QDate()
|
||
msg = ""
|
||
|
||
if y.isNull(): # conversion was unsuccessfull
|
||
selectionError = True
|
||
msg += "Enter the date and the date format, e.g. \"07.26.2011\" \"MM.dd.yyyy\"."
|
||
|
||
if ((comparison == 'begins with') or (comparison == 'contains')) and selectType != 10:
|
||
selectionError = True
|
||
msg = "\"" + comparison + "\" can only be used with string fields"
|
||
|
||
selected = []
|
||
|
||
if selectionError:
|
||
raise GeoAlgorithmExecutionException("Error in selection input: " + msg)
|
||
else:
|
||
for feature in inLayer.getFeatures():
|
||
aValue = feature.attributes()[selectindex]
|
||
|
||
if aValue == None:
|
||
continue
|
||
|
||
if selectType == 2:
|
||
x = int(aValue)
|
||
elif selectType == 6:
|
||
x = float(aValue)
|
||
elif selectType == 10: # 10: string, boolean
|
||
x = unicode(aValue)
|
||
elif selectType == 14: # date
|
||
x = aValue # should be date
|
||
|
||
match = False
|
||
|
||
if (comparison == '=='):
|
||
match = (x == y)
|
||
elif (comparison == '!='):
|
||
match = (x != y)
|
||
elif (comparison == '>'):
|
||
match = (x > y)
|
||
elif (comparison == '>='):
|
||
match = (x >= y)
|
||
elif (comparison == '<'):
|
||
match = (x < y)
|
||
elif (comparison == '<='):
|
||
match = (x <= y)
|
||
elif (comparison == 'begins with'):
|
||
match = x.startswith(y)
|
||
elif (comparison == 'contains'):
|
||
match = (x.find(y) >= 0)
|
||
|
||
if (match):
|
||
selected.append(feature.id())
|
||
|
||
inLayer.setSelectedFeatures(selected)
|
||
|
||
if inLayer.selectedFeatureCount() == 0:
|
||
ProcessingLog.addToLog(ProcessingLog.LOG_WARNING, self.commandLineName() + "(No selection in input layer \"" + self.getParameterValue(self.INPUT) + "\")")
|
||
#self.iface.messageBar().pushMessage("Eliminate", "No selection provided")
|
||
|
||
# keep references to the features to eliminate
|
||
featToEliminate = []
|
||
for aFeat in inLayer.selectedFeatures():
|
||
featToEliminate.append( aFeat )
|
||
|
||
#delete all features to eliminate in inLayer (we won't save this)
|
||
inLayer.startEditing()
|
||
inLayer.deleteSelectedFeatures()
|
||
|
||
# ANALYZE
|
||
if len( featToEliminate ) > 0: # prevent zero division
|
||
start = 20.00
|
||
add = 80.00 / len( featToEliminate )
|
||
else:
|
||
start = 100
|
||
|
||
progress.setPercentage( start )
|
||
madeProgress = True
|
||
|
||
# we go through the list and see if we find any polygons we can merge the selected with
|
||
# if we have no success with some we merge and then restart the whole story
|
||
while ( madeProgress ): #check if we made any progress
|
||
madeProgress = False
|
||
featNotEliminated = []
|
||
|
||
#iterate over the polygons to eliminate
|
||
for i in range( len( featToEliminate )):
|
||
feat = featToEliminate.pop()
|
||
geom2Eliminate = feat.geometry()
|
||
bbox = geom2Eliminate.boundingBox()
|
||
fit = inLayer.getFeatures( QgsFeatureRequest().setFilterRect( bbox ) )
|
||
mergeWithFid = None
|
||
mergeWithGeom = None
|
||
max = 0
|
||
selFeat = QgsFeature()
|
||
|
||
while fit.nextFeature( selFeat ):
|
||
selGeom = selFeat.geometry()
|
||
|
||
if geom2Eliminate.intersects( selGeom ): # we have a candidate
|
||
iGeom = geom2Eliminate.intersection( selGeom )
|
||
|
||
if boundary:
|
||
selValue = iGeom.length()
|
||
else: # largest area
|
||
# we need a common boundary in order to merge
|
||
if 0 < iGeom.length():
|
||
selValue = selGeom.area()
|
||
else:
|
||
selValue = 0
|
||
|
||
if selValue > max:
|
||
max = selValue
|
||
mergeWithFid = selFeat.id()
|
||
mergeWithGeom = QgsGeometry( selGeom ) # deep copy of the geometry
|
||
# end while fit
|
||
|
||
if mergeWithFid != None: # a successful candidate
|
||
newGeom = mergeWithGeom.combine( geom2Eliminate )
|
||
|
||
if inLayer.changeGeometry( mergeWithFid, newGeom ):
|
||
madeProgress = True
|
||
else:
|
||
raise GeoAlgorithmExecutionException("Could not replace geometry of feature with id %s" % (mergeWithFid))
|
||
|
||
start = start + add
|
||
progress.setPercentage( start )
|
||
else:
|
||
featNotEliminated.append( feat )
|
||
# end for featToEliminate
|
||
featToEliminate = featNotEliminated
|
||
#end while
|
||
|
||
# create output
|
||
provider = inLayer.dataProvider()
|
||
output = self.getOutputFromName( self.OUTPUT )
|
||
writer = output.getVectorWriter( provider.fields(),
|
||
provider.geometryType(),
|
||
inLayer.crs() )
|
||
|
||
#write all features that are left over to output layer
|
||
iterator = inLayer.getFeatures()
|
||
for feature in iterator:
|
||
writer.addFeature( feature )
|
||
|
||
# leave inLayer untouched
|
||
inLayer.rollBack()
|
||
|
||
for feature in featNotEliminated:
|
||
writer.addFeature( feature )
|