mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
aa60bfecb2
This commit sets a different way of handling SAGA versions and a new way of checking saga installations This is done to fix the messy situation that SAGA causes due to its API changing in each release.
97 lines
3.9 KiB
Python
97 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
SplitRGBBands.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 import QtGui
|
|
from processing.core.GeoAlgorithm import GeoAlgorithm
|
|
from processing.core.parameters import ParameterRaster
|
|
from processing.core.outputs import OutputRaster
|
|
from processing.tools.system import *
|
|
import SagaUtils
|
|
|
|
|
|
class SplitRGBBands(GeoAlgorithm):
|
|
|
|
INPUT = 'INPUT'
|
|
R = 'R'
|
|
G = 'G'
|
|
B = 'B'
|
|
|
|
def getIcon(self):
|
|
return QtGui.QIcon(os.path.dirname(__file__) + '/../../images/saga.png')
|
|
|
|
def defineCharacteristics(self):
|
|
self.name = 'Split RGB bands'
|
|
self.group = 'Grid - Tools'
|
|
self.addParameter(ParameterRaster(SplitRGBBands.INPUT, 'Input layer',
|
|
False))
|
|
self.addOutput(OutputRaster(SplitRGBBands.R, 'Output R band layer'))
|
|
self.addOutput(OutputRaster(SplitRGBBands.G, 'Output G band layer'))
|
|
self.addOutput(OutputRaster(SplitRGBBands.B, 'Output B band layer'))
|
|
|
|
def processAlgorithm(self, progress):
|
|
# TODO: check correct num of bands
|
|
input = self.getParameterValue(SplitRGBBands.INPUT)
|
|
temp = getTempFilename(None).replace('.', '')
|
|
basename = os.path.basename(temp)
|
|
validChars = \
|
|
'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789'
|
|
safeBasename = ''.join(c for c in basename if c in validChars)
|
|
temp = os.path.join(os.path.dirname(temp), safeBasename)
|
|
|
|
r = self.getOutputValue(SplitRGBBands.R)
|
|
g = self.getOutputValue(SplitRGBBands.G)
|
|
b = self.getOutputValue(SplitRGBBands.B)
|
|
commands = []
|
|
if isWindows():
|
|
commands.append('io_gdal 0 -GRIDS "' + temp + '" -FILES "' + input
|
|
+ '"')
|
|
commands.append('io_gdal 1 -GRIDS "' + temp
|
|
+ '_0001.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + r + '"'
|
|
)
|
|
commands.append('io_gdal 1 -GRIDS "' + temp
|
|
+ '_0002.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + g + '"'
|
|
)
|
|
commands.append('io_gdal 1 -GRIDS "' + temp
|
|
+ '_0003.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + b + '"'
|
|
)
|
|
else:
|
|
commands.append('libio_gdal 0 -GRIDS "' + temp + '" -FILES "'
|
|
+ input + '"')
|
|
commands.append('libio_gdal 1 -GRIDS "' + temp
|
|
+ '_0001.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + r + '"'
|
|
)
|
|
commands.append('libio_gdal 1 -GRIDS "' + temp
|
|
+ '_0002.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + g + '"'
|
|
)
|
|
commands.append('libio_gdal 1 -GRIDS "' + temp
|
|
+ '_0003.sgrd" -FORMAT 1 -TYPE 0 -FILE "' + b + '"'
|
|
)
|
|
|
|
SagaUtils.createSagaBatchJobFileFromSagaCommands(commands)
|
|
SagaUtils.executeSaga(progress)
|