Fix HBA pqr for core stations
parent
6a4a28b45e
commit
5dde31e22b
|
@ -310,9 +310,14 @@ def get_station_pqr(station_name: str, rcu_mode: Union[str, int], db):
|
||||||
>>> pqr[0, 0]
|
>>> pqr[0, 0]
|
||||||
1.7434713
|
1.7434713
|
||||||
|
|
||||||
>>> pqr = get_station_pqr("LV614", "5", db)
|
>>> get_station_pqr("LV614", "5", db).shape
|
||||||
>>> pqr.shape
|
|
||||||
(96, 3)
|
(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)
|
full_station_name = get_full_station_name(station_name, rcu_mode)
|
||||||
station_type = get_station_type(full_station_name)
|
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] + \
|
selected_dipoles = selected_dipole_config[station_type] + \
|
||||||
np.arange(len(selected_dipole_config[station_type])) * 16
|
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:
|
else:
|
||||||
raise RuntimeError("Station name did not contain LBA or HBA, could not load antenna positions")
|
raise RuntimeError("Station name did not contain LBA or HBA, could not load antenna positions")
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue