diff --git a/signalgen.py b/signalgen.py index bd88643..ea187e9 100755 --- a/signalgen.py +++ b/signalgen.py @@ -88,7 +88,8 @@ def addBurst(values): spectra = filterbankObj.get_data(0, samples) #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 bpWeights = create.filter_weights(spectra) @@ -121,9 +122,10 @@ def addBurst(values): gulp = samples ) - logger.info("Saving plot and filterbank.") #and save the new plot - show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) + if not skipplot: + logger.info("Saving plot...") + show_dynamic(injectedSpectra, f"{basename} Dynamic Spectra and {values.dm} DM Pulse", save=True) #now generate new filterbank file newName = f"{basename}_injected.fil" sigprocObj = make_sigproc_object( @@ -148,6 +150,7 @@ def addBurst(values): za_start=-1, ) sigprocObj.write_header(newName) + logger.info("Saving filterbank...") sigprocObj.append_spectra(injectedSpectra, newName) logger.info(f"{newName} successfully written.") @@ -174,11 +177,15 @@ if __name__ == "__main__": parser.add_argument( "-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(nsamp=int(3e4)) + parser.set_defaults(nsamp=int(3e5)) parser.set_defaults(listfile=None) parser.set_defaults(file=None) parser.set_defaults(plot=False) + parser.set_defaults(skipplot=False) values = parser.parse_args() #set working directory to ignored directory