Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
git-svn-id: http://sextante.googlecode.com/svn/trunk/soft/bindings/qgis-plugin@48 881b9c09-3ef8-f3c2-ec3d-21d735c97f4d
- Loading branch information
volayaf
committed
Mar 21, 2012
1 parent
b7d940e
commit 772e8e7
Showing
198 changed files
with
2,610 additions
and
214 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,227 @@ | ||
| import os | ||
| from qgis.core import * | ||
| from PyQt4.QtCore import * | ||
| from PyQt4.QtGui import * | ||
| from sextante.core.GeoAlgorithm import GeoAlgorithm | ||
| from sextante.parameters.ParameterTable import ParameterTable | ||
| from sextante.parameters.ParameterMultipleInput import ParameterMultipleInput | ||
| from sextante.parameters.ParameterRaster import ParameterRaster | ||
| from sextante.outputs.OutputRaster import OutputRaster | ||
| from sextante.parameters.ParameterVector import ParameterVector | ||
| from sextante.parameters.ParameterBoolean import ParameterBoolean | ||
| from sextante.outputs.OutputVector import OutputVector | ||
| from sextante.saga.SagaUtils import SagaUtils | ||
| from sextante.core.GeoAlgorithmExecutionException import GeoAlgorithmExecutionException | ||
| from sextante.core.SextanteLog import SextanteLog | ||
| from sextante.parameters.ParameterFactory import ParameterFactory | ||
| from sextante.outputs.OutputFactory import OutputFactory | ||
| from sextante.core.SextanteConfig import SextanteConfig | ||
| from sextante.core.QGisLayers import QGisLayers | ||
| from sextante.grass.GrassUtils import GrassUtils | ||
| import time | ||
| from sextante.core.SextanteUtils import SextanteUtils | ||
|
|
||
| class GrassAlgorithm(GeoAlgorithm): | ||
|
|
||
| def __init__(self, descriptionfile): | ||
| GeoAlgorithm.__init__(self) | ||
| self._descriptionFile = descriptionfile | ||
| self.defineCharacteristicsFromFile() | ||
| self.numExportedLayers = 0 | ||
| self.needsregion = False | ||
|
|
||
| def getIcon(self): | ||
| return QIcon(os.path.dirname(__file__) + "/../images/grass.png") | ||
|
|
||
| def defineCharacteristicsFromFile(self): | ||
| lines = open(self._descriptionFile) | ||
| line = lines.readline().strip("\n").strip() | ||
| self.name = line | ||
| line = lines.readline().strip("\n").strip() | ||
| self.group = line | ||
| while line != "": | ||
| try: | ||
| line = line.strip("\n").strip() | ||
| if line.startswith("Parameter"): | ||
| self.addParameter(ParameterFactory.getFromString(line)) | ||
| elif line.startswith("Region"): | ||
| self.needsregion = True | ||
| else: | ||
| self.addOutput(OutputFactory.getFromString(line)) | ||
| line = lines.readline().strip("\n").strip() | ||
| except Exception,e: | ||
| SextanteLog.addToLog(SextanteLog.LOG_ERROR, "Could not open GRASS algorithm: " + self._descriptionFile + "\n" + line) | ||
| raise e | ||
| lines.close() | ||
|
|
||
| def calculateResamplingExtent(self): | ||
| auto = SextanteConfig.getSetting(GrassUtils.GRASS_AUTO_REGION) | ||
| if auto: | ||
| first = True; | ||
| for param in self.parameters: | ||
| if isinstance(param, ParameterRaster): | ||
| if isinstance(param.value, QgsRasterLayer): | ||
| value = param.value | ||
| else: | ||
| value = QGisLayers.getObjectFromUri(param.value) | ||
| if first: | ||
| self.xmin = value.extent().xMinimum() | ||
| self.xmax = value.extent().xMaximum() | ||
| self.ymin = value.extent().yMinimum() | ||
| self.ymax = value.extent().yMaximum() | ||
| self.cellsize = (value.extent().xMaximum() - value.extent().xMinimum())/value.getRasterXDim() | ||
| first = False | ||
| else: | ||
| self.xmin = min(self.xmin, value.extent().xMinimum()) | ||
| self.xmax = max(self.xmax, value.extent().xMaximum()) | ||
| self.ymin = min(self.ymin, value.extent().yMinimum()) | ||
| self.ymax = max(self.ymax, value.extent().yMaximum()) | ||
| self.cellsize = max(self.cellsize, (value.extent().xMaximum() - value.extent().xMinimum())/value.getRasterXDim()) | ||
| else: | ||
| self.xmin = SextanteConfig.getSetting(SagaUtils.SAGA_RESAMPLING_REGION_XMIN) | ||
| self.xmax = SextanteConfig.getSetting(SagaUtils.SAGA_RESAMPLING_REGION_XMAX) | ||
| self.ymin = SextanteConfig.getSetting(SagaUtils.SAGA_RESAMPLING_REGION_YMIN) | ||
| self.ymax = SextanteConfig.getSetting(SagaUtils.SAGA_RESAMPLING_REGION_YMAX) | ||
| self.cellsize = SextanteConfig.getSetting(SagaUtils.SAGA_RESAMPLING_REGION_CELLSIZE) | ||
|
|
||
|
|
||
| def processAlgorithm(self, progress): | ||
| path = GrassUtils.grassPath() | ||
| if path == "": | ||
| raise GeoAlgorithmExecutionException("GRASS folder is not configured.\nPlease configure it before running GRASS algorithms.") | ||
| useSelection = SextanteConfig.getSetting(SagaUtils.SAGA_USE_SELECTED) | ||
|
|
||
| commands = [] | ||
| self.exportedLayers = {} | ||
| self.numExportedLayers = 0; | ||
|
|
||
| if self.needsregion: | ||
| self.calculateResamplingExtent() | ||
| GrassUtils.createTempMapset(); | ||
|
|
||
| if self.needsregion: | ||
| command = "g.region" | ||
| command += " n=" + str(self.ymax) | ||
| command +=" s=" + str(self.ymin) | ||
| command +=" e=" + str(self.xmax) | ||
| command +=" w=" + str(self.xmin) | ||
| command +=" res=" + str(self.cellsize); | ||
| commands.append(command) | ||
|
|
||
| #1: Export layer to grass mapset | ||
| for param in self.parameters: | ||
| if isinstance(param, ParameterRaster): | ||
| if param.value == None: | ||
| continue | ||
| value = param.value | ||
| commands.append(self.exportRasterLayer(value)) | ||
| if isinstance(param, ParameterVector): | ||
| if param.value == None: | ||
| continue | ||
| value = param.value | ||
| self.exportVectorLayer(value) | ||
| if isinstance(param, ParameterTable): | ||
| pass | ||
| if isinstance(param, ParameterMultipleInput): | ||
| if param.value == None: | ||
| continue | ||
| layers = param.value.split(";") | ||
| if layers == None or len(layers) == 0: | ||
| continue | ||
| if param.datatype == ParameterMultipleInput.TYPE_RASTER: | ||
| for layer in layers: | ||
| commands.append(self.exportRasterLayer(layer)) | ||
| elif param.datatype == ParameterMultipleInput.TYPE_VECTOR_ANY: | ||
| for layer in layers: | ||
| if (not value.endswith("shp")) or useSelection: | ||
| commands.append(self.exportRasterLayer(layer)) | ||
|
|
||
| #2: set parameters and outputs | ||
| command = self.name | ||
| for param in self.parameters: | ||
| if param.value == None: | ||
| continue | ||
| if isinstance(param, (ParameterRaster, ParameterVector)): | ||
| value = param.value | ||
| if value in self.exportedLayers.keys(): | ||
| command+=(" " + param.name + "=" + self.exportedLayers[value]) | ||
| else: | ||
| command+=(" " + param.name + "=" + value) | ||
| elif isinstance(param, ParameterMultipleInput): | ||
| s = param.value | ||
| for layer in self.exportedLayers.keys(): | ||
| s = s.replace(layer, self.exportedLayers[layer]) | ||
| s = s.replace(";",",") | ||
| command+=(" " + param.name + "=" + s); | ||
| elif isinstance(param, ParameterBoolean): | ||
| if param.value: | ||
| command += param.name | ||
| else: | ||
| command+=(" " + param.name + "=" + str(param.value)); | ||
|
|
||
| for out in self.outputs: | ||
| command+=(" " + out.name + "=" + out.name); | ||
|
|
||
| command += " --overwrite" | ||
| commands.append(command) | ||
|
|
||
| #3:Export resulting layers to a format that qgis can read | ||
| for out in self.outputs: | ||
| if isinstance(out, OutputRaster): | ||
| filename = out.value | ||
| #Raster layer output: adjust region to layer before exporting | ||
| commands.append("g.region rast=" + out.name) | ||
| command = "r.out.gdal -c createopt=\"TFW=YES,COMPRESS=LZW\"" | ||
| command += " input=" | ||
| command += out.name | ||
| command += " output=\"" + filename + "\"" | ||
| commands.append(command) | ||
| if isinstance(out, OutputVector): | ||
| command = "v.out.ogr -e -z input=" + out.name | ||
| command += " dsn=\"" + os.path.dirname(out.value) + "\"" | ||
| command += " format=ESRI_Shapefile" | ||
| command += " olayer=" + os.path.basename(out.value) | ||
| command += " type=auto" | ||
| commands.append(command) | ||
|
|
||
| #4 Run GRASS | ||
| GrassUtils.createGrassScript(commands) | ||
| loglines = [] | ||
| loglines.append("GRASS execution commands") | ||
| for line in commands: | ||
| loglines.append(line) | ||
| SextanteLog.addToLog(SextanteLog.LOG_INFO, loglines) | ||
| GrassUtils.executeGrass(progress); | ||
|
|
||
|
|
||
| def exportVectorLayer(self, filename): | ||
| destFilename = self.getTempFilename() | ||
| self.exportedLayers[filename]= destFilename | ||
| command = "v.in.ogr" | ||
| command += " min_area=-1" | ||
| command +=" dsn=\"" + os.path.dirname(filename) + "\"" | ||
| command +=" layer=" + os.path.basename(filename) | ||
| command +=" output=" + destFilename; | ||
| command +=" --overwrite -o" | ||
| return command | ||
|
|
||
|
|
||
| def exportRasterLayer(self, layer): | ||
| destFilename = self.getTempFilename() | ||
| self.exportedLayers[layer]= destFilename | ||
| command = "r.in.gdal" | ||
| command +=" input=\"" + layer + "\"" | ||
| command +=" band=0" | ||
| command +=" out=" + destFilename; | ||
| command +=" --overwrite -o" | ||
| return command | ||
|
|
||
|
|
||
| def getTempFilename(self): | ||
| self.numExportedLayers+=1 | ||
| filename = str(time.time()) + str(SextanteUtils.NUM_EXPORTED) | ||
| SextanteUtils.NUM_EXPORTED +=1 | ||
|
|
||
| return filename | ||
|
|
||
|
|
Oops, something went wrong.