j_hartling/paper_2025
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added loads of units in nearly all graphs.

Browse Source 
Overhauled fig_invariance_full.pdf.
Added some legends, somewhere.
This commit is contained in:
j-hartling
2026-04-28 19:43:05 +02:00
parent 7e1aa8721a
commit e70d100655
40 changed files with 965 additions and 471 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
python/fig_invariance_log-hp.py
View File

@@ -95,7 +95,7 @@ noise_grid_kwargs = dict(
top=edge_padding + block_height,
height_ratios=[1, 2, 1]
)
big_col_shift = -0.12
big_col_shift = -0.04
big_grid_kwargs = dict(
nrows=1,
ncols=3,
@@ -135,11 +135,10 @@ xlabels = dict(
big='scale $\\alpha$',
)
ylabels = dict(
env='$x_{\\text{env}}$',
log='$x_{\\text{dB}}$',
inv='$x_{\\text{adapt}}$',
env='$x_{\\text{env}}$\n$[\\text{a.u.}]$',
log='$x_{\\text{log}}$\n$[\\text{dB}]$',
inv='$x_{\\text{adapt}}$\n$[\\text{dB}]$',
big_pure='$\\sigma_x$',
big_log='$\\sigma_x\\,[\\text{dB}]$',
big_noise='$\\sigma_x\\,/\\,\\sigma_{\\eta}$' if relate_to_noise else None,
)
xlab_big_kwargs = dict(
@@ -148,23 +147,17 @@ xlab_big_kwargs = dict(
ha='center',
va='bottom',
)
ylab_big_left_kwargs = dict(
ylab_big_kwargs = dict(
x=-0.2,
fontsize=fs['lab_tex'],
ha='center',
va='bottom'
)
ylab_big_right_kwargs = dict(
x=1.2,
fontsize=fs['lab_tex'],
ha='center',
va='top'
)
ylab_snip_kwargs = dict(
x=0,
x=0.03,
fontsize=fs['lab_tex'],
rotation=0,
ha='left',
ha='center',
va='center',
)
yloc = dict(
@@ -231,19 +224,41 @@ bar_kwargs = dict(
va='center',
)
)
leg_kwargs = dict(
stage_leg_kwargs = dict(
ncols=1,
loc='upper left',
bbox_to_anchor=(0.05, 0.5, 0.5, 0.5),
frameon=False,
prop=dict(
size=20,
),
borderpad=0,
borderaxespad=0,
handlelength=1,
columnspacing=1,
handletextpad=0.5,
labelspacing=0
)
stage_leg_labels = dict(
env='$x_{\\text{env}}$',
log='$x_{\\text{log}}$',
inv='$x_{\\text{adapt}}$',
)
spec_leg_kwargs = dict(
ncols=2,
loc='upper right',
bbox_to_anchor=(0, 0.6, 1, 0.4),
frameon=False,
prop=dict(
size=12,
size=13.5,
style='italic',
),
borderpad=0,
borderaxespad=0,
handlelength=0.5,
columnspacing=1,
handletextpad=0.5,
labelspacing=0.25,
)
diag_kwargs = dict(
c=(0.3,) * 3,
@@ -374,17 +389,8 @@ for i, scales in enumerate([pure_scales, noise_scales, noise_scales]):
ax.set_ylim(0.9, 30)
big_axes[i] = ax
shift_subplot(big_axes[0], dx=big_col_shift)
ylabel(big_axes[0], ylabels['big_pure'], transform=big_axes[0].transAxes, **ylab_big_left_kwargs)
ylabel(big_axes[1], ylabels['big_noise'], transform=big_axes[1].transAxes, **ylab_big_left_kwargs)
big_twin = big_axes[0].twinx()
hide_axis(big_twin, 'left')
big_twin.spines['right'].set_visible(True)
big_twin.set_position(big_axes[0].get_position().bounds)
big_twin.set_ylim(scales[0], scales[-1])
big_twin.set_yscale('symlog', linthresh=scales[1], linscale=0.5)
ylabel(big_twin, ylabels['big_log'], transform=big_twin.transAxes, **ylab_big_right_kwargs)
color_axis(big_axes[0], colors['env'], side='left')
color_axis(big_twin, colors['log'], side='right')
ylabel(big_axes[0], ylabels['big_pure'], transform=big_axes[0].transAxes, **ylab_big_kwargs)
ylabel(big_axes[1], ylabels['big_noise'], transform=big_axes[1].transAxes, **ylab_big_kwargs)
super_xlabel(xlabels['big'], big_subfig, big_axes[0], big_axes[-1], **xlab_big_kwargs)
letter_subplots(big_axes, 'cde', **letter_big_kwargs)
@@ -421,9 +427,11 @@ plot_snippets(noise_axes[2, :], t_full, noise_data['snip_inv'],
time_bar(noise_axes[-1, -1], **bar_kwargs)
# Plot pure-song measures (ideal):
big_axes[0].plot(pure_scales, pure_data['measure_env'], c=colors['env'], lw=lw['big'])
big_axes[0].plot(pure_scales, pure_data['measure_log'], c=colors['log'], lw=lw['big'])
big_axes[0].plot(pure_scales, pure_data['measure_inv'], c=colors['inv'], lw=lw['big'])
big_axes[0].plot(pure_scales, pure_data['measure_env'], c=colors['env'], lw=lw['big'], label=stage_leg_labels['env'])
big_axes[0].plot(pure_scales, pure_data['measure_log'], c=colors['log'], lw=lw['big'], label=stage_leg_labels['log'])
big_axes[0].plot(pure_scales, pure_data['measure_inv'], c=colors['inv'], lw=lw['big'], label=stage_leg_labels['inv'])
legend = big_axes[0].legend(**stage_leg_kwargs)
[h.set_lw(lw['legend']) for h in legend.legend_handles]
# Plot noise-song measures (limited):
big_axes[1].plot(noise_scales, noise_data['measure_env'], c=colors['env'], lw=lw['big'])
@@ -458,7 +466,7 @@ for i, (species, measure) in enumerate(species_measures.items()):
transform=big_axes[2].get_xaxis_transform())
big_axes[2].vlines(scale, big_axes[2].get_ylim()[0], measure[ind],
color=color, **plateau_line_kwargs)
legend = big_axes[2].legend(**leg_kwargs)
legend = big_axes[2].legend(**spec_leg_kwargs)
[h.set_lw(lw['legend']) for h in legend.legend_handles]
if save_path is not None: