Function _tk

src/wrf/extension.py:259–274

Wrapper for dcomputetk. Located in wrf_user.f90.

(pressure, theta, outview=None)

@cast_type(arg_idxs=(0, 1))
@extract_and_transpose()
def _tk(pressure, theta, outview=None):
    """Wrapper for dcomputetk.

    Located in wrf_user.f90.

    """
    # No need to transpose here since operations on 1D array
    shape = pressure.shape
    if outview is None:
        outview = np.empty_like(pressure)
    result = dcomputetk(outview.ravel(order="A"),
                        pressure.ravel(order="A"),
                        theta.ravel(order="A"))
    result = np.reshape(result, shape, order="F")

    return result


@check_args(0, 2, (2, 2))