2
2
3
3
"""
4
4
***************************************************************************
5
- rasterize_over .py
5
+ rasterize_over_fixed_value .py
6
6
---------------------
7
7
Date : September 2013
8
8
Copyright : (C) 2013 by Alexander Bruy
32
32
QgsProcessingParameterRasterLayer ,
33
33
QgsProcessingParameterNumber ,
34
34
QgsProcessingParameterString ,
35
- QgsProcessingParameterBoolean )
35
+ QgsProcessingParameterBoolean ,
36
+ QgsProcessingOutputRasterLayer )
36
37
from processing .algs .gdal .GdalAlgorithm import GdalAlgorithm
37
38
from processing .algs .gdal .GdalUtils import GdalUtils
38
39
@@ -46,37 +47,35 @@ class rasterize_over_fixed_value(GdalAlgorithm):
46
47
ADD = 'ADD'
47
48
EXTRA = 'EXTRA'
48
49
BURN = 'BURN'
50
+ OUTPUT = 'OUTPUT'
49
51
50
52
def __init__ (self ):
51
53
super ().__init__ ()
52
54
53
55
def initAlgorithm (self , config = None ):
54
- self .units = [self .tr ("Pixels" ),
55
- self .tr ("Georeferenced units" )]
56
-
57
56
self .addParameter (QgsProcessingParameterFeatureSource (self .INPUT ,
58
57
self .tr ('Input vector layer' )))
59
-
60
- self .addParameter (QgsProcessingParameterRasterLayer (self .INPUT_RASTER , self .tr ('Input raster layer' )))
61
-
58
+ self .addParameter (QgsProcessingParameterRasterLayer (self .INPUT_RASTER ,
59
+ self .tr ('Input raster layer' )))
62
60
self .addParameter (QgsProcessingParameterNumber (self .BURN ,
63
61
self .tr ('A fixed value to burn' ),
64
62
type = QgsProcessingParameterNumber .Double ,
65
- defaultValue = 0.0 ,
66
- optional = False ))
67
-
68
- add_param = QgsProcessingParameterBoolean (self .ADD ,
69
- self .tr ('Add burn in values to existing raster values' ),
70
- defaultValue = False )
71
- add_param .setFlags (add_param .flags () | QgsProcessingParameterDefinition .FlagAdvanced )
72
- self .addParameter (add_param )
63
+ defaultValue = 0.0 ))
73
64
74
- extra_param = QgsProcessingParameterString (self .EXTRA ,
65
+ params = []
66
+ params .append (QgsProcessingParameterBoolean (self .ADD ,
67
+ self .tr ('Add burn in values to existing raster values' ),
68
+ defaultValue = False ))
69
+ params .append (QgsProcessingParameterString (self .EXTRA ,
75
70
self .tr ('Additional command-line parameters' ),
76
71
defaultValue = None ,
77
- optional = True )
78
- extra_param .setFlags (extra_param .flags () | QgsProcessingParameterDefinition .FlagAdvanced )
79
- self .addParameter (extra_param )
72
+ optional = True ))
73
+ for p in params :
74
+ p .setFlags (p .flags () | QgsProcessingParameterDefinition .FlagAdvanced )
75
+ self .addParameter (p )
76
+
77
+ self .addOutput (QgsProcessingOutputRasterLayer (self .OUTPUT ,
78
+ self .tr ('Rasterized' )))
80
79
81
80
def name (self ):
82
81
return 'rasterize_over_fixed_value'
@@ -99,14 +98,12 @@ def commandName(self):
99
98
def getConsoleCommands (self , parameters , context , feedback , executing = True ):
100
99
ogrLayer , layerName = self .getOgrCompatibleSource (self .INPUT , parameters , context , feedback , executing )
101
100
inLayer = self .parameterAsRasterLayer (parameters , self .INPUT_RASTER , context )
101
+ self .setOutputValue (self .OUTPUT , inLayer .source ())
102
102
103
103
arguments = ['-l' ]
104
104
arguments .append (layerName )
105
-
106
- fieldName = self .parameterAsString (parameters , self .FIELD , context )
107
-
108
105
arguments .append ('-burn' )
109
- arguments .append (self .parameterAsDouble (parameters , self .BURN , context ))
106
+ arguments .append (str ( self .parameterAsDouble (parameters , self .BURN , context ) ))
110
107
111
108
if self .parameterAsBool (parameters , self .ADD , context ):
112
109
arguments .append ('-add' )
0 commit comments