Need to generalise [this module](https://github.com/SheffieldML/advectionGP/blob/main/advectionGP/wind.py) (and update all the examples that use it).