Skip to content
Permalink
Browse files

[processing] add shortest path algorithm to Processing

  • Loading branch information
alexbruy committed Dec 13, 2016
1 parent a8b9723 commit a60048c126b699c79cf7293ca99b9c48c8b23458
Showing with 120 additions and 0 deletions.
  1. +120 −0 python/plugins/processing/algs/qgis/ShortestPath.py
@@ -0,0 +1,120 @@
# -*- coding: utf-8 -*-

"""
***************************************************************************
ShortestPath.py
---------------------
Date : November 2016
Copyright : (C) 2016 by Alexander Bruy
Email : alexander dot bruy 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__ = 'Alexander Bruy'
__date__ = 'November 2016'
__copyright__ = '(C) 2016, Alexander Bruy'

# This will get replaced with a git SHA1 when you do a git archive

__revision__ = '$Format:%H$'

import os

from qgis.PyQt.QtGui import QIcon

from qgis.analysis import (QgsLineVectorLayerDirector,
QgsDistanceArcProperter,
QgsGraphBuilder,
QgsGraphAnalyzer
)

from processing.core.GeoAlgorithm import GeoAlgorithm
from processing.core.parameters import (ParameterVector,
ParameterPoint,
ParameterNumber,
ParameterString,
ParameterTableField,
ParameterSelection
)
from processing.core.outputs import (OutputNumber,
OutputVector
)
from processing.tools import dataobjects

pluginPath = os.path.split(os.path.split(os.path.dirname(__file__))[0])[0]


class ShortestPath(GeoAlgorithm):

INPUT_VECTOR = 'INPUT_VECTOR'
START_POINT = 'START_POINT'
END_POINT = 'END_POINT'
DIRECTION_FIELD = 'DIRECTION_FIELD'
VALUE_DIRECT = 'VALUE_DIRECT'
VALUE_REVERSED = 'VALUE_REVERSED'
VALUE_BIDIRECTIONAL = 'VALUE_BIDIRECTIONAL'
DEFAULT_DIRECTION = 'DEFAULT_DIRECTION'
PATH_LENGTH = 'PATH_LENGTH'
OUTPUT_LAYER = 'OUTPUT_LAYER'

def getIcon(self):
return QIcon(os.path.join(pluginPath, 'images', 'networkanalysis.png'))

def defineCharacteristics(self):
self.DIRECTIONS = {1: self.tr('Direct direction'),
2: self.tr('Inverse direction'),
3: self.tr('Bidirectional')
}

self.UNITS = {0: self.tr('Meters'),
1: self.tr('Kilometers')
}

self.name, self.i18n_name = self.trAlgorithm('Shortest path')
self.group, self.i18n_group = self.trAlgorithm('Network analysis')

self.addParameter(ParameterVector(self.INPUT_VECTOR,
self.tr('Vector layer representing road network'),
[dataobjects.TYPE_VECTOR_LINE]))
self.addParameter(ParameterPoint(self.START_POINT,
self.tr('Start point')))
self.addParameter(ParameterPoint(self.END_POINT,
self.tr('End point')))

self.addParameter(ParameterTableField(self.DIRECTION_FIELD,
self.tr('Road direction field'),
self.INPUT_VECTOR))
self.addParameter(ParameterString(self.VALUE_DIRECT,
self.tr('Value for direct direction road'),
''))
self.addParameter(ParameterString(self.VALUE_REVERSED,
self.tr('Value for reversed direction road'),
''))
self.addParameter(ParameterString(self.VALUE_BIDIRECTIONAL,
self.tr('Value for bidirectional road'),
''))
self.addParameter(ParameterSelection(self.DEFAULT_DIRECTION,
self.tr('Default road direction'),
list(self.DIRECTIONS.keys())))

self.addOutput(OutputNumber(self.PATH_LENGTH,
self.tr('Path length')))
self.addOutput(OutputVector(self.OUTPUT_LAYER,
self.tr('Shortest path'),
datatype=[dataobjects.TYPE_VECTOR_LINE]))

def processAlgorithm(self, progress):
layer = dataobjects.getObjectFromUri(
self.getParameterValue(self.INPUT_VECTOR))
fielName = self.getParameterValue(self.RASTER_BAND)
columnPrefix = self.getParameterValue(self.COLUMN_PREFIX)
st = self.getParameterValue(self.STATISTICS)

vectorLayer = dataobjects.getObjectFromUri(vectorPath)

0 comments on commit a60048c

Please sign in to comment.
You can’t perform that action at this time.