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

more logging; nicer file names

Browse source
This commit is contained in:
Sakimori 2025-07-18 17:50:11 -04:00
parent 62c240d1e6
commit a8ed0977fe
No known key found for this signature in database
1 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
signalgen.py
View file

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