numo-narray 0.9.0.1-x86-mingw32

Sign up to get free protection for your applications and to get access to all the features.
Files changed (124) hide show
  1. checksums.yaml +7 -0
  2. data/Gemfile +4 -0
  3. data/README.md +47 -0
  4. data/Rakefile +41 -0
  5. data/ext/numo/narray/SFMT-params.h +97 -0
  6. data/ext/numo/narray/SFMT-params19937.h +46 -0
  7. data/ext/numo/narray/SFMT.c +620 -0
  8. data/ext/numo/narray/SFMT.h +157 -0
  9. data/ext/numo/narray/array.c +525 -0
  10. data/ext/numo/narray/data.c +901 -0
  11. data/ext/numo/narray/depend.erb +33 -0
  12. data/ext/numo/narray/extconf.rb +117 -0
  13. data/ext/numo/narray/gen/bit.erb.c +811 -0
  14. data/ext/numo/narray/gen/cogen.rb +18 -0
  15. data/ext/numo/narray/gen/def/dcomplex.rb +32 -0
  16. data/ext/numo/narray/gen/def/dfloat.rb +30 -0
  17. data/ext/numo/narray/gen/def/int16.rb +29 -0
  18. data/ext/numo/narray/gen/def/int32.rb +29 -0
  19. data/ext/numo/narray/gen/def/int64.rb +29 -0
  20. data/ext/numo/narray/gen/def/int8.rb +29 -0
  21. data/ext/numo/narray/gen/def/robject.rb +30 -0
  22. data/ext/numo/narray/gen/def/scomplex.rb +32 -0
  23. data/ext/numo/narray/gen/def/sfloat.rb +30 -0
  24. data/ext/numo/narray/gen/def/uint16.rb +29 -0
  25. data/ext/numo/narray/gen/def/uint32.rb +29 -0
  26. data/ext/numo/narray/gen/def/uint64.rb +29 -0
  27. data/ext/numo/narray/gen/def/uint8.rb +29 -0
  28. data/ext/numo/narray/gen/dtype.erb.c +328 -0
  29. data/ext/numo/narray/gen/tmpl/accum.c +36 -0
  30. data/ext/numo/narray/gen/tmpl/accum_binary.c +75 -0
  31. data/ext/numo/narray/gen/tmpl/accum_index.c +58 -0
  32. data/ext/numo/narray/gen/tmpl/allocate.c +35 -0
  33. data/ext/numo/narray/gen/tmpl/aref.c +51 -0
  34. data/ext/numo/narray/gen/tmpl/aset.c +61 -0
  35. data/ext/numo/narray/gen/tmpl/binary.c +53 -0
  36. data/ext/numo/narray/gen/tmpl/binary2.c +55 -0
  37. data/ext/numo/narray/gen/tmpl/binary_s.c +34 -0
  38. data/ext/numo/narray/gen/tmpl/bit_binary.c +94 -0
  39. data/ext/numo/narray/gen/tmpl/bit_count.c +82 -0
  40. data/ext/numo/narray/gen/tmpl/bit_unary.c +77 -0
  41. data/ext/numo/narray/gen/tmpl/cast.c +37 -0
  42. data/ext/numo/narray/gen/tmpl/cast_array.c +79 -0
  43. data/ext/numo/narray/gen/tmpl/cast_numeric.c +22 -0
  44. data/ext/numo/narray/gen/tmpl/coerce_cast.c +8 -0
  45. data/ext/numo/narray/gen/tmpl/cond_binary.c +51 -0
  46. data/ext/numo/narray/gen/tmpl/cond_unary.c +45 -0
  47. data/ext/numo/narray/gen/tmpl/cum.c +42 -0
  48. data/ext/numo/narray/gen/tmpl/each.c +43 -0
  49. data/ext/numo/narray/gen/tmpl/each_with_index.c +64 -0
  50. data/ext/numo/narray/gen/tmpl/extract.c +23 -0
  51. data/ext/numo/narray/gen/tmpl/eye.c +91 -0
  52. data/ext/numo/narray/gen/tmpl/fill.c +38 -0
  53. data/ext/numo/narray/gen/tmpl/format.c +60 -0
  54. data/ext/numo/narray/gen/tmpl/format_to_a.c +47 -0
  55. data/ext/numo/narray/gen/tmpl/head.c +25 -0
  56. data/ext/numo/narray/gen/tmpl/inspect.c +16 -0
  57. data/ext/numo/narray/gen/tmpl/map_with_index.c +94 -0
  58. data/ext/numo/narray/gen/tmpl/median.c +44 -0
  59. data/ext/numo/narray/gen/tmpl/minmax.c +47 -0
  60. data/ext/numo/narray/gen/tmpl/poly.c +49 -0
  61. data/ext/numo/narray/gen/tmpl/pow.c +74 -0
  62. data/ext/numo/narray/gen/tmpl/powint.c +17 -0
  63. data/ext/numo/narray/gen/tmpl/qsort.c +149 -0
  64. data/ext/numo/narray/gen/tmpl/rand.c +33 -0
  65. data/ext/numo/narray/gen/tmpl/rand_norm.c +46 -0
  66. data/ext/numo/narray/gen/tmpl/robj_allocate.c +32 -0
  67. data/ext/numo/narray/gen/tmpl/seq.c +61 -0
  68. data/ext/numo/narray/gen/tmpl/set2.c +56 -0
  69. data/ext/numo/narray/gen/tmpl/sort.c +36 -0
  70. data/ext/numo/narray/gen/tmpl/sort_index.c +86 -0
  71. data/ext/numo/narray/gen/tmpl/store.c +31 -0
  72. data/ext/numo/narray/gen/tmpl/store_array.c +5 -0
  73. data/ext/numo/narray/gen/tmpl/store_from.c +53 -0
  74. data/ext/numo/narray/gen/tmpl/store_numeric.c +22 -0
  75. data/ext/numo/narray/gen/tmpl/to_a.c +41 -0
  76. data/ext/numo/narray/gen/tmpl/unary.c +58 -0
  77. data/ext/numo/narray/gen/tmpl/unary2.c +58 -0
  78. data/ext/numo/narray/gen/tmpl/unary_s.c +57 -0
  79. data/ext/numo/narray/index.c +822 -0
  80. data/ext/numo/narray/kwarg.c +79 -0
  81. data/ext/numo/narray/math.c +140 -0
  82. data/ext/numo/narray/narray.c +1539 -0
  83. data/ext/numo/narray/ndloop.c +1928 -0
  84. data/ext/numo/narray/numo/compat.h +23 -0
  85. data/ext/numo/narray/numo/intern.h +112 -0
  86. data/ext/numo/narray/numo/narray.h +411 -0
  87. data/ext/numo/narray/numo/ndloop.h +99 -0
  88. data/ext/numo/narray/numo/template.h +140 -0
  89. data/ext/numo/narray/numo/types/bit.h +19 -0
  90. data/ext/numo/narray/numo/types/complex.h +410 -0
  91. data/ext/numo/narray/numo/types/complex_macro.h +205 -0
  92. data/ext/numo/narray/numo/types/dcomplex.h +11 -0
  93. data/ext/numo/narray/numo/types/dfloat.h +12 -0
  94. data/ext/numo/narray/numo/types/float_def.h +34 -0
  95. data/ext/numo/narray/numo/types/float_macro.h +277 -0
  96. data/ext/numo/narray/numo/types/int16.h +12 -0
  97. data/ext/numo/narray/numo/types/int32.h +12 -0
  98. data/ext/numo/narray/numo/types/int64.h +12 -0
  99. data/ext/numo/narray/numo/types/int8.h +12 -0
  100. data/ext/numo/narray/numo/types/int_macro.h +34 -0
  101. data/ext/numo/narray/numo/types/robj_macro.h +218 -0
  102. data/ext/numo/narray/numo/types/robject.h +21 -0
  103. data/ext/numo/narray/numo/types/scomplex.h +11 -0
  104. data/ext/numo/narray/numo/types/sfloat.h +13 -0
  105. data/ext/numo/narray/numo/types/uint16.h +12 -0
  106. data/ext/numo/narray/numo/types/uint32.h +12 -0
  107. data/ext/numo/narray/numo/types/uint64.h +12 -0
  108. data/ext/numo/narray/numo/types/uint8.h +12 -0
  109. data/ext/numo/narray/numo/types/uint_macro.h +31 -0
  110. data/ext/numo/narray/numo/types/xint_macro.h +133 -0
  111. data/ext/numo/narray/rand.c +87 -0
  112. data/ext/numo/narray/step.c +506 -0
  113. data/ext/numo/narray/struct.c +872 -0
  114. data/lib/2.1/numo/narray.so +0 -0
  115. data/lib/2.2/numo/narray.so +0 -0
  116. data/lib/2.3/numo/narray.so +0 -0
  117. data/lib/erbpp/line_number.rb +126 -0
  118. data/lib/erbpp/narray_def.rb +338 -0
  119. data/lib/erbpp.rb +286 -0
  120. data/lib/numo/narray.rb +6 -0
  121. data/numo-narray.gemspec +35 -0
  122. data/spec/bit_spec.rb +93 -0
  123. data/spec/narray_spec.rb +249 -0
  124. metadata +238 -0
@@ -0,0 +1,149 @@
1
+ /*
2
+ qsort.c
3
+ Numerical Array Extension for Ruby
4
+ (C) Copyright 2007-2011 by Masahiro TANAKA
5
+
6
+ This program is free software.
7
+ You can distribute/modify this program
8
+ under the same terms as Ruby itself.
9
+ NO WARRANTY.
10
+ */
11
+
12
+ /*
13
+ * qsort.c: standard quicksort algorithm
14
+ *
15
+ * Modifications from vanilla NetBSD source:
16
+ * Add do ... while() macro fix
17
+ * Remove __inline, _DIAGASSERTs, __P
18
+ * Remove ill-considered "swap_cnt" switch to insertion sort,
19
+ * in favor of a simple check for presorted input.
20
+ *
21
+ * CAUTION: if you change this file, see also qsort_arg.c
22
+ *
23
+ * $PostgreSQL: pgsql/src/port/qsort.c,v 1.12 2006/10/19 20:56:22 tgl Exp $
24
+ */
25
+
26
+ /* $NetBSD: qsort.c,v 1.13 2003/08/07 16:43:42 agc Exp $ */
27
+
28
+ /*-
29
+ * Copyright (c) 1992, 1993
30
+ * The Regents of the University of California. All rights reserved.
31
+ *
32
+ * Redistribution and use in source and binary forms, with or without
33
+ * modification, are permitted provided that the following conditions
34
+ * are met:
35
+ * 1. Redistributions of source code must retain the above copyright
36
+ * notice, this list of conditions and the following disclaimer.
37
+ * 2. Redistributions in binary form must reproduce the above copyright
38
+ * notice, this list of conditions and the following disclaimer in the
39
+ * documentation and/or other materials provided with the distribution.
40
+ * 3. Neither the name of the University nor the names of its contributors
41
+ * may be used to endorse or promote products derived from this software
42
+ * without specific prior written permission.
43
+ *
44
+ * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
45
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
46
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
47
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
48
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
49
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
50
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
51
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
52
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
53
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
54
+ * SUCH DAMAGE.
55
+ */
56
+
57
+ #ifndef QSORT_INCL
58
+ #define QSORT_INCL
59
+ #define Min(x, y) ((x) < (y) ? (x) : (y))
60
+
61
+ #define swap(type,a,b) \
62
+ do {type tmp=*(type*)(a); *(type*)(a)=*(type*)(b); *(type*)(b)=tmp;} while(0)
63
+
64
+ #define vecswap(type, a, b, n) if ((n)>0) swap(type,(a),(b))
65
+
66
+ #define MED3(a,b,c) \
67
+ (cmpgt(b,a) ? \
68
+ (cmpgt(c,b) ? b : (cmpgt(c,a) ? c : a)) \
69
+ : (cmpgt(b,c) ? b : (cmpgt(c,a) ? a : c)))
70
+ #endif
71
+
72
+ #undef qsort_dtype
73
+ #define qsort_dtype <%=dtype%>
74
+ #undef qsort_cast
75
+ #define qsort_cast <%=dcast%>
76
+
77
+ void
78
+ <%=tp%>_qsort(void *a, size_t n, ssize_t es)
79
+ {
80
+ char *pa, *pb, *pc, *pd, *pl, *pm, *pn;
81
+ int d, r, presorted;
82
+
83
+ loop:
84
+ if (n < 7) {
85
+ for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es)
86
+ for (pl = pm; pl > (char *) a && cmpgt(pl - es, pl);
87
+ pl -= es)
88
+ swap(qsort_dtype, pl, pl - es);
89
+ return;
90
+ }
91
+ presorted = 1;
92
+ for (pm = (char *) a + es; pm < (char *) a + n * es; pm += es) {
93
+ if (cmpgt(pm - es, pm)) {
94
+ presorted = 0;
95
+ break;
96
+ }
97
+ }
98
+ if (presorted)
99
+ return;
100
+ pm = (char *) a + (n / 2) * es;
101
+ if (n > 7) {
102
+ pl = (char *) a;
103
+ pn = (char *) a + (n - 1) * es;
104
+ if (n > 40) {
105
+ d = (n / 8) * es;
106
+ pl = MED3(pl, pl + d, pl + 2 * d);
107
+ pm = MED3(pm - d, pm, pm + d);
108
+ pn = MED3(pn - 2 * d, pn - d, pn);
109
+ }
110
+ pm = MED3(pl, pm, pn);
111
+ }
112
+ swap(qsort_dtype, a, pm);
113
+ pa = pb = (char *) a + es;
114
+ pc = pd = (char *) a + (n - 1) * es;
115
+ for (;;) {
116
+ while (pb <= pc && (r = cmp(pb, a)) <= 0) {
117
+ if (r == 0) {
118
+ swap(qsort_dtype, pa, pb);
119
+ pa += es;
120
+ }
121
+ pb += es;
122
+ }
123
+ while (pb <= pc && (r = cmp(pc, a)) >= 0) {
124
+ if (r == 0) {
125
+ swap(qsort_dtype, pc, pd);
126
+ pd -= es;
127
+ }
128
+ pc -= es;
129
+ }
130
+ if (pb > pc)
131
+ break;
132
+ swap(qsort_dtype, pb, pc);
133
+ pb += es;
134
+ pc -= es;
135
+ }
136
+ pn = (char *) a + n * es;
137
+ r = Min(pa - (char *) a, pb - pa);
138
+ vecswap(qsort_dtype, a, pb - r, r);
139
+ r = Min(pd - pc, pn - pd - es);
140
+ vecswap(qsort_dtype, pb, pn - r, r);
141
+ if ((r = pb - pa) > es)
142
+ <%=tp%>_qsort(a, r / es, es);
143
+ if ((r = pd - pc) > es) {
144
+ a = pn - r;
145
+ n = r / es;
146
+ goto loop;
147
+ }
148
+ }
149
+
@@ -0,0 +1,33 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i;
5
+ char *p1;
6
+ ssize_t s1;
7
+ size_t *idx1;
8
+ dtype x;
9
+
10
+ INIT_COUNTER(lp, i);
11
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
12
+ if (idx1) {
13
+ for (; i--;) {
14
+ x = m_rand;
15
+ SET_DATA_INDEX(p1,idx1,dtype,x);
16
+ }
17
+ } else {
18
+ for (; i--;) {
19
+ x = m_rand;
20
+ SET_DATA_STRIDE(p1,s1,dtype,x);
21
+ }
22
+ }
23
+ }
24
+
25
+ static VALUE
26
+ <%=c_func%>(VALUE self)
27
+ {
28
+ ndfunc_arg_in_t ain[1] = {{OVERWRITE,0}};
29
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 1, 0, ain, 0 };
30
+
31
+ na_ndloop(&ndf, 1, self);
32
+ return self;
33
+ }
@@ -0,0 +1,46 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i;
5
+ char *p1;
6
+ ssize_t s1;
7
+ size_t *idx1;
8
+ dtype y;
9
+ dtype a[2];
10
+
11
+ INIT_COUNTER(lp, i);
12
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
13
+ if (idx1) {
14
+ for (; i>1; i-=2) {
15
+ m_rand_norm(a);
16
+ *(dtype*)(p1+*idx1) = a[0];
17
+ *(dtype*)(p1+*(idx1+1)) = a[1];
18
+ idx1 += 2;
19
+ }
20
+ if (i>0) {
21
+ m_rand_norm(a);
22
+ *(dtype*)(p1+*idx1) = a[0];
23
+ }
24
+ } else {
25
+ for (; i>1; i-=2) {
26
+ m_rand_norm(a);
27
+ *(dtype*)(p1) = a[0];
28
+ *(dtype*)(p1+s1) = a[1];
29
+ p1 += s1*2;
30
+ }
31
+ if (i>0) {
32
+ m_rand_norm(a);
33
+ *(dtype*)(p1) = a[0];
34
+ }
35
+ }
36
+ }
37
+
38
+ static VALUE
39
+ <%=c_func%>(VALUE self)
40
+ {
41
+ ndfunc_arg_in_t ain[1] = {{OVERWRITE,0}};
42
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 1, 0, ain, 0 };
43
+
44
+ na_ndloop(&ndf, 1, self);
45
+ return self
46
+ }
@@ -0,0 +1,32 @@
1
+ static void
2
+ na_gc_mark_robj(narray_data_t* na)
3
+ {
4
+ size_t n, i;
5
+ VALUE *a;
6
+
7
+ if (na->ptr) {
8
+ a = (VALUE*)(na->ptr);
9
+ n = na->base.size;
10
+ for (i=0; i<n; i++) {
11
+ rb_gc_mark(a[i]);
12
+ }
13
+ }
14
+ }
15
+
16
+ void na_free(narray_data_t* na);
17
+
18
+ VALUE
19
+ <%=c_func%>(VALUE klass)
20
+ {
21
+ narray_data_t *na = ALLOC(narray_data_t);
22
+
23
+ na->base.ndim = 0;
24
+ na->base.type = NARRAY_DATA_T;
25
+ na->base.flag[0] = 0;
26
+ na->base.flag[1] = 0;
27
+ na->base.size = 0;
28
+ na->base.shape = NULL;
29
+ na->base.reduce = INT2FIX(0);
30
+ na->ptr = NULL;
31
+ return Data_Wrap_Struct(klass, na_gc_mark_robj, na_free, na);
32
+ }
@@ -0,0 +1,61 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i;
5
+ char *p1;
6
+ ssize_t s1;
7
+ size_t *idx1;
8
+ double x, beg, step, c;
9
+ dtype y;
10
+ seq_opt_t *g;
11
+
12
+ INIT_COUNTER(lp, i);
13
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
14
+ g = (seq_opt_t*)(lp->opt_ptr);
15
+ beg = g->beg;
16
+ step = g->step;
17
+ c = g->count;
18
+ if (idx1) {
19
+ for (; i--;) {
20
+ x = beg + step * c++;
21
+ y = m_from_double(x);
22
+ *(dtype*)(p1+*idx1) = y;
23
+ idx1++;
24
+ }
25
+ } else {
26
+ for (; i--;) {
27
+ x = beg + step * c++;
28
+ y = m_from_double(x);
29
+ *(dtype*)(p1) = y;
30
+ p1 += s1;
31
+ }
32
+ }
33
+ g->count = c;
34
+ }
35
+
36
+ /*
37
+ Set Sequence of numbers to self NArray.
38
+ @overload seq([beg,[step]])
39
+ @param [Numeric] beg begining of sequence. (default=0)
40
+ @param [Numeric] step step of sequence. (default=1)
41
+ @return [Numo::<%=class_name%>] self.
42
+ */
43
+ static VALUE
44
+ <%=c_func%>(int argc, VALUE *args, VALUE self)
45
+ {
46
+ seq_opt_t *g;
47
+ VALUE vbeg=Qnil, vstep=Qnil;
48
+ ndfunc_arg_in_t ain[1] = {{OVERWRITE,0}};
49
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 1, 0, ain, 0 };
50
+
51
+ g = ALLOCA_N(seq_opt_t,1);
52
+ g->beg = 0;
53
+ g->step = 1;
54
+ g->count = 0;
55
+ rb_scan_args(argc, args, "02", &vbeg, &vstep);
56
+ if (vbeg!=Qnil) {g->beg = NUM2DBL(vbeg);}
57
+ if (vstep!=Qnil) {g->step = NUM2DBL(vstep);}
58
+
59
+ na_ndloop3(&ndf, g, 1, self);
60
+ return self;
61
+ }
@@ -0,0 +1,56 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i;
5
+ char *p1, *p2;
6
+ ssize_t s1, s2;
7
+ size_t *idx1, *idx2;
8
+ dtype x;
9
+ <%=dtype%> y;
10
+ INIT_COUNTER(lp, i);
11
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
12
+ INIT_PTR_IDX(lp, 1, p2, s2, idx2);
13
+ if (idx1) {
14
+ if (idx2) {
15
+ for (; i--;) {
16
+ GET_DATA_INDEX(p1,idx1,dtype,x);
17
+ GET_DATA_INDEX(p2,idx2,<%=dtype%>,y);
18
+ x = m_<%=method%>(x,y);
19
+ SET_DATA_INDEX(p1,idx1,dtype,x);
20
+ }
21
+ } else {
22
+ for (; i--;) {
23
+ GET_DATA_INDEX(p1,idx1,dtype,x);
24
+ GET_DATA_STRIDE(p2,s2,<%=dtype%>,y);
25
+ x = m_<%=method%>(x,y);
26
+ SET_DATA_INDEX(p1,idx1,dtype,x);
27
+ }
28
+ }
29
+ } else {
30
+ if (idx2) {
31
+ for (; i--;) {
32
+ GET_DATA(p1,dtype,x);
33
+ GET_DATA_INDEX(p2,idx2,<%=dtype%>,y);
34
+ x = m_<%=method%>(x,y);
35
+ SET_DATA_STRIDE(p1,s1,dtype,x);
36
+ }
37
+ } else {
38
+ for (; i--;) {
39
+ GET_DATA(p1,dtype,x);
40
+ GET_DATA_STRIDE(p2,s2,<%=dtype%>,y);
41
+ x = m_<%=method%>(x,y);
42
+ SET_DATA_STRIDE(p1,s1,dtype,x);
43
+ }
44
+ }
45
+ }
46
+ }
47
+
48
+ static VALUE
49
+ <%=c_func%>(VALUE self, VALUE a1)
50
+ {
51
+ ndfunc_arg_in_t ain[2] = {{OVERWRITE,0},{<%=tpclass%>,0}};
52
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 2, 0, ain, 0 };
53
+
54
+ na_ndloop(&ndf, 2, self, a1);
55
+ return a1;
56
+ }
@@ -0,0 +1,36 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t n;
5
+ char *ptr;
6
+ ssize_t step;
7
+
8
+ INIT_COUNTER(lp, n);
9
+ INIT_PTR(lp, 0, ptr, step);
10
+ <%=tp%>_qsort(ptr, n, step);
11
+ }
12
+
13
+ /*
14
+ * call-seq:
15
+ * narray.sort() => narray
16
+ * narray.sort(dim0,dim1,...) => narray
17
+ *
18
+ * Return an index array of sort result.
19
+ *
20
+ * Numo::DFloat[3,4,1,2].sort => Numo::DFloat[1,2,3,4]
21
+ */
22
+ static VALUE
23
+ <%=c_func%>(int argc, VALUE *argv, VALUE self)
24
+ {
25
+ VALUE reduce;
26
+ ndfunc_arg_in_t ain[2] = {{OVERWRITE,0},{sym_reduce,0}};
27
+ ndfunc_t ndf = {<%=c_iter%>, STRIDE_LOOP_NIP|NDF_INPLACE|NDF_FLAT_REDUCE, 2,0, ain,0};
28
+
29
+ if (!TEST_INPLACE(self)) {
30
+ self = na_copy(self);
31
+ }
32
+ reduce = na_reduce_dimension(argc, argv, 1, &self); // v[0] = self
33
+
34
+ na_ndloop(&ndf, 2, self, reduce);
35
+ return self;
36
+ }
@@ -0,0 +1,86 @@
1
+
2
+ <% [64,32].each do |i| %>
3
+ #define idx_t int<%=i%>_t
4
+ static void
5
+ <%=tp%>_index<%=i%>_qsort(na_loop_t *const lp)
6
+ {
7
+ size_t i, n, idx;
8
+ char *d_ptr, *i_ptr, *o_ptr;
9
+ ssize_t d_step, i_step, o_step;
10
+ char **ptr;
11
+
12
+ INIT_COUNTER(lp, n);
13
+ INIT_PTR(lp, 0, d_ptr, d_step);
14
+ INIT_PTR(lp, 1, i_ptr, i_step);
15
+ INIT_PTR(lp, 2, o_ptr, o_step);
16
+
17
+ ptr = (char**)(lp->opt_ptr);
18
+
19
+ //printf("(ptr=%lx, d_ptr=%lx,d_step=%ld, i_ptr=%lx,i_step=%ld, o_ptr=%lx,o_step=%ld)\n",(size_t)ptr,(size_t)d_ptr,(ssize_t)d_step,(size_t)i_ptr,(ssize_t)i_step,(size_t)o_ptr,(ssize_t)o_step);
20
+
21
+ for (i=0; i<n; i++) {
22
+ ptr[i] = d_ptr + d_step * i;
23
+ //printf("(%ld,%.3f)",i,*(double*)ptr[i]);
24
+ }
25
+
26
+ <%=tp%>_index_qsort(ptr, n, sizeof(dtype*));
27
+
28
+ //d_ptr = lp->args[0].ptr;
29
+ //printf("(d_ptr=%lx)\n",(size_t)d_ptr);
30
+
31
+ for (i=0; i<n; i++) {
32
+ idx = (ptr[i] - d_ptr) / d_step;
33
+ *(idx_t*)o_ptr = *(idx_t*)(i_ptr + i_step * idx);
34
+ //printf("(idx[%ld]=%ld,%d)",i,idx,*(idx_t*)o_ptr);
35
+ o_ptr += o_step;
36
+ }
37
+ //printf("\n");
38
+ }
39
+ #undef idx_t
40
+ <% end %>
41
+
42
+ /*
43
+ * call-seq:
44
+ * narray.sort_index() => narray
45
+ * narray.sort_index(dim0,dim1,..) => narray
46
+ *
47
+ * Return an index array of sort result.
48
+ *
49
+ * Numo::NArray[3,4,1,2].sort_index => Numo::Int32[2,3,0,1]
50
+ */
51
+ static VALUE
52
+ <%=c_func%>(int argc, VALUE *argv, VALUE self)
53
+ {
54
+ size_t size;
55
+ narray_t *na;
56
+ VALUE idx, tmp, reduce, res;
57
+ char *buf;
58
+ ndfunc_arg_in_t ain[3] = {{cT,0},{0,0},{sym_reduce,0}};
59
+ ndfunc_arg_out_t aout[1] = {{0,0,0}};
60
+ ndfunc_t ndf = {0, STRIDE_LOOP_NIP|NDF_FLAT_REDUCE|NDF_CUM, 3,1, ain,aout};
61
+
62
+ GetNArray(self,na);
63
+ if (na->ndim==0) {
64
+ return INT2FIX(0);
65
+ }
66
+ if (na->size > (~(u_int32_t)0)) {
67
+ ain[1].type =
68
+ aout[0].type = numo_cInt64;
69
+ idx = rb_narray_new(numo_cInt64, na->ndim, na->shape);
70
+ ndf.func = <%=tp%>_index64_qsort;
71
+ } else {
72
+ ain[1].type =
73
+ aout[0].type = numo_cInt32;
74
+ idx = rb_narray_new(numo_cInt32, na->ndim, na->shape);
75
+ ndf.func = <%=tp%>_index32_qsort;
76
+ }
77
+ rb_funcall(idx, rb_intern("seq"), 0);
78
+
79
+ reduce = na_reduce_dimension(argc, argv, 1, &self); // v[0] = self
80
+
81
+ size = na->size*sizeof(void*);
82
+ buf = rb_alloc_tmp_buffer(&tmp, size);
83
+ res = na_ndloop3(&ndf, buf, 3, self, idx, reduce);
84
+ rb_free_tmp_buffer(&tmp);
85
+ return res;
86
+ }
@@ -0,0 +1,31 @@
1
+ /*
2
+ Store elements to Numo::<%=class_name%> from other.
3
+ @overload store(other)
4
+ @param [Object] other
5
+ @return [Numo::<%=class_name%>] self
6
+ */
7
+ static VALUE
8
+ <%=c_func%>(VALUE self, VALUE obj)
9
+ {
10
+ VALUE r;
11
+
12
+ <% Store.definitions.each do |x| %>
13
+ if (<%=x.condition%>) {
14
+ <%=x.c_func%>(self,obj);
15
+ return self;
16
+ }
17
+ <% end %>
18
+
19
+ if (IsNArray(obj)) {
20
+ r = rb_funcall(obj, rb_intern("coerce_cast"), 1, cT);
21
+ if (CLASS_OF(r)==cT) {
22
+ <%=c_func%>(self,r);
23
+ return self;
24
+ }
25
+ }
26
+
27
+ rb_raise(nary_eCastError, "unknown conversion from %s to %s",
28
+ rb_class2name(CLASS_OF(obj)),
29
+ rb_class2name(CLASS_OF(self)));
30
+ return self;
31
+ }
@@ -0,0 +1,5 @@
1
+ static VALUE
2
+ <%=c_func%>(VALUE self, VALUE obj)
3
+ {
4
+ return <%=find_tmpl("store").c_func%>(self,<%=find_tmpl("cast_array").c_func%>(obj));
5
+ }
@@ -0,0 +1,53 @@
1
+ static void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i, s1, s2;
5
+ char *p1, *p2;
6
+ size_t *idx1, *idx2;
7
+ <%=dtype%> x;
8
+ dtype y;
9
+
10
+ INIT_COUNTER(lp, i);
11
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
12
+ INIT_PTR_IDX(lp, 1, p2, s2, idx2);
13
+ if (idx2) {
14
+ if (idx1) {
15
+ for (; i--;) {
16
+ GET_DATA_INDEX(p2,idx2,<%=dtype%>,x);
17
+ y = <%=macro%>(x);
18
+ SET_DATA_INDEX(p1,idx1,dtype,y);
19
+ }
20
+ } else {
21
+ for (; i--;) {
22
+ GET_DATA_INDEX(p2,idx2,<%=dtype%>,x);
23
+ y = <%=macro%>(x);
24
+ SET_DATA_STRIDE(p1,s1,dtype,y);
25
+ }
26
+ }
27
+ } else {
28
+ if (idx1) {
29
+ for (; i--;) {
30
+ GET_DATA_STRIDE(p2,s2,<%=dtype%>,x);
31
+ y = <%=macro%>(x);
32
+ SET_DATA_INDEX(p1,idx1,dtype,y);
33
+ }
34
+ } else {
35
+ for (; i--;) {
36
+ GET_DATA_STRIDE(p2,s2,<%=dtype%>,x);
37
+ y = <%=macro%>(x);
38
+ SET_DATA_STRIDE(p1,s1,dtype,y);
39
+ }
40
+ }
41
+ }
42
+ }
43
+
44
+
45
+ static VALUE
46
+ <%=c_func%>(VALUE self, VALUE obj)
47
+ {
48
+ ndfunc_arg_in_t ain[2] = {{OVERWRITE,0},{Qnil,0}};
49
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP, 2, 0, ain, 0 };
50
+
51
+ na_ndloop(&ndf, 2, self, obj);
52
+ return self;
53
+ }
@@ -0,0 +1,22 @@
1
+ static VALUE
2
+ numo_<%=tp%>_new_dim0(dtype x)
3
+ {
4
+ VALUE v;
5
+ dtype *ptr;
6
+
7
+ v = rb_narray_new(cT, 0, NULL);
8
+ ptr = (dtype*)(char*)na_get_pointer_for_write(v);
9
+ *ptr = x;
10
+ na_release_lock(v);
11
+ return v;
12
+ }
13
+
14
+ static VALUE
15
+ <%=c_func%>(VALUE self, VALUE obj)
16
+ {
17
+ dtype x;
18
+ x = m_num_to_data(obj);
19
+ obj = numo_<%=tp%>_new_dim0(x);
20
+ <%=find_tmpl("store").c_func%>(self,obj);
21
+ return self;
22
+ }
@@ -0,0 +1,41 @@
1
+ void
2
+ <%=c_iter%>(na_loop_t *const lp)
3
+ {
4
+ size_t i, s1;
5
+ char *p1;
6
+ size_t *idx1;
7
+ dtype x;
8
+ volatile VALUE a, y;
9
+
10
+ INIT_COUNTER(lp, i);
11
+ INIT_PTR_IDX(lp, 0, p1, s1, idx1);
12
+ a = rb_ary_new2(i);
13
+ rb_ary_push(lp->args[1].value, a);
14
+ if (idx1) {
15
+ for (; i--;) {
16
+ GET_DATA_INDEX(p1,idx1,dtype,x);
17
+ y = m_data_to_num(x);
18
+ rb_ary_push(a,y);
19
+ }
20
+ } else {
21
+ for (; i--;) {
22
+ GET_DATA_STRIDE(p1,s1,dtype,x);
23
+ y = m_data_to_num(x);
24
+ rb_ary_push(a,y);
25
+ }
26
+ }
27
+ }
28
+
29
+ /*
30
+ Convert self to Array.
31
+ @overload <%=method%>
32
+ @return [Array]
33
+ */
34
+ static VALUE
35
+ <%=c_func%>(VALUE self)
36
+ {
37
+ ndfunc_arg_in_t ain[3] = {{Qnil,0},{sym_loop_opt},{sym_option}};
38
+ ndfunc_arg_out_t aout[1] = {{rb_cArray,0}}; // dummy?
39
+ ndfunc_t ndf = { <%=c_iter%>, FULL_LOOP_NIP, 3, 1, ain, aout };
40
+ return na_ndloop_cast_narray_to_rarray(&ndf, self, Qnil);
41
+ }