Packages
mmath
0.2.3
0.2.26
0.2.25
0.2.24
0.2.23
0.2.22
0.2.21
0.2.20
0.2.19
0.2.18
0.2.17
0.2.16
0.2.15
0.2.14
0.2.13
0.2.12
0.2.11
0.2.10
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.1
0.2.0
0.2.0-alpha9
0.2.0-alpha8
0.2.0-alpha7
0.2.0-alpha6
0.2.0-alpha5
0.2.0-alpha4
0.2.0-alpha3
0.2.0-alpha2
0.2.0-alpha10
0.2.0-alpha1
0.2.0-alpha
0.1.17-alpha
0.1.16
0.1.15
0.1.14
0.1.13
0.1.11
0.1.10
0.1.9
0.1.8
0.1.7
0.1.6
math library for metric sequences and binary arrays.
Current section
Files
Jump to
Current section
Files
c_src/comb_nif.c
#include "erl_nif.h"
#include "mmath.h"
#include <math.h>
static int
load(ErlNifEnv* env, void** priv, ERL_NIF_TERM load_info)
{
return 0;
}
static int
upgrade(ErlNifEnv* env, void** priv, void** old_priv, ERL_NIF_TERM load_info)
{
return 0;
}
// MACROS
#define COMB(name, fun) \
static ERL_NIF_TERM \
name##2 (ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) \
{ \
ErlNifBinary a; \
ErlNifBinary b; \
ERL_NIF_TERM r; \
ffloat* vs_a; \
ffloat* vs_b; \
ffloat* target; \
ffloat last_a = {0, 0}; \
ffloat last_b = {0, 0}; \
\
int count_a; \
int count_b; \
int count; \
int target_size; \
\
if (argc != 2) \
return enif_make_badarg(env); \
\
GET_BIN(0, a, count_a, vs_a); \
GET_BIN(1, b, count_b, vs_b); \
count = count_a > count_b ? count_a : count_b; \
\
target_size = count * sizeof(ffloat); \
if (! (target = (ffloat*) enif_make_new_binary(env, target_size, &r))) \
return enif_make_badarg(env); \
\
if (count_a == count_b) { \
for (int i = 0; i < count; i++) { \
target[i] = fun (vs_a[i], vs_b[i]); \
} \
} else { \
for (int i = 0; i < count; i++) { \
if (i < count_a) { \
last_a = vs_a[i]; \
} else { \
last_a.confidence = 0; \
} \
if (i < count_b) { \
last_b = vs_b[i]; \
} else { \
last_b.confidence = 0; \
} \
\
target[i] = fun (last_a, last_b); \
} \
} \
return r; \
}; \
static ERL_NIF_TERM \
name##3 (ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) \
{ \
ErlNifBinary a; \
ErlNifBinary b; \
ErlNifBinary c; \
ERL_NIF_TERM r; \
ffloat* vs_a; \
ffloat* vs_b; \
ffloat* vs_c; \
ffloat* target; \
ffloat last_a = {0, 0}; \
ffloat last_b = {0, 0}; \
ffloat last_c = {0, 0}; \
\
int count_a; \
int count_b; \
int count_c; \
int count; \
int target_size; \
\
if (argc != 3) \
return enif_make_badarg(env); \
\
GET_BIN(0, a, count_a, vs_a); \
GET_BIN(1, b, count_b, vs_b); \
GET_BIN(2, c, count_c, vs_c); \
\
count = count_a > count_b ? count_a : count_b; \
count = count > count_c ? count : count_c; \
\
target_size = count * sizeof(ffloat); \
if (! (target = (ffloat*) enif_make_new_binary(env, target_size, &r))) \
return enif_make_badarg(env); \
\
if (count_a == count_b && count_b == count_c) { \
for (int i = 0; i < count; i++) { \
target[i] = fun##3 (vs_a[i], vs_b[i], vs_c[i]); \
} \
} else { \
for (int i = 0; i < count; i++) { \
if (i < count_a) { \
last_a = vs_a[i]; \
} else { \
last_a.confidence = 0; \
} \
if (i < count_b){ \
last_b = vs_b[i]; \
} else { \
last_b.confidence = 0; \
} \
if (i < count_c){ \
last_c = vs_c[i]; \
} else { \
last_c.confidence = 0; \
} \
target[i] = fun##3 (last_a, last_b, last_c); \
} \
} \
return r; \
}
// SUM
COMB(sum, float_add)
// MIN
COMB(min, float_min)
// MAX
COMB(max, float_max)
static ErlNifFunc nif_funcs[] = {
{"sum", 2, sum2},
{"sum", 3, sum3},
{"min_", 2, min2},
{"min_", 3, min3},
{"max_", 2, max2},
{"max_", 3, max3}
};
// Initialize this NIF library.
//
// Args: (MODULE, ErlNifFunc funcs[], load, reload, upgrade, unload)
// Docs: http://erlang.org/doc/man/erl_nif.html#ERL_NIF_INIT
ERL_NIF_INIT(mmath_comb, nif_funcs, &load, NULL, &upgrade, NULL);