mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
32d6275fa3
This eliminates the need to export to a QGIS compatible format after generating a raster output. The SDAT format is the native SAGA format, and it is supported by GDAL, so it is also supported by QGIS. This was not used before, but now we use that to reduce the nuber of steps and provide a better integration. Conversion to other formats is now done by Processing itself, not by the SAGA algorithm. fixes #10735
132 lines
5.3 KiB
Python
132 lines
5.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
SagaAlgorithmProvider.py
|
|
---------------------
|
|
Date : August 2012
|
|
Copyright : (C) 2012 by Victor Olaya
|
|
Email : volayaf 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__ = '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 PyQt4.QtGui import QIcon
|
|
from processing.core.AlgorithmProvider import AlgorithmProvider
|
|
from processing.core.ProcessingConfig import ProcessingConfig, Setting
|
|
from processing.core.ProcessingLog import ProcessingLog
|
|
from SagaAlgorithm212 import SagaAlgorithm212
|
|
from SagaAlgorithm213 import SagaAlgorithm213
|
|
from SagaAlgorithm214 import SagaAlgorithm214
|
|
from SplitRGBBands import SplitRGBBands
|
|
import SagaUtils
|
|
from processing.tools.system import isWindows, isMac
|
|
|
|
pluginPath = os.path.normpath(os.path.join(
|
|
os.path.split(os.path.dirname(__file__))[0], os.pardir))
|
|
|
|
|
|
class SagaAlgorithmProvider(AlgorithmProvider):
|
|
|
|
supportedVersions = {"2.1.2": ("2.1.2", SagaAlgorithm212),
|
|
"2.1.3": ("2.1.3", SagaAlgorithm213),
|
|
"2.1.4": ("2.1.3", SagaAlgorithm214),
|
|
# to allow testing upcoming release
|
|
"2.2.0": ("2.1.3", SagaAlgorithm214)}
|
|
|
|
def __init__(self):
|
|
AlgorithmProvider.__init__(self)
|
|
self.activate = True
|
|
|
|
def initializeSettings(self):
|
|
AlgorithmProvider.initializeSettings(self)
|
|
if isWindows() or isMac():
|
|
ProcessingConfig.addSetting(Setting("SAGA",
|
|
SagaUtils.SAGA_FOLDER, self.tr('SAGA folder'), ''))
|
|
ProcessingConfig.addSetting(Setting("SAGA",
|
|
SagaUtils.SAGA_IMPORT_EXPORT_OPTIMIZATION,
|
|
self.tr('Enable SAGA Import/Export optimizations'), False))
|
|
ProcessingConfig.addSetting(Setting("SAGA",
|
|
SagaUtils.SAGA_LOG_COMMANDS,
|
|
self.tr('Log execution commands'), True))
|
|
ProcessingConfig.addSetting(Setting("SAGA",
|
|
SagaUtils.SAGA_LOG_CONSOLE,
|
|
self.tr('Log console output'), True))
|
|
ProcessingConfig.settingIcons["SAGA"] = self.getIcon()
|
|
ProcessingConfig.addSetting(Setting("SAGA", "ACTIVATE_SAGA",
|
|
self.tr('Activate'), self.activate))
|
|
|
|
def unload(self):
|
|
AlgorithmProvider.unload(self)
|
|
if isWindows() or isMac():
|
|
ProcessingConfig.removeSetting(SagaUtils.SAGA_FOLDER)
|
|
|
|
ProcessingConfig.removeSetting(SagaUtils.SAGA_LOG_CONSOLE)
|
|
ProcessingConfig.removeSetting(SagaUtils.SAGA_LOG_COMMANDS)
|
|
|
|
def _loadAlgorithms(self):
|
|
self.algs = []
|
|
version = SagaUtils.getSagaInstalledVersion(True)
|
|
if version is None:
|
|
ProcessingLog.addToLog(ProcessingLog.LOG_ERROR,
|
|
self.tr('Problem with SAGA installation: SAGA was not found or is not correctly installed'))
|
|
return
|
|
if version not in self.supportedVersions:
|
|
ProcessingLog.addToLog(ProcessingLog.LOG_ERROR,
|
|
self.tr('Problem with SAGA installation: installed SAGA version (%s) is not supported' % version))
|
|
return
|
|
|
|
folder = SagaUtils.sagaDescriptionPath()
|
|
folder = os.path.join(folder, self.supportedVersions[SagaUtils.getSagaInstalledVersion()][0])
|
|
for descriptionFile in os.listdir(folder):
|
|
if descriptionFile.endswith('txt'):
|
|
f = os.path.join(folder, descriptionFile)
|
|
self._loadAlgorithm(f)
|
|
self.algs.append(SplitRGBBands())
|
|
|
|
def _loadAlgorithm(self, descriptionFile):
|
|
try:
|
|
alg = self.supportedVersions[SagaUtils.getSagaInstalledVersion()][1](descriptionFile)
|
|
if alg.name.strip() != '':
|
|
self.algs.append(alg)
|
|
else:
|
|
ProcessingLog.addToLog(ProcessingLog.LOG_ERROR,
|
|
self.tr('Could not open SAGA algorithm: %s' % descriptionFile))
|
|
except Exception, e:
|
|
ProcessingLog.addToLog(ProcessingLog.LOG_ERROR,
|
|
self.tr('Could not open SAGA algorithm: %s\n%s' % (descriptionFile, str(e))))
|
|
|
|
def getDescription(self):
|
|
version = SagaUtils.getSagaInstalledVersion()
|
|
return 'SAGA (%s)' % version if version is not None else 'SAGA'
|
|
|
|
def getName(self):
|
|
return 'saga'
|
|
|
|
def getSupportedOutputVectorLayerExtensions(self):
|
|
return ['shp']
|
|
|
|
def getSupportedOutputRasterLayerExtensions(self):
|
|
return ['sdat']
|
|
|
|
def getSupportedOutputTableLayerExtensions(self):
|
|
return ['dbf']
|
|
|
|
def getIcon(self):
|
|
return QIcon(os.path.join(pluginPath, 'images', 'saga.png'))
|