zhetadelta/greenburstAux
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

added debug text

Browse source
This commit is contained in:
Abyss Halley 2025-08-22 12:56:45 -04:00
parent e67ee6a16a
commit 636c73310e
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
singlepulse.py
View file

@ -26,7 +26,7 @@ import astropy.units as u
#AH: Modifying this to support multiple directories #AH: Modifying this to support multiple directories
main_dir = os.path.join("ldata","trunk") main_dir = os.path.join("ldata","trunk")
pulsar_dir_list = [os.path.join(main_dir, name, "") for name in pulsar_dir_list = [os.path.join(main_dir, name) for name in
["data_2025-04-24_07-36-04", "data_2025-04-29_07-50-16", "data_2025-04-30_07-53-07", "data_2025-04-30_08-18-17", "data_2025-05-01_07-47-34"]] ["data_2025-04-24_07-36-04", "data_2025-04-29_07-50-16", "data_2025-04-30_07-53-07", "data_2025-04-30_08-18-17", "data_2025-05-01_07-47-34"]]
#AH: old variable: pulsar_dir #AH: old variable: pulsar_dir
pulsar_period = 1.34789947290 * 1000 #period of the pulsar. AH: in ms pulsar_period = 1.34789947290 * 1000 #period of the pulsar. AH: in ms
@ -328,11 +328,12 @@ def compute_snr_from_h5(h5_file, detrend_ft=True, mad_filter=True, save=False, s
for i, pulsar_dir in enumerate(pulsar_dir_list): for i, pulsar_dir in enumerate(pulsar_dir_list):
save_path = save_path_list[i] #AH: link directory and save path in a bit of a messy way oops save_path = save_path_list[i] #AH: link directory and save path in a bit of a messy way oops
h5_dir = os.path.join(pulsar_dir,"cands","") h5_dir = os.path.join(pulsar_dir,"cands","")
h5_files = sorted(glob.glob(f"{pulsar_dir}*.h5")) h5_files = sorted(glob.glob(f"{h5_dir}*.h5"))
results = [] results = []
for filepath in h5_files: for filepath in h5_files:
print(filepath)
try: try:
fluxes_mjy, flux_stds, widths_t, peak_mjds, time_peaks, time_pulse, dm = compute_snr_from_h5(filepath, fluxes_mjy, flux_stds, widths_t, peak_mjds, time_peaks, time_pulse, dm = compute_snr_from_h5(filepath,
detrend_ft=True, detrend_ft=True,