|
| 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() |
0 commit comments