zhetadelta/greenburstAux
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

added ability to zoom in on plots

Browse source
This commit is contained in:
Sakimori 2025-08-12 11:17:37 -04:00
parent 850c2e189f
commit ffad418316
No known key found for this signature in database
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
signalgen.py
View file

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