From a2cd042472ac6e4148afeb651337e812eb6b2092 Mon Sep 17 00:00:00 2001
From: Fabian Neumann <fabian.neumann@outlook.de>
Date: Mon, 14 Aug 2023 18:25:58 +0200
Subject: [PATCH] plot clustered network topology before optimisation

---
 Snakefile                               |  1 +
 rules/postprocess.smk                   | 20 +++++++
 scripts/plot_power_network_clustered.py | 79 +++++++++++++++++++++++++
 3 files changed, 100 insertions(+)
 create mode 100644 scripts/plot_power_network_clustered.py

diff --git a/Snakefile b/Snakefile
index 76ea7b80..23835a7e 100644
--- a/Snakefile
+++ b/Snakefile
@@ -75,6 +75,7 @@ if config["foresight"] == "perfect":
 rule all:
     input:
         RESULTS + "graphs/costs.pdf",
+        expand(RESULTS + "maps/power-network-{clusters}.pdf", **config["scenario"]),
         expand(
             RESULTS
             + "maps/elec_s{simpl}_{clusters}_l{ll}_{opts}_{sector_opts}-costs-all_{planning_horizons}.pdf",
diff --git a/rules/postprocess.smk b/rules/postprocess.smk
index f7c50733..d275279e 100644
--- a/rules/postprocess.smk
+++ b/rules/postprocess.smk
@@ -9,6 +9,26 @@ localrules:
 
 if config["foresight"] != "perfect":
 
+    rule plot_power_network_clustered:
+        params:
+            plotting=config["plotting"],
+        input:
+            network=RESOURCES + "networks/elec_s{simpl}_{clusters}.nc",
+            regions_onshore=RESOURCES + "regions_onshore_elec_s{simpl}_{clusters}.geojson",
+            rc="matplotlibrc",
+        output:
+            map=RESULTS + "maps/power-network-{clusters}.pdf",
+        threads: 1
+        resources:
+            mem_mb=4000,
+        benchmark:
+            BENCHMARKS + "plot_power_network_clustered/elec_s{simpl}_{clusters}"
+        conda:
+            "../envs/environment.yaml"
+        script:
+            "../scripts/plot_power_network_clustered.py"
+
+
     rule plot_power_network:
         params:
             plotting=config["plotting"],
diff --git a/scripts/plot_power_network_clustered.py b/scripts/plot_power_network_clustered.py
new file mode 100644
index 00000000..e1140fd6
--- /dev/null
+++ b/scripts/plot_power_network_clustered.py
@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+# SPDX-FileCopyrightText: : 2023-2024 PyPSA-Eur Authors
+#
+# SPDX-License-Identifier: MIT
+"""
+Plot clustered electricity transmission network.
+"""
+
+import pypsa
+import geopandas as gpd
+import matplotlib.pyplot as plt
+import cartopy.crs as ccrs
+from matplotlib.lines import Line2D
+from pypsa.plot import add_legend_lines
+
+from plot_power_network import load_projection
+
+
+if __name__ == "__main__":
+    if "snakemake" not in globals():
+        from _helpers import mock_snakemake
+
+        snakemake = mock_snakemake(
+            "plot_power_network_clustered",
+            clusters=128,
+            configfiles=["../../config/config.test.yaml"]
+        )
+
+    plt.style.use(snakemake.input.rc)
+
+    lw_factor = 2e3
+
+    n = pypsa.Network(snakemake.input.network)
+
+    regions = gpd.read_file(snakemake.input.regions_onshore).set_index('name')
+
+    proj = load_projection(snakemake.params.plotting)
+
+    fig, ax = plt.subplots(figsize=(8,8), subplot_kw={"projection": proj})
+    regions.to_crs(proj.proj4_init).plot(
+        ax=ax,
+        facecolor='none',
+        edgecolor='lightgray',
+        linewidth=0.75
+    )
+    n.plot(
+        ax=ax,
+        margin=0.06,
+        line_widths=n.lines.s_nom / lw_factor,
+        link_colors=n.links.p_nom.apply(
+            lambda x: "darkseagreen" if x > 0 else "skyblue"
+        ),
+        link_widths=2.,
+    )
+
+    sizes = [10, 20]
+    labels = [f"HVAC ({s} GW)" for s in sizes]
+    scale = 1e3 / lw_factor
+    sizes = [s * scale for s in sizes]
+
+    legend_kw = dict(
+        loc=[0.25, 0.9],
+        frameon=False,
+        labelspacing=0.5,
+        handletextpad=1,
+        fontsize=13,
+    )
+
+    add_legend_lines(
+        ax, sizes, labels, patch_kw=dict(color="rosybrown"), legend_kw=legend_kw
+    )
+
+    handles = [
+        Line2D([0], [0], color="darkseagreen", lw=2),
+        Line2D([0], [0], color="skyblue", lw=2),
+    ]
+    plt.legend(handles, ["HVDC existing", "HVDC planned"], frameon=False, loc=[0., 0.9], fontsize=13)
+
+    plt.savefig(snakemake.output.map, bbox_inches='tight')