Packages
mmath
0.2.13
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/trans_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;
}
#define MATHF(name, fun) \
static ERL_NIF_TERM \
name (ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) \
{ \
ErlNifBinary bin; \
ERL_NIF_TERM r; \
ffloat* vs; \
ErlNifSInt64 int_v; \
double m; \
ffloat* target; \
int count; \
\
if (argc != 2) \
return enif_make_badarg(env); \
\
GET_BIN(0, bin, count, vs); \
\
if (enif_get_int64(env, argv[1], &int_v)) { \
m = (double) int_v; \
} else if (!enif_get_double(env, argv[1], &m)) { \
return enif_make_badarg(env); \
} \
\
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r))) \
return enif_make_badarg(env); \
\
for (int i = 0; i < count; i++) { \
target[i] = fun (vs[i], m); \
} \
return r; \
}
MATHF(add, float_addc)
MATHF(sub, float_subc)
MATHF(mul, float_mulc)
MATHF(divide, float_divc)
MATHF(min, float_minc)
MATHF(max, float_maxc)
static ERL_NIF_TERM
derivate(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
ErlNifBinary bin;
ERL_NIF_TERM r;
ffloat* target;
ffloat* vs;
int count;
if (argc != 1)
return enif_make_badarg(env);
GET_BIN(0, bin, count, vs);
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r)))
return enif_make_badarg(env); // TODO return propper error
if (count == 0)
return r;
if (count == 1) {
target[0] = (ffloat) {.value = 0, .confidence = 0};
return r;
}
for (int i = 1; i < count; i++) {
target[i] = float_sub(vs[i], vs[i-1]);
}
target[0] = target[1];
target[0].confidence = 0;
return r;
}
static ERL_NIF_TERM
confidence(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
ErlNifBinary a;
ERL_NIF_TERM r;
ffloat* vs;
ffloat* target;
int count;
if (argc != 1)
return enif_make_badarg(env);
GET_BIN(0, a, count, vs);
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r)))
return enif_make_badarg(env); // TODO return propper error
for (int i = 0; i < count; i++) {
target[i] = (ffloat){
.value = vs[i].confidence,
.confidence = CERTAIN
};
}
return r;
}
static ERL_NIF_TERM
square_root(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
ErlNifBinary a;
ERL_NIF_TERM r;
ffloat* vs;
ffloat* target;
int count;
if (argc != 1)
return enif_make_badarg(env);
GET_BIN(0, a, count, vs);
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r)))
return enif_make_badarg(env); // TODO return propper error
for (int i = 0; i < count; i++) {
if (vs[i].value < 0) {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = sqrt(vs[i].value * -1) * - 1
};
} else {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = sqrt(vs[i].value)
};
}
}
return r;
}
static ERL_NIF_TERM
c_log10(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
ErlNifBinary a;
ERL_NIF_TERM r;
ffloat* vs;
ffloat* target;
int count;
if (argc != 1)
return enif_make_badarg(env);
GET_BIN(0, a, count, vs);
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r)))
return enif_make_badarg(env); // TODO return propper error
for (int i = 0; i < count; i++) {
if (vs[i].value < 0) {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = log10(vs[i].value * -1) * - 1
};
} else if (vs[i].value == 0) {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = 0
};
} else {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = log10(vs[i].value)
};
}
}
return r;
}
static ERL_NIF_TERM
c_abs(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
ErlNifBinary a;
ERL_NIF_TERM r;
ffloat* vs;
ffloat* target;
int count;
if (argc != 1)
return enif_make_badarg(env);
GET_BIN(0, a, count, vs);
if (! (target = (ffloat*) enif_make_new_binary(env, count * sizeof(ffloat), &r)))
return enif_make_badarg(env); // TODO return propper error
for (int i = 0; i < count; i++) {
if (vs[i].value >= 0) {
target[i] = vs[i];
} else {
target[i] = (ffloat){
.confidence = vs[i].confidence,
.value = vs[i].value * - 1.0
};
}
}
return r;
}
static ErlNifFunc nif_funcs[] = {
{"add", 2, add},
{"sub", 2, sub},
{"mul", 2, mul},
{"min", 2, min},
{"max", 2, max},
{"divide", 2, divide},
{"derivate", 1, derivate},
{"confidence", 1, confidence},
{"sqrt_scale", 1, square_root},
{"log10_scale", 1, c_log10},
{"abs", 1, c_abs}
};
// 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_trans, nif_funcs, &load, NULL, &upgrade, NULL);