update draw code

draw.py

@@ -35,7 +35,6 @@ def draw_results(results, filename, title_benefit, annot_benefit=False, figure_s
     new_Data = pd.DataFrame(index=[df.index[-1] + 1], columns=df.columns)
     for i in df.columns:
         new_Data[i] = df[i].iloc[-1]
-    # print(new_Data)
     df = pd.concat([df, new_Data])
 
     X, Y = np.meshgrid(np.arange(df.shape[1]), np.arange(df.shape[0]))
@@ -68,7 +67,6 @@ def draw_cost(costs, filename, title_cost, annot_cost=False, figure_size=(10, 10
     new_Data = pd.DataFrame(index=[df.index[-1] + 1], columns=df.columns)
     for i in df.columns:
         new_Data[i] = df[i].iloc[-1]
-    # print(new_Data)
     df = pd.concat([df, new_Data])
     X, Y = np.meshgrid(np.arange(df.shape[1]), np.arange(df.shape[0]))
 
@@ -96,13 +94,8 @@ def draw_overload(overload_cnt, filename, title_unmet, annot_unmet=False, figure
     df.index = df.index.map(int)
     df.columns = df.columns / 1000
     df.columns = df.columns.map(int)
-    print('unmet index')
-    print(df.index)
-    print('unmet columns')
-    print(df.columns)
     min_value = df.min().min()
     max_value = df.max().max()
-    # max_scale = max(abs(min_value/1000), abs(max_value/1000))
 
 
     df[df.columns[-1] + 1] = df.iloc[:, -1] 
@@ -136,6 +129,41 @@ def draw_overload(overload_cnt, filename, title_unmet, annot_unmet=False, figure
     plt.ylabel('PV Capacity (MW)')
     plt.savefig(filename)
 
+with open('config.json', 'r') as f:
+    js_data = json.load(f)
+
+data = pd.read_csv('combined_data.csv')
+time_interval = js_data["time_interval"]["numerator"] / js_data["time_interval"]["denominator"]
+
+pv_loss = js_data["pv"]["loss"]
+pv_cost_per_kW = js_data["pv"]["cost_per_kW"]
+pv_lifetime = js_data["pv"]["lifetime"]
+
+ess_loss = js_data["ess"]["loss"]
+ess_cost_per_kW = js_data["ess"]["cost_per_kW"]
+ess_lifetime = js_data["ess"]["lifetime"]
+
+grid_loss = js_data["grid"]["loss"]
+sell_price = js_data["grid"]["sell_price"]
+grid_capacity = js_data["grid"]["capacity"]
+
+pv_begin = js_data["pv_capacities"]["begin"]
+pv_end = js_data["pv_capacities"]["end"]
+pv_groups = js_data["pv_capacities"]["groups"]
+
+ess_begin = js_data["ess_capacities"]["begin"]
+ess_end = js_data["ess_capacities"]["end"]
+ess_groups = js_data["ess_capacities"]["groups"]
+
+annot_unmet = js_data["annotated"]["unmet_prob"]
+annot_benefit = js_data["annotated"]["benefit"]
+annot_cost = js_data["annotated"]["cost"]
+
+title_unmet = js_data["plot_title"]["unmet_prob"]
+title_cost = js_data["plot_title"]["cost"]
+title_benefit = js_data["plot_title"]["benefit"]
+
+figure_size = (js_data["figure_size"]["length"], js_data["figure_size"]["height"])
 
 directory = 'data/'
 
@@ -149,9 +177,9 @@ for f in split_files:
 costs_files = [f for f in split_files if f[-1].endswith('costs.json')]
 print(f'find costs files: {costs_files}')
 overload_files = [f for f in split_files if f[-1].endswith('overload_cnt.json')]
-print(f'find overload files: {overload_files}')
+print(f'find coverage/unmet files: {overload_files}')
 results_files = [f for f in split_files if f[-1].endswith('results.json')]
-print(f'find results files: {results_files}')
+print(f'find profit/benefit files: {results_files}')
 
 costs_dfs = [read_data(directory + '-'.join(f)) for f in costs_files]
 overload_dfs = [read_data(directory + '-'.join(f)) for f in overload_files]
@@ -160,17 +188,19 @@ results_dfs = [read_data(directory + '-'.join(f)) for f in results_files]
 for costs_df, overload_df, results_df in zip(costs_dfs, overload_dfs, results_dfs):
 
     draw_cost(costs_df, 
-              f'plots/costs-{int(costs_df.columns[-1])}-pv-{int(costs_df.index[-1])}.png', 
+              f'plots/costs-ess-{int(costs_df.columns[0])}-{int(costs_df.columns[-1])}-pv-{int(costs_df.index[0])}-{int(costs_df.index[-1])}.png', 
-              f'Costs for ESS-{int(costs_df.columns[-1])}-pv-{int(costs_df.index[-1])}MWh', annot_cost=False)
+              title_cost=title_cost, 
+              annot_cost=annot_cost)
 
     draw_overload(overload_df, 
-                  f'plots/overload-ess-{overload_df.columns[-1]}-pv-{overload_df.index[-1]}.png', 
+                  f'plots/overload-ess-{overload_df.columns[0]}-{overload_df.columns[-1]}-pv-{overload_df.index[0]}-{overload_df.index[-1]}.png', 
-                  f'Overload for ess-{overload_df.columns[-1]}MW pv-{overload_df.index[-1]}MWh', 
+                  title_unmet=title_unmet, 
                   annot_unmet=False)
 
     draw_results(results_df, 
-                 f'plots/results-{results_df.columns[-1]}-pv-{results_df.index[-1]}.png', 
+                 f'plots/results-ess-{results_df.columns[0]}-{results_df.columns[-1]}-pv-{results_df.index[0]}-{results_df.index[-1]}.png', 
-                 f'Results for ess-{results_df.columns[-1]}MW pv-{results_df.index[-1]}MWh', annot_benefit=False)
+                 title_benefit=title_benefit,
+                 annot_benefit=False)
 
 
 

main.py

@@ -111,7 +111,7 @@ def cal_profit(es: EnergySystem, saved_money):
 for ess_capacity in ess_capacities:
     print(f"ess_capacity:{ess_capacity}")
     for pv_capacity in pv_capacities:
-        print(f"pv_capacity:{ess_capacity}")
+        print(f"pv_capacity:{pv_capacity}")
         pv = pv_config(capacity=pv_capacity, 
                         cost_per_kW=pv_cost_per_kW,
                         lifetime=pv_lifetime,