From b4953afb38efcaea886e2026d35e3c0adf99e36d Mon Sep 17 00:00:00 2001 From: Lars Buntemeyer Date: Fri, 1 Dec 2023 15:55:38 +0100 Subject: [PATCH] added map_sst --- pyremo/preproc/cf.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) 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: