def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue( parameters[self.INTERPOLATION_DATA]) method = self.parameterAsEnum(parameters, self.METHOD, context) bbox = self.parameterAsExtent(parameters, self.EXTENT, context) pixel_size = self.parameterAsDouble(parameters, self.PIXEL_SIZE, context) output = self.parameterAsOutputLayer(parameters, self.OUTPUT, context) columns = self.parameterAsInt(parameters, self.COLUMNS, context) rows = self.parameterAsInt(parameters, self.ROWS, context) if columns == 0: columns = max(round(bbox.width() / pixel_size) + 1, 1) if rows == 0: rows = max(round(bbox.height() / pixel_size) + 1, 1) if interpolationData is None: raise QgsProcessingException( self.tr('You need to specify at least one input layer.')) layerData = [] layers = [] crs = QgsCoordinateReferenceSystem() for row in interpolationData.split(';'): v = row.split('::~::') data = QgsInterpolator.LayerData() # need to keep a reference until interpolation is complete layer = QgsProcessingUtils.variantToSource(v[0], context) data.source = layer layers.append(layer) if not crs.isValid(): crs = layer.sourceCrs() data.valueSource = int(v[1]) data.interpolationAttribute = int(v[2]) if v[3] == '0': data.sourceType = QgsInterpolator.SourcePoints elif v[3] == '1': data.sourceType = QgsInterpolator.SourceStructureLines else: data.sourceType = QgsInterpolator.SourceBreakLines layerData.append(data) if method == 0: interpolationMethod = QgsTinInterpolator.Linear else: interpolationMethod = QgsTinInterpolator.CloughTocher (triangulation_sink, triangulation_dest_id) = self.parameterAsSink( parameters, self.TRIANGULATION, context, QgsTinInterpolator.triangulationFields(), QgsWkbTypes.LineString, crs) interpolator = QgsTinInterpolator(layerData, interpolationMethod, feedback) if triangulation_sink is not None: interpolator.setTriangulationSink(triangulation_sink) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows) writer.writeFile(feedback) return {self.OUTPUT: output, self.TRIANGULATION: triangulation_dest_id}
def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue( parameters[self.INTERPOLATION_DATA]) method = self.parameterAsEnum(parameters, self.METHOD, context) columns = self.parameterAsInt(parameters, self.COLUMNS, context) rows = self.parameterAsInt(parameters, self.ROWS, context) cellsizeX = self.parameterAsDouble(parameters, self.CELLSIZE_X, context) cellsizeY = self.parameterAsDouble(parameters, self.CELLSIZE_Y, context) bbox = self.parameterAsExtent(parameters, self.EXTENT, context) output = self.parameterAsOutputLayer(parameters, self.OUTPUT, context) if interpolationData is None: raise QgsProcessingException( self.tr('You need to specify at least one input layer.')) if cellsizeX == 0.0 or cellsizeY == 0.0: raise QgsProcessingException( self.tr('Cellsize should be greater than 0.')) layerData = [] layers = [] crs = QgsCoordinateReferenceSystem() for row in interpolationData.split(';'): v = row.split(',') data = QgsInterpolator.LayerData() # need to keep a reference until interpolation is complete layer = QgsProcessingUtils.mapLayerFromString(v[0], context) data.vectorLayer = layer layers.append(layer) if not crs.isValid(): crs = layer.crs() data.zCoordInterpolation = bool(v[1]) data.interpolationAttribute = int(v[2]) if v[3] == '0': data.mInputType = QgsInterpolator.POINTS elif v[3] == '1': data.mInputType = QgsInterpolator.STRUCTURE_LINES else: data.mInputType = QgsInterpolator.BREAK_LINES layerData.append(data) if method == 0: interpolationMethod = QgsTINInterpolator.Linear else: interpolationMethod = QgsTINInterpolator.CloughTocher (triangulation_sink, triangulation_dest_id) = self.parameterAsSink( parameters, self.TRIANGULATION, context, QgsTINInterpolator.triangulationFields(), QgsWkbTypes.LineString, crs) interpolator = QgsTINInterpolator(layerData, interpolationMethod, feedback) if triangulation_sink is not None: interpolator.setTriangulationSink(triangulation_sink) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows, cellsizeX, cellsizeY) writer.writeFile(feedback) return {self.OUTPUT: output, self.TRIANGULATION: triangulation_dest_id}
def processAlgorithm(self, parameters, context, feedback): interpolationData = self.getParameterValue(self.INTERPOLATION_DATA) method = self.getParameterValue(self.METHOD) 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) triangulation = self.getOutputValue(self.TRIANULATION_FILE) if interpolationData is None: raise GeoAlgorithmExecutionException( self.tr('You need to specify at least one input layer.')) if cellsizeX == 0.0 or cellsizeY == 0.0: raise GeoAlgorithmExecutionException( self.tr('Cellsize should be greater than 0.')) xMin = float(extent[0]) xMax = float(extent[1]) yMin = float(extent[2]) yMax = float(extent[3]) bbox = QgsRectangle(xMin, yMin, xMax, yMax) layerData = [] layers = [] for row in interpolationData.split(';'): v = row.split(',') data = QgsInterpolator.LayerData() # need to keep a reference until interpolation is complete layer = QgsProcessingUtils.mapLayerFromString(v[0], context) data.vectorLayer = layer layers.append(layer) data.zCoordInterpolation = bool(v[1]) data.interpolationAttribute = int(v[2]) if v[3] == '0': data.mInputType = QgsInterpolator.POINTS elif v[3] == '1': data.mInputType = QgsInterpolator.STRUCTURE_LINES else: data.mInputType = QgsInterpolator.BREAK_LINES layerData.append(data) if method == 0: interpolationMethod = QgsTINInterpolator.Linear else: interpolationMethod = QgsTINInterpolator.CloughTocher interpolator = QgsTINInterpolator(layerData, interpolationMethod) interpolator.setExportTriangulationToFile(True) interpolator.setTriangulationFilePath(triangulation) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows, cellsizeX, cellsizeY) writer.writeFile()