/*
* call-seq:
* Numo::DataType.new(shape) => narray
* Numo::DataType.new(size1, size2, ...) => narray
*
* Constructs a narray using the given <i>DataType</i> and <i>shape</i> or
* <i>sizes</i>.
*/
static VALUE
na_initialize(VALUE self, VALUE args)
{
VALUE v;
size_t *shape=NULL;
int ndim;
if (RARRAY_LEN(args) == 1) {
v = RARRAY_AREF(args,0);
if (TYPE(v) != T_ARRAY) {
v = args;
}
} else {
v = args;
}
ndim = RARRAY_LEN(v);
if (ndim > NA_MAX_DIMENSION) {
rb_raise(rb_eArgError,"ndim=%d exceeds maximum dimension",ndim);
}
shape = ALLOCA_N(size_t, ndim);
// setup size_t shape[] from VALUE shape argument
na_array_to_internal_shape(self, v, shape);
na_setup(self, ndim, shape);
return self;
}
static VALUE
nstruct_add_type(VALUE type, int argc, VALUE *argv, VALUE nst)
{
VALUE ofs, size;
ID id;
int i;
VALUE name=Qnil;
size_t *shape=NULL;
int ndim=0;
ssize_t stride;
narray_view_t *nt;
int j;
for (i=0; i<argc; i++) {
switch(TYPE(argv[i])) {
case T_STRING:
case T_SYMBOL:
if (NIL_P(name)) {
name = argv[i];
break;
}
rb_raise(rb_eArgError,"multiple name in struct definition");
case T_ARRAY:
if (shape) {
rb_raise(rb_eArgError,"multiple shape in struct definition");
}
ndim = RARRAY_LEN(argv[i]);
if (ndim > NA_MAX_DIMENSION) {
rb_raise(rb_eArgError,"too large number of dimensions");
}
if (ndim == 0) {
rb_raise(rb_eArgError,"array is empty");
}
shape = ALLOCA_N(size_t, ndim);
na_array_to_internal_shape(Qnil, argv[i], shape);
break;
}
}
id = rb_to_id(name);
name = ID2SYM(id);
if (rb_obj_is_kind_of(type,cNArray)) {
narray_t *na;
GetNArray(type,na);
type = CLASS_OF(type);
ndim = na->ndim;
shape = na->shape;
}
type = rb_narray_view_new(type,ndim,shape);
GetNArrayView(type,nt);
nt->stridx = ALLOC_N(stridx_t,ndim);
stride = na_dtype_elmsz(CLASS_OF(type));
for (j=ndim; j--; ) {
SDX_SET_STRIDE(nt->stridx[j], stride);
stride *= shape[j];
}
ofs = rb_iv_get(nst, "__offset__");
nt->offset = NUM2SIZET(ofs);
size = rb_funcall(type, rb_intern("byte_size"), 0);
rb_iv_set(nst, "__offset__", rb_funcall(ofs,'+',1,size));
rb_ary_push(rb_iv_get(nst,"__members__"),
rb_ary_new3(4,name,type,ofs,size)); // <- field definition
return Qnil;
}
