Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
improved saga algorithm execution
git-svn-id: http://sextante.googlecode.com/svn/trunk/soft/bindings/qgis-plugin@18 881b9c09-3ef8-f3c2-ec3d-21d735c97f4d
  • Loading branch information
volayaf@gmail.com committed Jan 19, 2012
1 parent 9c5fcdf commit 955454e
Show file tree
Hide file tree
Showing 4 changed files with 47 additions and 42 deletions.
23 changes: 22 additions & 1 deletion src/sextante/core/SextanteUtils.py
@@ -1,4 +1,8 @@
import os
import time
from sextante.outputs.OutputTable import OutputTable
from sextante.outputs.OutputVector import OutputVector
from sextante.outputs.OutputRaster import OutputRaster
class SextanteUtils:

@staticmethod
Expand All @@ -17,10 +21,27 @@ def softwareFolder():
def isWindows():
return os.path =="nt"

@staticmethod
def tempFolder():
tempfolder = os.getenv('HOME') + os.sep + "sextante" + os.sep + "tempdata"
mkdir(tempfolder)

return tempfolder

@staticmethod
def setTempOutput(out):
pass
seconds = str(time.time())
if isinstance(out, OutputRaster):
ext = ".tif"
elif isinstance(out, OutputVector):
ext = ".shp"
elif isinstance(out, OutputTable):
ext = ".dbf"
else:
ext =""

filename = SextanteUtils.tempFolder() + os.sep + seconds + os.sep + ext
out.channel = filename


@staticmethod
Expand Down
1 change: 0 additions & 1 deletion src/sextante/gui/SextanteToolbox.py
Expand Up @@ -55,7 +55,6 @@ def setupUi(self, SextanteToolbox):
QtCore.QMetaObject.connectSlotsByName(SextanteToolbox)

def showPopupMenu(self,point):
treeidx=self.algorithmTree.indexAt(point)
item = self.algorithmTree.itemAt(point)
if isinstance(item, TreeAlgorithmItem):
alg = item.alg
Expand Down
56 changes: 18 additions & 38 deletions src/sextante/saga/SagaAlgorithm.py
Expand Up @@ -45,7 +45,6 @@ def defineCharacteristicsFromFile(self):
while line != "":
line = line.strip("\n").strip()
readLine = True;
#line = lines.readline()
if line.startswith("library name"):
self.undecoratedGroup = line.split("\t")[1]
self.group = SagaGroupNameDecorator.getDecoratedName(self.undecoratedGroup)
Expand All @@ -59,7 +58,7 @@ def defineCharacteristicsFromFile(self):
except Exception: #boolean params have a different syntax
paramName = line[1:line.index("\t")]
paramDescription = line[line.index("\t") + 1:].strip()
if SagaBlackList.isBlackListed(self.name, self.group):
if SagaBlackList.isBlackListed(self.name, self.undecoratedGroup):
raise UnwrappableSagaAlgorithmException()
line = lines.readline().lower()
if "data object" in line or "file" in line:
Expand Down Expand Up @@ -189,7 +188,7 @@ def defineCharacteristicsFromFile(self):
line = lines.readline()
lines.close()

def proccessAlgorithm(self):
def processAlgorithm(self):

commands = list()
self.exportedLayers = {}
Expand All @@ -202,7 +201,7 @@ def proccessAlgorithm(self):

#1: Export rasters to sgrd. only ASC and TIF are supported.
# Vector layers must be in shapefile format and tables in dbf format. We check that.
for param in self.parameters.values:
for param in self.parameters:
if isinstance(param, ParameterRaster):
if param.value == None:
continue;
Expand Down Expand Up @@ -271,11 +270,10 @@ def proccessAlgorithm(self):
for out in self.outputs.values:
if isinstance(out, OutputRaster):
filename = out.value
if not filename.endswith(".sgrd"):
filename += ".sgrd"
if not filename.endswith(".asc") or filename.endswith(".tif"):
filename += ".tif"
out.value = filename
#A LO MEJOR NO HACE FALTA ESTO!!!!!!!!
#filename = SextanteUtils.tempfolder + os.sep + os.path.dirname(filename) + ".sgrd"
filename = SextanteUtils.tempFolder + os.sep + os.path.basename(filename) + ".sgrd"
command+=(" -" + out.name + " " + filename);
if isinstance(out, OutputVector):
filename = out.value
Expand All @@ -292,38 +290,20 @@ def proccessAlgorithm(self):

commands.append(command)

#-------------------------------------- //3:Export resulting raster layers
#----- for (int i = 0; i < m_OutputObjects.getOutputObjectsCount(); i++) {
#--------------------- final Output out = m_OutputObjects.getOutput(i);
#------------------------------ if (out instanceof OutputRasterLayer) {
#--- final IOutputChannel channel = getOutputChannel(out.getName());
#-------------------- if (!(channel instanceof FileOutputChannel)) {
#-------------------- if (channel instanceof NullOutputChannel) {
#--------------------------------------------------- continue;
#-------------------------------------------------------------- }
#--------------------------------------------------------- else {
#-------------- throw new UnsupportedOutputChannelException();
#-------------------------------------------------------------- }
#----------------------------------------------------------------- }
#-------- final FileOutputChannel foc = (FileOutputChannel) channel;
#----------------------------- String sFilename = foc.getFilename();
#--- if (!sFilename.endsWith("asc") && !sFilename.endsWith("tif")) {
#-------------------------------- sFilename = sFilename + ".tif";
#----------------------------------------------------------------- }
# final String sFilename2 = SextanteGUI.getOutputFactory().getTempFolder() + File.separator
# + new File(sFilename).getName() + ".sgrd";
#---------------------------------- if (sFilename.endsWith("asc")) {
# commands.add("io_grid 0 -GRID " + sFilename2 + " -FORMAT 1 -FILE " + sFilename);
#----------------------------------------------------------------- }
#------------------------------------------------------------ else {
# commands.add("io_gdal 1 -GRIDS " + sFilename2 + " -FORMAT 1 -FILE " + sFilename);
#----------------------------------------------------------------- }
#-------------------------------------------------------------------- }
#----------------------------------------------------------------------- }

#3:Export resulting raster layers
for out in self.outputs.values:
if isinstance(out, OutputRaster):
filename = out.value
filename2 = SextanteUtils.tempFolder + os.sep + os.path.basename(filename) + ".sgrd"
if filename.endsWith("asc"):
commands.append("io_grid 0 -GRID " + filename2 + " -FORMAT 1 -FILE " + filename);
else:
commands.append("io_gdal 1 -GRIDS " + filename2 + " -FORMAT 1 -FILE " + filename);

#4 Run SAGA

from PyQt4.QtCore import *
from PyQt4.QtGui import *
QMessageBox.critical(None, "hola", str(commands))
SagaUtils.createSagaBatchJobFileFromSagaCommands(commands)
#SagaUtils.executeSaga(this);

Expand Down
9 changes: 7 additions & 2 deletions src/sextante/saga/SagaBlackList.py
Expand Up @@ -3,10 +3,15 @@ class SagaBlackList:
@staticmethod
def isBlackListed(name, group):

if "lectures_introduction" == group:
group = group.lower()

if "lectures" == group:
return True

if "io_" == group:
return True

if "pointcloud_viewer" == group:
if "pointcloud" in group:
return True

if group.startswith("docs"):
Expand Down

0 comments on commit 955454e

Please sign in to comment.