Add ClipTrajectoriesByPolygonLayer

Author: anitagraser

qgis_processing/createTrajectoriesAlgorithm.py

@@ -1,14 +1,6 @@
 import sys
 
-from qgis.PyQt.QtCore import QCoreApplication, QVariant
-from qgis.core import (
-    QgsField,
-    QgsGeometry,
-    QgsFeature,
-    QgsFeatureSink,
-    QgsProcessingParameterString,
-    QgsWkbTypes,
-)
+from qgis.PyQt.QtCore import QCoreApplication
 
 sys.path.append("..")
 
@@ -17,7 +9,6 @@
 
 
 class CreateTrajectoriesAlgorithm(TrajectoriesAlgorithm):
-
     def __init__(self):
         super().__init__()
 

qgis_processing/overlayAlgorithm.py

@@ -1,23 +1,12 @@
-import os
 import sys
 
 import shapely.wkt
 from shapely.geometry import Polygon
 
-from qgis.PyQt.QtCore import QCoreApplication, QVariant
-from qgis.PyQt.QtGui import QIcon
+from qgis.PyQt.QtCore import QCoreApplication
 from qgis.core import (
-    QgsField,
-    QgsFields,
-    QgsGeometry,
-    QgsFeature,
-    QgsFeatureSink,
-    QgsFeatureRequest,
-    QgsProcessing,
-    QgsProcessingAlgorithm,
-    QgsProcessingParameterFeatureSource,
-    QgsProcessingParameterString,
     QgsProcessingParameterExtent,
+    QgsProcessingParameterVectorLayer,
 )
 
 sys.path.append("..")
@@ -26,7 +15,18 @@
 from .trajectoriesAlgorithm import TrajectoriesAlgorithm
 
 
-class ClipTrajectoriesByExtentAlgorithm(TrajectoriesAlgorithm):
+class OverlayTrajectoriesAlgorithm(TrajectoriesAlgorithm):
+    def __init__(self):
+        super().__init__()
+
+    def group(self):
+        return self.tr("Trajectory overlay")
+
+    def groupId(self):
+        return "TrajectoryOverlay"
+
+
+class ClipTrajectoriesByExtentAlgorithm(OverlayTrajectoriesAlgorithm):
     EXTENT = "EXTENT"
 
     def __init__(self):
@@ -49,12 +49,6 @@ def tr(self, text):
     def displayName(self):
         return self.tr("Clip trajectories by extent")
 
-    def group(self):
-        return self.tr("Trajectory overlay")
-
-    def groupId(self):
-        return "TrajectoryOverlay"
-
     def shortHelpString(self):
         return self.tr(
             "<p>Creates a trajectory point layers with speed and direction information "
@@ -69,15 +63,59 @@ def shortHelpString(self):
     def helpUrl(self):
         return "https://movingpandas.org/units"
 
-    def createInstance(self):
-        return type(self)()
-
     def processTc(self, tc, parameters, context):
         extent = self.parameterAsExtent(parameters, self.EXTENT, context)
         extent = shapely.wkt.loads(extent.asWktPolygon())
-        tc = tc.clip(extent)
-        tc_to_sink(tc, self.sink_pts, self.fields_pts, self.timestamp_field)
-        for split in tc:
-            traj_to_sink(split, self.sink_trajs)
+        clipped = tc.clip(extent)
+        tc_to_sink(clipped, self.sink_pts, self.fields_pts, self.timestamp_field)
+        for traj in clipped:
+            traj_to_sink(traj, self.sink_trajs)
+
+
+class ClipTrajectoriesByPolygonLayer(OverlayTrajectoriesAlgorithm):
+    CLIP_LAYER = "CLIP_LAYER"
+
+    def __init__(self):
+        super().__init__()
+
+    def initAlgorithm(self, config=None):
+        super().initAlgorithm(config)
+        self.addParameter(
+            QgsProcessingParameterVectorLayer(
+                name=self.CLIP_LAYER,
+                description=self.tr("Overlay layer"),
+                optional=False,
+            )
+        )
+
+    def name(self):
+        return "clip_traj_vector"
+
+    def tr(self, text):
+        return QCoreApplication.translate("clip_traj_vector", text)
+
+    def displayName(self):
+        return self.tr("Clip trajectories by polygon layer")
+
+    def shortHelpString(self):
+        return self.tr(
+            "<p>Creates a trajectory point layers with speed and direction information "
+            "as well as a trajectory line layer clipped by the specified vector layer.</p>"
+            "<p><b>Speed</b> is calculated based on the input layer CRS information and "
+            "converted to the desired speed units. For more info on the supported units, "
+            "see https://movingpandas.org/units</p>"
+            "<p><b>Direction</b> is calculated between consecutive locations. Direction "
+            "values are in degrees, starting North turning clockwise.</p>"
+        )
+
+    def helpUrl(self):
+        return "https://movingpandas.org/units"
 
-        
+    def processTc(self, tc, parameters, context):
+        vlayer = self.parameterAsVectorLayer(parameters, self.CLIP_LAYER, context)
+        for feature in vlayer.getFeatures():
+            extent = shapely.wkt.loads(feature.geometry().asWkt())
+            clipped = tc.clip(extent)
+            tc_to_sink(clipped, self.sink_pts, self.fields_pts, self.timestamp_field)
+            for traj in clipped:
+                traj_to_sink(traj, self.sink_trajs)

qgis_processing/qgisUtils.py

@@ -1,9 +1,5 @@
 import sys
 import pandas as pd
-from geopandas import GeoDataFrame
-from PyQt5 import QtCore
-from shapely.geometry import Point
-from datetime import datetime
 from pyproj import CRS
 
 sys.path.append("..")
@@ -14,10 +10,7 @@
     QgsGeometry,
     QgsPointXY,
     QgsFeatureSink,
-    QgsFields,
-    QgsField,
 )
-from qgis.PyQt.QtCore import QVariant
 
 
 def trajectories_from_qgis_point_layer(
@@ -59,7 +52,10 @@ def feature_from_gdf_row(row):
 
 
 def tc_to_sink(tc, sink, fields, timestamp_field):
-    gdf = tc.to_point_gdf()
+    try:
+        gdf = tc.to_point_gdf()
+    except ValueError:  # when the tc is empty
+        return
     gdf[timestamp_field] = gdf.index.astype(str)
     names = [field.name() for field in fields]
     names.append("geometry")

qgis_processing/splitTrajectoriesAlgorithm.py

@@ -1,10 +1,13 @@
 import sys
 import pandas as pd
 
-from datetime import timedelta
 from movingpandas import TemporalSplitter, ObservationGapSplitter, StopSplitter
 
-from qgis.core import QgsProcessingParameterString, QgsProcessingParameterEnum, QgsProcessingParameterNumber
+from qgis.core import (
+    QgsProcessingParameterString,
+    QgsProcessingParameterEnum,
+    QgsProcessingParameterNumber,
+)
 
 sys.path.append("..")
 
@@ -124,7 +127,6 @@ def processTc(self, tc, parameters, context):
                 traj_to_sink(split, self.sink_trajs)
 
 
-
 class StopSplitterAlgorithm(SplitTrajectoriesAlgorithm):
     MAX_DIAMETER = "MAX_DIAMETER"
     MIN_DURATION = "MIN_DURATION"
@@ -145,7 +147,9 @@ def initAlgorithm(self, config=None):
         self.addParameter(
             QgsProcessingParameterString(
                 name=self.MIN_DURATION,
-                description=self.tr("Min stop duration (timedelta, e.g. 1 hours, 15 minutes)"),
+                description=self.tr(
+                    "Min stop duration (timedelta, e.g. 1 hours, 15 minutes)"
+                ),
                 defaultValue="15 minutes",
                 optional=False,
             )
@@ -174,11 +178,9 @@ def processTc(self, tc, parameters, context):
         min_duration = self.parameterAsString(parameters, self.MIN_DURATION, context)
         min_duration = pd.Timedelta(min_duration).to_pytimedelta()
         for traj in tc.trajectories:
-            splits = StopSplitter(traj).split(max_diameter=max_diameter, min_duration=min_duration)
+            splits = StopSplitter(traj).split(
+                max_diameter=max_diameter, min_duration=min_duration
+            )
             tc_to_sink(splits, self.sink_pts, self.fields_pts, self.timestamp_field)
             for split in splits:
                 traj_to_sink(split, self.sink_trajs)
-
-
-                
-   

qgis_processing/trajectoolsProvider.py

@@ -12,9 +12,12 @@
 from .splitTrajectoriesAlgorithm import (
     ObservationGapSplitterAlgorithm,
     TemporalSplitterAlgorithm,
-    StopSplitterAlgorithm
+    StopSplitterAlgorithm,
+)
+from .overlayAlgorithm import (
+    ClipTrajectoriesByExtentAlgorithm,
+    ClipTrajectoriesByPolygonLayer,
 )
-from .overlayAlgorithm import ClipTrajectoriesByExtentAlgorithm
 
 pluginPath = os.path.dirname(__file__)
 
@@ -52,7 +55,8 @@ def getAlgs(self):
             ObservationGapSplitterAlgorithm(),
             TemporalSplitterAlgorithm(),
             StopSplitterAlgorithm(),
-            ClipTrajectoriesByExtentAlgorithm()
+            ClipTrajectoriesByExtentAlgorithm(),
+            ClipTrajectoriesByPolygonLayer(),
         ]
         return algs
 

qgis_processing/trajectoriesAlgorithm.py

@@ -13,7 +13,7 @@
     QgsWkbTypes,
     QgsProcessingParameterString,
     QgsField,
-    QgsFields
+    QgsFields,
 )
 
 sys.path.append("..")
@@ -167,7 +167,7 @@ def get_pt_fields(self, fields_to_add=[]):
         for field in self.input_layer.fields():
             if field.name() == "fid":
                 continue
-            elif (field.name() == self.traj_id_field):  
+            elif field.name() == self.traj_id_field:
                 # we need to make sure the ID field is String
                 fields.append(QgsField(self.traj_id_field, QVariant.String))
             else: