added skip plotting option
This commit is contained in:
parent
bffdd1c699
commit
8bc3206d0c
15
signalgen.py
15
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
|
||||
|
|
Loading…
Reference in a new issue