ericteunis/m-thesis-introduction
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git synced 2024-11-13 18:13:31 +01:00
Code Issues Projects Releases Packages Wiki Activity

ZH: mutliple k script saves results into hdf5 file

Browse source 
+ slightly improve power per antenna plot if tx is plotted
This commit is contained in:
Eric Teunis de Boone 2023-01-09 11:51:03 +01:00
parent ed1eb3f5cb
commit ab07fe2126
1 changed files with 20 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
simulations/airshower_beacon_simulation/ca_period_from_shower.py
View file

@ -311,20 +311,27 @@ if __name__ == "__main__":
axs.set_title(f"Grid Run {r}")
axs.set_ylabel("vxvxB [km]")
axs.set_xlabel(" vxB [km]")
axs.set_aspect('equal', 'datalim')
sc = axs.scatter(xx/1e3, yy/1e3, c=maxima_per_loc, cmap='Spectral_r', alpha=0.6)
fig.colorbar(sc, ax=axs)
if fig_dir:
old_xlims = axs.get_xlim()
old_ylims = axs.get_ylim()
fig.tight_layout()
fig.savefig(path.join(fig_dir, __file__+f'.maxima.run{r}.pdf'))
if True:
axs.plot(tx.x/1e3, tx.y/1e3, marker='X', color='k')
fig.tight_layout()
fig.savefig(path.join(fig_dir, __file__+f'.maxima.run{r}.with_tx.pdf'))
axs.set_xlim(*old_xlims)
axs.set_ylim(*old_ylims)
fig.tight_layout()
## Save ks to file
best_idx = np.argmax(maxima_per_loc)
np.savetxt(fig_dir + __file__+f'.bestk.run{r}.txt', ks_per_loc[best_idx] )
print("Max at location: ", locs[best_idx])
print('Best k:', ks_per_loc[best_idx])
# Abort if no improvement
@ -332,12 +339,24 @@ if __name__ == "__main__":
print("No changes from previous grid, breaking")
break
old_ks_per_loc = ks_per_loc[best_idx]
# Save best ks to hdf5 antenna file
with h5py.File(antennas_fname, 'a') as fp:
group = fp.require_group('antennas')
for i, ant in enumerate(antennas):
h5ant = group[ant.name]
h5ant.attrs['best_k'] = old_ks_per_loc[i]
h5beacon_info = h5ant['beacon_info']
# find out freq_name
if freq_name is None:
freq_name = [ k for k in h5beacon_info.keys() if np.isclose(h5beacon_info[k].attrs['freq'], f_beacon)][0]
h5attrs = h5beacon_info[freq_name].attrs
h5attrs['best_k'] = old_ks_per_loc[i]
h5attrs['best_k_time'] = old_ks_per_loc[i]*dt/f_beacon
plt.show()