From 0ae016a7d0a236e6c25ddd0044d4b02e623af7b7 Mon Sep 17 00:00:00 2001 From: cpschau Date: Wed, 10 Apr 2024 17:10:56 +0200 Subject: [PATCH] write shapes to base network --- scripts/base_network.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/scripts/base_network.py b/scripts/base_network.py index 346f99a5..f78f4b04 100644 --- a/scripts/base_network.py +++ b/scripts/base_network.py @@ -697,6 +697,14 @@ def _adjust_capacities_of_under_construction_branches(n, config): return n +def _set_shapes(n, country_shapes, offshore_shapes): + # Write the geodataframes country_shapes and offshore_shapes to the network.shapes component + country_shapes = gpd.read_file(country_shapes).rename(columns={"name": "idx"}) + country_shapes["type"] = "country" + offshore_shapes = gpd.read_file(offshore_shapes).rename(columns={"name": "idx"}) + offshore_shapes["type"] = "offshore" + all_shapes = pd.concat([country_shapes, offshore_shapes]) + n.shapes = pd.concat([n.shapes, all_shapes]) def base_network( eg_buses, @@ -758,12 +766,16 @@ def base_network( n = _adjust_capacities_of_under_construction_branches(n, config) + _set_shapes(n, country_shapes, offshore_shapes) + return n if __name__ == "__main__": if "snakemake" not in globals(): from _helpers import mock_snakemake + import os + os.chdir(os.path.dirname(os.path.abspath(__file__))) snakemake = mock_snakemake("base_network") configure_logging(snakemake)