Function extract_dim

src/wrf/util.py:892–930

Return the dimension size for the specified dimension name. Args: wrfin (:class:`netCDF4.Dataset`, :class:`Nio.NioFile`, or an \ iterable): WRF-ARW NetCDF data as a :class:`netCDF4.Dataset`, :class:`Nio.NioFile` or an iterable sequence of the aforeme

(wrfin, dim)

def extract_dim(wrfin, dim):
    """Return the dimension size for the specified dimension name.

    Args:

        wrfin (:class:`netCDF4.Dataset`, :class:`Nio.NioFile`, or an \
            iterable): WRF-ARW NetCDF
            data as a :class:`netCDF4.Dataset`, :class:`Nio.NioFile`
            or an iterable sequence of the aforementioned types.

        dim (:obj:`str`): The dimension name.

    Returns:

        :obj:`int`:  The dimension size.

    """
    if is_multi_file(wrfin):
        if not is_mapping(wrfin):
            wrfin = next(iter(wrfin))
        else:
            entry = wrfin[next(iter(viewkeys(wrfin)))]
            return extract_dim(entry, dim)

    d = wrfin.dimensions[dim]
    if not isinstance(d, int):
        try:
            return len(d)  # netCDF4
        except TypeError:  # scipy.io.netcdf
            # Scipy can't handled unlimited dimensions, so now we have to
            # figure it out
            try:
                s = wrfin.variables["P"].shape
            except Exception:
                raise ValueError("unsupported NetCDF reader")
            else:
                return s[-4]

    return d  # PyNIO


def _combine_dict(wrfdict, varname, timeidx, method, meta, _key):