Skip to content

Commit f7575e0

Browse files
committedOct 19, 2016
[processing] expose IDW interpolation from Interpolation plugin in toolbox
1 parent 5d4cf2f commit f7575e0

File tree

3 files changed

+278
-1
lines changed

3 files changed

+278
-1
lines changed
 
Lines changed: 138 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,138 @@
1+
# -*- coding: utf-8 -*-
2+
3+
"""
4+
***************************************************************************
5+
IdwInterpolationAttribute.py
6+
---------------------
7+
Date : October 2016
8+
Copyright : (C) 2016 by Alexander Bruy
9+
Email : alexander dot bruy at gmail dot com
10+
***************************************************************************
11+
* *
12+
* This program is free software; you can redistribute it and/or modify *
13+
* it under the terms of the GNU General Public License as published by *
14+
* the Free Software Foundation; either version 2 of the License, or *
15+
* (at your option) any later version. *
16+
* *
17+
***************************************************************************
18+
"""
19+
20+
__author__ = 'Alexander Bruy'
21+
__date__ = 'October 2016'
22+
__copyright__ = '(C) 2016, Alexander Bruy'
23+
24+
# This will get replaced with a git SHA1 when you do a git archive
25+
26+
__revision__ = '$Format:%H$'
27+
28+
from qgis.core import QgsRectangle
29+
from qgis.analysis import (QgsInterpolator,
30+
QgsIDWInterpolator,
31+
QgsGridFileWriter
32+
)
33+
34+
from processing.core.GeoAlgorithm import GeoAlgorithm
35+
from processing.core.parameters import ParameterVector
36+
from processing.core.parameters import ParameterTableField
37+
from processing.core.parameters import ParameterSelection
38+
from processing.core.parameters import ParameterNumber
39+
from processing.core.parameters import ParameterExtent
40+
from processing.core.outputs import OutputRaster
41+
from processing.tools import dataobjects
42+
43+
44+
class IdwInterpolationAttribute(GeoAlgorithm):
45+
46+
INPUT_LAYER = 'INPUT_LAYER'
47+
ATTRIBUTE = 'ATTRIBUTE'
48+
LAYER_TYPE = 'LAYER_TYPE'
49+
DISTANCE_COEFFICIENT = 'DISTANCE_COEFFICIENT'
50+
COLUMNS = 'COLUMNS'
51+
ROWS = 'ROWS'
52+
CELLSIZE_X = 'CELLSIZE_X'
53+
CELLSIZE_Y = 'CELLSIZE_Y'
54+
EXTENT = 'EXTENT'
55+
OUTPUT_LAYER = 'OUTPUT_LAYER'
56+
57+
def defineCharacteristics(self):
58+
self.name, self.i18n_name = self.trAlgorithm('IDW interpolation (using attribute)')
59+
self.group, self.i18n_group = self.trAlgorithm('Interpolation')
60+
61+
self.TYPES = [self.tr('Points'),
62+
self.tr('Structure lines'),
63+
self.tr('Break lines')
64+
]
65+
66+
self.addParameter(ParameterVector(self.INPUT_LAYER,
67+
self.tr('Vector layer')))
68+
self.addParameter(ParameterTableField(self.ATTRIBUTE,
69+
self.tr('Interpolation attribute'),
70+
self.INPUT_LAYER,
71+
ParameterTableField.DATA_TYPE_NUMBER))
72+
self.addParameter(ParameterSelection(self.LAYER_TYPE,
73+
self.tr('Type'),
74+
self.TYPES,
75+
0))
76+
self.addParameter(ParameterNumber(self.DISTANCE_COEFFICIENT,
77+
self.tr('Distance coefficient P'),
78+
0.0, 99.99, 2.0))
79+
self.addParameter(ParameterNumber(self.COLUMNS,
80+
self.tr('Number of columns'),
81+
0, 10000000, 300))
82+
self.addParameter(ParameterNumber(self.ROWS,
83+
self.tr('Number of rows'),
84+
0, 10000000, 300))
85+
self.addParameter(ParameterNumber(self.CELLSIZE_X,
86+
self.tr('Cellsize X'),
87+
0.0, 999999.000000, 0.0))
88+
self.addParameter(ParameterNumber(self.CELLSIZE_Y,
89+
self.tr('Cellsize Y'),
90+
0.0, 999999.000000, 0.0))
91+
self.addParameter(ParameterExtent(self.EXTENT,
92+
self.tr('Extent')))
93+
self.addOutput(OutputRaster(self.OUTPUT_LAYER,
94+
self.tr('Interpolated')))
95+
96+
def processAlgorithm(self, progress):
97+
layer = dataobjects.getObjectFromUri(
98+
self.getParameterValue(self.INPUT_LAYER))
99+
fieldName = self.getParameterValue(self.ATTRIBUTE)
100+
layerType = self.getParameterValue(self.LAYER_TYPE)
101+
coefficient = self.getParameterValue(self.DISTANCE_COEFFICIENT)
102+
columns = self.getParameterValue(self.COLUMNS)
103+
rows = self.getParameterValue(self.ROWS)
104+
cellsizeX = self.getParameterValue(self.CELLSIZE_X)
105+
cellsizeY = self.getParameterValue(self.CELLSIZE_Y)
106+
extent = self.getParameterValue(self.EXTENT).split(',')
107+
output = self.getOutputValue(self.OUTPUT_LAYER)
108+
109+
xMin = float(extent[0])
110+
xMax = float(extent[1])
111+
yMin = float(extent[2])
112+
yMax = float(extent[3])
113+
bbox = QgsRectangle(xMin, yMin, xMax, yMax)
114+
115+
layerData = QgsInterpolator.LayerData()
116+
layerData.vectorLayer = layer
117+
layerData.zCoordInterpolation = False
118+
layerData.interpolationAttribute = layer.dataProvider().fieldNameIndex(fieldName)
119+
120+
if layerType == 0:
121+
layerData.mInputType = QgsInterpolator.POINTS
122+
elif layerType == 1:
123+
layerData.mInputType = QgsInterpolator.STRUCTURE_LINES
124+
else:
125+
layerData.mInputType = QgsInterpolator.BREAK_LINES
126+
127+
interpolator = QgsIDWInterpolator([layerData])
128+
interpolator.setDistanceCoefficient(coefficient)
129+
130+
writer = QgsGridFileWriter(interpolator,
131+
output,
132+
bbox,
133+
columns,
134+
rows,
135+
cellsizeX,
136+
cellsizeY)
137+
138+
writer.writeFile()
Lines changed: 135 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,135 @@
1+
# -*- coding: utf-8 -*-
2+
3+
"""
4+
***************************************************************************
5+
IdwInterpolationZValue.py
6+
---------------------
7+
Date : October 2016
8+
Copyright : (C) 2016 by Alexander Bruy
9+
Email : alexander dot bruy at gmail dot com
10+
***************************************************************************
11+
* *
12+
* This program is free software; you can redistribute it and/or modify *
13+
* it under the terms of the GNU General Public License as published by *
14+
* the Free Software Foundation; either version 2 of the License, or *
15+
* (at your option) any later version. *
16+
* *
17+
***************************************************************************
18+
"""
19+
20+
__author__ = 'Alexander Bruy'
21+
__date__ = 'October 2016'
22+
__copyright__ = '(C) 2016, Alexander Bruy'
23+
24+
# This will get replaced with a git SHA1 when you do a git archive
25+
26+
__revision__ = '$Format:%H$'
27+
28+
from qgis.core import QgsRectangle, QgsWkbTypes
29+
from qgis.analysis import (QgsInterpolator,
30+
QgsIDWInterpolator,
31+
QgsGridFileWriter
32+
)
33+
34+
from processing.core.GeoAlgorithm import GeoAlgorithm
35+
from processing.core.parameters import ParameterVector
36+
from processing.core.parameters import ParameterSelection
37+
from processing.core.parameters import ParameterNumber
38+
from processing.core.parameters import ParameterExtent
39+
from processing.core.outputs import OutputRaster
40+
from processing.tools import dataobjects
41+
42+
43+
class IdwInterpolationZValue(GeoAlgorithm):
44+
45+
INPUT_LAYER = 'INPUT_LAYER'
46+
LAYER_TYPE = 'LAYER_TYPE'
47+
DISTANCE_COEFFICIENT = 'DISTANCE_COEFFICIENT'
48+
COLUMNS = 'COLUMNS'
49+
ROWS = 'ROWS'
50+
CELLSIZE_X = 'CELLSIZE_X'
51+
CELLSIZE_Y = 'CELLSIZE_Y'
52+
EXTENT = 'EXTENT'
53+
OUTPUT_LAYER = 'OUTPUT_LAYER'
54+
55+
def defineCharacteristics(self):
56+
self.name, self.i18n_name = self.trAlgorithm('IDW interpolation (using Z-values)')
57+
self.group, self.i18n_group = self.trAlgorithm('Interpolation')
58+
59+
self.TYPES = [self.tr('Points'),
60+
self.tr('Structure lines'),
61+
self.tr('Break lines')
62+
]
63+
64+
self.addParameter(ParameterVector(self.INPUT_LAYER,
65+
self.tr('Vector layer')))
66+
self.addParameter(ParameterSelection(self.LAYER_TYPE,
67+
self.tr('Type'),
68+
self.TYPES,
69+
0))
70+
self.addParameter(ParameterNumber(self.DISTANCE_COEFFICIENT,
71+
self.tr('Distance coefficient P'),
72+
0.0, 99.99, 2.0))
73+
self.addParameter(ParameterNumber(self.COLUMNS,
74+
self.tr('Number of columns'),
75+
0, 10000000, 300))
76+
self.addParameter(ParameterNumber(self.ROWS,
77+
self.tr('Number of rows'),
78+
0, 10000000, 300))
79+
self.addParameter(ParameterNumber(self.CELLSIZE_X,
80+
self.tr('Cellsize X'),
81+
0.0, 999999.000000, 0.0))
82+
self.addParameter(ParameterNumber(self.CELLSIZE_Y,
83+
self.tr('Cellsize Y'),
84+
0.0, 999999.000000, 0.0))
85+
self.addParameter(ParameterExtent(self.EXTENT,
86+
self.tr('Extent')))
87+
self.addOutput(OutputRaster(self.OUTPUT_LAYER,
88+
self.tr('Interpolated')))
89+
90+
def processAlgorithm(self, progress):
91+
layer = dataobjects.getObjectFromUri(
92+
self.getParameterValue(self.INPUT_LAYER))
93+
layerType = self.getParameterValue(self.LAYER_TYPE)
94+
coefficient = self.getParameterValue(self.DISTANCE_COEFFICIENT)
95+
columns = self.getParameterValue(self.COLUMNS)
96+
rows = self.getParameterValue(self.ROWS)
97+
cellsizeX = self.getParameterValue(self.CELLSIZE_X)
98+
cellsizeY = self.getParameterValue(self.CELLSIZE_Y)
99+
extent = self.getParameterValue(self.EXTENT).split(',')
100+
output = self.getOutputValue(self.OUTPUT_LAYER)
101+
102+
if not QgsWkbTypes.hasZ(layer.wkbType()):
103+
raise GeoAlgorithmExecutionException(
104+
self.tr('Geometries in input layer does not have Z coordinates.'))
105+
106+
xMin = float(extent[0])
107+
xMax = float(extent[1])
108+
yMin = float(extent[2])
109+
yMax = float(extent[3])
110+
bbox = QgsRectangle(xMin, yMin, xMax, yMax)
111+
112+
layerData = QgsInterpolator.LayerData()
113+
layerData.vectorLayer = layer
114+
layerData.zCoordInterpolation = True
115+
layerData.interpolationAttribute = -1
116+
117+
if layerType == 0:
118+
layerData.mInputType = QgsInterpolator.POINTS
119+
elif layerType == 1:
120+
layerData.mInputType = QgsInterpolator.STRUCTURE_LINES
121+
else:
122+
layerData.mInputType = QgsInterpolator.BREAK_LINES
123+
124+
interpolator = QgsIDWInterpolator([layerData])
125+
interpolator.setDistanceCoefficient(coefficient)
126+
127+
writer = QgsGridFileWriter(interpolator,
128+
output,
129+
bbox,
130+
columns,
131+
rows,
132+
cellsizeX,
133+
cellsizeY)
134+
135+
writer.writeFile()

‎python/plugins/processing/algs/qgis/QGISAlgorithmProvider.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,6 +163,9 @@
163163
from .Ruggedness import Ruggedness
164164
from .Hillshade import Hillshade
165165
from .ReliefAuto import ReliefAuto
166+
from .IdwInterpolationZValue import IdwInterpolationZValue
167+
from .IdwInterpolationAttribute import IdwInterpolationAttribute
168+
166169

167170
pluginPath = os.path.normpath(os.path.join(
168171
os.path.split(os.path.dirname(__file__))[0], os.pardir))
@@ -220,7 +223,8 @@ def __init__(self):
220223
OffsetLine(), PolygonCentroids(), Translate(),
221224
SingleSidedBuffer(), PointsAlongGeometry(),
222225
Aspect(), Slope(), Ruggedness(), Hillshade(),
223-
ReliefAuto(),
226+
ReliefAuto(), IdwInterpolationZValue(),
227+
IdwInterpolationAttribute()
224228
]
225229

226230
if hasMatplotlib:

0 commit comments

Comments
 (0)
Please sign in to comment.