You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.1 KiB
46 lines
1.1 KiB
#include "neural_utils.h"
|
|
|
|
unsigned long int estimate_size(ErlNifEnv *env, ERL_NIF_TERM term) {
|
|
if (enif_is_atom(env, term)) {
|
|
return WORD_SIZE;
|
|
}
|
|
|
|
// Treating all numbers like longs.
|
|
if (enif_is_number(env, term)) {
|
|
return 2 * WORD_SIZE;
|
|
}
|
|
|
|
if (enif_is_binary(env, term)) {
|
|
ErlNifBinary bin;
|
|
enif_inspect_binary(env, term, &bin);
|
|
return bin.size + (6 * WORD_SIZE);
|
|
}
|
|
|
|
if (enif_is_list(env, term)) {
|
|
unsigned long int size = 0;
|
|
ERL_NIF_TERM it, curr;
|
|
it = term;
|
|
size += WORD_SIZE;
|
|
while (!enif_is_empty_list(env, it)) {
|
|
enif_get_list_cell(env, it, &curr, &it);
|
|
size += estimate_size(env, curr) + WORD_SIZE;
|
|
}
|
|
return size;
|
|
}
|
|
|
|
if (enif_is_tuple(env, term)) {
|
|
unsigned long int size = 0;
|
|
const ERL_NIF_TERM *tpl;
|
|
int arity;
|
|
enif_get_tuple(env, term, &arity, &tpl);
|
|
for (int i = 0; i < arity; ++i) {
|
|
size += estimate_size(env, tpl[i]);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
// Return 1 word by default
|
|
return WORD_SIZE;
|
|
}
|
|
|
|
|