refactor Hub lines alg

qgis · Oct 3, 2014 · b63a5d3 · b63a5d3
1 parent f614435
commit b63a5d3
Show file tree

Hide file tree

Showing 2 changed files with 109 additions and 0 deletions.
diff --git a/python/plugins/processing/algs/qgis/QGISAlgorithmProvider.py b/python/plugins/processing/algs/qgis/QGISAlgorithmProvider.py
@@ -81,6 +81,7 @@
 from mmqgisx.Grid import Grid
 from mmqgisx.Gridify import Gridify
 from mmqgisx.HubDistance import HubDistance
+from mmqgisx.HubLines import HubLines
 
 from ConcaveHull import ConcaveHull
 from Polygonize import Polygonize
@@ -148,6 +149,7 @@ def __init__(self):
                         DeleteColumn(), DeleteDuplicateGeometries(),
                         TextToFloat(), ExtractByAttribute(),
                         SelectByAttribute(), Grid(), Gridify(), HubDistance(),
+                        HubLines(),
                         #mmqgisx_delete_duplicate_geometries_algorithm(),
                         #mmqgisx_geometry_convert_algorithm(),
                         #mmqgisx_grid_algorithm(),

diff --git a/python/plugins/processing/algs/qgis/mmqgisx/HubLines.py b/python/plugins/processing/algs/qgis/mmqgisx/HubLines.py
@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+
+"""
+***************************************************************************
+    Gridify.py
+    ---------------------
+    Date                 : May 2010
+    Copyright            : (C) 2010 by Michael Minn
+    Email                : pyqgis at michaelminn 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__ = 'Michael Minn'
+__date__ = 'May 2010'
+__copyright__ = '(C) 2010, Michael Minn'
+
+# 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.core.GeoAlgorithmExecutionException import \
+        GeoAlgorithmExecutionException
+from processing.core.parameters import ParameterVector
+from processing.core.parameters import ParameterTableField
+from processing.core.parameters import ParameterSelection
+from processing.core.outputs import OutputVector
+
+from processing.tools import dataobjects, vector
+
+class HubLines(GeoAlgorithm):
+    HUBS = 'HUBS'
+    HUB_FIELD = 'HUB_FIELD'
+    SPOKES = 'SPOKES'
+    SPOKE_FIELD = 'SPOKE_FIELD'
+    OUTPUT = 'OUTPUT'
+
+    def defineCharacteristics(self):
+        self.name = 'Hub lines'
+        self.group = 'Vector analysis tools'
+
+        self.addParameter(ParameterVector(self.HUBS,
+            'Hub point layer', [ParameterVector.VECTOR_TYPE_ANY]))
+        self.addParameter(ParameterTableField(
+            self.HUB_FIELD, 'Hub ID field', self.HUBS))
+        self.addParameter(ParameterVector(self.SPOKES,
+            'Spoke point layer', [ParameterVector.VECTOR_TYPE_ANY]))
+        self.addParameter(ParameterTableField(
+            self.SPOKE_FIELD, 'Spoke ID field', self.SPOKES))
+
+        self.addOutput(OutputVector(self.OUTPUT, 'Output'))
+
+    def processAlgorithm(self, progress):
+        layerHub = dataobjects.getObjectFromUri(
+            self.getParameterValue(self.HUBS))
+        layerSpoke = dataobjects.getObjectFromUri(
+            self.getParameterValue(self.SPOKES))
+
+        fieldHub = self.getParameterValue(self.HUB_FIELD)
+        fieldSpoke = self.getParameterValue(self.SPOKE_FIELD)
+
+        if layerHub.source() == layerSpoke.source():
+            raise GeoAlgorithmExecutionException(
+                'Same layer given for both hubs and spokes')
+
+        writer = self.getOutputFromName(self.OUTPUT).getVectorWriter(
+            layerSpoke.pendingFields(), QGis.WKBLineString, layerSpoke.crs())
+
+        spokes = vector.features(layerSpoke)
+        hubs = vector.features(layerHub)
+
+        count = len(spokes)
+        total = 100.0 / float(count)
+
+        for count, spokepoint in enumerate(spokes):
+            p = spokepoint.geometry().boundingBox().center()
+            spokeX = p.x()
+            spokeY = p.y()
+            spokeId = unicode(spokepoint[fieldSpoke])
+
+            for hubpoint in hubs:
+                hubId = unicode(hubpoint[fieldHub])
+                if hubId == spokeId:
+                    p = hubpoint.geometry().boundingBox().center()
+                    hubX = p.x()
+                    hubY = p.y()
+
+                    f = QgsFeature()
+                    f.setAttributes(spokepoint.attributes())
+                    f.setGeometry(QgsGeometry.fromPolyline(
+                        [QgsPoint(spokeX, spokeY), QgsPoint(hubX, hubY)]))
+                    writer.addFeature(f)
+
+                    break
+
+            progress.setPercentage(int(count * total))
+
+        del writer