Make figure 5 and 6 match guidelines

Author: staadecker

papers/Martin_Staadecker_Value_of_LDES_and_Factors/LDES_paper_graphs/Figure LDES cycles.py

@@ -11,19 +11,19 @@
 
 from papers.Martin_Staadecker_Value_of_LDES_and_Factors.LDES_paper_graphs.util import (
     set_style,
-    get_set_e_scenarios,
+    get_set_e_scenarios, save_figure,
 )
 from switch_model.tools.graph.main import GraphTools
 
 # Prepare graph tools
-tools = GraphTools(scenarios=get_set_e_scenarios())
+tools = GraphTools(scenarios=get_set_e_scenarios(), set_style=False)
 tools.pre_graphing(multi_scenario=True)
 
 # %% CREATE FIGURE
 set_style()
 plt.close()
 fig = plt.figure()
-fig.set_size_inches(12, 12)
+fig.set_size_inches(6.850394, 6.850394)
 gs = matplotlib.gridspec.GridSpec(2, 2, figure=fig)
 ax1 = fig.add_subplot(gs[0, 0])
 ax2 = fig.add_subplot(gs[0, 1])
@@ -162,6 +162,8 @@
 demand = demand.groupby("timepoint", as_index=False).value.sum()
 demand = tools.transform.timestamp(demand, use_timepoint=True)
 demand = demand.set_index("datetime")["value"]
+demand *= 4 * 1e-6  # Each timestep is 4 hours, converting to TWh
+total_demand = demand.sum()
 demand = demand.resample(freq).mean()
 demand = demand * 60 / demand.max()
 
@@ -181,7 +183,7 @@
 
 lines = ax.get_lines()
 x_label = {
-    4.0: 135,
+    # 4.0: 135,
     8.0: 150,
     20.0: 170,
     24.0: 230,
@@ -193,14 +195,15 @@
     label = float(line.get_label())
     if label not in x_label.keys():
         continue
-    labellines.labelLine(line, state_of_charge.index[x_label[label]], label="S "+str(int(label)), align=False, color='k')
+    labellines.labelLine(line, state_of_charge.index[x_label[label]], linespacing=1, outline_width=1, label=str(int(label))+"TWh", align=False, color='k', fontsize="small")
 
 demand_lines = ax.plot(demand, c="dimgray", linestyle="--", alpha=0.5)
-ax.legend(demand_lines, ["Demand"])
+ax.legend(demand_lines, [f"Demand ({total_demand:.0f} TWh/year)"])
 
 ax.set_title("C. State of charge throughout the year")
-# %%
+# %% SAVE FIGURE
 plt.tight_layout()
+save_figure("figure-5-impact-of-ldes-on-grid.png")
 
 # %% CALCULATIONS
 cap_total = cap["Solar"] + cap["Wind"]