mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-25 00:58:06 -05:00
266 lines
11 KiB
Python
266 lines
11 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 *
|
|
|
|
from qgis.core import *
|
|
from random import *
|
|
import math, ftools_utils
|
|
from ui_frmRandPoints import Ui_Dialog
|
|
|
|
class Dialog(QDialog, Ui_Dialog):
|
|
def __init__(self, iface):
|
|
QDialog.__init__(self, iface.mainWindow())
|
|
self.iface = iface
|
|
self.setupUi(self)
|
|
QObject.connect(self.toolOut, SIGNAL("clicked()"), self.outFile)
|
|
QObject.connect(self.inShape, SIGNAL("currentIndexChanged(QString)"), self.update)
|
|
self.progressBar.setValue(0)
|
|
self.setWindowTitle(self.tr("Random Points"))
|
|
self.buttonOk = self.buttonBox_2.button( QDialogButtonBox.Ok )
|
|
self.populateLayers()
|
|
|
|
def populateLayers( self ):
|
|
layers = ftools_utils.getLayerNames([QGis.Polygon, "Raster"])
|
|
self.inShape.blockSignals(True)
|
|
self.inShape.clear()
|
|
self.inShape.blockSignals(False)
|
|
self.inShape.addItems(layers)
|
|
|
|
# If input layer is changed, update field list
|
|
def update(self, inputLayer):
|
|
self.cmbField.clear()
|
|
changedLayer = ftools_utils.getMapLayerByName(unicode(inputLayer))
|
|
if changedLayer.type() == changedLayer.VectorLayer:
|
|
self.rdoStratified.setEnabled(True)
|
|
self.rdoDensity.setEnabled(True)
|
|
self.rdoField.setEnabled(True)
|
|
self.label_4.setEnabled(True)
|
|
changedLayer = ftools_utils.getVectorLayerByName(inputLayer)
|
|
changedFields = ftools_utils.getFieldList(changedLayer)
|
|
for f in changedFields:
|
|
if f.typeName() == "Integer":
|
|
self.cmbField.addItem(unicode(f.name()))
|
|
else:
|
|
self.rdoUnstratified.setChecked(True)
|
|
self.rdoStratified.setEnabled(False)
|
|
self.rdoDensity.setEnabled(False)
|
|
self.rdoField.setEnabled(False)
|
|
self.spnStratified.setEnabled(False)
|
|
self.spnDensity.setEnabled(False)
|
|
self.cmbField.setEnabled(False)
|
|
self.label_4.setEnabled(False)
|
|
|
|
# when 'OK' button is pressed, gather required inputs, and initiate random points generation
|
|
def accept(self):
|
|
self.buttonOk.setEnabled( False )
|
|
if self.inShape.currentText() == "":
|
|
QMessageBox.information(self, self.tr("Random Points"), self.tr("No input layer specified"))
|
|
elif self.outShape.text() == "":
|
|
QMessageBox.information(self, self.tr("Random Points"), self.tr("Please specify output shapefile"))
|
|
else:
|
|
inName = self.inShape.currentText()
|
|
self.progressBar.setValue(1)
|
|
outPath = self.outShape.text()
|
|
self.progressBar.setValue(2.5)
|
|
if outPath.contains("\\"):
|
|
outName = outPath.right((outPath.length() - outPath.lastIndexOf("\\")) - 1)
|
|
else:
|
|
outName = outPath.right((outPath.length() - outPath.lastIndexOf("/")) - 1)
|
|
if outName.endsWith(".shp"):
|
|
outName = outName.left(outName.length() - 4)
|
|
self.progressBar.setValue(5)
|
|
mLayer = ftools_utils.getMapLayerByName(unicode(inName))
|
|
if mLayer.type() == mLayer.VectorLayer:
|
|
inLayer = ftools_utils.getVectorLayerByName(unicode(inName))
|
|
if self.rdoUnstratified.isChecked():
|
|
design = self.tr("unstratified")
|
|
value = self.spnUnstratified.value()
|
|
elif self.rdoStratified.isChecked():
|
|
design = self.tr("stratified")
|
|
value = self.spnStratified.value()
|
|
elif self.rdoDensity.isChecked():
|
|
design = self.tr("density")
|
|
value = self.spnDensity.value()
|
|
else:
|
|
design = self.tr("field")
|
|
value = unicode(self.cmbField.currentText())
|
|
elif mLayer.type() == mLayer.RasterLayer:
|
|
inLayer = ftools_utils.getRasterLayerByName(unicode(inName))
|
|
design = self.tr("unstratified")
|
|
value = self.spnUnstratified.value()
|
|
else:
|
|
QMessageBox.information(self, self.tr("Random Points"), self.tr("Unknown layer type..."))
|
|
minimum = 0.00
|
|
self.progressBar.setValue(10)
|
|
self.randomize(inLayer, outPath, minimum, design, value)
|
|
self.progressBar.setValue(100)
|
|
self.outShape.clear()
|
|
addToTOC = QMessageBox.question(self, self.tr("Random Points"),
|
|
self.tr("Created output point shapefile:\n%1\n\nWould you like to add the new layer to the TOC?").arg(outPath), QMessageBox.Yes, QMessageBox.No, QMessageBox.NoButton)
|
|
if addToTOC == QMessageBox.Yes:
|
|
self.vlayer = QgsVectorLayer(outPath, unicode(outName), "ogr")
|
|
QgsMapLayerRegistry.instance().addMapLayers([self.vlayer])
|
|
self.populateLayers()
|
|
self.progressBar.setValue(0)
|
|
self.buttonOk.setEnabled( True )
|
|
|
|
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 ) )
|
|
|
|
# combine all polygons in layer to create single polygon (slow for complex polygons)
|
|
def createSinglePolygon(self, vlayer):
|
|
provider = vlayer.dataProvider()
|
|
feat = QgsFeature()
|
|
geom = QgsGeometry()
|
|
fit = provider.getFeatures()
|
|
fit.nextFeature(feat)
|
|
geom = QgsGeometry(feat.geometry())
|
|
count = 10.00
|
|
add = ( 40.00 - 10.00 ) / provider.featureCount()
|
|
while fit.nextFeature(feat):
|
|
geom = geom.combine(QgsGeometry( feat.geometry() ))
|
|
count = count + add
|
|
self.progressBar.setValue(count)
|
|
return geom
|
|
|
|
# Generate list of random points
|
|
def simpleRandom(self, n, bound, xmin, xmax, ymin, ymax):
|
|
seed()
|
|
points = []
|
|
i = 1
|
|
count = 40.00
|
|
if n == 0:
|
|
return []
|
|
add = ( 70.00 - 40.00 ) / n
|
|
while i <= n:
|
|
pGeom = QgsGeometry().fromPoint(QgsPoint(xmin + (xmax-xmin) * random(), ymin + (ymax-ymin) * random()))
|
|
if pGeom.intersects(bound):
|
|
points.append(pGeom)
|
|
i = i + 1
|
|
count = count + add
|
|
self.progressBar.setValue(count)
|
|
return points
|
|
|
|
def vectorRandom(self, n, layer, xmin, xmax, ymin, ymax):
|
|
provider = layer.dataProvider()
|
|
index = ftools_utils.createIndex(provider)
|
|
seed()
|
|
points = []
|
|
feat = QgsFeature()
|
|
i = 1
|
|
count = 40.00
|
|
add = ( 70.00 - 40.00 ) / n
|
|
while i <= n:
|
|
point = QgsPoint(xmin + (xmax-xmin) * random(), ymin + (ymax-ymin) * random())
|
|
pGeom = QgsGeometry().fromPoint(point)
|
|
ids = index.intersects(pGeom.buffer(5,5).boundingBox())
|
|
for id in ids:
|
|
provider.getFeatures( QgsFeatureRequest().setFilterFid( int(id) ) ).nextFeature( feat )
|
|
tGeom = QgsGeometry(feat.geometry())
|
|
if pGeom.intersects(tGeom):
|
|
points.append(pGeom)
|
|
i = i + 1
|
|
count = count + add
|
|
self.progressBar.setValue(count)
|
|
break
|
|
return points
|
|
|
|
def randomize(self, inLayer, outPath, minimum, design, value):
|
|
outFeat = QgsFeature()
|
|
outFeat.initAttributes(1)
|
|
if design == self.tr("unstratified"):
|
|
ext = inLayer.extent()
|
|
if inLayer.type() == inLayer.RasterLayer:
|
|
points = self.simpleRandom(int(value), ext, ext.xMinimum(),
|
|
ext.xMaximum(), ext.yMinimum(), ext.yMaximum())
|
|
else:
|
|
points = self.vectorRandom(int(value), inLayer,
|
|
ext.xMinimum(), ext.xMaximum(), ext.yMinimum(), ext.yMaximum())
|
|
else: points = self.loopThruPolygons(inLayer, value, design)
|
|
crs = self.iface.mapCanvas().mapRenderer().destinationCrs()
|
|
if not crs.isValid(): crs = None
|
|
fields = QgsFields()
|
|
fields.append( QgsField("ID", QVariant.Int) )
|
|
outFeat.setFields(fields)
|
|
check = QFile(self.shapefileName)
|
|
if check.exists():
|
|
if not QgsVectorFileWriter.deleteShapeFile(self.shapefileName):
|
|
return
|
|
writer = QgsVectorFileWriter(self.shapefileName, self.encoding, fields, QGis.WKBPoint, crs)
|
|
idVar = 0
|
|
count = 70.00
|
|
add = ( 100.00 - 70.00 ) / len(points)
|
|
for i in points:
|
|
outFeat.setGeometry(i)
|
|
outFeat.setAttribute(0, QVariant(idVar))
|
|
writer.addFeature(outFeat)
|
|
idVar = idVar + 1
|
|
count = count + add
|
|
self.progressBar.setValue(count)
|
|
del writer
|
|
|
|
#
|
|
def loopThruPolygons(self, inLayer, numRand, design):
|
|
sProvider = inLayer.dataProvider()
|
|
sFeat = QgsFeature()
|
|
sGeom = QgsGeometry()
|
|
sPoints = []
|
|
if design == self.tr("field"):
|
|
i = 0
|
|
for attr in sProvider.fields():
|
|
if (unicode(numRand) == attr.name()):
|
|
index = i #get input field index
|
|
break
|
|
i += 1
|
|
count = 10.00
|
|
add = 60.00 / sProvider.featureCount()
|
|
sFit = sProvider.getFeatures()
|
|
while sFit.nextFeature(sFeat):
|
|
sGeom = sFeat.geometry()
|
|
if design == self.tr("density"):
|
|
sDistArea = QgsDistanceArea()
|
|
value = int(round(numRand * sDistArea.measure(sGeom)))
|
|
elif design == self.tr("field"):
|
|
sAtMap = sFeat.attributes()
|
|
value = sAtMap[index].toInt()[0]
|
|
else:
|
|
value = numRand
|
|
sExt = sGeom.boundingBox()
|
|
sPoints.extend(self.simpleRandom(value, sGeom, sExt.xMinimum(), sExt.xMaximum(), sExt.yMinimum(), sExt.yMaximum()))
|
|
count = count + add
|
|
self.progressBar.setValue(count)
|
|
return sPoints
|