QGIS/python/plugins/sextante/algs/ftools/SelectByLocation.py

96 lines
4.0 KiB
Python
Raw Normal View History

2012-10-04 19:33:47 +02:00
# -*- coding: utf-8 -*-
"""
***************************************************************************
SelectByLocation.py
---------------------
Date : August 2012
Copyright : (C) 2012 by Victor Olaya
Email : volayaf at gmail dot com
***************************************************************************
* *
* 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. *
* *
***************************************************************************
"""
__author__ = 'Victor Olaya'
__date__ = 'August 2012'
__copyright__ = '(C) 2012, Victor Olaya'
# This will get replaced with a git SHA1 when you do a git archive
__revision__ = '$Format:%H$'
2012-09-15 18:25:25 +03:00
from PyQt4.QtCore import *
from qgis.core import *
2012-10-03 19:03:39 +03:00
from sextante.core.GeoAlgorithm import GeoAlgorithm
2012-09-15 18:25:25 +03:00
from sextante.core.QGisLayers import QGisLayers
from sextante.parameters.ParameterSelection import ParameterSelection
2012-10-03 19:03:39 +03:00
from sextante.parameters.ParameterVector import ParameterVector
2013-02-03 10:26:43 +01:00
2012-10-03 19:03:39 +03:00
from sextante.outputs.OutputVector import OutputVector
from sextante.algs.ftools import FToolsUtils as utils
2012-09-15 18:25:25 +03:00
class SelectByLocation(GeoAlgorithm):
INPUT = "INPUT"
INTERSECT = "INTERSECT"
2013-02-28 22:08:32 +01:00
METHOD = "METHOD"
2012-09-15 18:25:25 +03:00
OUTPUT = "OUTPUT"
2012-10-03 19:03:39 +03:00
METHODS = ["creating new selection",
"adding to current selection",
"removing from current selection"]
#===========================================================================
# def getIcon(self):
# return QtGui.QIcon(os.path.dirname(__file__) + "/icons/select_location.png")
#===========================================================================
2012-09-15 18:25:25 +03:00
2012-10-03 19:03:39 +03:00
def defineCharacteristics(self):
self.name = "Select by location"
self.group = "Vector selection tools"
2012-10-03 19:03:39 +03:00
self.addParameter(ParameterVector(self.INPUT, "Layer to select from", ParameterVector.VECTOR_TYPE_ANY))
self.addParameter(ParameterVector(self.INTERSECT, "Additional layer (intersection layer)", ParameterVector.VECTOR_TYPE_ANY))
2013-02-28 22:08:32 +01:00
self.addParameter(ParameterSelection(self.METHOD, "Modify current selection by", self.METHODS, 0))
2012-10-03 19:03:39 +03:00
self.addOutput(OutputVector(self.OUTPUT, "Selection", True))
2012-09-15 18:25:25 +03:00
def processAlgorithm(self, progress):
2012-10-03 19:03:39 +03:00
filename = self.getParameterValue(self.INPUT)
inputLayer = QGisLayers.getObjectFromUri(filename)
2012-09-15 18:25:25 +03:00
method = self.getParameterValue(self.METHOD)
2012-10-03 19:03:39 +03:00
filename = self.getParameterValue(self.INTERSECT)
2012-09-15 18:25:25 +03:00
selectLayer = QGisLayers.getObjectFromUri(filename)
2013-02-03 10:26:43 +01:00
2013-02-28 22:08:32 +01:00
oldSelection = set(inputLayer.selectedFeaturesIds())
index = spatialIndex = utils.createSpatialIndex(inputLayer)
2013-02-28 22:08:32 +01:00
feat = QgsFeature()
2012-09-15 18:25:25 +03:00
geom = QgsGeometry()
2013-02-28 22:08:32 +01:00
selectedSet = []
current = 0
features = QGisLayers.features(selectLayer)
total = 100.0 / float(len(features))
for f in features:
geom = QgsGeometry(f.geometry())
intersects = index.intersects(geom.boundingBox())
for i in intersects:
request = QgsFeatureRequest().setFilterFid(i)
feat = inputLayer.getFeatures(request).next()
tmpGeom = QgsGeometry(feat.geometry())
if geom.intersects(tmpGeom):
selectedSet.append(feat.id())
current += 1
progress.setPercentage(int(current * total))
2012-10-03 19:03:39 +03:00
2012-09-15 18:25:25 +03:00
if method == 1:
2013-02-03 10:26:43 +01:00
selectedSet = list(oldSelection.union(selectedSet))
2012-09-15 18:25:25 +03:00
elif method == 2:
2013-02-03 10:26:43 +01:00
selectedSet = list(oldSelection.difference(selectedSet))
2012-10-03 19:03:39 +03:00
2012-09-15 18:25:25 +03:00
inputLayer.setSelectedFeatures(selectedSet)
self.setOutputValue(self.OUTPUT, filename)