diff --git a/pyremo/preproc/cf.py b/pyremo/preproc/cf.py index c9db7cf..55a27f5 100644 --- a/pyremo/preproc/cf.py +++ b/pyremo/preproc/cf.py @@ -15,7 +15,14 @@ import xarray as xr from cdo import Cdo -from .core import check_lev, convert_units, get_vc, horizontal_dims, open_mfdataset +from .core import ( + check_lev, + convert_units, + get_vc, + horizontal_dims, + map_sst, + open_mfdataset, +) cdo_exe = "cdo" @@ -187,7 +194,7 @@ def get_file(self, datetime=None, **kwargs): return sel.iloc[0].path -def gfile(ds, ref_ds=None, attrs=None, use_cftime=True, invertlev=None): +def gfile(ds, ref_ds=None, tos=None, attrs=None, use_cftime=True, invertlev=None): """Creates a global dataset ready for preprocessing. This function creates a homogenized global dataset. If neccessary, @@ -221,6 +228,8 @@ def gfile(ds, ref_ds=None, attrs=None, use_cftime=True, invertlev=None): ds["akgm"], ds["bkgm"] = get_vc(ds, invertlev) ds = check_lev(ds, invertlev) + if tos is not None: + ds["tos"] = map_sst(tos, ds) # ensure correct units ds = convert_units(ds) if "sftlf" in ds: