Sha256: c9a836bf84e47e38859de31f316e923f5ff40db5c6928ffa79064d65dcb795af
Contents?: true
Size: 1.83 KB
Versions: 17
Compression:
Stored size: 1.83 KB
Contents
typedef struct {
size_t count;
char *idx0;
char *idx1;
size_t elmsz;
} where_opt_t;
#define STORE_INT(ptr, esz, x) memcpy(ptr,&(x),esz)
static void
<%=c_iter%>(na_loop_t *const lp)
{
size_t i;
BIT_DIGIT *a;
size_t p;
ssize_t s;
size_t *idx;
BIT_DIGIT x=0;
char *idx1;
size_t count;
size_t e;
where_opt_t *g;
g = (where_opt_t*)(lp->opt_ptr);
count = g->count;
idx1 = g->idx1;
e = g->elmsz;
INIT_COUNTER(lp, i);
INIT_PTR_BIT_IDX(lp, 0, a, p, s, idx);
if (idx) {
for (; i--;) {
LOAD_BIT(a, p+*idx, x);
idx++;
if (x!=0) {
STORE_INT(idx1,e,count);
idx1 += e;
}
count++;
}
} else {
for (; i--;) {
LOAD_BIT(a, p, x);
p+=s;
if (x!=0) {
STORE_INT(idx1,e,count);
idx1 += e;
}
count++;
}
}
g->count = count;
g->idx1 = idx1;
}
/*
Returns the array of index where the bit is one (true).
@overload <%=op_map%>
@return [Numo::Int32,Numo::Int64]
*/
static VALUE
<%=c_func(0)%>(VALUE self)
{
volatile VALUE idx_1;
size_t size, n_1;
where_opt_t *g;
ndfunc_arg_in_t ain[1] = {{cT,0}};
ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 1, 0, ain, 0 };
size = RNARRAY_SIZE(self);
n_1 = NUM2SIZET(<%=find_tmpl("count_true").c_func%>(0, NULL, self));
g = ALLOCA_N(where_opt_t,1);
g->count = 0;
if (size>4294967295ul) {
idx_1 = nary_new(numo_cInt64, 1, &n_1);
g->elmsz = 8;
} else {
idx_1 = nary_new(numo_cInt32, 1, &n_1);
g->elmsz = 4;
}
g->idx1 = na_get_pointer_for_write(idx_1);
g->idx0 = NULL;
na_ndloop3(&ndf, g, 1, self);
na_release_lock(idx_1);
return idx_1;
}
Version data entries
17 entries across 17 versions & 1 rubygems