Fix HBA pqr for core stations

pull/3/merge
Tammo Jan Dijkema 2020-05-25 13:21:54 +02:00
parent 6a4a28b45e
commit 5dde31e22b
1 changed files with 13 additions and 3 deletions

View File

@ -310,9 +310,14 @@ def get_station_pqr(station_name: str, rcu_mode: Union[str, int], db):
>>> pqr[0, 0]
1.7434713
>>> pqr = get_station_pqr("LV614", "5", db)
>>> pqr.shape
>>> get_station_pqr("LV614", "5", db).shape
(96, 3)
>>> get_station_pqr("CS103", 5, db).shape
(48, 3)
>>> get_station_pqr("PL611", 6, db).shape
(48, 3)
"""
full_station_name = get_full_station_name(station_name, rcu_mode)
station_type = get_station_type(full_station_name)
@ -332,7 +337,12 @@ def get_station_pqr(station_name: str, rcu_mode: Union[str, int], db):
}
selected_dipoles = selected_dipole_config[station_type] + \
np.arange(len(selected_dipole_config[station_type])) * 16
station_pqr = db.hba_dipole_pqr(full_station_name)[selected_dipoles]
if station_type == 'core':
all_dipole_pqr = np.concatenate((db.hba_dipole_pqr(full_station_name + "0"),
db.hba_dipole_pqr(full_station_name + "1")))
else:
all_dipole_pqr = db.hba_dipole_pqr(full_station_name)
station_pqr = all_dipole_pqr[selected_dipoles]
else:
raise RuntimeError("Station name did not contain LBA or HBA, could not load antenna positions")