diff --git a/signalgen.py b/signalgen.py index 96348f0..a34731a 100755 --- a/signalgen.py +++ b/signalgen.py @@ -171,10 +171,14 @@ if __name__ == "__main__": parser.add_argument( "-n", "--nsamp", type=int, help="Number of samples to take of the generated pulse." ) + parser.add_argument( + "-p", "--plot", type=bool, help="Just plot file and quit." + ) parser.set_defaults(dm=250.0) parser.set_defaults(nsamp=int(3e4)) parser.set_defaults(listfile=None) parser.set_defaults(file=None) + parser.set_defaults(plot=False) values = parser.parse_args() #set working directory to ignored directory @@ -185,7 +189,12 @@ if __name__ == "__main__": if values.file is not None: #single file takes priority logging.info(f"Running with file {values.file}") - addBurst(values) + if values.plot: + filterbankObj = Your(values.file) + spectra = filterbankObj.get_data(0, 8096) + show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True) + else: + addBurst(values) elif values.listfile is not None: #list of files logging.info(f"Looking for directories in file {values.listfile}") processList(values)