static void
na_parse_narray_index(VALUE a, int orig_dim, ssize_t size, na_index_arg_t *q)
{
VALUE idx;
narray_t *na;
narray_data_t *nidx;
size_t k, n;
ssize_t *nidxp;
GetNArray(a,na);
if (NA_NDIM(na) != 1) {
rb_raise(rb_eIndexError, "should be 1-d NArray");
}
n = NA_SIZE(na);
idx = nary_new(cIndex,1,&n);
na_store(idx,a);
GetNArrayData(idx,nidx);
nidxp = (ssize_t*)nidx->ptr;
q->idx = ALLOC_N(size_t, n);
for (k=0; k<n; k++) {
q->idx[k] = na_range_check(nidxp[k], size, orig_dim);
}
q->n = n;
q->beg = 0;
q->step = 1;
q->reduce = 0;
q->orig_dim = orig_dim;
}
static VALUE
na_aref_md(int argc, VALUE *argv, VALUE self, int keep_dim, int result_nd)
{
VALUE args; // should be GC protected
narray_t *na1;
na_aref_md_data_t data;
VALUE store = 0;
VALUE idx;
narray_t *nidx;
GetNArray(self,na1);
args = rb_ary_new4(argc,argv);
if (argc == 1 && result_nd == 1) {
idx = argv[0];
if (rb_obj_is_kind_of(idx, rb_cArray)) {
idx = rb_apply(numo_cNArray,id_bracket,idx);
}
if (rb_obj_is_kind_of(idx, numo_cNArray)) {
GetNArray(idx,nidx);
if (NA_NDIM(nidx)>1) {
store = nary_new(CLASS_OF(self),NA_NDIM(nidx),NA_SHAPE(nidx));
idx = na_flatten(idx);
RARRAY_ASET(args,0,idx);
}
}
// flatten should be done only for narray-view with non-uniform stride.
if (na1->ndim > 1) {
self = na_flatten(self);
GetNArray(self,na1);
}
}
data.args = args;
data.self = self;
data.store = store;
data.ndim = result_nd;
data.q = na_allocate_index_args(result_nd);
data.na1 = na1;
data.keep_dim = keep_dim;
return rb_ensure(na_aref_md_protected, (VALUE)&data, na_aref_md_ensure, (VALUE)&data);
}
/*
Replaces the contents of self with the contents of other narray.
Used in dup and clone method.
@overload initialize_copy(other)
@param [Numo::NArray] other
@return [Numo::NArray] self
*/
static VALUE
na_initialize_copy(VALUE self, VALUE orig)
{
narray_t *na;
GetNArray(orig,na);
na_setup(self,NA_NDIM(na),NA_SHAPE(na));
na_store(self,orig);
na_copy_flags(orig,self);
return self;
}
/* method: shape() -- returns shape, array of the size of dimensions */
static VALUE
na_shape(VALUE self)
{
volatile VALUE v;
narray_t *na;
size_t i, n, c, s;
GetNArray(self,na);
n = NA_NDIM(na);
if (TEST_COLUMN_MAJOR(na)) {
c = n-1;
s = -1;
} else {
c = 0;
s = 1;
}
v = rb_ary_new2(n);
for (i=0; i<n; i++) {
rb_ary_push(v, SIZE2NUM(na->shape[c]));
c += s;
}
return v;
}
