carray 2.0.1 → 3.0.1

data/ext/ca_obj_unbound_repeat.c

@@ -1,529 +0,0 @@
-/* ---------------------------------------------------------------------------
-
-  ca_obj_unbound_repeat.c
-
-  This file is part of Ruby/CArray extension library.
-
-  Copyright (C) 2005-2025 Hiroki Motoyoshi
-
----------------------------------------------------------------------------- */
-
-#include "carray.h"
-
-const rb_data_type_t caunboundrepeat_data_type = {
-    .parent = &cavirtual_data_type,
-    .wrap_struct_name = "CAUnboundRepeat",
-    .function = {
-        .dmark = ca_mark,
-        .dfree = ca_free,
-        .dsize = NULL,
-        .dcompact = NULL
-    },
-    .flags = RUBY_TYPED_FREE_IMMEDIATELY
-};
-
-const rb_data_type_t caunboundrepeat_mask_data_type = {
-    .parent = &caunboundrepeat_data_type,
-    .wrap_struct_name = "CAUnboundRepeatMask",
-    .function = {
-        .dmark = NULL,
-        .dfree = ca_free_nop,
-        .dsize = NULL,
-        .dcompact = NULL
-    },
-    .flags = RUBY_TYPED_FREE_IMMEDIATELY
-};
-
-VALUE rb_cCAUnboundRepeat;
-VALUE rb_cCAUnboundRepeatMask;
-
-/* yard:
-  class CAUnboundRepeat < CArray 
-  end
-*/
-
-int
-ca_ubrep_setup (CAUnboundRepeat *ca, CArray *parent,
-                int32_t rep_ndim, ca_size_t *rep_dim)
-{
-  int8_t data_type;
-  ca_size_t bytes, elements;
-  int8_t i;
-
-  /* check arguments */
-
-  CA_CHECK_RANK(rep_ndim);
-
-  data_type = parent->data_type;
-  bytes     = parent->bytes;
-  elements  = parent->elements;
-
-  ca->obj_type  = CA_OBJ_UNBOUND_REPEAT;
-  ca->data_type = data_type;
-  ca->flags     = 0;
-  ca->ndim      = rep_ndim;
-  ca->bytes     = bytes;
-  ca->elements  = elements;
-  ca->ptr       = NULL;
-  ca->mask      = NULL;
-  ca->dim       = ALLOC_N(ca_size_t, rep_ndim);
-
-  ca->parent    = parent;
-  ca->attach    = 0;
-  ca->nosync    = 0;
-
-  ca->rep_ndim  = rep_ndim;
-  ca->rep_dim   = ALLOC_N(ca_size_t, rep_ndim);
-
-  for (i=0; i<rep_ndim; i++) {
-    ca->rep_dim[i] = rep_dim[i];
-    ca->dim[i] = ( rep_dim[i] ) ? rep_dim[i] : 1;      
-  }
-
-  if ( ca_has_mask(parent) ) {
-    ca_create_mask(ca);
-  }
-
-  return 0;
-}
-
-CAUnboundRepeat *
-ca_ubrep_new (CArray *parent, int32_t rep_ndim, ca_size_t *rep_dim)
-{
-  CAUnboundRepeat *ca = ALLOC(CAUnboundRepeat);
-  ca_ubrep_setup(ca, parent, rep_ndim, rep_dim);
-  return ca;
-}
-
-static void
-free_ca_ubrep (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  if ( ca != NULL ) {
-    ca_free(ca->mask);
-    xfree(ca->dim);
-    xfree(ca->rep_dim);
-    xfree(ca);
-  }
-}
-
-/* ------------------------------------------------------------------- */
-
-static void *
-ca_ubrep_func_clone (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  return ca_ubrep_new(ca->parent, ca->rep_ndim, ca->rep_dim);
-}
-
-static char *
-ca_ubrep_func_ptr_at_addr (void *ap, ca_size_t addr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  return ca_ptr_at_addr(ca->parent, addr);
-}
-
-static void
-ca_ubrep_func_fetch_addr (void *ap, ca_size_t addr, void *ptr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_fetch_addr(ca->parent, addr, ptr);
-}
-
-static void
-ca_ubrep_func_store_addr (void *ap, ca_size_t addr, void *ptr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_store_addr(ca->parent, addr, ptr);
-}
-
-static void
-ca_ubrep_func_allocate (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_allocate(ca->parent);
-  ca->ptr = ca->parent->ptr;
-}
-
-static void
-ca_ubrep_func_attach (void *ap)
-{
-  void ca_ubrep_attach (CAUnboundRepeat *cb);
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_attach(ca->parent);
-  ca->ptr = ca->parent->ptr;
-  return;
-}
-
-static void
-ca_ubrep_func_sync (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_sync(ca->parent);
-  return;
-}
-
-static void
-ca_ubrep_func_detach (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca->ptr = NULL;
-  ca_detach(ca->parent);
-  return;
-}
-
-static void
-ca_ubrep_func_copy_data (void *ap, void *ptr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  if ( ptr != ca->ptr ) {
-    ca_attach(ca->parent);
-    memmove(ptr, ca->parent->ptr, ca_length(ca));
-    ca_detach(ca->parent);
-  }
-}
-
-static void
-ca_ubrep_func_sync_data (void *ap, void *ptr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  if ( ptr != ca->ptr ) {
-    ca_allocate(ca->parent);
-    memmove(ptr, ca->parent->ptr, ca_length(ca));
-    ca_sync(ca->parent);
-    ca_detach(ca->parent);
-  }
-}
-
-static void
-ca_ubrep_func_fill_data (void *ap, void *ptr)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  if ( ptr != ca->ptr ) {
-    ca_allocate(ca->parent);
-    ca_func[CA_OBJ_ARRAY].fill_data(ca->parent, ptr);
-    ca_sync(ca->parent);
-    ca_detach(ca->parent);
-  }
-}
-
-static void
-ca_ubrep_func_create_mask (void *ap)
-{
-  CAUnboundRepeat *ca = (CAUnboundRepeat *) ap;
-  ca_update_mask(ca->parent);
-  if ( ! ca->parent->mask ) {
-    ca_create_mask(ca->parent);
-  }
-  ca->mask =
-    (CArray *) ca_ubrep_new(ca->parent->mask, ca->rep_ndim, ca->rep_dim);
-}
-
-ca_operation_function_t ca_ubrep_func = {
-  -1, /* CA_OBJ_UNBOUND_REPEAT */
-  CA_VIRTUAL_ARRAY,
-  free_ca_ubrep,
-  ca_ubrep_func_clone,
-  ca_ubrep_func_ptr_at_addr,
-  NULL, 
-  ca_ubrep_func_fetch_addr,
-  NULL, 
-  ca_ubrep_func_store_addr,
-  NULL, 
-  ca_ubrep_func_allocate,
-  ca_ubrep_func_attach,
-  ca_ubrep_func_sync,
-  ca_ubrep_func_detach,
-  ca_ubrep_func_copy_data,
-  ca_ubrep_func_sync_data,
-  ca_ubrep_func_fill_data,
-  ca_ubrep_func_create_mask,
-};
-
-/* ------------------------------------------------------------------- */
-
-VALUE
-rb_ca_ubrep_shave (VALUE self, VALUE other)
-{
-  CAUnboundRepeat *ca;
-  CArray *co;
-  int8_t ndim, i;
-  ca_size_t dim[CA_RANK_MAX];
-
-  rb_check_carray_object(self);
-  rb_check_carray_object(other);
-
-  TypedData_Get_Struct(self, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-  TypedData_Get_Struct(other, CArray, &carray_data_type, co);
-
-  if ( ca->elements != co->elements ) {
-    rb_raise(rb_eRuntimeError, "mismatch in # of elements");
-  }
-
-  ndim = 0;
-  for (i=0; i<ca->ndim; i++) {
-    if ( ca->rep_dim[i] ) {
-      dim[ndim] = ca->rep_dim[i];
-      ndim += 1;
-    }
-  }
-
-  return rb_ca_refer_new(other, co->data_type, ndim, dim, co->bytes, 0);
-}
-
-
-VALUE
-rb_ca_ubrep_new (VALUE cary, int32_t rep_ndim, ca_size_t *rep_dim)
-{
-  volatile VALUE obj;
-  CArray *parent;
-  CAUnboundRepeat *ca;
-  rb_check_carray_object(cary);
-  TypedData_Get_Struct(cary, CArray, &carray_data_type, parent);
-  ca = ca_ubrep_new(parent, rep_ndim, rep_dim);
-  obj = ca_wrap_struct(ca);
-  rb_ca_set_parent(obj, cary);
-  rb_ca_data_type_inherit(obj, cary);
-  return obj;
-}
-
-VALUE
-rb_ca_unbound_repeat (int argc, VALUE *argv, VALUE self)
-{
-  CArray *ca;
-  int8_t ndim;
-  ca_size_t dim[CA_RANK_MAX];
-  int32_t rep_ndim;
-  ca_size_t rep_dim[CA_RANK_MAX];
-  ca_size_t elements, count, i;
-
-  TypedData_Get_Struct(self, CArray, &carray_data_type, ca);
-
-  rep_ndim = argc;
-
-  count = 0;
-  ndim  = 0;
-  elements = 1;
-  for (i=0; i<rep_ndim; i++) {
-    if ( TYPE(argv[i]) == T_SYMBOL ) {
-      if ( argv[i] == ID2SYM(rb_intern("*")) ) {
-        rep_dim[i] = 0;
-      }
-      else {
-        rb_raise(rb_eArgError, "unknown symbol (!= ':*') in arguments");
-      }
-    }
-    else {
-      if ( ! NIL_P(argv[i]) ) {
-        rb_raise(rb_eArgError, "invalid argument");
-      }
-      rep_dim[i] = ca->dim[count];
-      dim[ndim]  = ca->dim[count];
-      elements  *= ca->dim[count];
-      count++; ndim++;
-    }
-  }
-
-  if ( elements != ca->elements ) {
-    rb_raise(rb_eArgError, "mismatch in entity elements (%lli for %lli)", elements, ca->elements);
-  }
-
-  if ( ndim != ca->ndim ) {
-    rb_raise(rb_eArgError, "invalid number of nil's (%i for %i)", ndim, ca->ndim);
-  }
-
-  return rb_ca_ubrep_new(self, rep_ndim, rep_dim);
-}
-
-static VALUE
-rb_ca_ubrep_s_allocate (VALUE klass)
-{
-  CAUnboundRepeat *ca;
-  return TypedData_Make_Struct(klass, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-}
-
-static VALUE
-rb_ca_ubrep_initialize_copy (VALUE self, VALUE other)
-{
-  CAUnboundRepeat *ca, *cs;
-
-  TypedData_Get_Struct(self,  CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-  TypedData_Get_Struct(other, CAUnboundRepeat, &caunboundrepeat_data_type, cs);
-
-  ca_ubrep_setup(ca, cs->parent, cs->rep_ndim, cs->rep_dim);
-
-  return self;
-}
-
-/*
-static CARepeat *
-ca_ubrep_bind (CAUnboundRepeat *ca, int32_t new_ndim, ca_size_t *new_dim)
-{
-  ca_size_t rep_spec[CA_RANK_MAX];
-  int i;
-  if ( ca->rep_ndim != new_ndim ) {
-    rb_raise(rb_eArgError, "invalid new_ndim");
-  }
-  for (i=0; i<new_ndim; i++) {
-    if ( ca->rep_dim[i] == 0 ) {
-      rep_spec[i] = new_dim[i];
-    }
-    else {
-      rep_spec[i] = 0;
-    }
-  }
-  return ca_repeat_new((CArray*)ca, new_ndim, rep_spec);
-}
-*/
-
-VALUE
-ca_ubrep_bind2 (VALUE self, int32_t new_ndim, ca_size_t *new_dim)
-{
-  CAUnboundRepeat *ca;
-  ca_size_t rep_spec[CA_RANK_MAX];
-  ca_size_t upr_spec[CA_RANK_MAX];
-  ca_size_t srp_spec[CA_RANK_MAX];
-  int uprep = 0, srp_ndim;
-  int ndim_real;
-  int i;
-
-  TypedData_Get_Struct(self, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-
-  if ( ca->rep_ndim != new_ndim ) {
-    rb_raise(rb_eArgError, "invalid new_ndim (%i <-> %i)",
-                           ca->rep_ndim, new_ndim);
-  }
-
-  srp_ndim = 0;
-  ndim_real = 0;
-  for (i=0; i<new_ndim; i++) {
-    if ( ca->rep_dim[i] == 0 ) {
-      if ( new_dim[i] == 0 ) {
-        uprep = 1;
-      }
-      else {
-        srp_spec[srp_ndim++] = new_dim[i];
-      }
-      rep_spec[i] = new_dim[i];
-      upr_spec[i] = new_dim[i];
-    }
-    else {
-      ndim_real++;
-      rep_spec[i] = 0;
-      srp_spec[srp_ndim++] = 0;
-      upr_spec[i] = ca->rep_dim[i];
-    }
-  }
-  
-  if ( uprep ) {
-    volatile VALUE rep;
-    if ( srp_ndim >= ndim_real ) {
-      rep = rb_ca_repeat_new(rb_ca_parent(self), srp_ndim, srp_spec);
-    }
-    else {
-      rep = rb_ca_parent(self);
-    }
-    return rb_ca_ubrep_new(rep, new_ndim, upr_spec);
-  }
-  else {
-    return rb_ca_repeat_new(rb_ca_parent(self), new_ndim, rep_spec);
-  }
-}
-
-/* @overload bind_with (other)
-
-*/
-
-VALUE
-ca_ubrep_bind_with (VALUE self, VALUE other)
-{
-  CAUnboundRepeat *ca, *cup;
-  CArray *co;
-
-  rb_check_carray_object(other);
-
-  TypedData_Get_Struct(self, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-  TypedData_Get_Struct(other, CArray, &carray_data_type, co);
-
-  if ( co->obj_type == CA_OBJ_UNBOUND_REPEAT ) {
-    TypedData_Get_Struct(other, CAUnboundRepeat, &caunboundrepeat_data_type, cup);
-    return ca_ubrep_bind2(self, cup->rep_ndim, cup->rep_dim);
-  }
-  else if ( ca_is_scalar(co) ) {
-    return self;
-  }
-  else {
-    return ca_ubrep_bind2(self, co->ndim, co->dim);
-  }
-}
-
-/* @overload bind (*index)
-
-*/
-
-static VALUE
-rb_ca_ubrep_bind (int argc, VALUE *argv, VALUE self)
-{
-  CAUnboundRepeat *ca;
-  ca_size_t rep_spec[CA_RANK_MAX];
-  int i;
-
-  TypedData_Get_Struct(self, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-
-  if ( ca->rep_ndim != argc ) {
-    rb_raise(rb_eArgError, "invalid new_ndim");
-  }
-  
-  for (i=0; i<argc; i++) {
-    if ( ca->rep_dim[i] == 0 ) {
-      rep_spec[i] = NUM2SIZE(argv[i]);
-    }
-    else {
-      rep_spec[i] = 0;
-    }
-  }
-
-  return rb_ca_repeat_new(rb_ca_parent(self), argc, rep_spec);
-}
-
-static VALUE
-rb_ca_ubrep_spec (VALUE self)
-{
-  volatile VALUE spec;
-  CAUnboundRepeat *ca;
-  int i;
-
-  TypedData_Get_Struct(self, CAUnboundRepeat, &caunboundrepeat_data_type, ca);
-
-  spec = rb_ary_new2(ca->rep_ndim);
-  for (i=0; i<ca->rep_ndim; i++) {
-    if ( ca->rep_dim[i] ) {
-      rb_ary_store(spec, i, SIZE2NUM(ca->rep_dim[i]));
-    }
-    else {
-      rb_ary_store(spec, i, ID2SYM(rb_intern("*")));
-    }
-  }
-
-  return spec;
-}
-
-void
-Init_ca_obj_unbound_repeat ()
-{
-  /* rb_cCAUnboundRepeat, CA_OBJ_UNBOUND_REPEAT are defined in rb_carray.c */
-
-  rb_define_const(rb_cObject, "CA_OBJ_UNBOUND_REPEAT", INT2NUM(CA_OBJ_UNBOUND_REPEAT));
-
-  rb_define_method(rb_cCArray, "unbound_repeat", rb_ca_unbound_repeat, -1);
-
-  rb_define_alloc_func(rb_cCAUnboundRepeat, rb_ca_ubrep_s_allocate);
-  rb_define_method(rb_cCAUnboundRepeat, "initialize_copy",
-                                      rb_ca_ubrep_initialize_copy, 1);
-
-  rb_define_method(rb_cCAUnboundRepeat, "bind", rb_ca_ubrep_bind, -1);
-  rb_define_method(rb_cCAUnboundRepeat, "bind_with", ca_ubrep_bind_with, 1);
-  rb_define_method(rb_cCAUnboundRepeat, "spec", rb_ca_ubrep_spec, 0);
-
-}
-