def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue( parameters[self.INTERPOLATION_DATA]) coefficient = self.parameterAsDouble(parameters, self.DISTANCE_COEFFICIENT, 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 = [] for i, row in enumerate(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 data.transformContext = context.transformContext() layers.append(layer) data.valueSource = int(v[1]) data.interpolationAttribute = int(v[2]) if data.valueSource == QgsInterpolator.ValueAttribute and data.interpolationAttribute == -1: raise QgsProcessingException( self. tr('Layer {} is set to use a value attribute, but no attribute was set' .format(i + 1))) if v[3] == '0': data.sourceType = QgsInterpolator.SourcePoints elif v[3] == '1': data.sourceType = QgsInterpolator.SourceStructureLines else: data.sourceType = QgsInterpolator.SourceBreakLines layerData.append(data) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows) writer.writeFile(feedback) return {self.OUTPUT: output}
def processAlgorithm(self, feedback): interpolationData = self.getParameterValue(self.INTERPOLATION_DATA) 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 interpolationData is None: raise GeoAlgorithmExecutionException( self.tr('You need to specify at least one input layer.')) 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 = dataobjects.getLayerFromString(v[0]) 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) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows, cellsizeX, cellsizeY) writer.writeFile()
def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue(parameters[self.INTERPOLATION_DATA]) coefficient = self.parameterAsDouble(parameters, self.DISTANCE_COEFFICIENT, 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 = [] 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) 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) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows) writer.writeFile(feedback) return {self.OUTPUT: output}
def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue( parameters[self.INTERPOLATION_DATA]) coefficient = self.parameterAsDouble(parameters, self.DISTANCE_COEFFICIENT, 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.')) 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) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows, cellsizeX, cellsizeY) writer.writeFile(feedback) return {self.OUTPUT: output}
def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue(parameters[self.INTERPOLATION_DATA]) coefficient = self.parameterAsDouble(parameters, self.DISTANCE_COEFFICIENT, 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.')) 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) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows, cellsizeX, cellsizeY) writer.writeFile(feedback) return {self.OUTPUT: output}
def interpolation(self, layer, attribute_for_interpolation, attribute_name, output_dir, resolution): #create interpolation-object layer_data = QgsInterpolator.LayerData() layer_data.source=layer layer_data.interpolationAttribute=attribute_for_interpolation layer_data.valueSource=0 layer_data.sourceType=0 # #add the given layer to the interpolation-object # layer_data.vectorLayer = layer # # #use the given attribute instead of the z coordinate for interpolation # layer_data.zCoordInterpolation=False # layer_data.interpolationAttribute = attribute_for_interpolation # layer_data.mInputType = 1 #interpolate the layer interpolator = QgsIDWInterpolator([layer_data]) #create the resulting raster rect = layer.extent() ncol = int((rect.xMaximum() - rect.xMinimum()) / resolution) nrows = int((rect.yMaximum() - rect.yMinimum()) / resolution) #create outut directory export_folder = QDir.toNativeSeparators(output_dir + "/batch_interpolation/") if not os.path.exists(export_folder): os.makedirs(export_folder) #write raster to file system export_path = QDir.toNativeSeparators(export_folder + layer.name() + "_" + attribute_name + ".tif") #QMessageBox.about(None,"sss","4") output = QgsGridFileWriter(interpolator, export_path, rect, ncol, nrows) #QMessageBox.about(None,"sss","5") a=output.writeFile()
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()
def processAlgorithm(self, parameters, context, feedback): interpolationData = ParameterInterpolationData.parseValue(parameters[self.INTERPOLATION_DATA]) coefficient = self.parameterAsDouble(parameters, self.DISTANCE_COEFFICIENT, context) columns = self.parameterAsInt(parameters, self.COLUMNS, context) rows = self.parameterAsInt(parameters, self.ROWS, 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.')) layerData = [] layers = [] 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) 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) interpolator = QgsIDWInterpolator(layerData) interpolator.setDistanceCoefficient(coefficient) writer = QgsGridFileWriter(interpolator, output, bbox, columns, rows) writer.writeFile(feedback) return {self.OUTPUT: output}
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()
def run(self): """Run method that performs all the real work""" # Add items to Management Practices & Soil Type k_lists = [] m_lists = [] k_lists = self.k_list() m_lists = self.m_list() self.dlg.comboBox.addItems(k_lists) self.dlg.comboBox_2.addItems(m_lists) # show dialog box self.dlg.show() # Run the dialog event loop result = self.dlg.exec_() # See if OK was pressed if result: # Save paths of input directories selectedBoundary = self.dlg.lineEdit.text() selectedDEM = self.dlg.lineEdit_2.text() selectedRLayer = self.dlg.lineEdit_3.text() selectedOutput = self.dlg.lineEdit_4.text() for letter in selectedOutput: if letter == "\\": selectedOutput = selectedOutput.replace(letter, "/") print(selectedBoundary) print(selectedDEM) print(selectedRLayer) print(selectedOutput) # Save indices for K and M selectedKLayer = self.dlg.comboBox.currentIndex() selectedMLayer = self.dlg.comboBox_2.currentIndex() boundary = QgsVectorLayer(selectedBoundary, 'Boundary', 'ogr') QgsMapLayerRegistry.instance().addMapLayer(boundary) entries = [] # Retrieve K and M values k_value = self.k_index(selectedKLayer) m_value = self.m_index(selectedMLayer) km_value = k_value * m_value km_value = str(km_value) # Process R index ## CSV to Layer uri = 'file:///' + selectedRLayer + '?delimiter=%s&xField=%s&yField=%s&crs=%s' % ( ",", "x", "y", "EPSG:4326") rainfall_unedited = QgsVectorLayer(uri, "rainfall", "delimitedtext") QgsMapLayerRegistry.instance().addMapLayer(rainfall_unedited) spatRef = QgsCoordinateReferenceSystem( 4326, QgsCoordinateReferenceSystem.EpsgCrsId) ## CSV points to editable shapefile rainfall_edited = QgsVectorFileWriter( selectedOutput + '/rainfall_edited.shp', None, rainfall_unedited.pendingFields(), QGis.WKBPoint, spatRef) pt = QgsPoint() outFeature = QgsFeature() for feat in rainfall_unedited.getFeatures(): attrs = feat.attributes() pt.setX(feat['x']) pt.setY(feat['y']) outFeature.setAttributes(attrs) outFeature.setGeometry(QgsGeometry.fromPoint(pt)) rainfall_edited.addFeature(outFeature) del rainfall_edited rainfall_edited2 = QgsVectorLayer( selectedOutput + '/rainfall_edited.shp', 'rainfall_edited', 'ogr') ## Add and calculate average field rainfall_edited2.startEditing() avgField = QgsField('average', QVariant.Double) rainfall_edited2.dataProvider().addAttributes([avgField]) rainfall_edited2.updateFields() idx = rainfall_edited2.fieldNameIndex('average') time_count = idx - 2 str_output = '' for i in range(1, time_count + 1): if i == 1: a = str(i) str_output += 'time' + a else: a = str(i) str_output += '+ time' + a e = QgsExpression(str_output) e.prepare(rainfall_edited2.pendingFields()) for f in rainfall_edited2.getFeatures(): f[idx] = e.evaluate(f) / time_count rainfall_edited2.updateFeature(f) rainfall_edited2.commitChanges() rainfall_edited3 = QgsVectorLayer( selectedOutput + '/rainfall_edited.shp', 'rainfall_edited', 'ogr') QgsMapLayerRegistry.instance().addMapLayer(rainfall_edited3) ## Interpolating average using IDW ### Parameters for interpolation idx = rainfall_edited3.fieldNameIndex('average') layer_data = qgis.analysis.QgsInterpolator.LayerData() layer_data.vectorLayer = rainfall_edited3 layer_data.zCoordInterpolation = False layer_data.interpolationAttribute = idx layer_data.mInputType = 1 idw_interpolator = QgsIDWInterpolator([layer_data]) ### Output parameter export_path = selectedOutput + "/interpolated_r_{}.asc".format(idx) rect = boundary.extent() res = 0.0001 ncol = (rect.xMaximum() - rect.xMinimum()) / res nrows = (rect.yMaximum() - rect.yMinimum()) / res interpolated_r = QgsGridFileWriter(idw_interpolator, export_path, rect, int(ncol), int(nrows), res, res) interpolated_r.writeFile(True) interpolated_r2 = QgsRasterLayer(export_path, "interpolated_r") ## Clip output to boundary clippedR = processing.runandload( 'gdalogr:cliprasterbymasklayer', interpolated_r2, #INPUT <ParameterRaster> boundary, #MASK <ParameterVector> "-9999", #NO_DATA <ParameterString> False, #ALPHA_BAND <ParameterBoolean> False, #CROP_TO_CUTLINE <ParameterBoolean> False, #KEEP_RESOLUTION <ParameterBoolean> 5, #RTYPE <ParameterSelection> 4, #COMPRESS <ParameterSelection> 1, #JPEGCOMPRESSION <ParameterNumber> 6, #ZLEVEL <ParameterNumber> 1, #PREDICTOR <ParameterNumber> False, #TILED <ParameterBoolean> 2, #BIGTIFF <ParameterSelection> False, #TFW <ParameterBoolean> "", #EXTRA <ParameterString> selectedOutput + '/clip_interpolated_r.tif') #OUTPUT <OutputRaster> r_layer = QgsRasterLayer( selectedOutput + '/clip_interpolated_r.tif', "R-index") boh4 = QgsRasterCalculatorEntry() boh4.ref = 'boh4@1' boh4.raster = r_layer boh4.bandNumber = 1 entries.append(boh4) # Process S index ## Load DEM bohLayer1 = QgsRasterLayer(selectedDEM, "DEM") boh2 = QgsRasterCalculatorEntry() boh2.ref = 'boh2@1' boh2.raster = bohLayer1 boh2.bandNumber = 1 entries.append(boh2) ## Clip output to boundary clippedOutput2 = processing.runandload( 'gdalogr:cliprasterbymasklayer', bohLayer1, # INPUT <ParameterRaster> boundary, # MASK <ParameterVector> "-9999", # NO_DATA <ParameterString> False, # ALPHA_BAND <ParameterBoolean> False, # CROP_TO_CUTLINE <ParameterBoolean> False, # KEEP_RESOLUTION <ParameterBoolean> 5, # RTYPE <ParameterSelection> 4, # COMPRESS <ParameterSelection> 1, # JPEGCOMPRESSION <ParameterNumber> 6, # ZLEVEL <ParameterNumber> 1, # PREDICTOR <ParameterNumber> False, # TILED <ParameterBoolean> 2, # BIGTIFF <ParameterSelection> False, # TFW <ParameterBoolean> "", # EXTRA <ParameterString> selectedOutput + '/clip_dem.tif') # OUTPUT <OutputRaster> bohLayer5 = QgsRasterLayer(selectedOutput + '/clip_dem.tif', "DEM-clipped") boh5 = QgsRasterCalculatorEntry() boh5.ref = 'boh5@1' boh5.raster = bohLayer5 boh5.bandNumber = 1 entries.append(boh5) ## GDAL algorithm for slope processing.runalg('gdalogr:slope', bohLayer5, 1, False, True, True, 111120, selectedOutput + '/slope(percent).tif') bohLayer6 = QgsRasterLayer(selectedOutput + '/slope(percent).tif', "slope(percent)") QgsMapLayerRegistry.instance().addMapLayer(bohLayer6) boh6 = QgsRasterCalculatorEntry() boh6.ref = 'boh6@1' boh6.raster = bohLayer6 boh6.bandNumber = 1 entries.append(boh6) # Process calculation with input extent and resolution calc = QgsRasterCalculator('(boh4@1 * boh6@1) *' + km_value, selectedOutput + '/soil_risk.tif', 'GTiff', bohLayer6.extent(), bohLayer6.width(), bohLayer6.height(), entries) calc.processCalculation() bohLayer4 = QgsRasterLayer(selectedOutput + '/soil_risk.tif', "Soil_Risk") QgsMapLayerRegistry.instance().addMapLayer(bohLayer4)