Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Option to plot production pie charts at a timestep, and user defined filename #29

Merged
merged 2 commits into from
Jan 21, 2025
Merged

Option to plot production pie charts at a timestep, and user defined filename #29

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e4208db
Option to plot production pie charts at a timestep, and user defined …
Mathias157 Jan 21, 2025
0b06f95
Updated documentation of generation_exclude variables
Mathias157 Jan 21, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
20 changes: 16 additions & 4 deletions src/pybalmorel/plotting/maps_balmorel.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,10 +63,11 @@ def plot_map(path_to_result: str,
**T (str, optional): Hour for FlowTime or UtilizationTime. Will pick one random if not specified.
**exo_end (str, optional): Show only exogenous or endogenous capacities. Choose from ['Both', 'Endogenous', 'Exogenous']. Defaults to 'Both'.
**generation_exclude_Import_Cap_H2 (bool, optional): Do not plot the capacities and production related to H2 Import (will be shown as line). Defaults to True.
**generation_exclude_H2Storage (bool, optional): Do not plot the capacities of the H2 storage. Defaults to True.
**generation_exclude_ElectricStorage (bool, optional): Do not plot the production of Electric storage. Defaults to True.
**generation_exclude_H2Storage (bool, optional): Do not plot capacity or production of the H2 storage. Defaults to True.
**generation_exclude_ElectricStorage (bool, optional): Do not plot capacity or production of Electric storage. Defaults to True.
**generation_exclude_Geothermal (bool, optional): Do not plot the production of Geothermal. Defaults to True.
**coordinates_geofile_offset (float, optional): Geofile coordinates offset from the min and max of the geofile. Defaults to 0.5.
**filename (str, optional): The name of the file to save, if save_fig = True. Defaults to .png if no extension is included.
Visual additional options:
**legend_show (bool, optional): Show legend_show or not. Defaults to True.
**show_country_out (bool, optional): Show countries outside the model or not. Defaults to True.
Expand Down Expand Up @@ -173,7 +174,7 @@ def plot_map(path_to_result: str,
if generation_commodity not in ['Electricity', 'Hydrogen']:
print(f'generation_commodity must be either "Electricity" or "Hydrogen", set to {commodity}')
generation_commodity = commodity
if generation not in ['Capacity', 'Production']: # Check that it's a possible type of generation display
if generation not in ['Capacity', 'Production', 'ProductionTime']: # Check that it's a possible type of generation display
raise ValueError('generation must be either "Capacity" or "Production"')
if lines in ['FlowTime', 'UtilizationTime']: # Check if there is a specified season and hour for flow maps
S = kwargs.get('S', '')
Expand Down Expand Up @@ -266,13 +267,16 @@ def plot_map(path_to_result: str,
pie_show_min = kwargs.get('pie_show_min', 0) # Minimum transmission capacity (GW) or flow (TWh) shown on map
pie_radius_dict = {'EU': {'pie_radius_min' : 0.2, 'pie_radius_max' : 1.4, 'pie_cluster_radius' : [0.2, 0.6, 1, 1.3],
'pie_cluster_groups' : {'Production' : {'Electricity' : [20,50,200,400], 'Hydrogen' : [10,50,100,250]},
'ProductionTime' : {'Electricity' : [20,50,200,400], 'Hydrogen' : [10,50,100,250]},
'Capacity' : {'Electricity' : [10,50,200,400], 'Hydrogen' : [1,10,30,60]}}},
'DK': {'pie_radius_min' : 0.05, 'pie_radius_max' : 0.5, 'pie_cluster_radius' : [0.05, 0.1, 0.25, 0.5],
'pie_cluster_groups' : {'Production' : {'Electricity' : [10,25,50,100], 'Hydrogen' : [0.5,1,2,50]},
'ProductionTime' : {'Electricity' : [10,25,50,100], 'Hydrogen' : [0.5,1,2,50]},
'Capacity' : {'Electricity' : [5,10,25,50], 'Hydrogen' : [0.1,0.2,0.5,1]}}}}
if choosen_map_coordinates not in ["EU", "DK"]:
pie_radius_dict[choosen_map_coordinates] = {'pie_radius_min' : 0.1, 'pie_radius_max' : 0.5, 'pie_cluster_radius' : [0.1, 0.2, 0.3, 0.5],
'pie_cluster_groups' : {'Production' : {'Electricity' : [10,25,50,100], 'Hydrogen' : [0.5,1,2,50]},
'ProductionTime' : {'Electricity' : [10,25,50,100], 'Hydrogen' : [0.5,1,2,50]},
'Capacity' : {'Electricity' : [5,10,25,50], 'Hydrogen' : [0.1,0.2,0.5,1]}}}
pie_radius_min = kwargs.get('pie_radius_min', pie_radius_dict[choosen_map_coordinates]['pie_radius_min']) # Minimum width of lines, used if cat is linear or log
pie_radius_max = kwargs.get('pie_radius_max', pie_radius_dict[choosen_map_coordinates]['pie_radius_max']) # Maximum width of lines, used if cat is linear or log
Expand Down Expand Up @@ -496,6 +500,8 @@ def df_creation(gdx_file, varname, system_directory):

var_list = []
var_list = var_list + ['G_CAP_YCRAF', 'PRO_YCRAGF']
if generation.lower() == 'productiontime':
var_list += ['PRO_YCRAGFST']
if commodity == 'Electricity':
if lines == 'Capacity' :
var_list = var_list + ['X_CAP_YCR']
Expand Down Expand Up @@ -573,6 +579,8 @@ def df_creation(gdx_file, varname, system_directory):
df_generation = all_df['G_CAP_YCRAF']
elif generation == 'Production':
df_generation = all_df['PRO_YCRAGF']
elif generation == 'ProductionTime':
df_generation = all_df['PRO_YCRAGFST'].query('SSS == "%s" and TTT == "%s"'%(S, T))
if generation_commodity == 'Electricity':
df_generation = df_generation[df_generation['COMMODITY'] == 'ELECTRICITY']
elif generation_commodity == 'Hydrogen':
Expand Down Expand Up @@ -1392,10 +1400,14 @@ def find_nearest(array, value):
### 3.8 Save the figure

# Output the figure in pdf
if save_fig:
if save_fig and kwargs.get('filename') == None:
output_dir = os.path.join(os.getcwd(), 'output')
os.makedirs(output_dir, exist_ok=True)
fig.savefig(os.path.join(output_dir, f'{commodity}_{lines}_{generation_commodity}_{generation}.pdf'), bbox_inches='tight')
elif save_fig and kwargs.get('filename') != None:
output_dir = os.path.join(os.getcwd(), 'output')
os.makedirs(output_dir, exist_ok=True)
fig.savefig(os.path.join(output_dir, kwargs.get('filename')), bbox_inches='tight')

return fig, ax

Expand Down
Loading