| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105 |
- """
- ***************************************************************************
- tpi.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'
- import os
- from qgis.core import (QgsProcessingException,
- QgsProcessingParameterDefinition,
- QgsProcessingParameterRasterLayer,
- QgsProcessingParameterBand,
- QgsProcessingParameterString,
- QgsProcessingParameterBoolean,
- QgsProcessingParameterRasterDestination)
- from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
- from processing.algs.gdal.GdalUtils import GdalUtils
- pluginPath = os.path.split(os.path.split(os.path.dirname(__file__))[0])[0]
- class tpi(GdalAlgorithm):
- INPUT = 'INPUT'
- BAND = 'BAND'
- COMPUTE_EDGES = 'COMPUTE_EDGES'
- OPTIONS = 'OPTIONS'
- OUTPUT = 'OUTPUT'
- def __init__(self):
- super().__init__()
- def initAlgorithm(self, config=None):
- self.addParameter(QgsProcessingParameterRasterLayer(self.INPUT, self.tr('Input layer')))
- self.addParameter(QgsProcessingParameterBand(self.BAND,
- self.tr('Band number'),
- 1,
- parentLayerParameterName=self.INPUT))
- self.addParameter(QgsProcessingParameterBoolean(self.COMPUTE_EDGES,
- self.tr('Compute edges'),
- defaultValue=False))
- options_param = QgsProcessingParameterString(self.OPTIONS,
- self.tr('Additional creation options'),
- defaultValue='',
- optional=True)
- options_param.setFlags(options_param.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
- options_param.setMetadata({
- 'widget_wrapper': {
- 'class': 'processing.algs.gdal.ui.RasterOptionsWidget.RasterOptionsWidgetWrapper'}})
- self.addParameter(options_param)
- self.addParameter(QgsProcessingParameterRasterDestination(self.OUTPUT, self.tr('Topographic Position Index')))
- def name(self):
- return 'tpitopographicpositionindex'
- def displayName(self):
- return self.tr('Topographic Position Index (TPI)')
- def group(self):
- return self.tr('Raster analysis')
- def groupId(self):
- return 'rasteranalysis'
- def commandName(self):
- return 'gdaldem'
- def getConsoleCommands(self, parameters, context, feedback, executing=True):
- arguments = ['TPI']
- inLayer = self.parameterAsRasterLayer(parameters, self.INPUT, context)
- if inLayer is None:
- raise QgsProcessingException(self.invalidRasterError(parameters, self.INPUT))
- arguments.append(inLayer.source())
- out = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
- self.setOutputValue(self.OUTPUT, out)
- arguments.append(out)
- arguments.append('-b')
- arguments.append(str(self.parameterAsInt(parameters, self.BAND, context)))
- if self.parameterAsBoolean(parameters, self.COMPUTE_EDGES, context):
- arguments.append('-compute_edges')
- options = self.parameterAsString(parameters, self.OPTIONS, context)
- if options:
- arguments.extend(GdalUtils.parseCreationOptions(options))
- return [self.commandName(), GdalUtils.escapeAndJoin(arguments)]
|
