From 4e078002af6e37a6b05dd65b5a22294c9b4e310b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jonas=20H=C3=B6rsch?= <jonas.hoersch@kit.edu>
Date: Tue, 14 Aug 2018 17:01:58 +0200
Subject: [PATCH] base_network: Add underwater_fraction to links

---
 scripts/base_network.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/scripts/base_network.py b/scripts/base_network.py
index 41e36a30..93329f06 100644
--- a/scripts/base_network.py
+++ b/scripts/base_network.py
@@ -260,6 +260,11 @@ def _set_countries_and_substations(n):
 
     return buses
 
+def _set_links_underwater_fraction(n):
+    offshore_shape = gpd.read_file(snakemake.input.offshore_shapes).set_index('id').unary_union
+    links = gpd.GeoSeries(n.links.geometry.map(loads))
+    n.links['underwater_fraction'] = links.intersection(offshore_shape).length / links.length
+
 def base_network():
     buses = _load_buses_from_eg()
 
@@ -293,6 +298,8 @@ def base_network():
 
     _set_countries_and_substations(n)
 
+    _set_links_underwater_fraction(n)
+
     return n
 
 if __name__ == "__main__":