carray 1.4.0 → 1.5.5

Sign up to get free protection for your applications and to get access to all the features.
Files changed (239) hide show
  1. checksums.yaml +4 -4
  2. data/.yardopts +32 -0
  3. data/LICENSE +22 -0
  4. data/NEWS.md +83 -0
  5. data/README.md +38 -18
  6. data/Rakefile +2 -2
  7. data/TODO.md +17 -0
  8. data/carray.gemspec +13 -18
  9. data/{ca_iter_block.c → ext/ca_iter_block.c} +16 -18
  10. data/{ca_iter_dimension.c → ext/ca_iter_dimension.c} +20 -21
  11. data/{ca_iter_window.c → ext/ca_iter_window.c} +12 -14
  12. data/{ca_obj_array.c → ext/ca_obj_array.c} +451 -176
  13. data/{ca_obj_bitarray.c → ext/ca_obj_bitarray.c} +18 -23
  14. data/{ca_obj_bitfield.c → ext/ca_obj_bitfield.c} +12 -16
  15. data/{ca_obj_block.c → ext/ca_obj_block.c} +47 -54
  16. data/{ca_obj_fake.c → ext/ca_obj_fake.c} +10 -12
  17. data/{ca_obj_farray.c → ext/ca_obj_farray.c} +21 -23
  18. data/{ca_obj_field.c → ext/ca_obj_field.c} +30 -32
  19. data/{ca_obj_grid.c → ext/ca_obj_grid.c} +32 -33
  20. data/{ca_obj_mapping.c → ext/ca_obj_mapping.c} +11 -13
  21. data/{ca_obj_object.c → ext/ca_obj_object.c} +40 -42
  22. data/{ca_obj_reduce.c → ext/ca_obj_reduce.c} +3 -5
  23. data/{ca_obj_refer.c → ext/ca_obj_refer.c} +44 -48
  24. data/{ca_obj_repeat.c → ext/ca_obj_repeat.c} +45 -47
  25. data/{ca_obj_select.c → ext/ca_obj_select.c} +4 -6
  26. data/{ca_obj_shift.c → ext/ca_obj_shift.c} +26 -28
  27. data/{ca_obj_transpose.c → ext/ca_obj_transpose.c} +26 -28
  28. data/{ca_obj_unbound_repeat.c → ext/ca_obj_unbound_repeat.c} +106 -160
  29. data/{ca_obj_window.c → ext/ca_obj_window.c} +33 -35
  30. data/{carray.h → ext/carray.h} +86 -59
  31. data/{carray_access.c → ext/carray_access.c} +194 -101
  32. data/{carray_attribute.c → ext/carray_attribute.c} +161 -207
  33. data/{carray_call_cfunc.c → ext/carray_call_cfunc.c} +1 -3
  34. data/{carray_cast.c → ext/carray_cast.c} +351 -150
  35. data/{carray_cast_func.rb → ext/carray_cast_func.rb} +1 -2
  36. data/{carray_class.c → ext/carray_class.c} +28 -36
  37. data/{carray_conversion.c → ext/carray_conversion.c} +63 -68
  38. data/{carray_copy.c → ext/carray_copy.c} +34 -50
  39. data/{carray_core.c → ext/carray_core.c} +75 -62
  40. data/ext/carray_data_type.c +66 -0
  41. data/{carray_element.c → ext/carray_element.c} +34 -53
  42. data/{carray_generate.c → ext/carray_generate.c} +71 -50
  43. data/{carray_iterator.c → ext/carray_iterator.c} +53 -53
  44. data/{carray_loop.c → ext/carray_loop.c} +77 -106
  45. data/{carray_mask.c → ext/carray_mask.c} +105 -114
  46. data/{carray_math.rb → ext/carray_math.rb} +29 -13
  47. data/ext/{mathfunc/carray_mathfunc.c → carray_mathfunc.c} +1 -3
  48. data/{carray_numeric.c → ext/carray_numeric.c} +43 -46
  49. data/{carray_operator.c → ext/carray_operator.c} +49 -36
  50. data/{carray_order.c → ext/carray_order.c} +232 -217
  51. data/{carray_sort_addr.c → ext/carray_sort_addr.c} +14 -21
  52. data/{carray_stat.c → ext/carray_stat.c} +6 -8
  53. data/{carray_stat_proc.rb → ext/carray_stat_proc.rb} +25 -27
  54. data/{carray_test.c → ext/carray_test.c} +63 -51
  55. data/{carray_undef.c → ext/carray_undef.c} +1 -11
  56. data/{carray_utils.c → ext/carray_utils.c} +12 -4
  57. data/{extconf.rb → ext/extconf.rb} +10 -7
  58. data/{mkmath.rb → ext/mkmath.rb} +2 -2
  59. data/{ruby_carray.c → ext/ruby_carray.c} +20 -8
  60. data/{ruby_ccomplex.c → ext/ruby_ccomplex.c} +2 -4
  61. data/{ruby_float_func.c → ext/ruby_float_func.c} +1 -3
  62. data/ext/version.h +16 -0
  63. data/{version.rb → ext/version.rb} +0 -0
  64. data/lib/carray.rb +51 -40
  65. data/lib/carray/{base/autoload.rb → autoload.rb} +1 -3
  66. data/lib/carray/autoload/autoload_base.rb +1 -1
  67. data/lib/carray/autoload/autoload_gem_numo_narray.rb +7 -6
  68. data/lib/carray/autoload/autoload_gem_random.rb +8 -0
  69. data/lib/carray/basic.rb +191 -0
  70. data/lib/carray/broadcast.rb +101 -0
  71. data/lib/carray/compose.rb +315 -0
  72. data/lib/carray/construct.rb +484 -0
  73. data/lib/carray/convert.rb +115 -0
  74. data/lib/carray/info.rb +1 -3
  75. data/lib/carray/{base/inspect.rb → inspect.rb} +9 -11
  76. data/lib/carray/io/imagemagick.rb +2 -4
  77. data/lib/carray/{base/iterator.rb → iterator.rb} +6 -6
  78. data/lib/carray/mask.rb +102 -0
  79. data/lib/carray/{base/math.rb → math.rb} +20 -52
  80. data/lib/carray/math/histogram.rb +8 -10
  81. data/lib/carray/math/recurrence.rb +1 -3
  82. data/lib/carray/mkmf.rb +9 -3
  83. data/lib/carray/object/ca_obj_iterator.rb +1 -3
  84. data/lib/carray/object/ca_obj_link.rb +1 -3
  85. data/lib/carray/object/ca_obj_pack.rb +9 -11
  86. data/lib/carray/obsolete.rb +256 -0
  87. data/lib/carray/ordering.rb +181 -0
  88. data/lib/carray/{base/serialize.rb → serialize.rb} +60 -76
  89. data/lib/carray/{base/string.rb → string.rb} +10 -64
  90. data/lib/carray/{base/struct.rb → struct.rb} +19 -21
  91. data/lib/carray/{io/table.rb → table.rb} +1 -10
  92. data/lib/carray/testing.rb +51 -0
  93. data/lib/carray/time.rb +76 -0
  94. data/lib/carray/transform.rb +109 -0
  95. data/misc/Methods.ja.md +182 -0
  96. data/{NOTE → misc/NOTE} +16 -38
  97. data/spec/Classes/CABitfield_spec.rb +58 -0
  98. data/spec/Classes/CABlockIterator_spec.rb +114 -0
  99. data/spec/Classes/CABlock_spec.rb +205 -0
  100. data/spec/Classes/CAField_spec.rb +39 -0
  101. data/spec/Classes/CAGrid_spec.rb +75 -0
  102. data/spec/Classes/CAMap_spec.rb +0 -0
  103. data/{test/test_CAMapping.rb → spec/Classes/CAMapping_spec.rb} +35 -36
  104. data/spec/Classes/CAObject_attribute_spec.rb +33 -0
  105. data/spec/Classes/CAObject_spec.rb +33 -0
  106. data/spec/Classes/CARefer_spec.rb +93 -0
  107. data/spec/Classes/CARepeat_spec.rb +65 -0
  108. data/spec/Classes/CASelect_spec.rb +22 -0
  109. data/spec/Classes/CAShift_spec.rb +16 -0
  110. data/spec/Classes/CAStruct_spec.rb +71 -0
  111. data/{test/test_CATranspose.rb → spec/Classes/CATranspose_spec.rb} +20 -21
  112. data/spec/Classes/CAUnboudRepeat_spec.rb +102 -0
  113. data/spec/Classes/CAWindow_spec.rb +54 -0
  114. data/spec/Classes/CAWrap_spec.rb +8 -0
  115. data/{test/test_CArray.rb → spec/Classes/CArray_spec.rb} +48 -92
  116. data/spec/Classes/CScalar_spec.rb +55 -0
  117. data/spec/Features/feature_130_spec.rb +19 -0
  118. data/spec/Features/feature_attributes_spec.rb +280 -0
  119. data/spec/Features/feature_boolean_spec.rb +98 -0
  120. data/spec/Features/feature_broadcast.rb +116 -0
  121. data/spec/Features/feature_cast_function.rb +19 -0
  122. data/spec/Features/feature_cast_spec.rb +33 -0
  123. data/spec/Features/feature_class_spec.rb +84 -0
  124. data/spec/Features/feature_complex_spec.rb +42 -0
  125. data/{test/test_composite.rb → spec/Features/feature_composite_spec.rb} +17 -18
  126. data/spec/Features/feature_convert_spec.rb +46 -0
  127. data/spec/Features/feature_copy_spec.rb +123 -0
  128. data/spec/Features/feature_creation_spec.rb +84 -0
  129. data/spec/Features/feature_element_spec.rb +144 -0
  130. data/spec/Features/feature_extream_spec.rb +54 -0
  131. data/spec/Features/feature_generate_spec.rb +74 -0
  132. data/spec/Features/feature_index_spec.rb +69 -0
  133. data/spec/Features/feature_mask_spec.rb +580 -0
  134. data/spec/Features/feature_math_spec.rb +97 -0
  135. data/spec/Features/feature_order_spec.rb +146 -0
  136. data/spec/Features/feature_ref_store_spec.rb +209 -0
  137. data/spec/Features/feature_serialization_spec.rb +125 -0
  138. data/spec/Features/feature_stat_spec.rb +397 -0
  139. data/spec/Features/feature_virtual_spec.rb +48 -0
  140. data/spec/Features/method_eq_spec.rb +81 -0
  141. data/spec/Features/method_is_nan_spec.rb +12 -0
  142. data/spec/Features/method_map_spec.rb +54 -0
  143. data/spec/Features/method_max_with.rb +20 -0
  144. data/spec/Features/method_min_with.rb +19 -0
  145. data/spec/Features/method_ne_spec.rb +18 -0
  146. data/spec/Features/method_project_spec.rb +188 -0
  147. data/spec/Features/method_ref_spec.rb +27 -0
  148. data/spec/Features/method_round_spec.rb +11 -0
  149. data/spec/Features/method_s_linspace_spec.rb +48 -0
  150. data/spec/Features/method_s_span_spec.rb +14 -0
  151. data/spec/Features/method_seq_spec.rb +47 -0
  152. data/spec/Features/method_sort_with.rb +43 -0
  153. data/spec/Features/method_sorted_with.rb +29 -0
  154. data/spec/Features/method_span_spec.rb +42 -0
  155. data/spec/Features/method_wrap_readonly_spec.rb +43 -0
  156. data/{test → spec/UnitTest}/test_CAVirtual.rb +0 -0
  157. data/spec/spec_all.rb +0 -1
  158. data/utils/convert_test.rb +73 -0
  159. data/utils/{extract_rdoc.rb → extract_yard.rb} +7 -12
  160. data/{devel → utils}/guess_shape.rb +0 -0
  161. data/utils/{diff_method.rb → monkey_patch_methods.rb} +17 -7
  162. metadata +159 -206
  163. data/COPYING +0 -56
  164. data/GPL +0 -340
  165. data/LEGAL +0 -50
  166. data/TODO +0 -5
  167. data/carray_random.c +0 -531
  168. data/devel/im2col.rb +0 -17
  169. data/ext/calculus/carray_calculus.c +0 -931
  170. data/ext/calculus/carray_interp.c +0 -358
  171. data/ext/calculus/extconf.rb +0 -12
  172. data/ext/calculus/lib/math/calculus.rb +0 -119
  173. data/ext/calculus/lib/math/interp/adapter_interp1d.rb +0 -31
  174. data/ext/mathfunc/extconf.rb +0 -18
  175. data/ext/mathfunc/test/test_hypot.rb +0 -5
  176. data/ext/mathfunc/test/test_j0.rb +0 -22
  177. data/ext/mathfunc/test/test_jn.rb +0 -8
  178. data/ext/mathfunc/test/test_sph.rb +0 -9
  179. data/lib/carray/autoload/autoload_io_table.rb +0 -1
  180. data/lib/carray/autoload/autoload_math_interp.rb +0 -4
  181. data/lib/carray/base/basic.rb +0 -1146
  182. data/lib/carray/base/obsolete.rb +0 -131
  183. data/lib/carray/math/interp.rb +0 -57
  184. data/lib/carray/math/interp/adapter_gsl_spline.rb +0 -47
  185. data/mt19937ar.c +0 -182
  186. data/mt19937ar.h +0 -86
  187. data/rdoc_main.rb +0 -27
  188. data/rdoc_math.rb +0 -5
  189. data/rdoc_stat.rb +0 -31
  190. data/spec/CABlockIterator/CABlockIterator_spec.rb +0 -113
  191. data/spec/CArray/bug/store_spec.rb +0 -27
  192. data/spec/CArray/index/repeat_spec.rb +0 -10
  193. data/spec/CArray/method/eq_spec.rb +0 -80
  194. data/spec/CArray/method/is_nan_spec.rb +0 -12
  195. data/spec/CArray/method/ne_spec.rb +0 -18
  196. data/spec/CArray/method/round_spec.rb +0 -11
  197. data/spec/CArray/object/_attribute_spec.rb +0 -32
  198. data/spec/CArray/object/s_new_spec.rb +0 -31
  199. data/spec/CArray/serialize/Serialization_spec.rb +0 -89
  200. data/test/test_130.rb +0 -23
  201. data/test/test_ALL.rb +0 -51
  202. data/test/test_CABitfield.rb +0 -59
  203. data/test/test_CABlock.rb +0 -208
  204. data/test/test_CAField.rb +0 -40
  205. data/test/test_CAGrid.rb +0 -76
  206. data/test/test_CAMmap.rb +0 -11
  207. data/test/test_CARefer.rb +0 -94
  208. data/test/test_CARepeat.rb +0 -66
  209. data/test/test_CASelect.rb +0 -23
  210. data/test/test_CAShift.rb +0 -17
  211. data/test/test_CAWindow.rb +0 -55
  212. data/test/test_CAWrap.rb +0 -9
  213. data/test/test_CComplex.rb +0 -83
  214. data/test/test_CScalar.rb +0 -91
  215. data/test/test_attribute.rb +0 -281
  216. data/test/test_block_iterator.rb +0 -17
  217. data/test/test_boolean.rb +0 -99
  218. data/test/test_cast.rb +0 -33
  219. data/test/test_class.rb +0 -85
  220. data/test/test_complex.rb +0 -43
  221. data/test/test_convert.rb +0 -79
  222. data/test/test_copy.rb +0 -141
  223. data/test/test_creation.rb +0 -85
  224. data/test/test_element.rb +0 -146
  225. data/test/test_extream.rb +0 -55
  226. data/test/test_generate.rb +0 -75
  227. data/test/test_index.rb +0 -71
  228. data/test/test_mask.rb +0 -578
  229. data/test/test_math.rb +0 -98
  230. data/test/test_narray.rb +0 -64
  231. data/test/test_order.rb +0 -147
  232. data/test/test_random.rb +0 -15
  233. data/test/test_ref_store.rb +0 -211
  234. data/test/test_stat.rb +0 -414
  235. data/test/test_struct.rb +0 -72
  236. data/test/test_virtual.rb +0 -49
  237. data/utils/create_rdoc.sh +0 -9
  238. data/utils/make_tgz.sh +0 -3
  239. data/version.h +0 -18
@@ -0,0 +1,484 @@
1
+ # ----------------------------------------------------------------------------
2
+ #
3
+ # carray/constructor.rb
4
+ #
5
+ # This file is part of Ruby/CArray extension library.
6
+ #
7
+ # Copyright (C) 2005-2020 Hiroki Motoyoshi
8
+ #
9
+ # ----------------------------------------------------------------------------
10
+
11
+ class CArray
12
+
13
+ #
14
+ # CArray.span(data_type, range[, step])
15
+ # CArray.span(range[, step]) -> data_type guessed by range.first type
16
+ #
17
+
18
+ def self.span (*argv)
19
+ if argv.first.is_a?(Range)
20
+ type = nil
21
+ else
22
+ type, = *CArray.guess_type_and_bytes(argv.shift, nil)
23
+ end
24
+ range, step = argv[0], argv[1]
25
+ start, stop = range.begin, range.end
26
+ if step == 0
27
+ raise "step should not be 0"
28
+ end
29
+ if not type
30
+ case start
31
+ when Integer
32
+ type = CA_INT32
33
+ when Float
34
+ type = CA_FLOAT64
35
+ else
36
+ type = CA_OBJECT
37
+ end
38
+ end
39
+ if type == CA_OBJECT and not step
40
+ return CA_OBJECT(range.to_a)
41
+ else
42
+ step ||= 1
43
+ if range.exclude_end?
44
+ n = ((stop - start).abs/step).floor
45
+ else
46
+ n = ((stop - start).abs/step).floor + 1
47
+ end
48
+ if start <= stop
49
+ return CArray.new(type, [n]).seq(start, step)
50
+ else
51
+ return CArray.new(type, [n]).seq(start, -step.abs)
52
+ end
53
+ end
54
+ end
55
+
56
+ def span! (range)
57
+ first = range.begin.to_r
58
+ last = range.end.to_r
59
+ if integer?
60
+ if range.exclude_end?
61
+ step = ((last-1)-first+1)/elements
62
+ else
63
+ step = (last-first+1)/elements
64
+ end
65
+ else
66
+ if range.exclude_end?
67
+ step = (last-first)/elements
68
+ else
69
+ step = (last-first)/(elements-1)
70
+ end
71
+ end
72
+ if integer? && step.denominator != 1
73
+ self[] = (first + seq * step).floor
74
+ else
75
+ seq!(first, step)
76
+ end
77
+ return self
78
+ end
79
+
80
+ def span (range)
81
+ return template.span!(range)
82
+ end
83
+
84
+ #
85
+ #
86
+ #
87
+
88
+ def scale! (xa, xb)
89
+ xa = xa.to_f
90
+ xb = xb.to_f
91
+ seq!(xa, (xb-xa)/(elements-1))
92
+ end
93
+
94
+ def scale (xa, xb)
95
+ template.scale!(xa, xb)
96
+ end
97
+
98
+ # @private
99
+ # Create new CArray object from the return value of the block
100
+ # with data type +type+. The dimensional size and the initialization value
101
+ # are guessed from the return value of the block.
102
+ # The block should return one of the following objects.
103
+ #
104
+ # * Numeric
105
+ # * Array
106
+ # * CArray
107
+ # * an object that has either method +to_ca+ or +to_a+ or +map+
108
+ #
109
+ # When the return value of the block is a Numeric or CScalar object,
110
+ # CScalar object is returned.
111
+ #
112
+ def self.__new__ (type, *args) # :nodoc:
113
+ case v = args.first
114
+ when CArray
115
+ return ( v.data_type == type ) ? v.to_ca : v.to_type(type)
116
+ when Array
117
+ return CArray.new(type, CArray.guess_array_shape(v)) { v }
118
+ when Range
119
+ return CArray.span(type, *args)
120
+ when String
121
+ if type == CA_OBJECT
122
+ return CScalar.new(CA_OBJECT) { v }
123
+ elsif type == CA_BOOLEAN
124
+ v = v.dup
125
+ v.tr!('^01',"1")
126
+ v.tr!('01',"\x0\x1")
127
+ return CArray.boolean(v.length).load_binary(v)
128
+ else
129
+ case v
130
+ when /;/
131
+ v = v.strip.split(/\s*;\s*/).
132
+ map{|s| s.split(/\s+|\s*,\s*/).map{|x| x=='_' ? UNDEF : x} }
133
+ else
134
+ v = v.strip.split(/\s+|\s*,\s*/).map{|x| x=='_' ? UNDEF : x}
135
+ end
136
+ return CArray.new(type, CArray.guess_array_shape(v)) { v }
137
+ end
138
+ when NilClass
139
+ return CArray.new(type, [0])
140
+ else
141
+ if v.respond_to?(:to_ca)
142
+ ca = v.to_ca
143
+ return ( ca.data_type == type ) ? ca : ca.to_type(type)
144
+ else
145
+ return CScalar.new(type) { v }
146
+ end
147
+ end
148
+ end
149
+
150
+ # @private
151
+ def self.__new_fixlen__ (bytes, v) # :nodoc:
152
+ case v
153
+ when CArray
154
+ return ( v.data_type == :fixlen ) ? v.to_ca : v.to_type(:fixlen, :bytes=>bytes)
155
+ when Array
156
+ unless bytes
157
+ bytes = v.map{|s| s.length}.max
158
+ end
159
+ return CArray.new(:fixlen, CArray.guess_array_shape(v), :bytes=>bytes) { v }
160
+ when NilClass
161
+ return CArray.new(type, [0])
162
+ else
163
+ if v.respond_to?(:to_ca)
164
+ ca = v.to_ca
165
+ return ( ca.data_type == :fixlen ) ? ca : ca.to_type(:fixlen, :bytes=>bytes)
166
+ else
167
+ return CScalar.new(:fixlen, :bytes=>bytes) { v }
168
+ end
169
+ end
170
+ end
171
+
172
+ end
173
+
174
+ #
175
+ # CA_INT8(data)
176
+ # :
177
+ # CA_CMPLX256(data)
178
+ #
179
+ # Create new CArray object from +data+ with data type
180
+ # which is guessed from the method name. +data+ should be one of the following
181
+ # objects.
182
+ #
183
+ # * Numeric
184
+ # * Array
185
+ # * CArray
186
+ # * an object that has either method +to_ca+ or +to_a+ or +map+
187
+ #
188
+ # When the block returns a Numeric or CScalar object,
189
+ # the resulted array is a CScalar object.
190
+
191
+ [
192
+ "CA_BOOLEAN",
193
+ "CA_INT8",
194
+ "CA_UINT8",
195
+ "CA_INT16",
196
+ "CA_UINT16",
197
+ "CA_INT32",
198
+ "CA_UINT32",
199
+ "CA_INT64",
200
+ "CA_UINT64",
201
+ "CA_FLOAT32",
202
+ "CA_FLOAT64",
203
+ "CA_FLOAT128",
204
+ "CA_CMPLX64",
205
+ "CA_CMPLX128",
206
+ "CA_CMPLX256",
207
+ "CA_OBJECT",
208
+ "CA_BYTE",
209
+ "CA_SHORT",
210
+ "CA_INT",
211
+ "CA_FLOAT",
212
+ "CA_DOUBLE",
213
+ "CA_COMPLEX",
214
+ "CA_DCOMPLEX",
215
+ "CA_SIZE",
216
+ ].each do |name|
217
+ eval %{
218
+ def #{name} (*val)
219
+ CArray.__new__(#{name}, *val)
220
+ end
221
+ }
222
+ end
223
+
224
+ def CA_FIXLEN (val, options = {})
225
+ CArray.__new_fixlen__(options[:bytes], val)
226
+ end
227
+
228
+ class CArray
229
+
230
+ module DataTypeNewConstructor
231
+ end
232
+
233
+ module DataTypeExtension
234
+ end
235
+
236
+ class Boolean
237
+ extend DataTypeNewConstructor
238
+ extend DataTypeExtension
239
+ TypeSymbol = :boolean
240
+ DataType = CA_BOOLEAN
241
+ end
242
+
243
+ class UInt8
244
+ extend DataTypeNewConstructor
245
+ extend DataTypeExtension
246
+ TypeSymbol = :uint8
247
+ DataType = CA_UINT8
248
+ end
249
+
250
+ class UInt16
251
+ extend DataTypeNewConstructor
252
+ extend DataTypeExtension
253
+ TypeSymbol = :uint16
254
+ DataType = CA_UINT16
255
+ end
256
+
257
+ class UInt32
258
+ extend DataTypeNewConstructor
259
+ extend DataTypeExtension
260
+ TypeSymbol = :uint32
261
+ DataType = CA_UINT32
262
+ end
263
+
264
+ class UInt64
265
+ extend DataTypeNewConstructor
266
+ extend DataTypeExtension
267
+ TypeSymbol = :uint64
268
+ DataType = CA_UINT64
269
+ end
270
+
271
+ class Int8
272
+ extend DataTypeNewConstructor
273
+ extend DataTypeExtension
274
+ TypeSymbol = :int8
275
+ DataType = CA_INT8
276
+ end
277
+
278
+ class Int16
279
+ extend DataTypeNewConstructor
280
+ extend DataTypeExtension
281
+ TypeSymbol = :int16
282
+ DataType = CA_INT16
283
+ end
284
+
285
+ class Int32
286
+ extend DataTypeNewConstructor
287
+ extend DataTypeExtension
288
+ TypeSymbol = :int32
289
+ DataType = CA_INT32
290
+ end
291
+
292
+ class Int64
293
+ extend DataTypeNewConstructor
294
+ extend DataTypeExtension
295
+ TypeSymbol = :int64
296
+ DataType = CA_INT64
297
+ end
298
+
299
+ class Float32
300
+ extend DataTypeNewConstructor
301
+ extend DataTypeExtension
302
+ TypeSymbol = :float32
303
+ DataType = CA_FLOAT32
304
+ end
305
+
306
+ SFloat = Float32
307
+
308
+ class Float64
309
+ extend DataTypeNewConstructor
310
+ extend DataTypeExtension
311
+ TypeSymbol = :float64
312
+ DataType = CA_FLOAT64
313
+ end
314
+
315
+ DFloat = Float64
316
+
317
+ class Complex64
318
+ extend DataTypeNewConstructor
319
+ extend DataTypeExtension
320
+ TypeSymbol = :complex64
321
+ DataType = CA_CMPLX64
322
+ end
323
+
324
+ SComplex = Complex64
325
+
326
+ class Complex128
327
+ extend DataTypeNewConstructor
328
+ extend DataTypeExtension
329
+ TypeSymbol = :complex128
330
+ DataType = CA_CMPLX128
331
+ end
332
+
333
+ DComplex = Complex128
334
+
335
+ class Object
336
+ extend DataTypeNewConstructor
337
+ extend DataTypeExtension
338
+ TypeSymbol = :object
339
+ DataType = CA_OBJECT
340
+ end
341
+
342
+ RObject = Object
343
+
344
+ class Fixlen
345
+ extend DataTypeNewConstructor
346
+ extend DataTypeExtension
347
+ TypeSymbol = :fixlen
348
+ DataType = CA_FIXLEN
349
+ end
350
+
351
+ end
352
+
353
+ class CArray
354
+ extend DataTypeExtension
355
+ TypeSymbol = nil
356
+ DataType = nil
357
+ end
358
+
359
+ class CArray
360
+
361
+ module DataTypeNewConstructor
362
+
363
+ def new (*shape)
364
+ CArray.new(self::DataType, shape)
365
+ end
366
+
367
+ end
368
+
369
+ module DataTypeExtension
370
+
371
+ def guess_data_type_from_values (*values)
372
+ if values.all? {|v| v == true || v == false }
373
+ CA_BOOLEAN
374
+ elsif values.all? { |v| v.is_a?(Integer) }
375
+ CA_INT64
376
+ elsif values.all? { |v| v.is_a?(Float) }
377
+ CA_FLOAT64
378
+ elsif values.all? { |v| v.is_a?(Complex) }
379
+ CA_CMPLX128
380
+ else
381
+ CA_OBJECT
382
+ end
383
+ end
384
+
385
+ private :guess_data_type_from_values
386
+
387
+ def zeros (*shape)
388
+ CArray.new(self::DataType || CA_FLOAT64, shape).zero
389
+ end
390
+
391
+ def ones (*shape)
392
+ CArray.new(self::DataType || CA_FLOAT64 , shape).one
393
+ end
394
+
395
+ def eye (n, m = nil, k = 0)
396
+ m ||= n
397
+ mat = CArray.new(self::DataType || CA_FLOAT64, [n, m])
398
+ start = ( k == 0 ) ? k : ( k > 0 ? k : m - k - 1 )
399
+ mat[[start..-1,m+1]] = 1
400
+ mat
401
+ end
402
+
403
+ def identity (n)
404
+ mat = CArray.new(self::DataType || CA_FLOAT64, [n, n])
405
+ mat[[nil,n+1]] = 1
406
+ mat
407
+ end
408
+
409
+ def linspace (x1, x2, n = 100)
410
+ data_type = self::DataType
411
+ unless data_type
412
+ guess = guess_data_type_from_values(x1, x2)
413
+ guess = CA_FLOAT64 if guess == CA_INT64
414
+ data_type = guess
415
+ end
416
+ CArray.new(data_type, [n]).span(x1..x2)
417
+ end
418
+
419
+ def arange (*args)
420
+ case args.size
421
+ when 3
422
+ start, stop, step = *args
423
+ when 2
424
+ start, stop = *args
425
+ step = 1
426
+ when 1
427
+ start = 0
428
+ stop, = *args
429
+ step = 1
430
+ end
431
+ data_type = self::DataType
432
+ data_type ||= guess_data_type_from_values(start, stop, step)
433
+ CArray.__new__(data_type, start..stop-step, step)
434
+ end
435
+
436
+ def full (shape, fill_value)
437
+ data_type = self::DataType
438
+ data_type ||= guess_data_type_from_values(fill_value)
439
+ shape = [shape] unless shape.is_a?(Array)
440
+ CArray.new(data_type, shape).fill(fill_value)
441
+ end
442
+
443
+ end
444
+
445
+ end
446
+
447
+ class CArray
448
+
449
+ def self.meshgrid (*axes, indexing: "xy", copy: true, sparse: false, &block)
450
+ case indexing
451
+ when "xy"
452
+ ### no operation
453
+ when "ij"
454
+ axes = axes.map{|axis| axis.seq }
455
+ else
456
+ raise ArgumentError, %{indexing option should be one of "xy" and "ij"}
457
+ end
458
+ shape = axes.map(&:size).reverse
459
+ if sparse ### => CAUnboundRepeat
460
+ list = axes.map.with_index do |axis, k|
461
+ extended_shape = (shape.size-1).downto(0).map { |i| ( i == k ) ? nil : :* }
462
+ if copy
463
+ axis[*extended_shape].to_ca
464
+ else
465
+ axis[*extended_shape]
466
+ end
467
+ end
468
+ else ### => CARepeat
469
+ naxes = shape.size
470
+ list = axes.map.with_index do |axis, k|
471
+ extended_shape = shape.dup
472
+ extended_shape[naxes - k - 1] = :%
473
+ if copy
474
+ axis[*extended_shape].to_ca
475
+ else
476
+ axis[*extended_shape]
477
+ end
478
+ end
479
+ end
480
+ return block.call(*list) if block
481
+ return list
482
+ end
483
+
484
+ end