Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
[processing] add gdal TRI algorithm (addresses #7201)
  • Loading branch information
alexbruy committed Oct 13, 2013
1 parent bcad38b commit 9288a5d
Show file tree
Hide file tree
Showing 2 changed files with 75 additions and 1 deletion.
3 changes: 2 additions & 1 deletion python/plugins/processing/gdal/GdalOgrAlgorithmProvider.py
Expand Up @@ -57,6 +57,7 @@
from processing.gdal.hillshade import hillshade
from processing.gdal.slope import slope
from processing.gdal.aspect import aspect
from processing.gdal.tri import tri

from processing.gdal.ogr2ogr import Ogr2Ogr
from processing.gdal.ogrinfo import OgrInfo
Expand Down Expand Up @@ -108,7 +109,7 @@ def createAlgsList(self):
rgb2pct(), pct2rgb(), merge(), polygonize(), gdaladdo(),
ClipByExtent(), ClipByMask(), contour(), rasterize(), proximity(),
sieve(), fillnodata(), ExtractProjection(), gdal2xyz(),
hillshade(), slope(), aspect(),
hillshade(), slope(), aspect(), tri(),
# ----- OGR tools -----
OgrInfo(), Ogr2Ogr(), OgrSql(),
]
Expand Down
73 changes: 73 additions & 0 deletions python/plugins/processing/gdal/tri.py
@@ -0,0 +1,73 @@
# -*- coding: utf-8 -*-

"""
***************************************************************************
tri.py
---------------------
Date : October 2013
Copyright : (C) 2013 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__ = 'October 2013'
__copyright__ = '(C) 2013, Alexander Bruy'

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

__revision__ = '$Format:%H$'


from PyQt4.QtGui import *

from processing.core.GeoAlgorithm import GeoAlgorithm
from processing.parameters.ParameterRaster import ParameterRaster
from processing.parameters.ParameterBoolean import ParameterBoolean
from processing.parameters.ParameterNumber import ParameterNumber
from processing.outputs.OutputRaster import OutputRaster
from processing.gdal.GdalUtils import GdalUtils
from processing.tools.system import *


class tri(GeoAlgorithm):

INPUT = 'INPUT'
BAND = 'BAND'
COMPUTE_EDGES = 'COMPUTE_EDGES'
OUTPUT = 'OUTPUT'

#def getIcon(self):
# filepath = os.path.dirname(__file__) + '/icons/dem.png'
# return QIcon(filepath)

def defineCharacteristics(self):
self.name = 'TRI (Terrain Ruggedness Index)'
self.group = '[GDAL] Analysis'
self.addParameter(ParameterRaster(self.INPUT, 'Input layer'))
self.addParameter(ParameterNumber(self.BAND, 'Band number', 1, 99, 1))
self.addParameter(ParameterBoolean(self.COMPUTE_EDGES, 'Compute edges',
False))

self.addOutput(OutputRaster(self.OUTPUT, 'Output file'))

def processAlgorithm(self, progress):
arguments = ['TRI']
arguments.append(unicode(self.getParameterValue(self.INPUT)))
arguments.append(unicode(self.getOutputValue(self.OUTPUT)))

arguments.append('-b')
arguments.append(str(self.getParameterValue(self.BAND)))

if self.getParameterValue(self.COMPUTE_EDGES):
arguments.append('-compute_edges')

GdalUtils.runGdal(['gdaldem',
GdalUtils.escapeAndJoin(arguments)], progress)

0 comments on commit 9288a5d

Please sign in to comment.