#ifndef EVISION_BACKEND_CEIL_H
#define EVISION_BACKEND_CEIL_H

#include <erl_nif.h>
#include "../../ArgInfo.hpp"

// @evision c: mat_ceil, evision_cv_mat_ceil, 1
// @evision nif: def mat_ceil(_opts \\ []), do: :erlang.nif_error(:undefinedined)
static ERL_NIF_TERM evision_cv_mat_ceil(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]) {
    using namespace cv;
    ERL_NIF_TERM error_term = 0;
    std::map<std::string, ERL_NIF_TERM> erl_terms;
    int nif_opts_index = 0;
    evision::nif::parse_arg(env, nif_opts_index, argv, erl_terms);

    {
        Mat img;

        if (evision_to_safe(env, evision_get_kw(env, erl_terms, "img"), img, ArgInfo("img", 0))) {
            int type = img.type();
            uint8_t depth = type & CV_MAT_DEPTH_MASK;
            if (depth == CV_32F) {
                auto ptr = img.ptr<float>();
                size_t count = img.total();
                for (size_t i = 0; i < count; ++i) {
                    ptr[i] = ceilf(ptr[i]);
                }
                return evision_from(env, img);
            } else if (depth == CV_64F) {
                auto ptr = img.ptr<double>();
                size_t count = img.total();
                for (size_t i = 0; i < count; ++i) {
                    ptr[i] = ceil(ptr[i]);
                }
                return evision_from(env, img);
            } else {
                return evision_from(env, img);
            }
        }
    }

    if (error_term != 0) return error_term;
    else return enif_make_badarg(env);
}

#endif // EVISION_BACKEND_CEIL_H