#beam #mwa #primary #array #radio #telescope #model

bin+lib mwa_hyperbeam

Primary beam code for the Murchison Widefield Array (MWA) radio telescope

19 unstable releases (6 breaking)

new 0.9.2 Jun 21, 2024
0.8.0 May 24, 2024
0.7.1 Jan 29, 2024
0.7.0 Oct 31, 2023
0.3.3 Mar 10, 2021

#17 in Science

Download history 13/week @ 2024-02-29 2/week @ 2024-03-07 42/week @ 2024-03-28 12/week @ 2024-04-04 3/week @ 2024-04-18 12/week @ 2024-04-25 1/week @ 2024-05-02 120/week @ 2024-05-09 34/week @ 2024-05-16 143/week @ 2024-05-23 80/week @ 2024-06-13

231 downloads per month
Used in mwa_hyperdrive

MPL-2.0 license

380KB
7.5K SLoC

mwa_hyperbeam

Primary beam code for the Murchison Widefield Array (MWA) radio telescope.

This code exists to provide a single correct, convenient implementation of Marcin Sokolowski's Full Embedded Element (FEE) primary beam model of the MWA, a.k.a. "the 2016 beam". This code should be used over all others. If there are soundness issues, please raise them here so everyone can benefit.

See the changelog for the latest changes to the code.

Polarisation order

See this document for details on the polarisation order of the beam-response Jones matrices. If the parallactic-angle correction is applied, then it is possible for the code to re-order the Jones matrices.

Usage

hyperbeam requires the MWA FEE HDF5 file. This can be obtained with:

wget http://ws.mwatelescope.org/static/mwa_full_embedded_element_pattern.h5

When making a new beam object, hyperbeam needs to know where this HDF5 file is. The easiest thing to do is set the environment variable MWA_BEAM_FILE:

export MWA_BEAM_FILE=/path/to/mwa_full_embedded_element_pattern.h5

(On Pawsey systems, this should be export MWA_BEAM_FILE=/pawsey/mwa/mwa_full_embedded_element_pattern.h5)

hyperbeam can be used by any programming language providing FFI via C. In other words, most languages. See Rust, C and Python examples of usage in the examples directory. A simple Python example is:

>>> import mwa_hyperbeam
>>> beam = mwa_hyperbeam.FEEBeam()
>>> help(beam.calc_jones)
Help on built-in function calc_jones:

calc_jones(az_rad, za_rad, freq_hz, delays, amps, norm_to_zenith, latitude_rad, iau_order) method of builtins.FEEBeam instance
    Calculate the beam-response Jones matrix for a given direction and
    pointing. If `latitude_rad` is *not* supplied, the result will match
    the original specification of the FEE beam code (possibly more useful
    for engineers).

    Astronomers are more likely to want to specify `latitude_rad` (which
    will apply the parallactic-angle correction using the Earth latitude
    provided for the telescope) and `iau_order`. If `latitude_rad` is not
    given, then `iau_reorder` does nothing. See this document for more
    information:
    <https://github.com/MWATelescope/mwa_hyperbeam/blob/main/fee_pols.pdf>

    `delays` and `amps` apply to each dipole in an MWA tile in the M&C
    order; see
    <https://wiki.mwatelescope.org/pages/viewpage.action?pageId=48005139>.
    `delays` *must* have 16 elements, whereas `amps` can have 16 or 32
    elements; if 16 are given, then these map 1:1 with dipoles, otherwise
    the first 16 are for X dipole elements, and the next 16 are for Y.

>>> print(beam.calc_jones(0, 0.7, 167e6, [0]*16, [1]*16, True, -0.4660608448386394, True))
[-1.51506097e-01-4.35034884e-02j -9.76099405e-06-1.21699926e-05j
  1.73003520e-05-1.53580286e-05j -2.23184781e-01-4.51051073e-02j]

CUDA / HIP

hyperbeam also can also be run on NVIDIA GPUs, or AMD GPUs. To see an example of usage, see any of the examples with "cuda" or "hip" in the name. GPU functionality is provided with Cargo features; see installing from source instructions below.

Installation

Python PyPI

If you're using Python version >=3.6:

pip install mwa_hyperbeam

Pre-compiled

Have a look at the GitHub releases page. There is a Python wheel for all versions of Python 3.6+, as well as shared and static objects for C-style linking. To get an idea of how to link hyperbeam, see the fee.c file in the examples directory.

Because these hyperbeam objects have the HDF5 and ERFA libraries compiled in, their respective licenses are also distributed.

From source

Prerequisites

  • Cargo and a Rust compiler. rustup is recommended:

    https://www.rust-lang.org/tools/install

    The Rust compiler must be at least version 1.64.0:

    $ rustc -V
    rustc 1.64.0 (a55dd71d5 2022-09-19)
    
  • hdf5

    • Optional; use the hdf5-static or all-static features.
      • Requires CMake version 3.10 or higher.
    • Ubuntu: libhdf5-dev
    • Arch: hdf5
    • The C library dir can be specified manually with HDF5_DIR
      • If this is not specified, pkg-config is used to find the library.

Clone the repo, and run:

cargo build --release

For usage with other languages, an include file will be in the include directory, along with C-compatible shared and static objects in the target/release directory.

CUDA

Are you running hyperbeam on a desktop GPU? Then you probably want to compile with single-precision floats:

cargo build --release --features=cuda,gpu-single
cargo build --release --features=hip,gpu-single

Otherwise, go ahead with double-precision floats:

cargo build --release --features=cuda
cargo build --release --features=hip

Desktop GPUs (e.g. NVIDIA GeForce RTX 2070) have significantly less double-precision compute capability than "data center" GPUs (e.g. NVIDIA V100). Allowing hyperbeam to switch on the float type allows the user to decide between the performance and precision compromise.

CUDA can also be linked statically:

cargo build --release --features=cuda,cuda-static

HIP

The situation with HIP is similar to that of CUDA; use the hip feature and use gpu-single if you want the code to use single-precision floats. HIP does not appear to offer static libraries, so no static feature is provided.

Static dependencies

To make hyperbeam without a dependence on a system HDF5 library, give the build command a feature flag:

cargo build --release --features=hdf5-static

This will automatically compile the HDF5 source code and "bake" it into the hyperbeam products, meaning that HDF5 is not needed as a system dependency. CMake version 3.10 or higher is needed to build the HDF5 source.

To compile all C libraries statically:

cargo build --release --features=all-static

Python

To install hyperbeam to your currently-in-use virtualenv or conda environment, you'll need the Python package maturin (can get it with pip), then run:

maturin develop --release -b pyo3 --features=python --strip

If you don't have or don't want to install HDF5 as a system dependency, include the hdf5-static feature:

maturin develop --release -b pyo3 --features=python,hdf5-static --strip

Comparing with other FEE beam codes

Below is a table comparing other implementations of the FEE beam code. All benchmarks were done with unique azimuth and zenith angle directions, and all on the same system. The CPU is a Ryzen 9 3900X, which has 12 cores and SMT (24 threads). The CUDA benchmarks uses an NVIDIA GeForce RTX 2070. All benchmarks were done in serial, unless indicated by "parallel". Python times were taken by running time.time() before and after the calculations. Memory usage is measured by running time -v on the command (not the time associated with your shell; this is usually at /usr/bin/time).

Code Number of directions Duration Max. memory usage
mwa_pb 500 98.8 ms 134.6 MiB
100000 13.4 s 5.29 GiB
1000000 139.8 s 51.6 GiB
mwa-reduce (C++) 500 115.2 ms 48.9 MiB
10000 2.417 s 6.02 GiB
mwa_hyperbeam 500 10.0 ms 9.75 MiB
100000 1.82 s 11.3 MiB
1000000 18.1 s 25.0 MiB
mwa_hyperbeam (parallel) 1000000 1.55 s 88.8 MiB
mwa_hyperbeam (via python) 500 20.5 ms 44.2 MiB
100000 3.70 s 45.4 MiB
1000000 37.2 s 59.0 MiB
mwa_hyperbeam (via python, parallel) 1000000 2.49 s 246.6 MiB
mwa_hyperbeam (CUDA, single precision) 1000000 450 ms 253.8 MiB
1e8 3.08 s 14.26 GiB

Not sure what's up with the C++ code. Maybe I'm calling CalcJonesArray wrong, but it uses a huge amount of memory. In any case, hyperbeam seems to be roughly 10x faster. If you know how to compare with Everybeam, please let me know.

Troubleshooting

Run your code with hyperbeam again, but this time with the debug build. This should be as simple as running:

cargo build

and then using the results in ./target/debug.

If that doesn't help reveal the problem, report the version of the software used, your usage and the program output in a new GitHub issue.

hyperbeam?

AERODACTYL used HYPER BEAM!

Dependencies

~9–26MB
~403K SLoC