Function af_anisotropic_diffusion

src/api/c/anisotropic_diffusion.cpp:61–104

}

af_err af_anisotropic_diffusion(af_array* out, const af_array in,
                                const float dt, const float K,
                                const unsigned iterations,
                                const af_flux_function fftype,
                                const af_diffusion_eq eq) {
    try {
        const ArrayInfo& info = getInfo(in);

        const af::dim4& inputDimensions = info.dims();
        const af_dtype inputType        = info.getType();
        const unsigned inputNumDims     = inputDimensions.ndims();

        DIM_ASSERT(1, (inputNumDims >= 2));

        ARG_ASSERT(3, (K > 0 || K < 0));
        ARG_ASSERT(4, (iterations > 0));

        const af_flux_function F =
            (fftype == AF_FLUX_DEFAULT ? AF_FLUX_EXPONENTIAL : fftype);

        auto input = castArray<float>(in);

        af_array output = nullptr;
        switch (inputType) {
            case f64:
                output = diffusion<double>(input, dt, K, iterations, F, eq);
                break;
            case f32:
            case s32:
            case u32:
            case s16:
            case u16:
            case s8:
            case u8:
                output = diffusion<float>(input, dt, K, iterations, F, eq);
                break;
            default: TYPE_ERROR(1, inputType);
        }
        std::swap(*out, output);
    }
    CATCHALL;

    return AF_SUCCESS;
}