mirror of
https://github.com/qgis/QGIS.git
synced 2025-04-13 00:03:09 -04:00
[processing] expose IDW interpolation from Interpolation plugin in toolbox
This commit is contained in:
Alexander Bruy
parent
5d4cf2fd4d
commit
f7575e08f5
138
python/plugins/processing/algs/qgis/IdwInterpolationAttribute.py
Normal file
138
python/plugins/processing/algs/qgis/IdwInterpolationAttribute.py
Normal file
@ -0,0 +1,138 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
***************************************************************************
|
||||
IdwInterpolationAttribute.py
|
||||
---------------------
|
||||
Date : October 2016
|
||||
Copyright : (C) 2016 by Alexander Bruy
|
||||
Email : alexander dot bruy 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__ = 'Alexander Bruy'
|
||||
__date__ = 'October 2016'
|
||||
__copyright__ = '(C) 2016, Alexander Bruy'
|
||||
|
||||
# This will get replaced with a git SHA1 when you do a git archive
|
||||
|
||||
__revision__ = '$Format:%H$'
|
||||
|
||||
from qgis.core import QgsRectangle
|
||||
from qgis.analysis import (QgsInterpolator,
|
||||
QgsIDWInterpolator,
|
||||
QgsGridFileWriter
|
||||
)
|
||||
|
||||
from processing.core.GeoAlgorithm import GeoAlgorithm
|
||||
from processing.core.parameters import ParameterVector
|
||||
from processing.core.parameters import ParameterTableField
|
||||
from processing.core.parameters import ParameterSelection
|
||||
from processing.core.parameters import ParameterNumber
|
||||
from processing.core.parameters import ParameterExtent
|
||||
from processing.core.outputs import OutputRaster
|
||||
from processing.tools import dataobjects
|
||||
|
||||
|
||||
class IdwInterpolationAttribute(GeoAlgorithm):
|
||||
|
||||
INPUT_LAYER = 'INPUT_LAYER'
|
||||
ATTRIBUTE = 'ATTRIBUTE'
|
||||
LAYER_TYPE = 'LAYER_TYPE'
|
||||
DISTANCE_COEFFICIENT = 'DISTANCE_COEFFICIENT'
|
||||
COLUMNS = 'COLUMNS'
|
||||
ROWS = 'ROWS'
|
||||
CELLSIZE_X = 'CELLSIZE_X'
|
||||
CELLSIZE_Y = 'CELLSIZE_Y'
|
||||
EXTENT = 'EXTENT'
|
||||
OUTPUT_LAYER = 'OUTPUT_LAYER'
|
||||
|
||||
def defineCharacteristics(self):
|
||||
self.name, self.i18n_name = self.trAlgorithm('IDW interpolation (using attribute)')
|
||||
self.group, self.i18n_group = self.trAlgorithm('Interpolation')
|
||||
|
||||
self.TYPES = [self.tr('Points'),
|
||||
self.tr('Structure lines'),
|
||||
self.tr('Break lines')
|
||||
]
|
||||
|
||||
self.addParameter(ParameterVector(self.INPUT_LAYER,
|
||||
self.tr('Vector layer')))
|
||||
self.addParameter(ParameterTableField(self.ATTRIBUTE,
|
||||
self.tr('Interpolation attribute'),
|
||||
self.INPUT_LAYER,
|
||||
ParameterTableField.DATA_TYPE_NUMBER))
|
||||
self.addParameter(ParameterSelection(self.LAYER_TYPE,
|
||||
self.tr('Type'),
|
||||
self.TYPES,
|
||||
0))
|
||||
self.addParameter(ParameterNumber(self.DISTANCE_COEFFICIENT,
|
||||
self.tr('Distance coefficient P'),
|
||||
0.0, 99.99, 2.0))
|
||||
self.addParameter(ParameterNumber(self.COLUMNS,
|
||||
self.tr('Number of columns'),
|
||||
0, 10000000, 300))
|
||||
self.addParameter(ParameterNumber(self.ROWS,
|
||||
self.tr('Number of rows'),
|
||||
0, 10000000, 300))
|
||||
self.addParameter(ParameterNumber(self.CELLSIZE_X,
|
||||
self.tr('Cellsize X'),
|
||||
0.0, 999999.000000, 0.0))
|
||||
self.addParameter(ParameterNumber(self.CELLSIZE_Y,
|
||||
self.tr('Cellsize Y'),
|
||||
0.0, 999999.000000, 0.0))
|
||||
self.addParameter(ParameterExtent(self.EXTENT,
|
||||
self.tr('Extent')))
|
||||
self.addOutput(OutputRaster(self.OUTPUT_LAYER,
|
||||
self.tr('Interpolated')))
|
||||
|
||||
def processAlgorithm(self, progress):
|
||||
layer = dataobjects.getObjectFromUri(
|
||||
self.getParameterValue(self.INPUT_LAYER))
|
||||
fieldName = self.getParameterValue(self.ATTRIBUTE)
|
||||
layerType = self.getParameterValue(self.LAYER_TYPE)
|
||||
coefficient = self.getParameterValue(self.DISTANCE_COEFFICIENT)
|
||||
columns = self.getParameterValue(self.COLUMNS)
|
||||
rows = self.getParameterValue(self.ROWS)
|
||||
cellsizeX = self.getParameterValue(self.CELLSIZE_X)
|
||||
cellsizeY = self.getParameterValue(self.CELLSIZE_Y)
|
||||
extent = self.getParameterValue(self.EXTENT).split(',')
|
||||
output = self.getOutputValue(self.OUTPUT_LAYER)
|
||||
|
||||
xMin = float(extent[0])
|
||||
xMax = float(extent[1])
|
||||
yMin = float(extent[2])
|
||||
yMax = float(extent[3])
|
||||
bbox = QgsRectangle(xMin, yMin, xMax, yMax)
|
||||
|
||||
layerData = QgsInterpolator.LayerData()
|
||||
layerData.vectorLayer = layer
|
||||
layerData.zCoordInterpolation = False
|
||||
layerData.interpolationAttribute = layer.dataProvider().fieldNameIndex(fieldName)
|
||||
|
||||
if layerType == 0:
|
||||
layerData.mInputType = QgsInterpolator.POINTS
|
||||
elif layerType == 1:
|
||||
layerData.mInputType = QgsInterpolator.STRUCTURE_LINES
|
||||
else:
|
||||
layerData.mInputType = QgsInterpolator.BREAK_LINES
|
||||
|
||||
interpolator = QgsIDWInterpolator([layerData])
|
||||
interpolator.setDistanceCoefficient(coefficient)
|
||||
|
||||
writer = QgsGridFileWriter(interpolator,
|
||||
output,
|
||||
bbox,
|
||||
columns,
|
||||
rows,
|
||||
cellsizeX,
|
||||
cellsizeY)
|
||||
|
||||
writer.writeFile()
|
||||
135
python/plugins/processing/algs/qgis/IdwInterpolationZValue.py
Normal file
135
python/plugins/processing/algs/qgis/IdwInterpolationZValue.py
Normal file
@ -0,0 +1,135 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
"""
|
||||
***************************************************************************
|
||||
IdwInterpolationZValue.py
|
||||
---------------------
|
||||
Date : October 2016
|
||||
Copyright : (C) 2016 by Alexander Bruy
|
||||
Email : alexander dot bruy 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__ = 'Alexander Bruy'
|
||||
__date__ = 'October 2016'
|
||||
__copyright__ = '(C) 2016, Alexander Bruy'
|
||||
|
||||
# This will get replaced with a git SHA1 when you do a git archive
|
||||
|
||||
__revision__ = '$Format:%H$'
|
||||
|
||||
from qgis.core import QgsRectangle, QgsWkbTypes
|
||||
from qgis.analysis import (QgsInterpolator,
|
||||
QgsIDWInterpolator,
|
||||
QgsGridFileWriter
|
||||
)
|
||||
|
||||
from processing.core.GeoAlgorithm import GeoAlgorithm
|
||||
from processing.core.parameters import ParameterVector
|
||||
from processing.core.parameters import ParameterSelection
|
||||
from processing.core.parameters import ParameterNumber
|
||||
from processing.core.parameters import ParameterExtent
|
||||
from processing.core.outputs import OutputRaster
|
||||
from processing.tools import dataobjects
|
||||
|
||||
|
||||
class IdwInterpolationZValue(GeoAlgorithm):
|
||||
|
||||
INPUT_LAYER = 'INPUT_LAYER'
|
||||
LAYER_TYPE = 'LAYER_TYPE'
|
||||
DISTANCE_COEFFICIENT = 'DISTANCE_COEFFICIENT'
|
||||
COLUMNS = 'COLUMNS'
|
||||
ROWS = 'ROWS'
|
||||
CELLSIZE_X = 'CELLSIZE_X'
|
||||
CELLSIZE_Y = 'CELLSIZE_Y'
|
||||
EXTENT = 'EXTENT'
|
||||
OUTPUT_LAYER = 'OUTPUT_LAYER'
|
||||
|
||||
def defineCharacteristics(self):
|
||||
self.name, self.i18n_name = self.trAlgorithm('IDW interpolation (using Z-values)')
|
||||
self.group, self.i18n_group = self.trAlgorithm('Interpolation')
|
||||
|
||||
self.TYPES = [self.tr('Points'),
|
||||
self.tr('Structure lines'),
|
||||
self.tr('Break lines')
|
||||
]
|
||||
|
||||
self.addParameter(ParameterVector(self.INPUT_LAYER,
|
||||
self.tr('Vector layer')))
|
||||
self.addParameter(ParameterSelection(self.LAYER_TYPE,
|
||||
self.tr('Type'),
|
||||
self.TYPES,
|
||||
0))
|
||||
self.addParameter(ParameterNumber(self.DISTANCE_COEFFICIENT,
|
||||
self.tr('Distance coefficient P'),
|
||||
0.0, 99.99, 2.0))
|
||||
self.addParameter(ParameterNumber(self.COLUMNS,
|
||||
self.tr('Number of columns'),
|
||||
0, 10000000, 300))
|
||||
self.addParameter(ParameterNumber(self.ROWS,
|
||||
self.tr('Number of rows'),
|
||||
0, 10000000, 300))
|
||||
self.addParameter(ParameterNumber(self.CELLSIZE_X,
|
||||
self.tr('Cellsize X'),
|
||||
0.0, 999999.000000, 0.0))
|
||||
self.addParameter(ParameterNumber(self.CELLSIZE_Y,
|
||||
self.tr('Cellsize Y'),
|
||||
0.0, 999999.000000, 0.0))
|
||||
self.addParameter(ParameterExtent(self.EXTENT,
|
||||
self.tr('Extent')))
|
||||
self.addOutput(OutputRaster(self.OUTPUT_LAYER,
|
||||
self.tr('Interpolated')))
|
||||
|
||||
def processAlgorithm(self, progress):
|
||||
layer = dataobjects.getObjectFromUri(
|
||||
self.getParameterValue(self.INPUT_LAYER))
|
||||
layerType = self.getParameterValue(self.LAYER_TYPE)
|
||||
coefficient = self.getParameterValue(self.DISTANCE_COEFFICIENT)
|
||||
columns = self.getParameterValue(self.COLUMNS)
|
||||
rows = self.getParameterValue(self.ROWS)
|
||||
cellsizeX = self.getParameterValue(self.CELLSIZE_X)
|
||||
cellsizeY = self.getParameterValue(self.CELLSIZE_Y)
|
||||
extent = self.getParameterValue(self.EXTENT).split(',')
|
||||
output = self.getOutputValue(self.OUTPUT_LAYER)
|
||||
|
||||
if not QgsWkbTypes.hasZ(layer.wkbType()):
|
||||
raise GeoAlgorithmExecutionException(
|
||||
self.tr('Geometries in input layer does not have Z coordinates.'))
|
||||
|
||||
xMin = float(extent[0])
|
||||
xMax = float(extent[1])
|
||||
yMin = float(extent[2])
|
||||
yMax = float(extent[3])
|
||||
bbox = QgsRectangle(xMin, yMin, xMax, yMax)
|
||||
|
||||
layerData = QgsInterpolator.LayerData()
|
||||
layerData.vectorLayer = layer
|
||||
layerData.zCoordInterpolation = True
|
||||
layerData.interpolationAttribute = -1
|
||||
|
||||
if layerType == 0:
|
||||
layerData.mInputType = QgsInterpolator.POINTS
|
||||
elif layerType == 1:
|
||||
layerData.mInputType = QgsInterpolator.STRUCTURE_LINES
|
||||
else:
|
||||
layerData.mInputType = QgsInterpolator.BREAK_LINES
|
||||
|
||||
interpolator = QgsIDWInterpolator([layerData])
|
||||
interpolator.setDistanceCoefficient(coefficient)
|
||||
|
||||
writer = QgsGridFileWriter(interpolator,
|
||||
output,
|
||||
bbox,
|
||||
columns,
|
||||
rows,
|
||||
cellsizeX,
|
||||
cellsizeY)
|
||||
|
||||
writer.writeFile()
|
||||
@ -163,6 +163,9 @@ from .Slope import Slope
|
||||
from .Ruggedness import Ruggedness
|
||||
from .Hillshade import Hillshade
|
||||
from .ReliefAuto import ReliefAuto
|
||||
from .IdwInterpolationZValue import IdwInterpolationZValue
|
||||
from .IdwInterpolationAttribute import IdwInterpolationAttribute
|
||||
|
||||
|
||||
pluginPath = os.path.normpath(os.path.join(
|
||||
os.path.split(os.path.dirname(__file__))[0], os.pardir))
|
||||
@ -220,7 +223,8 @@ class QGISAlgorithmProvider(AlgorithmProvider):
|
||||
OffsetLine(), PolygonCentroids(), Translate(),
|
||||
SingleSidedBuffer(), PointsAlongGeometry(),
|
||||
Aspect(), Slope(), Ruggedness(), Hillshade(),
|
||||
ReliefAuto(),
|
||||
ReliefAuto(), IdwInterpolationZValue(),
|
||||
IdwInterpolationAttribute()
|
||||
]
|
||||
|
||||
if hasMatplotlib:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user