1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
"""
***************************************************************************
r_colors.py
-----------
Date : February 2016
Copyright : (C) 2016 by Médéric Ribreux
Email : medspx at medspx dot fr
***************************************************************************
* *
* 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__ = "Médéric Ribreux"
__date__ = "February 2016"
__copyright__ = "(C) 2016, Médéric Ribreux"
import os
from grassprovider.grass_utils import GrassUtils
from processing.tools.system import getTempFilename
def checkParameterValuesBeforeExecuting(alg, parameters, context):
"""Verify if we have the right parameters"""
txtRules = alg.parameterAsString(parameters, "rules_txt", context)
rules = alg.parameterAsString(parameters, "rules", context)
if txtRules and rules:
return False, alg.tr("You need to set either inline rules or a rules file!")
rules_or_txtRules = bool(txtRules or rules)
color = bool(alg.parameterAsEnum(parameters, "color", context))
raster = bool(alg.parameterAsString(parameters, "raster", context))
if not sum([rules_or_txtRules, color, raster]) == 1:
return False, alg.tr(
"The color table, color rules and raster map parameters are mutually exclusive. You need to set one and only one of them!"
)
return True, None
def processInputs(alg, parameters, context, feedback):
# import all rasters with their color tables (and their bands)
# We need to import all the bands and color tables of the input rasters
rasters = alg.parameterAsLayerList(parameters, "map", context)
for idx, layer in enumerate(rasters):
layerName = f"map_{idx}"
# Add a raster layer
alg.loadRasterLayer(layerName, layer, context, False, None)
# Optional raster layer to copy from
raster = alg.parameterAsString(parameters, "raster", context)
if raster:
alg.loadRasterLayerFromParameter("raster", parameters, context, False, None)
alg.postInputs(context)
def processCommand(alg, parameters, context, feedback):
# Handle inline rules
txtRules = alg.parameterAsString(parameters, "txtrules", context)
if txtRules:
# Creates a temporary txt file
tempRulesName = getTempFilename(context=context)
# Inject rules into temporary txt file
with open(tempRulesName, "w") as tempRules:
tempRules.write(txtRules)
alg.removeParameter("txtrules")
parameters["rules"] = tempRulesName
if alg.parameterAsEnum(parameters, "color", context) == 0:
alg.removeParameter("color")
alg.processCommand(parameters, context, feedback, True)
def processOutputs(alg, parameters, context, feedback):
createOpt = alg.parameterAsString(parameters, alg.GRASS_RASTER_FORMAT_OPT, context)
metaOpt = alg.parameterAsString(parameters, alg.GRASS_RASTER_FORMAT_META, context)
# Export all rasters with their color tables (and their bands)
rasters = alg.parameterAsLayerList(parameters, "map", context)
outputDir = alg.parameterAsString(parameters, "output_dir", context)
for idx, raster in enumerate(rasters):
rasterName = f"map_{idx}"
fileName = os.path.join(outputDir, rasterName)
outFormat = GrassUtils.getRasterFormatFromFilename(fileName)
alg.exportRasterLayer(
alg.exportedLayers[rasterName],
fileName,
True,
outFormat,
createOpt,
metaOpt,
)
|
