Enhance drains_pwn script by adding conductance properties and integrating open water drainage features

Author: bdestombe

src/nhflodata/data/mockup/drains_pwn/v1.0.0/drains_pwn.geojson

@@ -5,9 +5,10 @@
 "xy_coordinate_resolution": 0.01,
 "bbox": [ 107080.87, 521548.54, 107453.90, 521643.63 ],                                                                            
 "features": [
-{ "type": "Feature", "properties": { "name": "drain_dwk", "elevation": 0.73, "comment": "wordt met pomp bemalen" }, "bbox": [ 107080.87, 521548.54, 107253.51, 521590.54 ], "geometry": { "type": "LineString", "coordinates": [ [ 107080.87, 521548.54 ], [ 107253.51, 521590.54 ] ] } },
-{ "type": "Feature", "properties": { "name": "drain_pomgebouw_ZUID", "elevation": -0.62, "comment": "via put uitloop op -0.5mNAP" }, "bbox": [ 107341.82, 521617.28, 107406.43, 521639.18 ], "geometry": { "type": "LineString", "coordinates": [ [ 107342.3, 521639.18 ], [ 107341.82, 521617.28 ], [ 107394.21, 521617.91 ], [ 107406.43, 521627.28 ] ] } },
-{ "type": "Feature", "properties": { "name": "drain_pomgebouw_NRD", "elevation": -0.8, "comment": "via put uitloop op -0.5mNAP" }, "bbox": [ 107347.06, 521635.69, 107407.07, 521643.63 ], "geometry": { "type": "LineString", "coordinates": [ [ 107347.06, 521643.15 ], [ 107398.49, 521643.63 ], [ 107407.07, 521635.69 ] ] } },
-{ "type": "Feature", "properties": { "name": "drain_dienstgebouw", "elevation": -0.844, "comment": "uitloop onder water op -0.5mNAP" }, "bbox": [ 107422.94, 521619.66, 107453.9, 521636.01 ], "geometry": { "type": "LineString", "coordinates": [ [ 107453.9, 521636.01 ], [ 107453.74, 521620.29 ], [ 107436.12, 521619.66 ], [ 107422.94, 521629.18 ] ] } }
+{ "type": "Feature", "properties": { "name": "drain_dwk", "elevation": 0.73, "conductance_per_meter": 1.0, "conductance_per_squared_meter": null, "bgt-identificatie": "", "mover_lake_name": "", "comment": "wordt met pomp bemalen" }, "bbox": [ 107080.87, 521548.54, 107253.51, 521590.54 ], "geometry": { "type": "LineString", "coordinates": [ [ 107080.87, 521548.54 ], [ 107253.51, 521590.54 ] ] } },
+{ "type": "Feature", "properties": { "name": "drain_pomgebouw_ZUID", "elevation": -0.62, "conductance_per_meter": 1.0, "conductance_per_squared_meter": null, "bgt-identificatie": "", "mover_lake_name": "", "comment": "via put uitloop op -0.5mNAP" }, "bbox": [ 107341.82, 521617.28, 107406.43, 521639.18 ], "geometry": { "type": "LineString", "coordinates": [ [ 107342.3, 521639.18 ], [ 107341.82, 521617.28 ], [ 107394.21, 521617.91 ], [ 107406.43, 521627.28 ] ] } },
+{ "type": "Feature", "properties": { "name": "drain_pomgebouw_NRD", "elevation": -0.8, "conductance_per_meter": 1.0, "conductance_per_squared_meter": null, "bgt-identificatie": "", "mover_lake_name": "", "comment": "via put uitloop op -0.5mNAP" }, "bbox": [ 107347.06, 521635.69, 107407.07, 521643.63 ], "geometry": { "type": "LineString", "coordinates": [ [ 107347.06, 521643.15 ], [ 107398.49, 521643.63 ], [ 107407.07, 521635.69 ] ] } },
+{ "type": "Feature", "properties": { "name": "drain_dienstgebouw", "elevation": -0.844, "conductance_per_meter": 1.0, "conductance_per_squared_meter": null, "bgt-identificatie": "", "mover_lake_name": "", "comment": "uitloop onder water op -0.5mNAP" }, "bbox": [ 107422.94, 521619.66, 107453.9, 521636.01 ], "geometry": { "type": "LineString", "coordinates": [ [ 107453.9, 521636.01 ], [ 107453.74, 521620.29 ], [ 107436.12, 521619.66 ], [ 107422.94, 521629.18 ] ] } },
+{ "type": "Feature", "properties": { "name": "bassin", "elevation": -0.5, "conductance_per_meter": null, "conductance_per_squared_meter": 1.0, "bgt-identificatie": "", "mover_lake_name": "vijver 4", "comment": "" }, "bbox": [ 107403.66, 521624.72, 107425.88, 521641.05 ], "geometry": { "type": "Polygon", "coordinates": [ [ [ 107403.66, 521638.85 ], [ 107409.53, 521640.66 ], [ 107425.54, 521641.05 ], [ 107425.88, 521627.06 ], [ 107423.49, 521624.98 ], [ 107412.31, 521624.72 ], [ 107404.06, 521625.46 ], [ 107403.66, 521638.85 ] ] ] } }
 ]
 }

src/nhflodata/data/mockup/drains_pwn/v1.0.0/drains_pwn.py

@@ -1,26 +1,63 @@
 from pathlib import Path
 
 import geopandas as gpd
+import numpy as np
+import pandas as pd
 from nhflotools.geoconverter.utils import round_bounds, validate_crs, validate_geometry
 
 from nhflodata import get_abs_data_path
 
-gdf_dir = get_abs_data_path("drains_en_damwand_bergen")
-gdf = gpd.read_file(gdf_dir / "drains_en_damwand_bergen.geojson")
-gdf = gdf.loc[gdf.naam.str.startswith("drain")]
-gdf["elevation"] = gdf[["diepte_van", "diepte_tot"]].min(axis=1)
-gdf = gdf.rename(columns={"naam": "name", "opmerking": "comment"})
 columns_to_keep = [
     "name",
     "elevation",  # Elevation of the bottom of the drain
+    "conductance_per_meter",
+    "conductance_per_squared_meter",
+    "bgt-identificatie",
+    "mover_lake_name",
     "comment",
     "geometry",
 ]
-gdf = gdf[columns_to_keep]
+
+# %% Create line drains input
+gdf_dir = get_abs_data_path("drains_en_damwand_bergen")
+gdf = gpd.read_file(gdf_dir / "drains_en_damwand_bergen.geojson")
+gdf = gdf.loc[gdf.naam.str.startswith("drain")]
+gdf["elevation"] = gdf[["diepte_van", "diepte_tot"]].min(axis=1)
+gdf = gdf.rename(columns={"naam": "name", "opmerking": "comment"})
 
 # Add conductances
 # m3/day per meter head difference per meter drain length
 gdf["conductance_per_meter"] = 1.0  # Default conductance
+gdf["conductance_per_squared_meter"] = np.nan  # Default conductance for open water
+gdf["bgt-identificatie"] = ""  # Placeholder for BGT identification
+gdf["mover_lake_name"] = ""  # Placeholder for mover lake boundary name, to move drained water to.
+gdf = gdf[columns_to_keep]
+
+# %% Add open water drainage
+# m3/day per meter head difference  per squaredd meter
+bergen_oppwater_path = get_abs_data_path("oppervlaktewater_bergen")
+oppwater = gpd.read_file(bergen_oppwater_path / "oppervlaktewater_bergen.geojson")
+
+open_water_drainage = oppwater.loc[oppwater["naam"].eq("bassin")]
+gdf = pd.concat(
+    (
+        gdf,
+        gpd.GeoDataFrame(
+            data={
+                "name": open_water_drainage["naam"],
+                "elevation": open_water_drainage["peil_circa"],
+                "comment": "",
+                "geometry": open_water_drainage["geometry"],
+                "conductance_per_meter": np.nan,  # Default conductance for open water
+                "conductance_per_squared_meter": 1.0,  # Default conductance for open water
+                "bgt-identificatie": "",
+                "mover_lake_name": "vijver 4",
+            }
+        ),
+    ),
+    axis=0,
+    ignore_index=True,
+)
 
 # Checks
 validate_crs(gdf)