Skip to content

Extension to xarray providing functionality to work with spherical harmonic datasets

License

Notifications You must be signed in to change notification settings

ITC-Water-Resources/shxarray

Repository files navigation

Xarray Extension for working with spherical harmonic data

Build PyPI version Documentation Status

This extension adds functionality to work with Spherical Harmonics to Xarray.

Features and functionality (in progress)

  • Gravity functionals: (convert from Stokes coefficients to various gravity functionals, such as equivalent water heights, geoid changes, etc.)
  • Filter (e.g. Gaussian or anisotropic filter applied in the spectral domain)
  • The use of Xarray like operations allow for applying functionality to multi-dimensional datasets

Getting started

The tutorials in the documentation will gradually epxand and provide Jupyter notebook workflows to start with.

Installation

You can install this package from PyPi using:

pip install shxarray

Development Installation

If you want to help in the development of this package, it's best to clone the repository to allow for modifications and pull requests. The extension makes use of Cython generated code to speed up spherical harmonic synthesis and analysis.

  1. Create your own virtual environment with venv or Anaconda (Optional but recommended, when a user installation is desired)
  2. Clone this repository git clone https://github.com/ITC-Water-Resources/shxarray.git
  3. Change to the repository directory cd shaxarray
  4. Set the environment variable export USE_CYTHON=1 (Optional and only in the case Cython code is being developed or needs to be regenerated)
  5. Install using pip pip install . or use pip install -e . for an editable install

Cython build tip on an editable install

From the repository root directory, regenerating the shared library running

python ./setup.py build_ext

will be much faster than using

pip install -e .

This will build the shared library in for example ./build/lib.linux-x86_64-cpython-3xx/shxarray/shlib.cpython-3xx-x86_64-linux-gnu.so. To make sure changes are picked up in your editable install you can create a symbolic link in the Python part of the library e.g. :

cd src/shxarray/
ln -sf ../../build/lib.linux-x86_64-cpython-311/shxarray/shlib.cpython-311-x86_64-linux-gnu.so

Building against Numpy 2

For portablity, the current C++ files are build against Numpy version 1.26 and are incompatible with a Numpy 2.0 installation. To build against Numpy 2.0, (1) change the Numpy version numbers in pyproject.toml, (2) set the environment variable export USE_CYTHON=1, and (3) build the module as described above.

Contributing

This repository is under development and contributions and feedback are welcome.

Contributors