numo-narray 0.9.0.1

data/ext/numo/narray/numo/compat.h

@@ -0,0 +1,23 @@
+#ifndef COMPAT_H
+#define COMPAT_H
+
+#if !defined RSTRING_LEN
+#define RSTRING_LEN(a) RSTRING(a)->len
+#endif
+#if !defined RSTRING_PTR
+#define RSTRING_PTR(a) RSTRING(a)->ptr
+#endif
+#if !defined RARRAY_LEN
+#define RARRAY_LEN(a) RARRAY(a)->len
+#endif
+#if !defined RARRAY_PTR
+#define RARRAY_PTR(a) RARRAY(a)->ptr
+#endif
+#if !defined RARRAY_AREF
+#define RARRAY_AREF(a,i) RARRAY_PTR(a)[i]
+#endif
+#if !defined RARRAY_ASET
+#define RARRAY_ASET(a,i,v) (RARRAY_PTR(a)[i] = v)
+#endif
+
+#endif /* ifndef COMPAT_H */

data/ext/numo/narray/numo/intern.h

@@ -0,0 +1,112 @@
+/*
+  intern.h
+  Numerical Array Extension for Ruby
+    (C) Copyright 1999-2011 by Masahiro TANAKA
+
+  This program is free software.
+  You can distribute/modify this program
+  under the same terms as Ruby itself.
+  NO WARRANTY.
+*/
+#ifndef INTERN_H
+#define INTERN_H
+
+VALUE rb_narray_new(VALUE elem, int ndim, size_t *shape);
+VALUE rb_narray_view_new(VALUE elem, int ndim, size_t *shape);
+VALUE rb_narray_debug_info(VALUE);
+
+VALUE na_original_data(VALUE self);
+VALUE na_make_view(VALUE self);
+VALUE na_make_view_struct(VALUE self, VALUE dtype, VALUE offset);
+void na_array_to_internal_shape(VALUE self, VALUE ary, size_t *shape);
+
+VALUE na_reduce_dimension(int argc, VALUE *argv, int naryc, VALUE *naryv);
+
+VALUE na_check_ladder(VALUE self, int start_dim);
+void na_setup_shape(narray_t *na, int ndim, size_t *shape);
+
+VALUE na_transpose_map(VALUE self, int *map);
+VALUE na_flatten_dim(VALUE self, int sd);
+VALUE na_flatten_by_reduce(int argc, VALUE *argv, VALUE self);
+
+VALUE na_sort_main(int argc, VALUE *argv, volatile VALUE self, na_iter_func_t iter_func);
+VALUE na_median_main(int argc, VALUE *argv, volatile VALUE self, na_iter_func_t iter_func);
+
+VALUE na_sort_index_main(int argc, VALUE *argv, VALUE self, void (*func_qsort)());
+
+char *na_get_pointer(VALUE);
+char *na_get_pointer_for_write(VALUE);
+char *na_get_pointer_for_read(VALUE);
+char *na_get_pointer_at_origin_for_read(VALUE);
+
+size_t na_get_offset(VALUE self);
+
+VALUE na_s_allocate(VALUE klass);
+VALUE na_s_allocate_view(VALUE klass);
+
+void na_alloc_shape(narray_t *na, int ndim);
+void na_alloc_index(narray_t *na);
+
+void na_copy_flags(VALUE src, VALUE dst);
+
+VALUE na_flatten(VALUE);
+VALUE na_dup(VALUE);
+VALUE na_copy(VALUE);
+VALUE na_upcast(VALUE type1, VALUE type2);
+
+stridx_t *na_get_stride(VALUE v);
+
+void na_release_lock(VALUE);
+
+void na_index_arg_to_internal_order(int argc, VALUE *argv, VALUE self);
+
+//ndfunc_t *ndfunc_alloc(na_iter_func_t func, int has_loop, int narg, int nres, ...);
+//ndfunc_t *ndfunc_alloc2(na_iter_func_t func, unsigned int flag, int narg, int nres, VALUE *etypes);
+//void ndfunc_free(ndfunc_t* nf);
+
+VALUE na_ndloop(ndfunc_t *nf, int argc, ...);
+VALUE na_ndloop2(ndfunc_t *nf, VALUE args);
+VALUE na_ndloop3(ndfunc_t *nf, void *ptr, int argc, ...);
+VALUE na_ndloop4(ndfunc_t *nf, void *ptr, VALUE args);
+
+VALUE na_ndloop_cast_narray_to_rarray(ndfunc_t *nf, VALUE nary, VALUE fmt);
+VALUE na_ndloop_cast_rarray_to_narray(ndfunc_t *nf, VALUE rary, VALUE nary);
+VALUE na_ndloop_cast_rarray_to_narray2(ndfunc_t *nf, VALUE rary, VALUE nary, VALUE opt);
+
+VALUE na_ndloop_inspect(VALUE nary, na_text_func_t func, VALUE opt);
+
+VALUE na_ndloop_with_index(ndfunc_t *nf, int argc, ...);
+
+
+VALUE na_info_str(VALUE);
+
+size_t na_get_elmsz(VALUE nary);
+size_t na_dtype_elmsz(VALUE klass);
+
+bool na_test_reduce(VALUE reduce, int dim);
+
+//size_t *na_mdarray_investigate(VALUE ary, int *ndim, VALUE *type);
+VALUE na_ary_composition(VALUE ary);
+
+//void na_copy_bytes(na_loop_t *const itr);
+
+VALUE na_debug_set(VALUE mod, VALUE flag);
+
+void nary_step_array_index(VALUE self, size_t ary_size, size_t *plen, ssize_t *pbeg, ssize_t *pstep);
+
+VALUE na_store(VALUE self, VALUE src);
+
+void nary_step_sequence(VALUE self, size_t *plen, double *pbeg, double *pstep);
+
+VALUE nary_init_accum_aref0(VALUE self, VALUE reduce);
+
+VALUE nary_mathcast(int argc, VALUE *argv);
+
+
+ssize_t
+na_get_scalar_position(VALUE self, int argc, VALUE *argv, ssize_t stride);
+VALUE
+na_aref_main(int nidx, VALUE *idx, VALUE self, int keep_dim);
+
+
+#endif /* ifndef INTERN_H */

data/ext/numo/narray/numo/narray.h

@@ -0,0 +1,411 @@
+/*
+  narray.h
+  Numerical Array Extension for Ruby
+    (C) Copyright 1999-2003 by Masahiro TANAKA
+
+  This program is free software.
+  You can distribute/modify this program
+  under the same terms as Ruby itself.
+  NO WARRANTY.
+*/
+#ifndef NARRAY_H
+#define NARRAY_H
+
+#define NARRAY_VERSION "0.9.0.1"
+#define NARRAY_VERSION_CODE 901
+
+#include <math.h>
+#include "numo/compat.h"
+#include "numo/template.h"
+
+#include "extconf.h"
+
+#ifdef HAVE_STDBOOL_H
+# include <stdbool.h>
+#endif
+
+#ifdef HAVE_STDINT_H
+# include <stdint.h>
+#endif
+
+#ifdef HAVE_SYS_TYPES_H
+# include <sys/types.h>
+#endif
+
+#ifndef HAVE_U_INT8_T
+# ifdef HAVE_UINT8_T
+    typedef uint8_t u_int8_t;
+# endif
+#endif
+
+#ifndef HAVE_U_INT16_T
+# ifdef HAVE_UINT16_T
+    typedef uint16_t u_int16_t;
+# endif
+#endif
+
+#ifndef HAVE_U_INT32_T
+# ifdef HAVE_UINT32_T
+    typedef uint32_t u_int32_t;
+# endif
+#endif
+
+#ifndef HAVE_U_INT64_T
+# ifdef HAVE_UINT64_T
+    typedef uint64_t u_int64_t;
+# endif
+#endif
+
+#if   SIZEOF_VOIDP==SIZEOF_LONG
+# define NUM2SIZE(x) NUM2ULONG(x)
+# define NUM2SSIZE(x) NUM2LONG(x)
+# define SIZE2NUM(x) ULONG2NUM(x)
+# define SSIZE2NUM(x) LONG2NUM(x)
+# define SZF "l"
+#elif SIZEOF_VOIDP==SIZEOF_LONG_LONG
+# define NUM2SIZE(x) NUM2ULL(x)
+# define NUM2SSIZE(x) NUM2LL(x)
+# define SIZE2NUM(x) ULL2NUM(x)
+# define SSIZE2NUM(x) LL2NUM(x)
+# define SZF "ll"
+#endif
+
+#if   SIZEOF_LONG==8
+# define NUM2INT64(x) NUM2LONG(x)
+# define INT642NUM(x) LONG2NUM(x)
+# define NUM2UINT64(x) NUM2ULONG(x)
+# define UINT642NUM(x) ULONG2NUM(x)
+# ifndef PRId64
+#  define PRId64 "ld"
+# endif
+# ifndef PRIu64
+#  define PRIu64 "lu"
+# endif
+#elif SIZEOF_LONG_LONG==8
+# define NUM2INT64(x) NUM2LL(x)
+# define INT642NUM(x) LL2NUM(x)
+# define NUM2UINT64(x) NUM2ULL(x)
+# define UINT642NUM(x) ULL2NUM(x)
+# ifndef PRId64
+#  define PRId64 "lld"
+# endif
+# ifndef PRIu64
+#  define PRIu64 "llu"
+# endif
+#endif
+
+#if   SIZEOF_LONG==4
+# define NUM2INT32(x) NUM2LONG(x)
+# define INT322NUM(x) LONG2NUM(x)
+# define NUM2UINT32(x) NUM2ULONG(x)
+# define UINT322NUM(x) ULONG2NUM(x)
+# ifndef PRId32
+#  define PRId32 "ld"
+# endif
+# ifndef PRIu32
+#  define PRIu32 "lu"
+# endif
+#elif SIZEOF_INT==4
+# define NUM2INT32(x) NUM2INT(x)
+# define INT322NUM(x) UINT2NUM(x)
+# define NUM2UINT32(x) NUM2UINT(x)
+# define UINT322NUM(x) UINT2NUM(x)
+# ifndef PRId32
+#  define PRId32 "d"
+# endif
+# ifndef PRIu32
+#  define PRIu32 "u"
+# endif
+#endif
+
+#ifndef HAVE_TYPE_BOOL
+  typedef int bool;
+#endif
+#ifndef FALSE                   /* in case these macros already exist */
+# define FALSE   0              /* values of bool */
+#endif
+#ifndef TRUE
+# define TRUE    1
+#endif
+
+typedef struct { float dat[2]; }  scomplex;
+typedef struct { double dat[2]; } dcomplex;
+typedef int fortran_integer;
+
+#define REAL(x) ((x).dat[0])
+#define IMAG(x) ((x).dat[1])
+
+extern int na_debug_flag;
+
+#ifndef NARRAY_C
+extern VALUE numo_cNArray;
+extern VALUE rb_mNumo;
+extern VALUE nary_eCastError;
+extern VALUE nary_eShapeError;
+extern VALUE nary_eOperationError;
+extern VALUE nary_eDimensionError;
+
+//EXTERN const int na_sizeof[NA_NTYPES+1];
+#endif
+
+#define cNArray numo_cNArray
+#define mNumo rb_mNumo
+#define na_upcast(x,y) numo_na_upcast(x,y)
+
+/* global variables within this module */
+extern VALUE numo_cBit;
+extern VALUE numo_cDFloat;
+extern VALUE numo_cSFloat;
+extern VALUE numo_cDComplex;
+extern VALUE numo_cSComplex;
+extern VALUE numo_cInt64;
+extern VALUE numo_cInt32;
+extern VALUE numo_cInt16;
+extern VALUE numo_cInt8;
+extern VALUE numo_cUInt64;
+extern VALUE numo_cUInt32;
+extern VALUE numo_cUInt16;
+extern VALUE numo_cUInt8;
+extern VALUE numo_cRObject;
+extern VALUE na_cStep;
+#ifndef HAVE_RB_CCOMPLEX
+extern VALUE rb_cComplex;
+#endif
+
+extern VALUE sym_reduce;
+extern VALUE sym_option;
+extern VALUE sym_loop_opt;
+extern VALUE sym_init;
+
+#define NARRAY_DATA_T     0x1
+#define NARRAY_VIEW_T     0x2
+#define NARRAY_FILEMAP_T  0x3
+
+typedef struct RNArray {
+    unsigned char ndim;     // # of dimensions
+    unsigned char type;
+    unsigned char flag[2];  // flags
+    size_t   size;          // # of total elements
+    size_t  *shape;         // # of elements for each dimension
+    VALUE    reduce;
+} narray_t;
+
+
+typedef struct RNArrayData {
+    narray_t base;
+    char    *ptr;
+} narray_data_t;
+
+
+typedef union {
+    ssize_t stride;
+    size_t *index;
+} stridx_t;
+
+typedef struct RNArrayView {
+    narray_t base;
+    VALUE    data;       // data object
+    size_t   offset;     // offset of start point from data pointer
+                         // :in units of elm.unit_bits
+                         // address_unit  pointer_unit access_unit data_unit
+                         // elm.step_unit = elm.bit_size / elm.access_unit
+                         // elm.step_unit = elm.size_bits / elm.unit_bits
+    stridx_t *stridx;    // stride or indices of data pointer for each dimension
+} narray_view_t;
+
+
+typedef struct RNArrayFileMap {
+    narray_t base;
+    char    *ptr;
+#ifdef WIN32
+    HANDLE hFile;
+    HANDLE hMap;
+#else // POSIX mmap
+    int prot;
+    int flag;
+#endif
+} narray_filemap_t;
+
+
+typedef struct {
+    int     ndim;
+    size_t *shape;
+    VALUE   dtype;
+} na_compose_t;
+
+
+static inline narray_t *
+na_get_narray_t(VALUE obj)
+{
+    narray_t *na;
+
+    Check_Type(obj, T_DATA);
+    na = (narray_t*)DATA_PTR(obj);
+    return na;
+}
+
+static inline narray_t *
+_na_get_narray_t(VALUE obj, unsigned char na_type)
+{
+    narray_t *na;
+
+    Check_Type(obj, T_DATA);
+    na = (narray_t*)DATA_PTR(obj);
+    if (na->type != na_type) {
+	rb_bug("unknown type 0x%x (0x%x given)", na_type, na->type);
+    }
+    return na;
+}
+
+#define na_get_narray_data_t(obj) (narray_data_t*)_na_get_narray_t(obj,NARRAY_DATA_T)
+#define na_get_narray_view_t(obj) (narray_view_t*)_na_get_narray_t(obj,NARRAY_VIEW_T)
+#define na_get_narray_filemap_t(obj) (narray_filemap_t*)_na_get_narray_t(obj,NARRAY_FILEMAP_T)
+
+#define GetNArray(obj,var)  Data_Get_Struct(obj, narray_t, var)
+#define GetNArrayView(obj,var)  Data_Get_Struct(obj, narray_view_t, var)
+#define GetNArrayData(obj,var)  Data_Get_Struct(obj, narray_data_t, var)
+
+#define SDX_IS_STRIDE(x) ((x).stride&0x1)
+#define SDX_IS_INDEX(x)  (!SDX_IS_STRIDE(x))
+#define SDX_GET_STRIDE(x) ((x).stride>>1)
+#define SDX_GET_INDEX(x)  ((x).index)
+
+#define SDX_SET_STRIDE(x,s) ((x).stride=((s)<<1)|0x1)
+#define SDX_SET_INDEX(x,idx) ((x).index=idx)
+
+#define RNARRAY(val)		((narray_t*)DATA_PTR(val))
+#define RNARRAY_DATA(val)	((narray_data_t*)DATA_PTR(val))
+#define RNARRAY_VIEW(val)	((narray_view_t*)DATA_PTR(val))
+#define RNARRAY_FILEMAP(val)	((narray_filemap_t*)DATA_PTR(val))
+
+#define RNARRAY_NDIM(val)	(RNARRAY(val)->ndim)
+#define RNARRAY_TYPE(val)	(RNARRAY(val)->type)
+#define RNARRAY_FLAG(val)	(RNARRAY(val)->flag)
+#define RNARRAY_SIZE(val)	(RNARRAY(val)->size)
+#define RNARRAY_SHAPE(val)	(RNARRAY(val)->shape)
+#define RNARRAY_REDUCE(val)	(RNARRAY(val)->reduce)
+
+#define RNARRAY_DATA_PTR(val)	 (RNARRAY_DATA(val)->ptr)
+#define RNARRAY_VIEW_DATA(val)	 (RNARRAY_VIEW(val)->data)
+#define RNARRAY_VIEW_OFFSET(val) (RNARRAY_VIEW(val)->offset)
+#define RNARRAY_VIEW_STRIDX(val) (RNARRAY_VIEW(val)->stridx)
+
+#define NA_NDIM(na)	(((narray_t*)na)->ndim)
+#define NA_TYPE(na)	(((narray_t*)na)->type)
+#define NA_FLAG(na)	(((narray_t*)na)->flag)
+#define NA_FLAG0(na)	(((narray_t*)na)->flag[0])
+#define NA_FLAG1(na)	(((narray_t*)na)->flag[1])
+#define NA_SIZE(na)	(((narray_t*)na)->size)
+#define NA_SHAPE(na)	(((narray_t*)na)->shape)
+#define NA_REDUCE(na)	(((narray_t*)na)->reduce)
+
+#define NA_DATA_PTR(na)         (((narray_data_t*)na)->ptr)
+#define NA_VIEW_DATA(na)	(((narray_view_t*)na)->data)
+#define NA_VIEW_OFFSET(na)	(((narray_view_t*)na)->offset)
+#define NA_VIEW_STRIDX(na)	(((narray_view_t*)na)->stridx)
+
+#define NA_FILEMAP_PTR(na)	(((narray_filemap_t*)na)->ptr)
+
+
+#define NA_FL0_TEST(x,f) (NA_FLAG0(x)&(f))
+#define NA_FL1_TEST(x,f) (NA_FLAG1(x)&(f))
+
+#define NA_FL0_SET(x,f) do {NA_FLAG0(x) |= (f);} while(0)
+#define NA_FL1_SET(x,f) do {NA_FLAG1(x) |= (f);} while(0)
+
+#define NA_FL0_UNSET(x,f) do {NA_FLAG0(x) &= ~(f);} while(0)
+#define NA_FL1_UNSET(x,f) do {NA_FLAG1(x) &= ~(f);} while(0)
+
+#define NA_FL0_REVERSE(x,f) do {NA_FLAG0(x) ^= (f);} while(0)
+#define NA_FL1_REVERSE(x,f) do {NA_FLAG1(x) ^= (f);} while(0)
+
+#define NA_TEST_LOCK(x)  NA_FL0_TEST(x,NA_FL_LOCK)
+#define NA_SET_LOCK(x)   NA_FL0_SET(x,NA_FL_LOCK)
+#define NA_UNSET_LOCK(x) NA_FL0_UNSET(x,NA_FL_LOCK)
+
+
+/* FLAGS
+   - row-major / column-major
+   - Overwrite or not
+   - byteswapp
+   - Extensible?
+   - matrix or not
+*/
+
+#define NA_FL_LOCK         (0x1<<0)
+#define NA_FL_COLUMN_MAJOR (0x1<<1)
+#define NA_FL_BYTE_SWAPPED (0x1<<2)
+#define NA_FL_INPLACE      (0x1<<3)
+
+#define TEST_COLUMN_MAJOR(x)   NA_FL0_TEST(x,NA_FL_COLUMN_MAJOR)
+#define TEST_ROW_MAJOR(x)    (!NA_FL0_TEST(x,NA_FL_COLUMN_MAJOR))
+#define TEST_BYTE_SWAPPED(x)   NA_FL0_TEST(x,NA_FL_BYTE_SWAPPED)
+#define TEST_HOST_ORDER(x)   (!TEST_BYTE_SWAPPED(x))
+
+#define TEST_INPLACE(x)        NA_FL0_TEST(x,NA_FL_INPLACE)
+#define SET_INPLACE(x)         NA_FL0_SET(x,NA_FL_INPLACE)
+#define UNSET_INPLACE(x)       NA_FL0_UNSET(x,NA_FL_INPLACE)
+
+#define REVERSE_BYTE_SWAPPED(x)	NA_FL0_REVERSE((x),NA_FL_BYTE_SWAPPED)
+
+#ifdef DYNAMIC_ENDIAN
+#else
+#ifdef WORDS_BIGENDIAN
+#else // LITTLE_ENDIAN
+
+#define TEST_NETWORK_ORDER(x) TEST_BYTE_SWAPPED(x)
+#define TEST_VACS_ORDER(x)    TEST_HOST_ORDER(x)
+#endif
+#endif
+
+
+#define IsNArray(obj) (rb_obj_is_kind_of(obj,cNArray)==Qtrue)
+
+#define DEBUG_PRINT(v) puts(StringValueCStr(rb_funcall(v,rb_intern("inspect"),0)))
+
+
+
+#define NA_IsNArray(obj) \
+  (rb_obj_is_kind_of(obj,cNArray)==Qtrue)
+#define NA_IsArray(obj) \
+  (TYPE(obj)==T_ARRAY || rb_obj_is_kind_of(obj,cNArray)==Qtrue)
+
+#define NUM2REAL(v)  NUM2DBL( rb_funcall((v),na_id_real,0) )
+#define NUM2IMAG(v)  NUM2DBL( rb_funcall((v),na_id_imag,0) )
+
+#define NA_MAX_DIMENSION (int)(sizeof(VALUE)*8-2)
+
+/* Function Prototypes */
+
+
+typedef struct {
+    double beg;
+    double step;
+    double count;
+} seq_opt_t;
+
+typedef struct {
+    u_int64_t max;
+    int64_t sign;
+    int shift;
+} rand_opt_t;
+
+
+typedef unsigned int BIT_DIGIT;
+//#define BYTE_BIT_DIGIT sizeof(BIT_DIGIT)
+#define NB     (sizeof(BIT_DIGIT)*8)
+#define BALL   (~(BIT_DIGIT)0)
+#define SLB(n) (((n)==NB)?~(BIT_DIGIT)0:(~(~(BIT_DIGIT)0<<(n))))
+
+//#include "template.h"
+
+#define ELEMENT_BIT_SIZE  "ELEMENT_BIT_SIZE"
+#define ELEMENT_BYTE_SIZE "ELEMENT_BYTE_SIZE"
+#define CONTIGUOUS_STRIDE "CONTIGUOUS_STRIDE"
+
+#include "numo/ndloop.h"
+//#include "ndfunc.h"
+
+#include "numo/intern.h"
+
+#endif /* ifndef NARRAY_H */