From a8ed0977feb9b550656e915b88334a57c3c3ecd2 Mon Sep 17 00:00:00 2001 From: Sakimori Date: Fri, 18 Jul 2025 17:50:11 -0400 Subject: [PATCH] more logging; nicer file names --- signalgen.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/signalgen.py b/signalgen.py index d242c51..ea7b5bc 100755 --- a/signalgen.py +++ b/signalgen.py @@ -75,15 +75,17 @@ def addBurst(values): --listfile will enter into this function multiple times for each line. """ filterbankObj = Your(values.file) + values.file = values.file.strip(".fil") #removing extension for text manipulation later + basename = path.basename(values.file) samples = 8096 spectra = filterbankObj.get_data(0, samples) #save pre-injection spectra plot - show_dynamic(spectra, f"{path.basename(values.file)} Pre-injection Dynamic Spectra", save=True) + show_dynamic(spectra, f"{basename} Pre-injection Dynamic Spectra", save=True) #get bandpass and store in bpWeights bpWeights = create.filter_weights(spectra) - logger.info(f"{path.basename(values.file)} loaded. Sampling pulse {values.nsamp} times.") + logger.info(f"{basename} loaded. Sampling pulse {values.nsamp} times.") #create pulse #first version is very simple, plan on adding more complex injections in future @@ -102,6 +104,7 @@ def addBurst(values): ) pulse = pulseObj.sample_pulse(nsamp=values.nsamp) #30000 by default + logger.info("Injecting pulse") #inject pulse injectedSpectra = inject.inject_constant_into_file( yr_input = filterbankObj, @@ -110,11 +113,12 @@ def addBurst(values): gulp = samples ) + logger.info("Saving plot and filterbank.") #and save the new plot and filterbank file - show_dynamic(spectra, f"{path.basename(values.file)} Dynamic Spectra and {values.dm} DM Pulse", save=True) - filWriter = Writer(filterbankObj, outdir="./", outname = f"{path.basename(values.file)}_injected") + show_dynamic(spectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) + filWriter = Writer(filterbankObj, outdir="./", outname = f"{basename}_injected") filWriter.to_fil(data=injectedSpectra) - logger.info(f"{path.basename(values.file)}_injected.fil successfully written.") + logger.info(f"{basename}_injected.fil successfully written.")