spacecruft
/
pystrf
1
0
Fork 0
Code Releases Activity

Added reading sites.txt

merge-requests/2/head
Michał Drzał 2022-06-21 14:27:17 +02:00 committed by GitHub
parent 3b7c8fd706
commit 4f3cbb84b5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
rfplot.py
View File

@ -1,8 +1,10 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import sys import sys
import argparse import argparse
import os
import numpy as np import numpy as np
import pandas as pd
from strf.rfio import Spectrogram from strf.rfio import Spectrogram
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
@ -26,6 +28,26 @@ if __name__ == "__main__":
parser.add_argument('-C', type=int, help='Site ID') parser.add_argument('-C', type=int, help='Site ID')
args = parser.parse_args() args = parser.parse_args()
if "ST_DATADIR" not in os.environ:
print("ST_DATADIR variable not found")
sys.exit(1)
sites = os.path.join(os.environ["ST_DATADIR"], "data", "sites.txt")
if not os.path.exists(sites):
print(f"Sites file not available under {sites}")
sys.exit(1)
sites_df = pd.read_fwf(sites,colspecs=[(0,4), (5,7), (8,18), (18,30), (30,38), (38,100)])
matching_site = sites_df[sites_df["# No"] == args.C]
if len(matching_site) == 0:
print(f"Site with no: {args.C} does not exist")
sys.exit(1)
print("Selected site:")
matching_site = matching_site.to_dict(orient='records')[0]
print(matching_site)
# Read spectrogram # Read spectrogram
s = Spectrogram(args.p, args.P, args.s, args.l, args.C) s = Spectrogram(args.p, args.P, args.s, args.l, args.C)