mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-25 00:58:06 -05:00
251 lines
9.3 KiB
Python
251 lines
9.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
#-----------------------------------------------------------
|
|
#
|
|
# fTools
|
|
# Copyright (C) 2008-2011 Carson Farmer
|
|
# EMAIL: carson.farmer (at) gmail.com
|
|
# WEB : http://www.ftools.ca/fTools.html
|
|
#
|
|
# A collection of data management and analysis tools for vector data
|
|
#
|
|
#-----------------------------------------------------------
|
|
#
|
|
# licensed under the terms of GNU GPL 2
|
|
#
|
|
# 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.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program; if not, write to the Free Software Foundation, Inc.,
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
#
|
|
#---------------------------------------------------------------------
|
|
|
|
from PyQt4.QtCore import *
|
|
from PyQt4.QtGui import *
|
|
import ftools_utils
|
|
from qgis.core import *
|
|
from ui_frmPointsInPolygon import Ui_Dialog
|
|
|
|
class Dialog(QDialog, Ui_Dialog):
|
|
|
|
def __init__(self, iface):
|
|
QDialog.__init__(self, iface.mainWindow())
|
|
self.iface = iface
|
|
self.setupUi(self)
|
|
|
|
self.setWindowTitle(self.tr("Count Points in Polygon"))
|
|
self.btnOk = self.buttonBox.button( QDialogButtonBox.Ok )
|
|
self.btnClose = self.buttonBox.button( QDialogButtonBox.Close )
|
|
|
|
QObject.connect(self.toolOut, SIGNAL("clicked()"), self.outFile)
|
|
self.progressBar.setValue(0)
|
|
self.populateLayers()
|
|
|
|
def populateLayers( self ):
|
|
layers = ftools_utils.getLayerNames([QGis.Polygon])
|
|
self.inPolygon.clear()
|
|
self.inPolygon.addItems(layers)
|
|
|
|
self.inPoint.clear()
|
|
layers = ftools_utils.getLayerNames([QGis.Point])
|
|
self.inPoint.addItems(layers)
|
|
|
|
def outFile(self):
|
|
self.outShape.clear()
|
|
(self.shapefileName, self.encoding) = ftools_utils.saveDialog(self)
|
|
if self.shapefileName is None or self.encoding is None:
|
|
return
|
|
self.outShape.setText(QString(self.shapefileName))
|
|
|
|
def accept(self):
|
|
if self.inPolygon.currentText() == "":
|
|
QMessageBox.information(self, self.tr("Count Points In Polygon"),
|
|
self.tr("Please specify input polygon vector layer"))
|
|
elif self.inPoint.currentText() == "":
|
|
QMessageBox.information(self, self.tr("Count Points In Polygon"),
|
|
self.tr("Please specify input point vector layer"))
|
|
elif self.lnField.text() == "":
|
|
QMessageBox.information(self, self.tr("Count Points In Polygon"),
|
|
self.tr("Please specify output count field"))
|
|
elif self.outShape.text() == "":
|
|
QMessageBox.information(self, self.tr("Count Points In Polygon"),
|
|
self.tr("Please specify output shapefile"))
|
|
else:
|
|
inPoly = ftools_utils.getVectorLayerByName(self.inPolygon.currentText())
|
|
inPnts = ftools_utils.getVectorLayerByName(self.inPoint.currentText())
|
|
|
|
polyProvider = inPoly.dataProvider()
|
|
pointProvider = inPnts.dataProvider()
|
|
if polyProvider.crs() <> pointProvider.crs():
|
|
QMessageBox.warning(self, self.tr("CRS warning!"),
|
|
self.tr("Warning: Input layers have non-matching CRS.\nThis may cause unexpected results."))
|
|
|
|
self.btnOk.setEnabled(False)
|
|
|
|
self.workThread = PointsInPolygonThread(inPoly, inPnts, self.lnField.text(), self.outShape.text(), self.encoding)
|
|
|
|
QObject.connect(self.workThread, SIGNAL("rangeChanged(int)"), self.setProgressRange)
|
|
QObject.connect(self.workThread, SIGNAL("updateProgress()"), self.updateProgress)
|
|
QObject.connect(self.workThread, SIGNAL("processingFinished()"), self.processFinished)
|
|
QObject.connect(self.workThread, SIGNAL("processingInterrupted()"), self.processInterrupted)
|
|
|
|
self.btnClose.setText(self.tr("Cancel"))
|
|
QObject.disconnect(self.buttonBox, SIGNAL("rejected()"), self.reject)
|
|
QObject.connect(self.btnClose, SIGNAL("clicked()"), self.stopProcessing)
|
|
|
|
self.workThread.start()
|
|
|
|
def setProgressRange(self, maxValue):
|
|
self.progressBar.setRange(0, maxValue)
|
|
self.progressBar.setValue(0)
|
|
|
|
def updateProgress(self):
|
|
self.progressBar.setValue(self.progressBar.value() + 1)
|
|
|
|
def processFinished(self):
|
|
self.stopProcessing()
|
|
|
|
addToTOC = QMessageBox.question(self, self.tr("Count Points in Polygon"),
|
|
self.tr("Created output shapefile:\n%1\n\nWould you like to add the new layer to the TOC?").arg(self.outShape.text()),
|
|
QMessageBox.Yes, QMessageBox.No, QMessageBox.NoButton)
|
|
if addToTOC == QMessageBox.Yes:
|
|
fileInfo = QFileInfo( self.outShape.text() )
|
|
layerName = fileInfo.completeBaseName()
|
|
layer = QgsVectorLayer(self.outShape.text(), layerName, "ogr")
|
|
QgsMapLayerRegistry.instance().addMapLayers([layer])
|
|
self.populateLayers()
|
|
|
|
self.restoreGui()
|
|
|
|
def processInterrupted(self):
|
|
self.restoreGui()
|
|
|
|
def stopProcessing(self):
|
|
if self.workThread != None:
|
|
self.workThread.stop()
|
|
self.workThread = None
|
|
|
|
def restoreGui(self):
|
|
self.progressBar.setRange(0, 1)
|
|
self.progressBar.setValue(0)
|
|
self.outShape.clear()
|
|
|
|
QObject.disconnect(self.btnClose, SIGNAL("clicked()"), self.stopProcessing)
|
|
QObject.connect(self.buttonBox, SIGNAL("rejected()"), self.reject)
|
|
self.btnClose.setText(self.tr("Close"))
|
|
self.btnOk.setEnabled(True)
|
|
|
|
class PointsInPolygonThread(QThread):
|
|
def __init__( self, inPoly, inPoints, fieldName, outPath, encoding ):
|
|
QThread.__init__( self, QThread.currentThread() )
|
|
self.mutex = QMutex()
|
|
self.stopMe = 0
|
|
self.interrupted = False
|
|
|
|
self.layerPoly = inPoly
|
|
self.layerPoints = inPoints
|
|
self.fieldName = fieldName
|
|
self.outPath = outPath
|
|
self.encoding = encoding
|
|
|
|
def run(self):
|
|
self.mutex.lock()
|
|
self.stopMe = 0
|
|
self.mutex.unlock()
|
|
|
|
interrupted = False
|
|
|
|
polyProvider = self.layerPoly.dataProvider()
|
|
pointProvider = self.layerPoints.dataProvider()
|
|
|
|
allAttrs = polyProvider.attributeIndexes()
|
|
polyProvider.select(allAttrs)
|
|
|
|
fieldList = ftools_utils.getFieldList(self.layerPoly)
|
|
index = polyProvider.fieldNameIndex(unicode(self.fieldName))
|
|
if index == -1:
|
|
index = polyProvider.fieldCount()
|
|
field = QgsField(unicode(self.fieldName), QVariant.Double, "real", 24, 15, self.tr("point count field"))
|
|
fieldList[index] = field
|
|
|
|
sRs = polyProvider.crs()
|
|
if QFile(self.outPath).exists():
|
|
if not QgsVectorFileWriter.deleteShapeFile(self.outPath):
|
|
return
|
|
|
|
writer = QgsVectorFileWriter(self.outPath, self.encoding, fieldList,
|
|
polyProvider.geometryType(), sRs)
|
|
|
|
spatialIndex = ftools_utils.createIndex( pointProvider )
|
|
pointProvider.rewind()
|
|
pointProvider.select()
|
|
|
|
self.emit(SIGNAL("rangeChanged(int)"), polyProvider.featureCount() )
|
|
|
|
polyFeat = QgsFeature()
|
|
pntFeat = QgsFeature()
|
|
outFeat = QgsFeature()
|
|
inGeom = QgsGeometry()
|
|
while polyProvider.nextFeature(polyFeat):
|
|
inGeom = polyFeat.geometry()
|
|
atMap = polyFeat.attributeMap()
|
|
outFeat.setAttributeMap(atMap)
|
|
outFeat.setGeometry(inGeom)
|
|
|
|
count = 0
|
|
pointList = []
|
|
hasIntersection = True
|
|
pointList = spatialIndex.intersects(inGeom.boundingBox())
|
|
if len(pointList) > 0:
|
|
hasIntersection = True
|
|
else:
|
|
hasIntersection = False
|
|
|
|
if hasIntersection:
|
|
for p in pointList:
|
|
pointProvider.featureAtId(p, pntFeat , True)
|
|
tmpGeom = QgsGeometry(pntFeat.geometry())
|
|
if inGeom.intersects(tmpGeom):
|
|
count += 1
|
|
|
|
self.mutex.lock()
|
|
s = self.stopMe
|
|
self.mutex.unlock()
|
|
if s == 1:
|
|
interrupted = True
|
|
break
|
|
|
|
outFeat.addAttribute(index, QVariant(count))
|
|
writer.addFeature(outFeat)
|
|
|
|
self.emit( SIGNAL( "updateProgress()" ) )
|
|
|
|
self.mutex.lock()
|
|
s = self.stopMe
|
|
self.mutex.unlock()
|
|
if s == 1:
|
|
interrupted = True
|
|
break
|
|
|
|
del writer
|
|
|
|
if not interrupted:
|
|
self.emit( SIGNAL( "processingFinished()" ) )
|
|
else:
|
|
self.emit( SIGNAL( "processingInterrupted()" ) )
|
|
|
|
def stop(self):
|
|
self.mutex.lock()
|
|
self.stopMe = 1
|
|
self.mutex.unlock()
|
|
|
|
QThread.wait( self )
|