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
import sys
import argparse
import os
import numpy as np
import pandas as pd
from strf.rfio import Spectrogram
import matplotlib.pyplot as plt
@ -26,6 +28,26 @@ if __name__ == "__main__":
parser.add_argument('-C', type=int, help='Site ID')
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
s = Spectrogram(args.p, args.P, args.s, args.l, args.C)