Source code for pycif.plugins.datastreams.fields.lmdz_inicond_ico.read

from __future__ import annotations

import datetime

import numpy as np
import xarray as xr




def read(
    self,
    name: str,
    varnames: str,
    dates: list[tuple[datetime.datetime, datetime.datetime]],
    files: list[str],
    tracer: object | None = None,
    **kwargs,
) -> xr.DataArray:
    varnames = varnames if varnames else name

    if len(files) > 1:
        raise ValueError("multiple files provided for initial conditions")
    (path,) = files

    with xr.open_dataset(path) as ds:
        data = ds[varnames].values

    xmod = xr.DataArray(
        data[np.newaxis, :, np.newaxis, :],
        coords={"time": [dates[0][0]]},
        dims=("time", "lev", "lat", "lon"),
    )

    return xmod