sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-04-18 00:03:05 -04:00
Code Issues Packages Projects Releases Wiki Activity

Uses spatial index to select intersecting features (used selections before). Tool should run faster and with fewer map canvas glitches.

Browse Source 
git-svn-id: http://svn.osgeo.org/qgis/trunk@13036 c8812cc2-4d05-0410-92ff-de0c093fc19c
This commit is contained in:
cfarmer 2010-03-10 00:49:23 +00:00
parent cd23b747a9
commit 63c36711a0
1 changed files with 171 additions and 215 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

90
python/plugins/fTools/tools/doSpatialJoin.py
View File

@ -1,3 +1,4 @@
# -*- coding: utf-8 -*-
#----------------------------------------------------------- #-----------------------------------------------------------
# #
# Spatial Join # Spatial Join
@ -50,11 +51,9 @@ class Dialog(QDialog, Ui_Dialog):
# populate layer list # populate layer list
self.progressBar.setValue(0) self.progressBar.setValue(0)
mapCanvas = self.iface.mapCanvas() mapCanvas = self.iface.mapCanvas()
for i in range(mapCanvas.layerCount()): layers = ftools_utils.getLayerNames([QGis.Point, QGis.Line, QGis.Polygon])
layer = mapCanvas.layer(i) self.inShape.addItems(layers)
if layer.type() == layer.VectorLayer: self.joinShape.addItems(layers)
self.inShape.addItem(layer.name())
self.joinShape.addItem(layer.name())
def accept(self): def accept(self):
if self.inShape.currentText() == "": if self.inShape.currentText() == "":
@ -103,22 +102,22 @@ class Dialog(QDialog, Ui_Dialog):
self.outShape.setText( QString( self.shapefileName ) ) self.outShape.setText( QString( self.shapefileName ) )
def compute(self, inName, joinName, outName, summary, sumList, keep, progressBar): def compute(self, inName, joinName, outName, summary, sumList, keep, progressBar):
layer1 = self.getVectorLayerByName(inName) layer1 = ftools_utils.getVectorLayerByName(inName)
provider1 = layer1.dataProvider() provider1 = layer1.dataProvider()
allAttrs = provider1.attributeIndexes() allAttrs = provider1.attributeIndexes()
provider1.select(allAttrs) provider1.select(allAttrs)
fieldList1 = self.getFieldList(layer1).values() fieldList1 = ftools_utils.getFieldList(layer1).values()
layer2 = self.getVectorLayerByName(joinName) layer2 = ftools_utils.getVectorLayerByName(joinName)
provider2 = layer2.dataProvider() provider2 = layer2.dataProvider()
allAttrs = provider2.attributeIndexes() allAttrs = provider2.attributeIndexes()
provider2.select(allAttrs) provider2.select(allAttrs)
fieldList2 = self.getFieldList(layer2) fieldList2 = ftools_utils.getFieldList(layer2)
fieldList = [] fieldList = []
if provider1.crs() <> provider2.crs(): if provider1.crs() <> provider2.crs():
QMessageBox.warning(self, self.tr("CRS warning!"), self.tr("Warning: Input layers have non-matching CRS.\nThis may cause unexpected results.")) QMessageBox.warning(self, self.tr("CRS warning!"), self.tr("Warning: Input layers have non-matching CRS.\nThis may cause unexpected results."))
if not summary: if not summary:
fieldList2 = self.testForUniqueness(fieldList1, fieldList2.values()) fieldList2 = ftools_utils.testForUniqueness(fieldList1, fieldList2.values())
seq = range(0, len(fieldList1) + len(fieldList2)) seq = range(0, len(fieldList1) + len(fieldList2))
fieldList1.extend(fieldList2) fieldList1.extend(fieldList2)
fieldList1 = dict(zip(seq, fieldList1)) fieldList1 = dict(zip(seq, fieldList1))
@ -132,7 +131,7 @@ class Dialog(QDialog, Ui_Dialog):
fieldList.append(field) fieldList.append(field)
field = QgsField("COUNT", QVariant.Double, "real", 24, 16, self.tr("Summary field") ) field = QgsField("COUNT", QVariant.Double, "real", 24, 16, self.tr("Summary field") )
fieldList.append(field) fieldList.append(field)
fieldList2 = self.testForUniqueness(fieldList1, fieldList) fieldList2 = ftools_utils.testForUniqueness(fieldList1, fieldList)
fieldList1.extend(fieldList) fieldList1.extend(fieldList)
seq = range(0, len(fieldList1)) seq = range(0, len(fieldList1))
fieldList1 = dict(zip(seq, fieldList1)) fieldList1 = dict(zip(seq, fieldList1))
@ -147,13 +146,13 @@ class Dialog(QDialog, Ui_Dialog):
#writer = QgsVectorFileWriter(outName, "UTF-8", fieldList1, provider1.geometryType(), sRs) #writer = QgsVectorFileWriter(outName, "UTF-8", fieldList1, provider1.geometryType(), sRs)
inFeat = QgsFeature() inFeat = QgsFeature()
outFeat = QgsFeature() outFeat = QgsFeature()
joinFeat = QgsFeature() inFeatB = QgsFeature()
inGeom = QgsGeometry() inGeom = QgsGeometry()
progressBar.setValue(15) progressBar.setValue(15)
start = 15.00 start = 15.00
add = 85.00 / provider1.featureCount() add = 85.00 / provider1.featureCount()
provider1.rewind() provider1.rewind()
index = ftools_utils.createIndex(provider2)
while provider1.nextFeature(inFeat): while provider1.nextFeature(inFeat):
inGeom = inFeat.geometry() inGeom = inFeat.geometry()
atMap1 = inFeat.attributeMap() atMap1 = inFeat.attributeMap()
@ -162,24 +161,28 @@ class Dialog(QDialog, Ui_Dialog):
joinList = [] joinList = []
if inGeom.type() == QGis.Point: if inGeom.type() == QGis.Point:
#(check, joinList) = layer2.featuresInRectangle(inGeom.buffer(10,2).boundingBox(), True, True) #(check, joinList) = layer2.featuresInRectangle(inGeom.buffer(10,2).boundingBox(), True, True)
layer2.select(inGeom.buffer(10,2).boundingBox(), False) #layer2.select(inGeom.buffer(10,2).boundingBox(), False)
joinList = layer2.selectedFeatures() #joinList = layer2.selectedFeatures()
joinList = index.intersects( inGeom.buffer(10,2).boundingBox() )
if len(joinList) > 0: check = 0 if len(joinList) > 0: check = 0
else: check = 1 else: check = 1
else: else:
#(check, joinList) = layer2.featuresInRectangle(inGeom.boundingBox(), True, True) #(check, joinList) = layer2.featuresInRectangle(inGeom.boundingBox(), True, True)
layer2.select(inGeom.boundingBox(), False) #layer2.select(inGeom.boundingBox(), False)
joinList = layer2.selectedFeatures() #joinList = layer2.selectedFeatures()
joinList = index.intersects( inGeom.boundingBox() )
if len(joinList) > 0: check = 0 if len(joinList) > 0: check = 0
else: check = 1 else: check = 1
if check == 0: if check == 0:
count = 0 count = 0
for i in joinList: for i in joinList:
tempGeom = i.geometry() #tempGeom = i.geometry()
if inGeom.intersects(tempGeom): provider2.featureAtId(int(i), inFeatB , True, allAttrs)
tmpGeom = QgsGeometry( inFeatB.geometry() )
if inGeom.intersects(tmpGeom):
count = count + 1 count = count + 1
none = False none = False
atMap2 = i.attributeMap() atMap2 = inFeatB.attributeMap()
if not summary: if not summary:
atMap = atMap1.values() atMap = atMap1.values()
atMap2 = atMap2.values() atMap2 = atMap2.values()
@ -212,50 +215,3 @@ class Dialog(QDialog, Ui_Dialog):
start = start + add start = start + add
progressBar.setValue(start) progressBar.setValue(start)
del writer del writer
def testForUniqueness(self, fieldList1, fieldList2):
changed = True
while changed:
changed = False
for i in fieldList1:
for j in fieldList2:
if j.name() == i.name():
j = self.createUniqueFieldName(j)
changed = True
return fieldList2
def createUniqueFieldName(self, field):
check = field.name().right(2)
if check.startsWith("_"):
(val,test) = check.right(1).toInt()
if test:
if val < 2:
val = 2
else:
val = val + 1
field.setName(field.name().left(len(field.name())-1) + unicode(val))
else:
field.setName(field.name() + "_2")
else:
field.setName(field.name() + "_2")
return field
def getVectorLayerByName(self, myName):
mc = self.iface.mapCanvas()
nLayers = mc.layerCount()
for l in range(nLayers):
layer = mc.layer(l)
if layer.name() == unicode(myName):
vlayer = QgsVectorLayer(unicode(layer.source()), unicode(myName), unicode(layer.dataProvider().name()))
if vlayer.isValid():
return vlayer
else:
QMessageBox.information(self, self.tr("Spatial Join"), self.tr("Vector layer is not valid"))
def getFieldList(self, vlayer):
fProvider = vlayer.dataProvider()
feat = QgsFeature()
allAttrs = fProvider.attributeIndexes()
fProvider.select(allAttrs)
myFields = fProvider.fields()
return myFields