Add missing concentration tables in QGIS.

Deltares · Sep 23, 2024 · 6e1c220 · 6e1c220
1 parent cae6838
commit 6e1c220
Showing 1 changed file with 58 additions and 0 deletions.
diff --git a/ribasim_qgis/core/nodes.py b/ribasim_qgis/core/nodes.py
@@ -357,6 +357,64 @@ def attributes(cls) -> list[QgsField]:
         ]
 
 
+class BasinConcentrationExternal(Input):
+    @classmethod
+    def input_type(cls) -> str:
+        return "Basin / concentration_external"
+
+    @classmethod
+    def geometry_type(cls) -> str:
+        return "No Geometry"
+
+    @classmethod
+    def attributes(cls) -> list[QgsField]:
+        return [
+            QgsField("time", QVariant.DateTime),
+            QgsField("node_id", QVariant.Int),
+            QgsField("substance", QVariant.String),
+            QgsField("concentration", QVariant.Double),
+        ]
+
+
+class BasinConcentrationState(Input):
+    @classmethod
+    def input_type(cls) -> str:
+        return "Basin / concentration_state"
+
+    @classmethod
+    def geometry_type(cls) -> str:
+        return "No Geometry"
+
+    @classmethod
+    def attributes(cls) -> list[QgsField]:
+        return [
+            QgsField("time", QVariant.DateTime),
+            QgsField("node_id", QVariant.Int),
+            QgsField("substance", QVariant.String),
+            QgsField("concentration", QVariant.Double),
+        ]
+
+
+class BasinConcentration(Input):
+    @classmethod
+    def input_type(cls) -> str:
+        return "Basin / concentration"
+
+    @classmethod
+    def geometry_type(cls) -> str:
+        return "No Geometry"
+
+    @classmethod
+    def attributes(cls) -> list[QgsField]:
+        return [
+            QgsField("time", QVariant.DateTime),
+            QgsField("node_id", QVariant.Int),
+            QgsField("substance", QVariant.String),
+            QgsField("drainage", QVariant.Double),
+            QgsField("precipitation", QVariant.Double),
+        ]
+
+
 class BasinSubgridLevel(Input):
     @classmethod
     def input_type(cls) -> str: