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

11
signalgen.py
View file

@ -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)