Add a style to gate layer in count trajectories and add a way to create a valid gate layer

.gitignore

@@ -6,3 +6,4 @@ start_ide.bat
 */.pytest_cache
 __pycache__
 fvh3t.egg-info
+symbology-style.db

fvh3t/core/gate_layer.py

@@ -60,7 +60,8 @@ def gates(self) -> tuple[Gate, ...]:
         return self.__gates
 
     def as_line_layer(self, traveler_class: str, start_time: QDateTime, end_time: QDateTime) -> QgsVectorLayer | None:
-        line_layer = QgsVectorLayer("LineString?crs=3067", "Line Layer", "memory")
+        line_layer = QgsVectorLayer("LineString", "Line Layer", "memory")
+        line_layer.setCrs(self.__layer.crs())
 
         line_layer.startEditing()
 

fvh3t/core/qgis_layer_utils.py

@@ -0,0 +1,45 @@
+from qgis.core import (
+    QgsCoordinateReferenceSystem,
+    QgsFeatureRenderer,
+    QgsField,
+    QgsFieldConstraints,
+    QgsReadWriteContext,
+    QgsVectorLayer,
+    edit,
+)
+from qgis.PyQt.QtCore import QVariant
+from qgis.PyQt.QtXml import QDomDocument
+
+from fvh3t.qgis_plugin_tools.tools.resources import resources_path
+
+
+class QgisLayerUtils:
+    @staticmethod
+    def set_gate_style(layer: QgsVectorLayer) -> None:
+        doc = QDomDocument()
+
+        with open(resources_path("style", "gate_style.xml")) as style_file:
+            doc.setContent(style_file.read())
+
+        renderer = QgsFeatureRenderer.load(doc.documentElement(), QgsReadWriteContext())
+        layer.setRenderer(renderer)
+
+    @staticmethod
+    def create_gate_layer(crs: QgsCoordinateReferenceSystem) -> QgsVectorLayer:
+        layer = QgsVectorLayer("LineString", "Gate Layer", "memory")
+        layer.setCrs(crs)
+
+        with edit(layer):
+            layer.addAttribute(QgsField("name", QVariant.String))
+            layer.addAttribute(QgsField("counts_negative", QVariant.Bool))
+            layer.addAttribute(QgsField("counts_positive", QVariant.Bool))
+
+        layer.setFieldConstraint(1, QgsFieldConstraints.ConstraintExpression)
+        layer.setConstraintExpression(1, '"counts_negative" OR "counts_positive"')
+
+        layer.setFieldConstraint(2, QgsFieldConstraints.ConstraintExpression)
+        layer.setConstraintExpression(2, '"counts_negative" OR "counts_positive"')
+
+        QgisLayerUtils.set_gate_style(layer)
+
+        return layer

fvh3t/fvh3t_processing/count_trajectories.py

@@ -12,11 +12,13 @@
     QgsProcessingParameterFeatureSink,
     QgsProcessingParameterString,
     QgsProcessingParameterVectorLayer,
+    QgsProcessingUtils,
     QgsUnitTypes,
 )
 from qgis.PyQt.QtCore import QCoreApplication, QDateTime
 
 from fvh3t.core.gate_layer import GateLayer
+from fvh3t.core.qgis_layer_utils import QgisLayerUtils
 from fvh3t.core.trajectory_layer import TrajectoryLayer
 
 
@@ -29,6 +31,8 @@ class CountTrajectories(QgsProcessingAlgorithm):
     OUTPUT_GATES = "OUTPUT_GATES"
     OUTPUT_TRAJECTORIES = "OUTPUT_TRAJECTORIES"
 
+    gate_dest_id: str | None = None
+
     def __init__(self) -> None:
         super().__init__()
 
@@ -221,7 +225,7 @@ def processAlgorithm(  # noqa N802
             msg = "Gate layer is None"
             raise ValueError(msg)
 
-        (sink, gate_dest_id) = self.parameterAsSink(
+        (sink, self.gate_dest_id) = self.parameterAsSink(
             parameters,
             self.OUTPUT_GATES,
             context,
@@ -233,4 +237,11 @@ def processAlgorithm(  # noqa N802
         for feature in exported_gate_layer.getFeatures():
             sink.addFeature(feature, QgsFeatureSink.FastInsert)
 
-        return {self.OUTPUT_TRAJECTORIES: traj_dest_id, self.OUTPUT_GATES: gate_dest_id}
+        return {self.OUTPUT_TRAJECTORIES: traj_dest_id, self.OUTPUT_GATES: self.gate_dest_id}
+
+    def postProcessAlgorithm(self, context: QgsProcessingContext, feedback: QgsProcessingFeedback) -> dict[str, Any]:  # noqa: N802
+        if self.gate_dest_id:
+            layer = QgsProcessingUtils.mapLayerFromString(self.gate_dest_id, context)
+            QgisLayerUtils.set_gate_style(layer)
+
+        return super().postProcessAlgorithm(context, feedback)

fvh3t/plugin.py

@@ -2,16 +2,17 @@
 
 from typing import Callable
 
-from qgis.core import QgsApplication
+from qgis.core import QgsApplication, QgsProject, QgsVectorLayer
 from qgis.PyQt.QtCore import QCoreApplication, QTranslator
 from qgis.PyQt.QtGui import QIcon
 from qgis.PyQt.QtWidgets import QAction, QWidget
 from qgis.utils import iface
 
+from fvh3t.core.qgis_layer_utils import QgisLayerUtils
 from fvh3t.fvh3t_processing.traffic_trajectory_toolkit_provider import TTTProvider
 from fvh3t.qgis_plugin_tools.tools.custom_logging import setup_logger, teardown_logger
-from fvh3t.qgis_plugin_tools.tools.i18n import setup_translation
-from fvh3t.qgis_plugin_tools.tools.resources import plugin_name
+from fvh3t.qgis_plugin_tools.tools.i18n import setup_translation, tr
+from fvh3t.qgis_plugin_tools.tools.resources import plugin_name, resources_path
 
 
 class Plugin:
@@ -109,11 +110,11 @@ def initProcessing(self):  # noqa N802
     def initGui(self) -> None:  # noqa N802
         """Create the menu entries and toolbar icons inside the QGIS GUI."""
         self.add_action(
-            "",
-            text=Plugin.name,
-            callback=self.run,
+            resources_path("icons", "add_gate.png"),
+            text=tr("Create gate layer"),
+            callback=self.create_gate_layer,
             parent=iface.mainWindow(),
-            add_to_toolbar=False,
+            add_to_toolbar=True,
         )
         self.initProcessing()
 
@@ -128,6 +129,7 @@ def unload(self) -> None:
         teardown_logger(Plugin.name)
         QgsApplication.processingRegistry().removeProvider(self.provider)
 
-    def run(self) -> None:
-        """Run method that performs all the real work"""
-        print("Hello QGIS plugin")  # noqa: T201
+    def create_gate_layer(self) -> None:
+        layer: QgsVectorLayer = QgisLayerUtils.create_gate_layer(QgsProject.instance().crs())
+
+        QgsProject.instance().addMapLayer(layer)