The sxs
python package provides a high-level interface for using data
produced by the SXS collaboration. In particular, the function sxs.load
can
automatically find, download, and load data, returning objects that provide
common interfaces to the various types of data, without forcing the user to
worry about details like data formats or where to find the data. It can also
automatically select the newest or highest-resolution dataset for a given
simulation, or return a range of versions or resolutions. Currently, the
high-level objects encapsulate
- Simulations — a catalog of all simulations produced by the SXS collaboration
- Simulation — an object encapsulating all data for a single simulation
- Metadata — data describing the simulation parameters
- Horizons — time-series data describing the apparent horizons
- Waveforms — time-series data describing the extrapolated gravitational-wave modes
Because this package is pure python code, installation is very simple. In particular, with a reasonably modern installation, you can just run a command like
conda install -c conda-forge sxs
or
python -m pip install sxs
Here, conda
requires the conda
installation of python, which is the most recommended approach for scientific
python; the second command assumes that you have an appropriate python
environment set up in some other way. Either of these commands will download
and install the sxs
package and its most vital requirements.
If you want to install all the goodies that enable things like jupyter notebooks with plots and interactive tables, you could run
conda install -c conda-forge sxs-ecosystem
or
python -m pip install sxs[ecosystem]
You will probably also want to set some sensible defaults to automatically download and cache data:
python -c "import sxs; sxs.write_config(download=True, cache=True)"
This will write a configuration file in the directory returned by
sxs.sxs_directory("config")
, and downloaded data will be cached in the
directory returned by sxs.sxs_directory("cache")
. See that function's
documentation
for details.
An extensive demonstration of this package's capabilities is available
here, in the form of
interactive jupyter notebooks that are actually running this code and some
pre-downloaded data. The following is just a very brief overview of the sxs
package's main components.
There are five important objects to understand in this package:
import sxs
simulations = sxs.load("simulations")
sxs_bbh_1234 = sxs.load("SXS:BBH:1234")
metadata = sxs_bbh_1234.metadata
horizons = sxs_bbh_1234.horizons
h = sxs_bbh_1234.h
The simulations
object contains
information about every simulation in the catalog, including all
available data files, and information about how to get them. You
probably don't need to actually know about details like where to get
the data, but simulations
can help you find the simulations you care
about. It is a dict
object, where the keys are names of simulations
(like "SXS:BBH:0123") and the values are the same types as the
metadata
object,
which contains metadata about that simulation — things like mass
ratio, spins, etc. This metadata
reflects the actual output of the
simulations, which leads to some inconsistencies in their formats. A
more consistent interface (though it is biased toward returning NaNs
where a human might glean more information) is provided by
simulations.dataframe
, which returns a
pandas
DataFrame
with specific
data types for each column.
The actual data itself is primarily contained in the next two objects. The
horizons
object
has three attributes — horizons.A
, horizons.B
, and horizons.C
— typically
representing the original two horizons of the black-hole binary and the common
horizon that forms at merger. In matter simulations, one or more of these may
be None
. Otherwise, each of these three is a
HorizonQuantities
object, containing several timeseries relating to mass, spin, and position.
Finally, the
waveform
encapsulates the modes of the waveform and the corresponding time information,
along with relevant metadata like data type, spin weight, etc., and useful
features like numpy-array-style slicing.