Packages

evision

1.0.1-rc.0
1.0.1-rc.0 1.0.0 1.0.0-rc.0 0.2.17 0.2.17-rc2 0.2.17-rc1 0.2.16 0.2.16-pre.2 0.2.16-pre 0.2.15 0.2.14 0.2.13 0.2.12 0.2.11 0.2.9 0.2.8 0.2.7 0.2.6 0.2.5 0.2.4 0.2.3 0.2.2 0.2.2-rc2 0.2.1 0.2.0 0.1.39 0.1.38 0.1.37 0.1.36 0.1.35 0.1.34 0.1.33 0.1.32 retired 0.1.31 0.1.30 0.1.29 0.1.28 0.1.27 0.1.26 0.1.26-rc3 0.1.26-rc2 0.1.26-rc1 0.1.26-rc0 0.1.25 0.1.24 0.1.23 0.1.22 0.1.21 0.1.20 0.1.19 0.1.18 0.1.17 0.1.16 0.1.15 0.1.14 0.1.13 0.1.12 0.1.11 0.1.10 0.1.9 0.1.8 0.1.7

OpenCV-Erlang/Elixir binding.

HexDocs

Current section

Files

Jump to
Readme
62 Versions
5 Dependencies
Dependants
Files
Activity
evision c_src modules evision_backend select.h
Raw

c_src/modules/evision_backend/select.h 

#ifndef EVISION_BACKEND_SELECT_H

#define EVISION_BACKEND_SELECT_H



#include <cstring>

#include <erl_nif.h>

#include "../../ArgInfo.hpp"

#include "../evision_mat_utils.hpp"

#include "parallel.h"



// Elementwise select (Nx.select): result = mask ? on_true : on_false. All three

// have the same shape (broadcast by the caller); on_true/on_false are already the

// output type and mask is u8. Type-agnostic byte copy: start from on_true, then

// overwrite the elements where the mask is zero with on_false.

// @evision c: mat_select, evision_cv_mat_select, 1

// @evision nif: def mat_select(_opts \\ []), do: :erlang.nif_error(:undefined)

static ERL_NIF_TERM evision_cv_mat_select(ErlNifEnv *env, int argc, const ERL_NIF_TERM argv[]) {

    using namespace cv;

    using namespace evision::nif;

    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 mask, on_true, on_false;



        if (evision_to_safe(env, evision_get_kw(env, erl_terms, "mask"), mask, ArgInfo("mask", ArgInfo::INPUT_ONLY)) &&

            evision_to_safe(env, evision_get_kw(env, erl_terms, "on_true"), on_true, ArgInfo("on_true", ArgInfo::INPUT_ONLY)) &&

            evision_to_safe(env, evision_get_kw(env, erl_terms, "on_false"), on_false, ArgInfo("on_false", ArgInfo::INPUT_ONLY))) {

            Mat mc = mask.isContinuous() ? mask : mask.clone();

            Mat fc = on_false.isContinuous() ? on_false : on_false.clone();

            Mat dst = on_true.clone();



            const unsigned char *mp = mc.data;

            const unsigned char *fp = fc.data;

            unsigned char *dp = dst.data;

            size_t elem_size = dst.elemSize();

            size_t n = dst.total();

            evision_parallel_for((int64_t)n, (int64_t)n, EVISION_PARALLEL_SIMPLE_MIN_WORK, [&](int64_t begin, int64_t end) {

                for (int64_t i = begin; i < end; i++) {

                    if (mp[(size_t)i] == 0)

                        std::memcpy(dp + (size_t)i * elem_size, fp + (size_t)i * elem_size, elem_size);

                }

            });

            return evision_from(env, dst);

        }

    }



    return enif_make_badarg(env);

}



#endif // EVISION_BACKEND_SELECT_H