mirror of
https://github.com/qgis/QGIS.git
synced 2025-03-10 00:13:55 -04:00
dc2b29ce6a
(fix #20586). Without this parameter it is not possible to remove collars surrounding input raster which may overlap with other input rasters. As this is very frequent case algorithm is useless without such parameter. To keep API compatibility new parameter is optional and not used by default.
177 lines
8.0 KiB
Python
177 lines
8.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
***************************************************************************
|
|
merge.py
|
|
---------------------
|
|
Date : October 2014
|
|
Copyright : (C) 2014 by Radoslaw Guzinski
|
|
Email : rmgu at dhi-gras 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__ = 'Radoslaw Guzinski'
|
|
__date__ = 'October 2014'
|
|
__copyright__ = '(C) 2014, Radoslaw Guzinski'
|
|
|
|
# This will get replaced with a git SHA1 when you do a git archive
|
|
|
|
__revision__ = '$Format:%H$'
|
|
|
|
import os
|
|
|
|
from qgis.PyQt.QtCore import QCoreApplication
|
|
from qgis.PyQt.QtGui import QIcon
|
|
|
|
from qgis.core import (QgsProcessingAlgorithm,
|
|
QgsProcessing,
|
|
QgsProcessingParameterDefinition,
|
|
QgsProperty,
|
|
QgsProcessingParameterMultipleLayers,
|
|
QgsProcessingParameterEnum,
|
|
QgsProcessingParameterBoolean,
|
|
QgsProcessingParameterRasterDestination,
|
|
QgsProcessingParameterCrs,
|
|
QgsProcessingParameterString,
|
|
QgsProcessingOutputLayerDefinition,
|
|
QgsProcessingUtils)
|
|
from processing.algs.gdal.GdalAlgorithm import GdalAlgorithm
|
|
from processing.algs.gdal.GdalUtils import GdalUtils
|
|
|
|
pluginPath = os.path.split(os.path.split(os.path.dirname(__file__))[0])[0]
|
|
|
|
|
|
class buildvrt(GdalAlgorithm):
|
|
|
|
INPUT = 'INPUT'
|
|
OUTPUT = 'OUTPUT'
|
|
RESOLUTION = 'RESOLUTION'
|
|
SEPARATE = 'SEPARATE'
|
|
PROJ_DIFFERENCE = 'PROJ_DIFFERENCE'
|
|
ADD_ALPHA = 'ADD_ALPHA'
|
|
ASSIGN_CRS = 'ASSIGN_CRS'
|
|
RESAMPLING = 'RESAMPLING'
|
|
SRC_NODATA = 'SRC_NODATA'
|
|
|
|
RESOLUTION_OPTIONS = ['average', 'highest', 'lowest']
|
|
RESAMPLING_OPTIONS = ['nearest', 'bilinear', 'cubic', 'cubicspline', 'lanczos', 'average', 'mode']
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
|
|
def initAlgorithm(self, config=None):
|
|
|
|
class ParameterVrtDestination(QgsProcessingParameterRasterDestination):
|
|
|
|
def __init__(self, name, description):
|
|
super().__init__(name, description)
|
|
|
|
def clone(self):
|
|
copy = ParameterVrtDestination(self.name(), self.description())
|
|
return copy
|
|
|
|
def type(self):
|
|
return 'vrt_destination'
|
|
|
|
def defaultFileExtension(self):
|
|
return 'vrt'
|
|
|
|
self.addParameter(QgsProcessingParameterMultipleLayers(self.INPUT,
|
|
self.tr('Input layers'),
|
|
QgsProcessing.TypeRaster))
|
|
self.addParameter(QgsProcessingParameterEnum(self.RESOLUTION,
|
|
self.tr('Resolution'),
|
|
options=self.RESOLUTION_OPTIONS,
|
|
defaultValue=0))
|
|
self.addParameter(QgsProcessingParameterBoolean(self.SEPARATE,
|
|
self.tr('Place each input file into a separate band'),
|
|
defaultValue=True))
|
|
self.addParameter(QgsProcessingParameterBoolean(self.PROJ_DIFFERENCE,
|
|
self.tr('Allow projection difference'),
|
|
defaultValue=False))
|
|
|
|
add_alpha_param = QgsProcessingParameterBoolean(self.ADD_ALPHA,
|
|
self.tr('Add alpha mask band to VRT when source raster has none'),
|
|
defaultValue=False)
|
|
add_alpha_param.setFlags(add_alpha_param.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
|
|
self.addParameter(add_alpha_param)
|
|
|
|
assign_crs = QgsProcessingParameterCrs(self.ASSIGN_CRS,
|
|
self.tr('Override projection for the output file'),
|
|
defaultValue=None, optional=True)
|
|
assign_crs.setFlags(assign_crs.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
|
|
self.addParameter(assign_crs)
|
|
|
|
resampling = QgsProcessingParameterEnum(self.RESAMPLING,
|
|
self.tr('Resampling algorithm'),
|
|
options=self.RESAMPLING_OPTIONS,
|
|
defaultValue=0)
|
|
resampling.setFlags(resampling.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
|
|
self.addParameter(resampling)
|
|
|
|
src_nodata_param = QgsProcessingParameterString(self.SRC_NODATA,
|
|
self.tr('Nodata value(s) for input bands'),
|
|
defaultValue=None,
|
|
optional=True)
|
|
src_nodata_param.setFlags(src_nodata_param.flags() | QgsProcessingParameterDefinition.FlagAdvanced)
|
|
self.addParameter(src_nodata_param)
|
|
|
|
self.addParameter(ParameterVrtDestination(self.OUTPUT, QCoreApplication.translate("ParameterVrtDestination", 'Virtual')))
|
|
|
|
def name(self):
|
|
return 'buildvirtualraster'
|
|
|
|
def displayName(self):
|
|
return QCoreApplication.translate("buildvrt", 'Build virtual raster')
|
|
|
|
def icon(self):
|
|
return QIcon(os.path.join(pluginPath, 'images', 'gdaltools', 'vrt.png'))
|
|
|
|
def group(self):
|
|
return QCoreApplication.translate("buildvrt", 'Raster miscellaneous')
|
|
|
|
def groupId(self):
|
|
return 'rastermiscellaneous'
|
|
|
|
def commandName(self):
|
|
return "gdalbuildvrt"
|
|
|
|
def getConsoleCommands(self, parameters, context, feedback, executing=True):
|
|
arguments = []
|
|
arguments.append('-resolution')
|
|
arguments.append(self.RESOLUTION_OPTIONS[self.parameterAsEnum(parameters, self.RESOLUTION, context)])
|
|
if self.parameterAsBool(parameters, buildvrt.SEPARATE, context):
|
|
arguments.append('-separate')
|
|
if self.parameterAsBool(parameters, buildvrt.PROJ_DIFFERENCE, context):
|
|
arguments.append('-allow_projection_difference')
|
|
if self.parameterAsBool(parameters, buildvrt.ADD_ALPHA, context):
|
|
arguments.append('-addalpha')
|
|
crs = self.parameterAsCrs(parameters, self.ASSIGN_CRS, context)
|
|
if crs.isValid():
|
|
arguments.append('-a_srs')
|
|
arguments.append(GdalUtils.gdal_crs_string(crs))
|
|
arguments.append('-r')
|
|
arguments.append(self.RESAMPLING_OPTIONS[self.parameterAsEnum(parameters, self.RESAMPLING, context)])
|
|
|
|
if self.SRC_NODATA in parameters and parameters[self.SRC_NODATA] not in (None, ''):
|
|
nodata = self.parameterAsString(parameters, self.SRC_NODATA, context)
|
|
arguments.append('-srcnodata "{}"'.format(nodata))
|
|
|
|
# Always write input files to a text file in case there are many of them and the
|
|
# length of the command will be longer then allowed in command prompt
|
|
list_file = GdalUtils.writeLayerParameterToTextFile(filename='buildvrtInputFiles.txt', alg=self, parameters=parameters, parameter_name=self.INPUT, context=context, executing=executing, quote=False)
|
|
arguments.append('-input_file_list')
|
|
arguments.append(list_file)
|
|
|
|
out = self.parameterAsOutputLayer(parameters, self.OUTPUT, context)
|
|
arguments.append(out)
|
|
|
|
return [self.commandName(), GdalUtils.escapeAndJoin(arguments)]
|