numo-narray-alt 0.9.14 → 0.10.1

data/ext/numo/narray/src/t_scomplex.c

@@ -38,8 +38,11 @@ static ID id_to_a;
 VALUE cT;
 extern VALUE cRT;
 
+#include "mh/store.h"
+#include "mh/s_cast.h"
 #include "mh/extract.h"
 #include "mh/aref.h"
+#include "mh/aset.h"
 #include "mh/coerce_cast.h"
 #include "mh/to_a.h"
 #include "mh/fill.h"
@@ -118,8 +121,12 @@ extern VALUE cRT;
 #include "mh/math/atanh.h"
 #include "mh/math/sinc.h"
 
+DEF_NARRAY_CMP_STORE_METHOD_FUNC(scomplex, numo_cSComplex)
+DEF_NARRAY_S_CAST_METHOD_FUNC(scomplex, numo_cSComplex)
 DEF_NARRAY_EXTRACT_METHOD_FUNC(scomplex)
 DEF_NARRAY_AREF_METHOD_FUNC(scomplex)
+DEF_CMP_EXTRACT_DATA_FUNC(scomplex, numo_cSComplex)
+DEF_NARRAY_ASET_METHOD_FUNC(scomplex)
 DEF_NARRAY_COERCE_CAST_METHOD_FUNC(scomplex)
 DEF_NARRAY_TO_A_METHOD_FUNC(scomplex)
 DEF_NARRAY_FILL_METHOD_FUNC(scomplex)
@@ -198,8 +205,6 @@ DEF_NARRAY_FLT_ACOSH_METHOD_FUNC(scomplex, numo_cSComplex)
 DEF_NARRAY_FLT_ATANH_METHOD_FUNC(scomplex, numo_cSComplex)
 DEF_NARRAY_FLT_SINC_METHOD_FUNC(scomplex, numo_cSComplex)
 
-static VALUE scomplex_store(VALUE, VALUE);
-
 static size_t scomplex_memsize(const void* ptr) {
   size_t size = sizeof(narray_data_t);
   const narray_data_t* na = (const narray_data_t*)ptr;
@@ -229,11 +234,9 @@ static void scomplex_free(void* ptr) {
     }
     na->ptr = NULL;
   }
-  if (na->base.size > 0) {
-    if (na->base.shape != NULL && na->base.shape != &(na->base.size)) {
-      xfree(na->base.shape);
-      na->base.shape = NULL;
-    }
+  if (na->base.shape != NULL && na->base.shape != &(na->base.size)) {
+    xfree(na->base.shape);
+    na->base.shape = NULL;
   }
   xfree(na);
 }
@@ -301,1141 +304,6 @@ static VALUE scomplex_allocate(VALUE self) {
   return self;
 }
 
-static VALUE scomplex_new_dim0(dtype x) {
-  VALUE v;
-  dtype* ptr;
-
-  v = nary_new(cT, 0, NULL);
-  ptr = (dtype*)(char*)na_get_pointer_for_write(v);
-  *ptr = x;
-  na_release_lock(v);
-  return v;
-}
-
-static VALUE scomplex_store_numeric(VALUE self, VALUE obj) {
-  dtype x;
-  x = m_num_to_data(obj);
-  obj = scomplex_new_dim0(x);
-  scomplex_store(self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_bit(na_loop_t* const lp) {
-  size_t i;
-  char* p1;
-  size_t p2;
-  ssize_t s1, s2;
-  size_t *idx1, *idx2;
-  BIT_DIGIT *a2, x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_BIT_IDX(lp, 1, a2, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        LOAD_BIT(a2, p2 + *idx2, x);
-        idx2++;
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        LOAD_BIT(a2, p2 + *idx2, x);
-        idx2++;
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        LOAD_BIT(a2, p2, x);
-        p2 += s2;
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        LOAD_BIT(a2, p2, x);
-        p2 += s2;
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_bit(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_bit, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_dcomplex(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  dcomplex x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, dcomplex, x);
-        y = m_from_dcomplex(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, dcomplex, x);
-        y = m_from_dcomplex(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, dcomplex, x);
-        y = m_from_dcomplex(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, dcomplex, x);
-        y = m_from_dcomplex(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_dcomplex(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_dcomplex, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_scomplex(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  scomplex x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, scomplex, x);
-        y = m_from_scomplex(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, scomplex, x);
-        y = m_from_scomplex(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, scomplex, x);
-        y = m_from_scomplex(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, scomplex, x);
-        y = m_from_scomplex(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_scomplex(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_scomplex, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_dfloat(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  double x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, double, x);
-        y = m_from_real(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, double, x);
-        y = m_from_real(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, double, x);
-        y = m_from_real(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, double, x);
-        y = m_from_real(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_dfloat(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_dfloat, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_sfloat(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  float x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, float, x);
-        y = m_from_real(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, float, x);
-        y = m_from_real(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, float, x);
-        y = m_from_real(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, float, x);
-        y = m_from_real(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_sfloat(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_sfloat, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_int64(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  int64_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int64_t, x);
-        y = m_from_int64(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int64_t, x);
-        y = m_from_int64(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int64_t, x);
-        y = m_from_int64(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int64_t, x);
-        y = m_from_int64(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_int64(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_int64, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_int32(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  int32_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int32_t, x);
-        y = m_from_int32(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int32_t, x);
-        y = m_from_int32(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int32_t, x);
-        y = m_from_int32(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int32_t, x);
-        y = m_from_int32(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_int32(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_int32, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_int16(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  int16_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_int16(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_int16, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_int8(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  int8_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_int8(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_int8, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_uint64(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  u_int64_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int64_t, x);
-        y = m_from_uint64(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int64_t, x);
-        y = m_from_uint64(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int64_t, x);
-        y = m_from_uint64(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int64_t, x);
-        y = m_from_uint64(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_uint64(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_uint64, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_uint32(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  u_int32_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int32_t, x);
-        y = m_from_uint32(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int32_t, x);
-        y = m_from_uint32(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int32_t, x);
-        y = m_from_uint32(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int32_t, x);
-        y = m_from_uint32(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_uint32(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_uint32, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_uint16(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  u_int16_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int16_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_uint16(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_uint16, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_uint8(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  u_int8_t x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, u_int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, u_int8_t, x);
-        y = m_from_sint(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_uint8(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_uint8, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_robject(na_loop_t* const lp) {
-  size_t i, s1, s2;
-  char *p1, *p2;
-  size_t *idx1, *idx2;
-  VALUE x;
-  dtype y;
-
-  INIT_COUNTER(lp, i);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  INIT_PTR_IDX(lp, 1, p2, s2, idx2);
-  if (idx2) {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, VALUE, x);
-        y = m_num_to_data(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_INDEX(p2, idx2, VALUE, x);
-        y = m_num_to_data(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  } else {
-    if (idx1) {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, VALUE, x);
-        y = m_num_to_data(x);
-        SET_DATA_INDEX(p1, idx1, dtype, y);
-      }
-    } else {
-      for (; i--;) {
-        GET_DATA_STRIDE(p2, s2, VALUE, x);
-        y = m_num_to_data(x);
-        SET_DATA_STRIDE(p1, s1, dtype, y);
-      }
-    }
-  }
-}
-
-static VALUE scomplex_store_robject(VALUE self, VALUE obj) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { Qnil, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_robject, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop(&ndf, 2, self, obj);
-  return self;
-}
-
-static void iter_scomplex_store_array(na_loop_t* const lp) {
-  size_t i, n;
-  size_t i1, n1;
-  VALUE v1, *ptr;
-  char* p1;
-  size_t s1, *idx1;
-  VALUE x;
-  double y;
-  dtype z;
-  size_t len, c;
-  double beg, step;
-
-  INIT_COUNTER(lp, n);
-  INIT_PTR_IDX(lp, 0, p1, s1, idx1);
-  v1 = lp->args[1].value;
-  i = 0;
-
-  if (lp->args[1].ptr) {
-    if (v1 == Qtrue) {
-      iter_scomplex_store_scomplex(lp);
-      i = lp->args[1].shape[0];
-      if (idx1) {
-        idx1 += i;
-      } else {
-        p1 += s1 * i;
-      }
-    }
-    goto loop_end;
-  }
-
-  ptr = &v1;
-
-  switch (TYPE(v1)) {
-  case T_ARRAY:
-    n1 = RARRAY_LEN(v1);
-    ptr = RARRAY_PTR(v1);
-    break;
-  case T_NIL:
-    n1 = 0;
-    break;
-  default:
-    n1 = 1;
-  }
-
-  if (idx1) {
-    for (i = i1 = 0; i1 < n1 && i < n; i++, i1++) {
-      x = ptr[i1];
-      if (rb_obj_is_kind_of(x, rb_cRange)
-#ifdef HAVE_RB_ARITHMETIC_SEQUENCE_EXTRACT
-          || rb_obj_is_kind_of(x, rb_cArithSeq)
-#else
-          || rb_obj_is_kind_of(x, rb_cEnumerator)
-#endif
-      ) {
-        nary_step_sequence(x, &len, &beg, &step);
-        for (c = 0; c < len && i < n; c++, i++) {
-          y = beg + step * c;
-          z = m_from_double(y);
-          SET_DATA_INDEX(p1, idx1, dtype, z);
-        }
-      } else if (TYPE(x) != T_ARRAY) {
-        z = m_num_to_data(x);
-        SET_DATA_INDEX(p1, idx1, dtype, z);
-      }
-    }
-  } else {
-    for (i = i1 = 0; i1 < n1 && i < n; i++, i1++) {
-      x = ptr[i1];
-      if (rb_obj_is_kind_of(x, rb_cRange)
-#ifdef HAVE_RB_ARITHMETIC_SEQUENCE_EXTRACT
-          || rb_obj_is_kind_of(x, rb_cArithSeq)
-#else
-          || rb_obj_is_kind_of(x, rb_cEnumerator)
-#endif
-      ) {
-        nary_step_sequence(x, &len, &beg, &step);
-        for (c = 0; c < len && i < n; c++, i++) {
-          y = beg + step * c;
-          z = m_from_double(y);
-          SET_DATA_STRIDE(p1, s1, dtype, z);
-        }
-      } else if (TYPE(x) != T_ARRAY) {
-        z = m_num_to_data(x);
-        SET_DATA_STRIDE(p1, s1, dtype, z);
-      }
-    }
-  }
-
-loop_end:
-  z = m_zero;
-  if (idx1) {
-    for (; i < n; i++) {
-      SET_DATA_INDEX(p1, idx1, dtype, z);
-    }
-  } else {
-    for (; i < n; i++) {
-      SET_DATA_STRIDE(p1, s1, dtype, z);
-    }
-  }
-}
-
-static VALUE scomplex_store_array(VALUE self, VALUE rary) {
-  ndfunc_arg_in_t ain[2] = { { OVERWRITE, 0 }, { rb_cArray, 0 } };
-  ndfunc_t ndf = { iter_scomplex_store_array, FULL_LOOP, 2, 0, ain, 0 };
-
-  na_ndloop_store_rarray(&ndf, self, rary);
-  return self;
-}
-
-/*
-  Store elements to Numo::SComplex from other.
-  @overload store(other)
-    @param [Object] other
-    @return [Numo::SComplex] self
-*/
-static VALUE scomplex_store(VALUE self, VALUE obj) {
-  VALUE r, klass;
-
-  klass = rb_obj_class(obj);
-
-  if (klass == numo_cSComplex) {
-    scomplex_store_scomplex(self, obj);
-    return self;
-  }
-
-  if (IS_INTEGER_CLASS(klass) || klass == rb_cFloat || klass == rb_cComplex) {
-    scomplex_store_numeric(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cBit) {
-    scomplex_store_bit(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cDComplex) {
-    scomplex_store_dcomplex(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cDFloat) {
-    scomplex_store_dfloat(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cSFloat) {
-    scomplex_store_sfloat(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cInt64) {
-    scomplex_store_int64(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cInt32) {
-    scomplex_store_int32(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cInt16) {
-    scomplex_store_int16(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cInt8) {
-    scomplex_store_int8(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cUInt64) {
-    scomplex_store_uint64(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cUInt32) {
-    scomplex_store_uint32(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cUInt16) {
-    scomplex_store_uint16(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cUInt8) {
-    scomplex_store_uint8(self, obj);
-    return self;
-  }
-
-  if (klass == numo_cRObject) {
-    scomplex_store_robject(self, obj);
-    return self;
-  }
-
-  if (klass == rb_cArray) {
-    scomplex_store_array(self, obj);
-    return self;
-  }
-
-  if (IsNArray(obj)) {
-    r = rb_funcall(obj, rb_intern("coerce_cast"), 1, cT);
-    if (rb_obj_class(r) == cT) {
-      scomplex_store(self, r);
-      return self;
-    }
-  }
-
-  rb_raise(
-    nary_eCastError, "unknown conversion from %s to %s", rb_class2name(rb_obj_class(obj)),
-    rb_class2name(rb_obj_class(self))
-  );
-
-  return self;
-}
-
-/*
-  Convert a data value of obj (with a single element) to dtype.
-*/
-static dtype scomplex_extract_data(VALUE obj) {
-  narray_t* na;
-  dtype x;
-  char* ptr;
-  size_t pos;
-  VALUE r, klass;
-
-  if (IsNArray(obj)) {
-    GetNArray(obj, na);
-    if (na->size != 1) {
-      rb_raise(nary_eShapeError, "narray size should be 1");
-    }
-    klass = rb_obj_class(obj);
-    ptr = na_get_pointer_for_read(obj);
-    pos = na_get_offset(obj);
-
-    if (klass == numo_cSComplex) {
-      {
-        scomplex* p = (scomplex*)(ptr + pos);
-        x = c_new(REAL(*p), IMAG(*p));
-      };
-      return x;
-    }
-
-    if (klass == numo_cBit) {
-      {
-        BIT_DIGIT b;
-        LOAD_BIT(ptr, pos, b);
-        x = m_from_sint(b);
-      };
-      return x;
-    }
-
-    if (klass == numo_cDComplex) {
-      {
-        dcomplex* p = (dcomplex*)(ptr + pos);
-        x = c_new(REAL(*p), IMAG(*p));
-      };
-      return x;
-    }
-
-    if (klass == numo_cDFloat) {
-      x = m_from_real(*(double*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cSFloat) {
-      x = m_from_real(*(float*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cInt64) {
-      x = m_from_int64(*(int64_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cInt32) {
-      x = m_from_int32(*(int32_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cInt16) {
-      x = m_from_sint(*(int16_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cInt8) {
-      x = m_from_sint(*(int8_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cUInt64) {
-      x = m_from_uint64(*(u_int64_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cUInt32) {
-      x = m_from_uint32(*(u_int32_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cUInt16) {
-      x = m_from_sint(*(u_int16_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cUInt8) {
-      x = m_from_sint(*(u_int8_t*)(ptr + pos));
-      return x;
-    }
-
-    if (klass == numo_cRObject) {
-      x = m_num_to_data(*(VALUE*)(ptr + pos));
-      return x;
-    }
-
-    // coerce
-    r = rb_funcall(obj, rb_intern("coerce_cast"), 1, cT);
-    if (rb_obj_class(r) == cT) {
-      return scomplex_extract_data(r);
-    }
-
-    rb_raise(
-      nary_eCastError, "unknown conversion from %s to %s", rb_class2name(rb_obj_class(obj)),
-      rb_class2name(cT)
-    );
-  }
-  if (TYPE(obj) == T_ARRAY) {
-    if (RARRAY_LEN(obj) != 1) {
-      rb_raise(nary_eShapeError, "array size should be 1");
-    }
-    return m_num_to_data(RARRAY_AREF(obj, 0));
-  }
-  return m_num_to_data(obj);
-}
-
-static VALUE scomplex_cast_array(VALUE rary) {
-  VALUE nary;
-  narray_t* na;
-
-  nary = na_s_new_like(cT, rary);
-  GetNArray(nary, na);
-  if (na->size > 0) {
-    scomplex_store_array(nary, rary);
-  }
-  return nary;
-}
-
-/*
-  Cast object to Numo::SComplex.
-  @overload [](elements)
-  @overload cast(array)
-    @param [Numeric,Array] elements
-    @param [Array] array
-    @return [Numo::SComplex]
-*/
-static VALUE scomplex_s_cast(VALUE type, VALUE obj) {
-  VALUE v;
-  narray_t* na;
-  dtype x;
-
-  if (rb_obj_class(obj) == cT) {
-    return obj;
-  }
-  if (RTEST(rb_obj_is_kind_of(obj, rb_cNumeric))) {
-    x = m_num_to_data(obj);
-    return scomplex_new_dim0(x);
-  }
-  if (RTEST(rb_obj_is_kind_of(obj, rb_cArray))) {
-    return scomplex_cast_array(obj);
-  }
-  if (IsNArray(obj)) {
-    GetNArray(obj, na);
-    v = nary_new(cT, NA_NDIM(na), NA_SHAPE(na));
-    if (NA_SIZE(na) > 0) {
-      scomplex_store(v, obj);
-    }
-    return v;
-  }
-  if (rb_respond_to(obj, id_to_a)) {
-    obj = rb_funcall(obj, id_to_a, 0);
-    if (TYPE(obj) != T_ARRAY) {
-      rb_raise(rb_eTypeError, "`to_a' did not return Array");
-    }
-    return scomplex_cast_array(obj);
-  }
-
-  rb_raise(nary_eCastError, "cannot cast to %s", rb_class2name(type));
-  return Qnil;
-}
-
-/*
-  Multi-dimensional element assignment.
-  @overload []=(dim0,...,dimL,val)
-    @param [Numeric,Range,Array,Numo::Int32,Numo::Int64,Numo::Bit,TrueClass,FalseClass,Symbol]
-    dim0,...,dimL  multi-dimensional indices.
-    @param [Numeric,Numo::NArray,Array] val  Value(s) to be set to self.
-    @return [Numeric,Numo::NArray,Array] returns `val` (last argument).
-  @see Numo::NArray#[]=
-  @see #[]
-*/
-static VALUE scomplex_aset(int argc, VALUE* argv, VALUE self) {
-  int nd;
-  size_t pos;
-  char* ptr;
-  VALUE a;
-  dtype x;
-
-  argc--;
-  if (argc == 0) {
-    scomplex_store(self, argv[argc]);
-  } else {
-    nd = na_get_result_dimension(self, argc, argv, sizeof(dtype), &pos);
-    if (nd) {
-      a = na_aref_main(argc, argv, self, 0, nd);
-      scomplex_store(a, argv[argc]);
-    } else {
-      x = scomplex_extract_data(argv[argc]);
-      ptr = na_get_pointer_for_read_write(self) + pos;
-      *(dtype*)ptr = x;
-    }
-  }
-  return argv[argc];
-}
-
 VALUE mTM;
 
 void Init_numo_scomplex(void) {
@@ -1469,12 +337,7 @@ void Init_numo_scomplex(void) {
   rb_define_const(cT, "UPCAST", hCast);
   rb_hash_aset(hCast, rb_cArray, cT);
 
-#ifdef RUBY_INTEGER_UNIFICATION
   rb_hash_aset(hCast, rb_cInteger, cT);
-#else
-  rb_hash_aset(hCast, rb_cFixnum, cT);
-  rb_hash_aset(hCast, rb_cBignum, cT);
-#endif
   rb_hash_aset(hCast, rb_cFloat, cT);
   rb_hash_aset(hCast, rb_cComplex, cT);
   rb_hash_aset(hCast, numo_cRObject, numo_cRObject);
@@ -1514,9 +377,21 @@ void Init_numo_scomplex(void) {
    *   otherwise returns self.
    */
   rb_define_method(cT, "extract", scomplex_extract, 0);
-
+  /**
+   * Store elements to Numo::SComplex from other.
+   * @overload store(other)
+   *   @param [Object] other
+   *   @return [Numo::SComplex] self
+   */
   rb_define_method(cT, "store", scomplex_store, 1);
-
+  /**
+   * Cast object to Numo::SComplex.
+   * @overload [](elements)
+   * @overload cast(array)
+   *   @param [Numeric,Array] elements
+   *   @param [Array] array
+   *   @return [Numo::SComplex]
+   */
   rb_define_singleton_method(cT, "cast", scomplex_s_cast, 1);
   /**
    * Multi-dimensional element reference.
@@ -1528,6 +403,16 @@ void Init_numo_scomplex(void) {
    *   @see #[]=
    */
   rb_define_method(cT, "[]", scomplex_aref, -1);
+  /**
+   * Multi-dimensional element assignment.
+   * @overload []=(dim0,...,dimL,val)
+   *   @param [Numeric,Range,Array,Numo::Int32,Numo::Int64,Numo::Bit,TrueClass,FalseClass,
+   *     Symbol] dim0,...,dimL  multi-dimensional indices.
+   *   @param [Numeric,Numo::NArray,Array] val  Value(s) to be set to self.
+   *   @return [Numeric,Numo::NArray,Array] returns `val` (last argument).
+   * @see Numo::NArray#[]=
+   * @see #[]
+   */
   rb_define_method(cT, "[]=", scomplex_aset, -1);
   /**
    * return NArray with cast to the type of self.