pytroll / pyorbital

Orbital and astronomy computations in python
http://pyorbital.readthedocs.org/
GNU General Public License v3.0
224 stars 77 forks source link

Add support for dask and xarray #88

Open djhoese opened 2 years ago

djhoese commented 2 years ago

Feature description

As discussed in https://github.com/pytroll/satpy/pull/1910, calculations will generally perform better if they are wrapped in a da.map_blocks call when dask arrays are involved. The Satpy PR implements basic wrappers for this. In the long run it would be best if pyorbital had this builtin. @mraspaud had the idea of using a decorator that would detect if it was being given dask arrays or numpy arrays and wrap the function call in map_blocks if using dask arrays. Additionally, I think this functionality could be expanded to include xarray DataArrays.