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

added skip plotting option

Browse source
This commit is contained in:
Abyss Halley 2025-07-22 10:40:50 -04:00
parent bffdd1c699
commit 8bc3206d0c
1 changed files with 11 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
signalgen.py
View file

@ -88,7 +88,8 @@ def addBurst(values):
spectra = filterbankObj.get_data(0, samples) spectra = filterbankObj.get_data(0, samples)
#save pre-injection spectra plot #save pre-injection spectra plot
#show_dynamic(spectra, f"{basename} Pre-injection Dynamic Spectra", save=True) if not skipplot:
show_dynamic(spectra, f"{basename} Pre-injection Dynamic Spectra", save=True)
#get bandpass and store in bpWeights #get bandpass and store in bpWeights
bpWeights = create.filter_weights(spectra) bpWeights = create.filter_weights(spectra)
@ -121,8 +122,9 @@ def addBurst(values):
gulp = samples gulp = samples
) )
logger.info("Saving plot and filterbank.")
#and save the new plot #and save the new plot
if not skipplot:
logger.info("Saving plot...")
show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True)
#now generate new filterbank file #now generate new filterbank file
newName = f"{basename}_injected.fil" newName = f"{basename}_injected.fil"
@ -148,6 +150,7 @@ def addBurst(values):
za_start=-1, za_start=-1,
) )
sigprocObj.write_header(newName) sigprocObj.write_header(newName)
logger.info("Saving filterbank...")
sigprocObj.append_spectra(injectedSpectra, newName) sigprocObj.append_spectra(injectedSpectra, newName)
logger.info(f"{newName} successfully written.") logger.info(f"{newName} successfully written.")
@ -174,11 +177,15 @@ if __name__ == "__main__":
parser.add_argument( parser.add_argument(
"-p", "--plot", action="store_true", help="Just plot file and quit." "-p", "--plot", action="store_true", help="Just plot file and quit."
) )
parser.add_argument(
"-s", "--skipplot", action="store_true", help="Skip plotting for large filterbanks."
)
parser.set_defaults(dm=250.0) parser.set_defaults(dm=250.0)
parser.set_defaults(nsamp=int(3e4)) parser.set_defaults(nsamp=int(3e5))
parser.set_defaults(listfile=None) parser.set_defaults(listfile=None)
parser.set_defaults(file=None) parser.set_defaults(file=None)
parser.set_defaults(plot=False) parser.set_defaults(plot=False)
parser.set_defaults(skipplot=False)
values = parser.parse_args() values = parser.parse_args()
#set working directory to ignored directory #set working directory to ignored directory