mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
181 lines
8.1 KiB
Python
181 lines
8.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
doDEM.py
|
|
---------------------
|
|
Date : March 2011
|
|
Copyright : (C) 2011 by Giuseppe Sucameli
|
|
Email : brush dot tyler 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__ = 'Giuseppe Sucameli'
|
|
__date__ = 'March 2011'
|
|
__copyright__ = '(C) 2011, Giuseppe Sucameli'
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
__revision__ = '$Format:%H$'
|
|
|
|
from PyQt4.QtCore import SIGNAL
|
|
from PyQt4.QtGui import QWidget
|
|
|
|
from ui_widgetDEM import Ui_GdalToolsWidget as Ui_Widget
|
|
from widgetPluginBase import GdalToolsBasePluginWidget as BasePluginWidget
|
|
import GdalTools_utils as Utils
|
|
|
|
|
|
class GdalToolsDialog(QWidget, Ui_Widget, BasePluginWidget):
|
|
|
|
def __init__(self, iface):
|
|
QWidget.__init__(self)
|
|
self.iface = iface
|
|
self.modes = ("hillshade", "slope", "aspect", "color-relief", "TRI", "TPI", "roughness")
|
|
|
|
self.setupUi(self)
|
|
BasePluginWidget.__init__(self, self.iface, "gdaldem")
|
|
|
|
self.outSelector.setType(self.outSelector.FILE)
|
|
self.configSelector.setType(self.configSelector.FILE)
|
|
|
|
# set the default QSpinBoxes and QProgressBar value
|
|
self.bandSpin.setValue(1)
|
|
|
|
self.hillshadeZFactorSpin.setValue(1)
|
|
self.hillshadeScaleSpin.setValue(1)
|
|
self.hillshadeAltitudeSpin.setValue(45.0)
|
|
self.hillshadeAzimuthSpin.setValue(315.0)
|
|
self.slopeScaleSpin.setValue(1)
|
|
|
|
# set the default color configuration file to terrain
|
|
import os.path
|
|
colorConfigFile = os.path.join(os.path.dirname(__file__), "terrain.txt")
|
|
self.configSelector.setFilename(colorConfigFile)
|
|
|
|
self.outputFormat = Utils.fillRasterOutputFormat()
|
|
self.creationOptionsWidget.setFormat(self.outputFormat)
|
|
|
|
self.setParamsStatus([
|
|
(self.inSelector, SIGNAL("filenameChanged()")),
|
|
(self.outSelector, SIGNAL("filenameChanged()")),
|
|
(self.computeEdgesCheck, SIGNAL("stateChanged(int)"), None, 1800),
|
|
(self.bandSpin, SIGNAL("valueChanged(int)"), self.bandCheck),
|
|
(self.algorithmCheck, SIGNAL("stateChanged(int)"), None, 1800),
|
|
(self.creationOptionsWidget, SIGNAL("optionsChanged()")),
|
|
(self.creationOptionsGroupBox, SIGNAL("toggled(bool)")),
|
|
(self.modeCombo, SIGNAL("currentIndexChanged(int)")),
|
|
([self.hillshadeZFactorSpin, self.hillshadeScaleSpin, self.hillshadeAltitudeSpin, self.hillshadeAzimuthSpin], SIGNAL("valueChanged(double)")),
|
|
(self.slopeScaleSpin, SIGNAL("valueChanged(double)")),
|
|
(self.slopePercentCheck, SIGNAL("stateChanged(int)")),
|
|
([self.aspectTrigonometricCheck, self.aspectZeroForFlatCheck], SIGNAL("stateChanged(int)")),
|
|
(self.configSelector, SIGNAL("filenameChanged()")),
|
|
([self.colorExactRadio, self.colorNearestRadio], SIGNAL("toggled(bool)"), self.colorMatchGroupBox),
|
|
(self.colorAlphaCheck, SIGNAL("stateChanged(int)"))
|
|
])
|
|
|
|
self.connect(self.inSelector, SIGNAL("selectClicked()"), self.fillInputFileEdit)
|
|
self.connect(self.outSelector, SIGNAL("selectClicked()"), self.fillOutputFileEdit)
|
|
self.connect(self.configSelector, SIGNAL("selectClicked()"), self.fillColorConfigFileEdit)
|
|
self.connect(self.modeCombo, SIGNAL("currentIndexChanged(int)"), self.showModeParams)
|
|
|
|
def showModeParams(self, index):
|
|
self.stackedWidget.setVisible(index < 4)
|
|
self.algorithmCheck.setVisible(index < 3)
|
|
if index >= 3:
|
|
self.algorithmCheck.setChecked(False)
|
|
|
|
def onLayersChanged(self):
|
|
self.inSelector.setLayers(Utils.LayerRegistry.instance().getRasterLayers())
|
|
|
|
def fillInputFileEdit(self):
|
|
lastUsedFilter = Utils.FileFilter.lastUsedRasterFilter()
|
|
inputFile = Utils.FileDialog.getOpenFileName(self, self.tr("Select the file for DEM"), Utils.FileFilter.allRastersFilter(), lastUsedFilter)
|
|
if not inputFile:
|
|
return
|
|
Utils.FileFilter.setLastUsedRasterFilter(lastUsedFilter)
|
|
|
|
self.inSelector.setFilename(inputFile)
|
|
self.getArguments()
|
|
|
|
def fillOutputFileEdit(self):
|
|
lastUsedFilter = Utils.FileFilter.lastUsedRasterFilter()
|
|
outputFile = Utils.FileDialog.getSaveFileName(self, self.tr("Select the raster file to save the results to"), Utils.FileFilter.saveRastersFilter(), lastUsedFilter)
|
|
if not outputFile:
|
|
return
|
|
Utils.FileFilter.setLastUsedRasterFilter(lastUsedFilter)
|
|
|
|
self.outputFormat = Utils.fillRasterOutputFormat(lastUsedFilter, outputFile)
|
|
self.outSelector.setFilename(outputFile)
|
|
self.creationOptionsWidget.setFormat(self.outputFormat)
|
|
|
|
def fillColorConfigFileEdit(self):
|
|
configFile = Utils.FileDialog.getOpenFileName(self, self.tr("Select the color configuration file"))
|
|
if not configFile:
|
|
return
|
|
self.configSelector.setFilename(configFile)
|
|
|
|
def getArguments(self):
|
|
mode = self.modes[self.modeCombo.currentIndex()]
|
|
arguments = []
|
|
arguments.append(mode)
|
|
arguments.append(self.getInputFileName())
|
|
if mode == "color-relief":
|
|
arguments.append(self.configSelector.filename())
|
|
outputFn = self.getOutputFileName()
|
|
arguments.append(outputFn)
|
|
if mode == "hillshade":
|
|
arguments.extend(["-z", self.hillshadeZFactorSpin.value()])
|
|
arguments.extend(["-s", self.hillshadeScaleSpin.value()])
|
|
arguments.extend(["-az", self.hillshadeAzimuthSpin.value()])
|
|
arguments.extend(["-alt", self.hillshadeAltitudeSpin.value()])
|
|
elif mode == "slope":
|
|
if self.slopePercentCheck.isChecked():
|
|
arguments.append("-p")
|
|
arguments.extend(["-s", self.slopeScaleSpin.value()])
|
|
elif mode == "aspect":
|
|
if self.aspectTrigonometricCheck.isChecked():
|
|
arguments.append("-trigonometric")
|
|
if self.aspectZeroForFlatCheck.isChecked():
|
|
arguments.append("-zero_for_flat")
|
|
elif mode == "color-relief":
|
|
if self.colorAlphaCheck.isChecked():
|
|
arguments.append("-alpha")
|
|
if self.colorMatchGroupBox.isChecked():
|
|
if self.colorExactRadio.isChecked():
|
|
arguments.append("-exact_color_entry")
|
|
elif self.colorNearestRadio.isChecked():
|
|
arguments.append("-nearest_color_entry")
|
|
if self.algorithmCheck.isChecked():
|
|
arguments.extend(["-alg", "ZevenbergenThorne"])
|
|
if self.computeEdgesCheck.isChecked():
|
|
arguments.append("-compute_edges")
|
|
if self.bandCheck.isChecked():
|
|
arguments.extend(["-b", self.bandSpin.value()])
|
|
if outputFn:
|
|
arguments.extend(["-of", self.outputFormat])
|
|
if self.creationOptionsGroupBox.isChecked():
|
|
for opt in self.creationOptionsWidget.options():
|
|
arguments.extend(["-co", opt])
|
|
# set creation options filename/layer for validation
|
|
if self.inSelector.layer():
|
|
self.creationOptionsWidget.setRasterLayer(self.inSelector.layer())
|
|
else:
|
|
self.creationOptionsWidget.setRasterFileName(self.getInputFileName())
|
|
|
|
return arguments
|
|
|
|
def getInputFileName(self):
|
|
return self.inSelector.filename()
|
|
|
|
def getOutputFileName(self):
|
|
return self.outSelector.filename()
|
|
|
|
def addLayerIntoCanvas(self, fileInfo):
|
|
self.iface.addRasterLayer(fileInfo.filePath())
|