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

removed try except

Browse source
This commit is contained in:
Abyss Halley 2025-08-22 13:09:26 -04:00
parent 72d172ccbc
commit f72f7edeb2
1 changed files with 23 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
singlepulse.py
View file

@ -25,7 +25,7 @@ from astropy.coordinates import Angle, Latitude, Longitude
import astropy.units as u 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
@ -325,37 +325,35 @@ 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 pulsar_dir in 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","")
print(h5_dir)
h5_files = sorted(glob.glob(f"{h5_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, mad_filter=True,
mad_filter=True, save=True,
save=True, show=False,
show=False, save_path=save_path)
save_path=save_path)
for i in range(0,(len(fluxes_mjy))):
for i in range(0,(len(fluxes_mjy))): results.append({
results.append({ "mjd": peak_mjds[i],
"mjd": peak_mjds[i], "flux_std": flux_stds[i],
"flux_std": flux_stds[i], "flux_mJy": fluxes_mjy[i],
"flux_mJy": fluxes_mjy[i], "width_s": widths_t[i],
"width_s": widths_t[i], "Pulse time (ms)": time_peaks[i],
"Pulse time (ms)": time_peaks[i], "tcand": time_pulse,
"tcand": time_pulse, "dm": dm,
"dm": dm, "filename": os.path.basename(filepath)
"filename": os.path.basename(filepath) })
}) #except Exception as e:
except Exception as e: #print(f"Error processing {filepath}: {e}")
print(f"Error processing {filepath}: {e}")
df = pd.DataFrame(results) df = pd.DataFrame(results)
#%% #%%