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.")