2015-08-22 14:29:41 +02:00

101 lines
3.7 KiB
Python

# -*- coding: utf-8 -*-
"""
***************************************************************************
translate.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$'
from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
from processing.core.parameters import ParameterRaster
from processing.core.parameters import ParameterBoolean
from processing.core.parameters import ParameterSelection
from processing.core.parameters import ParameterString
from processing.core.outputs import OutputRaster
from processing.algs.gdal.GdalUtils import GdalUtils
class gdaladdo(GdalAlgorithm):
INPUT = 'INPUT'
LEVELS = 'LEVELS'
CLEAN = 'CLEAN'
RESAMPLING_METHOD = 'RESAMPLING_METHOD'
FORMAT = 'FORMAT'
OUTPUT = 'OUTPUT'
METHODS = [
'nearest',
'average',
'gauss',
'cubic',
'average_mp',
'average_magphase',
'mode',
]
FORMATS = ['Internal (if possible)', 'External (GTiff .ovr)',
'External (ERDAS Imagine .aux)']
def commandLineName(self):
return "gdalogr:overviews"
def defineCharacteristics(self):
self.name, self.i18n_name = self.trAlgorithm('Build overviews (pyramids)')
self.group, self.i18n_group = self.trAlgorithm('[GDAL] Miscellaneous')
self.addParameter(ParameterRaster(
self.INPUT, self.tr('Input layer'), False))
self.addParameter(ParameterString(self.LEVELS,
self.tr('Overview levels'), '2 4 8 16'))
self.addParameter(ParameterBoolean(self.CLEAN,
self.tr('Remove all existing overviews'), False))
self.addParameter(ParameterSelection(self.RESAMPLING_METHOD,
self.tr('Resampling method'), self.METHODS, 0))
self.addParameter(ParameterSelection(self.FORMAT,
self.tr('Overview format'), self.FORMATS, 0))
self.addOutput(OutputRaster(self.OUTPUT, self.tr('Pyramidized'), True))
def getConsoleCommands(self):
inFile = self.getParameterValue(self.INPUT)
clearOverviews = self.getParameterValue(self.CLEAN)
ovrFormat = self.getParameterValue(self.FORMAT)
arguments = []
arguments.append(inFile)
if clearOverviews:
arguments.append('-clean')
arguments.append('-r')
arguments.append(self.METHODS[self.getParameterValue(self.RESAMPLING_METHOD)])
if ovrFormat == 1:
# external .ovr
arguments.append('-ro')
elif ovrFormat == 2:
# external .aux
arguments.extend('--config USE_RRD YES'.split(' '))
arguments.extend(self.getParameterValue(self.LEVELS).split(' '))
self.setOutputValue(self.OUTPUT, inFile)
return ['gdaladdo', GdalUtils.escapeAndJoin(arguments)]