diff --git a/signalgen.py b/signalgen.py index 60b5662..fd382d1 100755 --- a/signalgen.py +++ b/signalgen.py @@ -37,7 +37,7 @@ rng = np.random.default_rng() def show_dynamic( dynamic_spectra: np.ndarray, title: Union[str, None] = None, - save: Union[bool, None] = False, + save: Union[bool, None] = False ) -> None: """ Show a dynamic spectra by first flattening it @@ -264,6 +264,9 @@ if __name__ == "__main__": parser.add_argument( "-o", "--output", dest="output", type=str, help="Set output directory." ) + parser.add_argument( + "-s", "--start", type=int, help="Zoomed plot start time bin." + ) parser.set_defaults(dm=250.0) parser.set_defaults(pWidth=0.001) parser.set_defaults(nsamp=int(3e5)) @@ -273,6 +276,7 @@ if __name__ == "__main__": parser.set_defaults(plot=False) parser.set_defaults(rsamp=False) parser.set_defaults(output=None) + parser.set_defaults(start=-1) values = parser.parse_args() #set working directory to ignored directory or set output @@ -290,7 +294,10 @@ if __name__ == "__main__": logging.info(f"Running with file {values.file}") if values.plot: filterbankObj = Your(values.file) - spectra = filterbankObj.get_data(0, 524288) + if values.start == -1: + spectra = filterbankObj.get_data(0, 524288) + else: + spectra = filterbankObj.get_data(values.start, 8900) show_dynamic(spectra, f"{values.file} Dynamic Spectra", save=True) else: addBurst(values)