diff --git a/mlair/helpers/geofunctions.py b/mlair/helpers/geofunctions.py index 3082b5633ae954611a859f23c42a91f4e9cb00de..17923cd273f983a2c40f4833acaf395c7a4d8132 100644 --- a/mlair/helpers/geofunctions.py +++ b/mlair/helpers/geofunctions.py @@ -8,6 +8,7 @@ import xarray as xr from mlair.helpers.helpers import convert2xrda from typing import Union, Tuple +import logging xr_int_float = Union[xr.DataArray, xr.Dataset, np.ndarray, int, float] tuple_of_2xr_int_float = Tuple[xr_int_float, xr_int_float] @@ -416,7 +417,12 @@ class VectorRotateLambertConformal2latlon(VectorRotate): :return: :rtype: """ - return vgrd * da.sin(self._alpha) + ugrd * da.cos(self._alpha) + ull = vgrd * da.sin(self._alpha) + ugrd * da.cos(self._alpha) + try: + ull.attrs = ugrd.attrs + except Exception as e: + logging.debug(f"Can not set attrs for ull due to: {e}") + return ull def vgrd2vll(self, ugrd, vgrd): """ @@ -429,12 +435,12 @@ class VectorRotateLambertConformal2latlon(VectorRotate): :return: :rtype: """ - return vgrd * da.cos(self._alpha) - ugrd * da.sin(self._alpha) - - - - - + vll = vgrd * da.cos(self._alpha) - ugrd * da.sin(self._alpha) + try: + vll.attrs = vgrd.attrs + except Exception as e: + logging.debug(f"Can not set attrs for vll due to: {e}") + return vll if __name__ == '__main__':