marcel/survBot
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 1 Wiki Activity

[minor] add axis-shading for a better distinction of different axes

Browse Source
This commit is contained in:
Marcel Paffrath 2023-12-18 16:06:42 +01:00
parent fde000ec0d
commit 3c6ea1ffd0
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
utils.py
View File

@ -187,13 +187,15 @@ def set_axis_ylabels(fig, parameters):
ax.set_ylabel(channel_name) ax.set_ylabel(channel_name)
def set_axis_color(fig, color='0.8'): def set_axis_color(fig, color='0.8', shade_color='0.95'):
""" """
Set all axes of figure to specific color Set all axes (frame) of figure to specific color. Shade every second axis.
""" """
for ax in fig.axes: for i, ax in enumerate(fig.axes):
for key in ['bottom', 'top', 'right', 'left']: for key in ['bottom', 'top', 'right', 'left']:
ax.spines[key].set_color(color) ax.spines[key].set_color(color)
if i % 2:
ax.set_facecolor(shade_color)
def set_axis_yticks(fig, parameters): def set_axis_yticks(fig, parameters):
@ -258,7 +260,7 @@ def get_warn_states_pbox(soh_key: str, parameters: dict) -> list:
return [key for key in pb_dict.keys() if key > 1] return [key for key in pb_dict.keys() if key > 1]
def annotate_voltage_states(ax, parameters, pb_key): def annotate_voltage_states(ax, parameters, pb_key, color='0.75'):
for voltage, voltage_dict in parameters.get('POWBOX').get(pb_key).items(): for voltage, voltage_dict in parameters.get('POWBOX').get(pb_key).items():
if float(voltage) < 1: if float(voltage) < 1:
continue continue
@ -269,5 +271,5 @@ def annotate_voltage_states(ax, parameters, pb_key):
out_string += ' | ' out_string += ' | '
out_string += f'{key}: {val}' out_string += f'{key}: {val}'
ax.annotate(out_string, (ax.get_xlim()[-1], voltage), color='0.8', fontsize='xx-small', ax.annotate(out_string, (ax.get_xlim()[-1], voltage), color=color, fontsize='xx-small',
horizontalalignment='right') horizontalalignment='right')