added skip plotting option

This commit is contained in:
Abyss Halley 2025-07-22 10:40:50 -04:00
parent bffdd1c699
commit 8bc3206d0c

View file

@ -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,8 +122,9 @@ def addBurst(values):
gulp = samples
)
logger.info("Saving plot and filterbank.")
#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)
#now generate new filterbank file
newName = f"{basename}_injected.fil"
@ -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