ZH: plotting improvements

This commit is contained in:
Eric-Teunis de Boone 2023-04-13 12:28:04 +02:00
parent ee13fe91f9
commit 1130f2c679
4 changed files with 12 additions and 7 deletions

View file

@ -283,17 +283,20 @@ if __name__ == "__main__":
fig, axs = rit.slice_figure(ev, X, xx, yy, p, mode='sp', scatter_kwargs=dict(
vmax=1e5,
vmin=0,
s=150,
s=250,
cmap='inferno',
# edgecolor='black',
))
suptitle = fig._suptitle.get_text()
fig.suptitle("")
axs.set_title(suptitle +"\n" +plot_titling[case])
#axs.set_aspect('equal', 'datalim')
axs.set_title("Shower plane slice\n" + plot_titling[case] + "\n" + suptitle)
axs.set_aspect('equal', 'datalim')
axs.legend(title=snr_str)
axs.set_xlim(1.1*min(xx)/1e3, 1.1*max(xx)/1e3)
axs.set_ylim(1.1*min(yy)/1e3, 1.1*max(yy)/1e3)
if fig_dir:
fig.tight_layout()
fig.savefig(path.join(fig_dir, path.basename(__file__) + f'.X{X}.{case}.{scalename}.pdf'))