ericteunis/m-thesis-introduction
1
0
Fork 0
mirror of https://gitlab.science.ru.nl/mthesis-edeboone/m-thesis-introduction.git synced 2024-12-22 03:23:34 +01:00
Code Issues Projects Releases Packages Wiki Activity

ZH: findks: introduce tqdm + suppress many figures warning

Browse source
This commit is contained in:
Eric Teunis de Boone 2023-05-01 18:19:37 +02:00
parent 7eaa97d0cb
commit 208bd9a35d
1 changed files with 10 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
airshower_beacon_simulation/ca_period_from_shower.py
View file

@ -20,6 +20,11 @@ import aa_generate_beacon as beacon
import lib import lib
from lib import rit from lib import rit
try:
from tqdm import tqdm
except:
tqdm = lambda x: x
def find_best_period_shifts_summing_at_location(test_loc, antennas, allowed_ks, period=1, dt=None, period_shift_first_trace=0, plot_iteration_with_shifted_trace=None, fig_dir=None, fig_distinguish=None,snr_str=None, shower_plane_loc=None): def find_best_period_shifts_summing_at_location(test_loc, antennas, allowed_ks, period=1, dt=None, period_shift_first_trace=0, plot_iteration_with_shifted_trace=None, fig_dir=None, fig_distinguish=None,snr_str=None, shower_plane_loc=None):
""" """
Find the best sample_shift for each antenna by summing the antenna traces Find the best sample_shift for each antenna by summing the antenna traces
@ -175,6 +180,8 @@ if __name__ == "__main__":
if os.name == 'posix' and "DISPLAY" not in os.environ: if os.name == 'posix' and "DISPLAY" not in os.environ:
matplotlib.use('Agg') matplotlib.use('Agg')
plt.rcParams.update({'figure.max_open_warning': 0})
atm = AtmoCal() atm = AtmoCal()
from scriptlib import MyArgumentParser from scriptlib import MyArgumentParser
@ -427,10 +434,11 @@ if __name__ == "__main__":
yy = [] yy = []
N_loc = len(maxima_per_loc) N_loc = len(maxima_per_loc)
for i, (x_, y_) in enumerate(product(x,y)): for i, (x_, y_) in tqdm(enumerate(product(x,y)), total=N_loc, ):
tmp_fig_subdir = None tmp_fig_subdir = None
if i % 10 ==0: if i % 10 ==0:
print(f"Testing location {i} out of {N_loc}") if hasattr(tqdm, '__code__') and tqdm.__code__.co-name == '<lambda>':
print(f"Testing location {i} out of {N_loc}")
tmp_fig_subdir = fig_subdir tmp_fig_subdir = fig_subdir
test_loc = (x_+xoff)* ev.uAxB + (y_+yoff)*ev.uAxAxB + dXref *ev.uA test_loc = (x_+xoff)* ev.uAxB + (y_+yoff)*ev.uAxAxB + dXref *ev.uA