Packages
evision
1.0.0-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.
Current section
Files
Jump to
Current section
Files
c_src/modules/evision_mat_api.h
#ifndef EVISION_MAT_API_H
#define EVISION_MAT_API_H
#pragma once
#include <erl_nif.h>
#include "../evision_consts.h"
static ERL_NIF_TERM __evision_get_mat_type(ErlNifEnv *env, int type) {
uint8_t depth = type & CV_MAT_DEPTH_MASK;
switch ( depth ) {
case CV_8U: return enif_make_tuple2(env, kAtomU, enif_make_uint64(env, 8));
case CV_8S: return enif_make_tuple2(env, kAtomS, enif_make_uint64(env, 8));
case CV_16U: return enif_make_tuple2(env, kAtomU, enif_make_uint64(env, 16));
case CV_16S: return enif_make_tuple2(env, kAtomS, enif_make_uint64(env, 16));
case CV_32S: return enif_make_tuple2(env, kAtomS, enif_make_uint64(env, 32));
case CV_32F: return enif_make_tuple2(env, kAtomF, enif_make_uint64(env, 32));
case CV_64F: return enif_make_tuple2(env, kAtomF, enif_make_uint64(env, 64));
case CV_16F: return enif_make_tuple2(env, kAtomF, enif_make_uint64(env, 16));
case CV_64S: return enif_make_tuple2(env, kAtomS, enif_make_uint64(env, 64));
case CV_32U: return enif_make_tuple2(env, kAtomU, enif_make_uint64(env, 32));
case CV_64U: return enif_make_tuple2(env, kAtomU, enif_make_uint64(env, 64));
case CV_16BF: return enif_make_tuple2(env, kAtomBF, enif_make_uint64(env, 16));
default: return enif_make_tuple2(env, kAtomUser, enif_make_uint64(env, depth));
}
}
static ERL_NIF_TERM _evision_get_mat_type(ErlNifEnv *env, const cv::Mat& img) {
int type = img.type();
return __evision_get_mat_type(env, type);
}
static ERL_NIF_TERM _evision_get_mat_shape(ErlNifEnv *env, const cv::Mat& img) {
cv::MatSize size = img.size;
int channels = img.channels();
int dims = img.dims;
int include_channels = 0;
if (img.channels() != 1) {
dims += 1;
include_channels = 1;
}
ERL_NIF_TERM* shape = (ERL_NIF_TERM *)enif_alloc(sizeof(ERL_NIF_TERM) * dims);
for (int i = 0; i < img.dims; i++) {
shape[i] = enif_make_int(env, size[i]);
}
if (include_channels) {
shape[dims - 1] = enif_make_int(env, channels);
}
#ifdef GLEAM_EVISION
ERL_NIF_TERM ret = enif_make_list_from_array(env, shape, dims);
#else
ERL_NIF_TERM ret = enif_make_tuple_from_array(env, shape, dims);
#endif
enif_free(shape);
return ret;
}
static ERL_NIF_TERM _evision_make_mat_resource_into_map(ErlNifEnv *env, const cv::Mat& m, ERL_NIF_TERM res_term) {
const size_t num_items = 7;
size_t item_index = 0;
ERL_NIF_TERM keys[num_items];
ERL_NIF_TERM values[num_items];
keys[item_index] = kAtomChannels;
values[item_index] = enif_make_int(env, m.channels());
item_index++;
keys[item_index] = kAtomDims;
values[item_index] = enif_make_int(env, m.dims);
item_index++;
keys[item_index] = kAtomType;
values[item_index] = _evision_get_mat_type(env, m);
item_index++;
keys[item_index] = kAtomRawType;
values[item_index] = enif_make_int(env, m.type());
item_index++;
keys[item_index] = kAtomShape;
values[item_index] = _evision_get_mat_shape(env, m);
item_index++;
keys[item_index] = kAtomRef;
values[item_index] = res_term;
item_index++;
keys[item_index] = kAtomClass;
values[item_index] = kAtomEvisionMatModule;
item_index++;
ERL_NIF_TERM map;
if (enif_make_map_from_arrays(env, keys, values, item_index, &map)) {
return map;
} else {
return evision::nif::error(env, "error when making map from arrays");
}
}
#endif // EVISION_MAT_API_H