casangi / xradio

Xarray Radio Astronomy Data IO
Other
9 stars 5 forks source link

Performance Improvements of Unique Baselines and Baseline Indices #115

Closed max-nag closed 6 months ago

max-nag commented 6 months ago

The function to get (unique) baselines has been optimised, reducing the computation time from 45s to 0.8s for a 71G LOFAR MeasurementSet. In addition, the function now returns a 2D array of antenna pairs, representing baselines, instead of using strings. The pairing functions used in this function have been added with tests, and a new function has been added to get baseline indices, which is also an optimisation. A function has also been added to get unique values of 1D arrays, which is around 3x faster on average relative the the NumPy method.

CLAassistant commented 6 months ago

CLA assistant check
All committers have signed the CLA.