mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
132 lines
5.7 KiB
Python
132 lines
5.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
GridMetrics.py
|
|
---------------------
|
|
Date : August 2012
|
|
Copyright : (C) 2012 by Victor Olaya
|
|
Email : volayaf at gmail dot com
|
|
---------------------
|
|
Date : June 2014
|
|
Copyright : (C) 2014 by Agresta S. Coop.
|
|
Email : iescamochero at agresta dot org
|
|
***************************************************************************
|
|
* *
|
|
* 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$'
|
|
|
|
import os
|
|
from processing.core.parameters import ParameterFile
|
|
from processing.core.parameters import ParameterNumber
|
|
from processing.core.parameters import ParameterBoolean
|
|
from processing.core.parameters import ParameterString
|
|
from processing.core.outputs import OutputFile
|
|
from FusionUtils import FusionUtils
|
|
from FusionAlgorithm import FusionAlgorithm
|
|
|
|
|
|
class GridMetrics(FusionAlgorithm):
|
|
|
|
INPUT = 'INPUT'
|
|
OUTPUT_CSV_ELEVATION = 'OUTPUT_CSV_ELEVATION'
|
|
OUTPUT_CSV_INTENSITY = 'OUTPUT_CSV_INTENSITY'
|
|
OUTPUT_TXT_ELEVATION = 'OUTPUT_TXT_ELEVATION'
|
|
OUTPUT_TXT_INTENSITY = 'OUTPUT_TXT_INTENSITY'
|
|
GROUND = 'GROUND'
|
|
HEIGHT = 'HEIGHT'
|
|
CELLSIZE = 'CELLSIZE'
|
|
OUTLIER = 'OUTLIER'
|
|
FIRST = 'FIRST'
|
|
MINHT = 'MINHT'
|
|
CLASS = 'CLASS'
|
|
|
|
def defineCharacteristics(self):
|
|
self.name = 'Grid Metrics'
|
|
self.group = 'Points'
|
|
self.addParameter(ParameterFile(
|
|
self.INPUT, self.tr('Input LAS layer')))
|
|
self.addParameter(ParameterFile(
|
|
self.GROUND, self.tr('Input ground DTM layer')))
|
|
self.addParameter(ParameterNumber(
|
|
self.HEIGHT, self.tr('Height break')))
|
|
self.addParameter(ParameterNumber(
|
|
self.CELLSIZE, self.tr('Cellsize')))
|
|
|
|
self.addOutput(OutputFile(
|
|
self.OUTPUT_CSV_ELEVATION, self.tr('Output table with grid metrics')))
|
|
|
|
output_csv_intensity = OutputFile(
|
|
self.OUTPUT_CSV_INTENSITY, self.tr('OUTPUT CSV INTENSITY'))
|
|
output_csv_intensity.hidden = True
|
|
self.addOutput(output_csv_intensity)
|
|
|
|
output_txt_elevation = OutputFile(
|
|
self.OUTPUT_TXT_ELEVATION, self.tr('OUTPUT CSV INTENSITY'))
|
|
output_txt_elevation.hidden = True
|
|
self.addOutput(output_txt_elevation)
|
|
|
|
output_txt_intensity = OutputFile(
|
|
self.OUTPUT_TXT_INTENSITY, self.tr('OUTPUT CSV INTENSITY'))
|
|
output_txt_intensity.hidden = True
|
|
self.addOutput(output_txt_intensity)
|
|
|
|
outlier = ParameterString(
|
|
self.OUTLIER, self.tr('Outlier:low,high'), '', False, True)
|
|
outlier.isAdvanced = True
|
|
self.addParameter(outlier)
|
|
first = ParameterBoolean(self.FIRST, self.tr('First'), False)
|
|
first.isAdvanced = True
|
|
self.addParameter(first)
|
|
minht = ParameterString(self.MINHT, self.tr('Htmin'), '', False, True)
|
|
minht.isAdvanced = True
|
|
self.addParameter(minht)
|
|
class_var = ParameterString(
|
|
self.CLASS, self.tr('Class (set blank if not used)'), '', False, True)
|
|
class_var.isAdvanced = True
|
|
self.addParameter(class_var)
|
|
|
|
def processAlgorithm(self, progress):
|
|
commands = [os.path.join(FusionUtils.FusionPath(), 'GridMetrics.exe')]
|
|
commands.append('/verbose')
|
|
outlier = self.getParameterValue(self.OUTLIER)
|
|
if str(outlier).strip() != '':
|
|
commands.append('/outlier:' + str(outlier))
|
|
first = self.getParameterValue(self.FIRST)
|
|
if first:
|
|
commands.append('/first:' + str(first))
|
|
minht = self.getParameterValue(self.MINHT)
|
|
if str(minht).strip() != '':
|
|
commands.append('/minht:' + str(minht))
|
|
class_var = self.getParameterValue(self.CLASS)
|
|
if str(class_var).strip() != '':
|
|
commands.append('/class:' + str(class_var))
|
|
commands.append(self.getParameterValue(self.GROUND))
|
|
commands.append(str(self.getParameterValue(self.HEIGHT)))
|
|
commands.append(str(self.getParameterValue(self.CELLSIZE)))
|
|
commands.append(self.getOutputValue(self.OUTPUT_CSV_ELEVATION))
|
|
files = self.getParameterValue(self.INPUT).split(';')
|
|
if len(files) == 1:
|
|
commands.append(self.getParameterValue(self.INPUT))
|
|
else:
|
|
FusionUtils.createFileList(files)
|
|
commands.append(FusionUtils.tempFileListFilepath())
|
|
FusionUtils.runFusion(commands, progress)
|
|
basePath = self.getOutputValue(self.OUTPUT_CSV_ELEVATION)
|
|
basePath = os.path.join(os.path.dirname(basePath), os.path.splitext(os.path.basename(basePath))[0])
|
|
self.setOutputValue(self.OUTPUT_CSV_ELEVATION, basePath + '_all_returns_elevation_stats.csv')
|
|
self.setOutputValue(self.OUTPUT_CSV_INTENSITY, basePath + '_all_returns_intensity_stats.csv')
|
|
self.setOutputValue(self.OUTPUT_TXT_ELEVATION, basePath + '_all_returns_elevation_stats_ascii_header.txt')
|
|
self.setOutputValue(self.OUTPUT_TXT_INTENSITY, basePath + '_all_returns_intensity_stats_ascii_header.txt')
|