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

added simple plot command

Browse source
This commit is contained in:
Sakimori 2025-07-21 17:00:56 -04:00
parent a071724015
commit bcbdf32d37
No known key found for this signature in database
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
signalgen.py
View file

@ -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,6 +189,11 @@ if __name__ == "__main__":
if values.file is not None: #single file takes priority
logging.info(f"Running with file {values.file}")
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}")