[processing] added Chainage algorithm as script

qgis · Apr 24, 2014 · 8436027 · 8436027
1 parent 1610d63
commit 8436027
Showing 1 changed file with 46 additions and 0 deletions.
diff --git a/python/plugins/processing/script/scripts/Chainage.py b/python/plugins/processing/script/scripts/Chainage.py
@@ -0,0 +1,46 @@
+##[Example scripts]=group
+##lines=vector
+##distance=number 1
+##startpoint=number 0
+##endpoint=number 0
+##output=output vector
+
+from qgis.core import *
+from PyQt4.QtCore import *
+from processing.core.VectorWriter import VectorWriter
+
+
+def create_points(feat):  
+    geom = feat.geometry()  
+    length = geom.length()
+    currentdistance = 0
+
+    if endpoint > 0:
+        length = endpoint
+
+    out = QgsFeature()
+
+    while startpoint + currentdistance <= length:        
+        point = geom.interpolate(startpoint + currentdistance)                
+        currentdistance = currentdistance + distance
+        out.setGeometry(point)    
+        attrs = feat.attributes()                  
+        attrs.append(currentdistance)    
+        out.setAttributes(attrs)
+        writer.addFeature(out)    
+
+
+layer = processing.getObject(lines)
+fields = layer.dataProvider().fields()
+fields.append(QgsField('Distance', QVariant.Double))
+writer = VectorWriter(output, None, fields, QGis.WKBPoint,
+                      layer.crs())
+
+feats = processing.features(layer)
+nFeat = len(feats)
+for i, feat in enumerate(feats):
+    progress.setPercentage(int(100 * i / nFeat))    
+    create_points(feat)
+
+del writer
+