sharpetronics/QGIS
1
0
Fork 0
mirror of https://github.com/qgis/QGIS.git synced 2025-02-28 00:17:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

[processing] Make set m value accept dynamic m values

Browse Source
This commit is contained in:
Nyall Dawson 2017-11-27 08:03:23 +10:00
parent c5f65359d8
commit 8c9147eabd
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
python/plugins/processing/algs/qgis/SetMValue.py
View File

@ -29,6 +29,8 @@ import os
from qgis.core import (QgsGeometry,
QgsWkbTypes,
QgsPropertyDefinition,
QgsProcessingParameters,
QgsProcessingParameterNumber)
@ -47,6 +49,8 @@ class SetMValue(QgisFeatureBasedAlgorithm):
def __init__(self):
super().__init__()
self.m_value = 0
self.dynamic_m = False
self.m_property = None
def name(self):
return 'setmvalue'
@ -61,14 +65,21 @@ class SetMValue(QgisFeatureBasedAlgorithm):
return self.tr('set,add,m,measure,values').split(',')
def initParameters(self, config=None):
self.addParameter(QgsProcessingParameterNumber(self.M_VALUE,
self.tr('M Value'), QgsProcessingParameterNumber.Double, defaultValue=0.0))
m_param = QgsProcessingParameterNumber(self.M_VALUE,
self.tr('M Value'), QgsProcessingParameterNumber.Double, defaultValue=0.0)
m_param.setIsDynamic(True)
m_param.setDynamicLayerParameterName('INPUT')
m_param.setDynamicPropertyDefinition(QgsPropertyDefinition(self.M_VALUE, self.tr("M Value"), QgsPropertyDefinition.Double))
self.addParameter(m_param)
def outputWkbType(self, inputWkb):
return QgsWkbTypes.addM(inputWkb)
def prepareAlgorithm(self, parameters, context, feedback):
self.m_value = self.parameterAsDouble(parameters, self.M_VALUE, context)
self.dynamic_m = QgsProcessingParameters.isDynamic(parameters, self.M_VALUE)
if self.dynamic_m:
self.m_property = parameters[self.M_VALUE]
return True
def processFeature(self, feature, context, feedback):
@ -79,7 +90,10 @@ class SetMValue(QgisFeatureBasedAlgorithm):
# addMValue won't alter existing M values, so drop them first
new_geom.dropMValue()
new_geom.addMValue(self.m_value)
m = self.m_value
if self.dynamic_m:
m, ok = self.m_property.valueAsDouble(context.expressionContext(), m)
new_geom.addMValue(m)
feature.setGeometry(QgsGeometry(new_geom))