4513
|
1 // N-D Array manipulations. |
4511
|
2 /* |
|
3 |
7017
|
4 Copyright (C) 1996, 1997, 2003, 2004, 2005, 2006, 2007 John W. Eaton |
4511
|
5 |
|
6 This file is part of Octave. |
|
7 |
|
8 Octave is free software; you can redistribute it and/or modify it |
|
9 under the terms of the GNU General Public License as published by the |
7016
|
10 Free Software Foundation; either version 3 of the License, or (at your |
|
11 option) any later version. |
4511
|
12 |
|
13 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
16 for more details. |
|
17 |
|
18 You should have received a copy of the GNU General Public License |
7016
|
19 along with Octave; see the file COPYING. If not, see |
|
20 <http://www.gnu.org/licenses/>. |
4511
|
21 |
|
22 */ |
|
23 |
|
24 #ifdef HAVE_CONFIG_H |
|
25 #include <config.h> |
|
26 #endif |
|
27 |
4634
|
28 #include <cfloat> |
5164
|
29 |
4780
|
30 #include <vector> |
4634
|
31 |
4588
|
32 #include "Array-util.h" |
4513
|
33 #include "dNDArray.h" |
4511
|
34 #include "mx-base.h" |
4773
|
35 #include "f77-fcn.h" |
4513
|
36 #include "lo-error.h" |
4511
|
37 #include "lo-ieee.h" |
4634
|
38 #include "lo-mappers.h" |
4511
|
39 |
4773
|
40 #if defined (HAVE_FFTW3) |
|
41 #include "oct-fftw.h" |
|
42 |
|
43 ComplexNDArray |
|
44 NDArray::fourier (int dim) const |
|
45 { |
|
46 dim_vector dv = dims (); |
|
47 |
|
48 if (dim > dv.length () || dim < 0) |
|
49 return ComplexNDArray (); |
|
50 |
5275
|
51 octave_idx_type stride = 1; |
|
52 octave_idx_type n = dv(dim); |
4773
|
53 |
|
54 for (int i = 0; i < dim; i++) |
|
55 stride *= dv(i); |
|
56 |
5275
|
57 octave_idx_type howmany = numel () / dv (dim); |
4773
|
58 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); |
5275
|
59 octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv (dim) / stride); |
|
60 octave_idx_type dist = (stride == 1 ? n : 1); |
4773
|
61 |
|
62 const double *in (fortran_vec ()); |
|
63 ComplexNDArray retval (dv); |
|
64 Complex *out (retval.fortran_vec ()); |
|
65 |
|
66 // Need to be careful here about the distance between fft's |
5275
|
67 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
68 octave_fftw::fft (in + k * stride * n, out + k * stride * n, |
|
69 n, howmany, stride, dist); |
|
70 |
|
71 return retval; |
|
72 } |
|
73 |
|
74 ComplexNDArray |
4816
|
75 NDArray::ifourier (int dim) const |
4773
|
76 { |
|
77 dim_vector dv = dims (); |
|
78 |
|
79 if (dim > dv.length () || dim < 0) |
|
80 return ComplexNDArray (); |
|
81 |
5275
|
82 octave_idx_type stride = 1; |
|
83 octave_idx_type n = dv(dim); |
4773
|
84 |
|
85 for (int i = 0; i < dim; i++) |
|
86 stride *= dv(i); |
|
87 |
5275
|
88 octave_idx_type howmany = numel () / dv (dim); |
4773
|
89 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); |
5275
|
90 octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv (dim) / stride); |
|
91 octave_idx_type dist = (stride == 1 ? n : 1); |
4773
|
92 |
|
93 ComplexNDArray retval (*this); |
|
94 Complex *out (retval.fortran_vec ()); |
|
95 |
|
96 // Need to be careful here about the distance between fft's |
5275
|
97 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
98 octave_fftw::ifft (out + k * stride * n, out + k * stride * n, |
|
99 n, howmany, stride, dist); |
|
100 |
|
101 return retval; |
|
102 } |
|
103 |
|
104 ComplexNDArray |
|
105 NDArray::fourier2d (void) const |
|
106 { |
|
107 dim_vector dv = dims(); |
|
108 if (dv.length () < 2) |
|
109 return ComplexNDArray (); |
|
110 |
|
111 dim_vector dv2(dv(0), dv(1)); |
|
112 const double *in = fortran_vec (); |
|
113 ComplexNDArray retval (dv); |
|
114 Complex *out = retval.fortran_vec (); |
5275
|
115 octave_idx_type howmany = numel() / dv(0) / dv(1); |
|
116 octave_idx_type dist = dv(0) * dv(1); |
4773
|
117 |
5275
|
118 for (octave_idx_type i=0; i < howmany; i++) |
4773
|
119 octave_fftw::fftNd (in + i*dist, out + i*dist, 2, dv2); |
|
120 |
|
121 return retval; |
|
122 } |
|
123 |
|
124 ComplexNDArray |
|
125 NDArray::ifourier2d (void) const |
|
126 { |
|
127 dim_vector dv = dims(); |
|
128 if (dv.length () < 2) |
|
129 return ComplexNDArray (); |
|
130 |
|
131 dim_vector dv2(dv(0), dv(1)); |
|
132 ComplexNDArray retval (*this); |
|
133 Complex *out = retval.fortran_vec (); |
5275
|
134 octave_idx_type howmany = numel() / dv(0) / dv(1); |
|
135 octave_idx_type dist = dv(0) * dv(1); |
4773
|
136 |
5275
|
137 for (octave_idx_type i=0; i < howmany; i++) |
4773
|
138 octave_fftw::ifftNd (out + i*dist, out + i*dist, 2, dv2); |
|
139 |
|
140 return retval; |
|
141 } |
|
142 |
|
143 ComplexNDArray |
|
144 NDArray::fourierNd (void) const |
|
145 { |
|
146 dim_vector dv = dims (); |
|
147 int rank = dv.length (); |
|
148 |
|
149 const double *in (fortran_vec ()); |
|
150 ComplexNDArray retval (dv); |
|
151 Complex *out (retval.fortran_vec ()); |
|
152 |
|
153 octave_fftw::fftNd (in, out, rank, dv); |
|
154 |
|
155 return retval; |
|
156 } |
|
157 |
|
158 ComplexNDArray |
|
159 NDArray::ifourierNd (void) const |
|
160 { |
|
161 dim_vector dv = dims (); |
|
162 int rank = dv.length (); |
|
163 |
|
164 ComplexNDArray tmp (*this); |
|
165 Complex *in (tmp.fortran_vec ()); |
|
166 ComplexNDArray retval (dv); |
|
167 Complex *out (retval.fortran_vec ()); |
|
168 |
|
169 octave_fftw::ifftNd (in, out, rank, dv); |
|
170 |
|
171 return retval; |
|
172 } |
|
173 |
|
174 #else |
|
175 |
|
176 extern "C" |
|
177 { |
|
178 // Note that the original complex fft routines were not written for |
|
179 // double complex arguments. They have been modified by adding an |
|
180 // implicit double precision (a-h,o-z) statement at the beginning of |
|
181 // each subroutine. |
|
182 |
|
183 F77_RET_T |
5275
|
184 F77_FUNC (cffti, CFFTI) (const octave_idx_type&, Complex*); |
4773
|
185 |
|
186 F77_RET_T |
5275
|
187 F77_FUNC (cfftf, CFFTF) (const octave_idx_type&, Complex*, Complex*); |
4773
|
188 |
|
189 F77_RET_T |
5275
|
190 F77_FUNC (cfftb, CFFTB) (const octave_idx_type&, Complex*, Complex*); |
4773
|
191 } |
|
192 |
|
193 ComplexNDArray |
|
194 NDArray::fourier (int dim) const |
|
195 { |
|
196 dim_vector dv = dims (); |
|
197 |
|
198 if (dim > dv.length () || dim < 0) |
|
199 return ComplexNDArray (); |
|
200 |
|
201 ComplexNDArray retval (dv); |
5275
|
202 octave_idx_type npts = dv(dim); |
|
203 octave_idx_type nn = 4*npts+15; |
4773
|
204 Array<Complex> wsave (nn); |
|
205 Complex *pwsave = wsave.fortran_vec (); |
|
206 |
|
207 OCTAVE_LOCAL_BUFFER (Complex, tmp, npts); |
|
208 |
5275
|
209 octave_idx_type stride = 1; |
4773
|
210 |
|
211 for (int i = 0; i < dim; i++) |
|
212 stride *= dv(i); |
|
213 |
5275
|
214 octave_idx_type howmany = numel () / npts; |
4773
|
215 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); |
5275
|
216 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
217 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
218 |
|
219 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
220 |
5275
|
221 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
222 { |
5275
|
223 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
224 { |
|
225 OCTAVE_QUIT; |
|
226 |
5275
|
227 for (octave_idx_type i = 0; i < npts; i++) |
4773
|
228 tmp[i] = elem((i + k*npts)*stride + j*dist); |
|
229 |
|
230 F77_FUNC (cfftf, CFFTF) (npts, tmp, pwsave); |
|
231 |
5275
|
232 for (octave_idx_type i = 0; i < npts; i++) |
4773
|
233 retval ((i + k*npts)*stride + j*dist) = tmp[i]; |
|
234 } |
|
235 } |
|
236 |
|
237 return retval; |
|
238 } |
|
239 |
|
240 ComplexNDArray |
|
241 NDArray::ifourier (int dim) const |
|
242 { |
|
243 dim_vector dv = dims (); |
|
244 |
|
245 if (dim > dv.length () || dim < 0) |
|
246 return ComplexNDArray (); |
|
247 |
|
248 ComplexNDArray retval (dv); |
5275
|
249 octave_idx_type npts = dv(dim); |
|
250 octave_idx_type nn = 4*npts+15; |
4773
|
251 Array<Complex> wsave (nn); |
|
252 Complex *pwsave = wsave.fortran_vec (); |
|
253 |
|
254 OCTAVE_LOCAL_BUFFER (Complex, tmp, npts); |
|
255 |
5275
|
256 octave_idx_type stride = 1; |
4773
|
257 |
|
258 for (int i = 0; i < dim; i++) |
|
259 stride *= dv(i); |
|
260 |
5275
|
261 octave_idx_type howmany = numel () / npts; |
4773
|
262 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany)); |
5275
|
263 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
264 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
265 |
|
266 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
267 |
5275
|
268 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
269 { |
5275
|
270 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
271 { |
|
272 OCTAVE_QUIT; |
|
273 |
5275
|
274 for (octave_idx_type i = 0; i < npts; i++) |
4773
|
275 tmp[i] = elem((i + k*npts)*stride + j*dist); |
|
276 |
|
277 F77_FUNC (cfftb, CFFTB) (npts, tmp, pwsave); |
|
278 |
5275
|
279 for (octave_idx_type i = 0; i < npts; i++) |
4773
|
280 retval ((i + k*npts)*stride + j*dist) = tmp[i] / |
|
281 static_cast<double> (npts); |
|
282 } |
|
283 } |
|
284 |
|
285 return retval; |
|
286 } |
|
287 |
|
288 ComplexNDArray |
|
289 NDArray::fourier2d (void) const |
|
290 { |
|
291 dim_vector dv = dims(); |
|
292 dim_vector dv2 (dv(0), dv(1)); |
|
293 int rank = 2; |
|
294 ComplexNDArray retval (*this); |
5275
|
295 octave_idx_type stride = 1; |
4773
|
296 |
|
297 for (int i = 0; i < rank; i++) |
|
298 { |
5275
|
299 octave_idx_type npts = dv2(i); |
|
300 octave_idx_type nn = 4*npts+15; |
4773
|
301 Array<Complex> wsave (nn); |
|
302 Complex *pwsave = wsave.fortran_vec (); |
|
303 Array<Complex> row (npts); |
|
304 Complex *prow = row.fortran_vec (); |
|
305 |
5275
|
306 octave_idx_type howmany = numel () / npts; |
4773
|
307 howmany = (stride == 1 ? howmany : |
|
308 (howmany > stride ? stride : howmany)); |
5275
|
309 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
310 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
311 |
|
312 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
313 |
5275
|
314 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
315 { |
5275
|
316 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
317 { |
|
318 OCTAVE_QUIT; |
|
319 |
5275
|
320 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
321 prow[l] = retval ((l + k*npts)*stride + j*dist); |
|
322 |
|
323 F77_FUNC (cfftf, CFFTF) (npts, prow, pwsave); |
|
324 |
5275
|
325 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
326 retval ((l + k*npts)*stride + j*dist) = prow[l]; |
|
327 } |
|
328 } |
|
329 |
|
330 stride *= dv2(i); |
|
331 } |
|
332 |
|
333 return retval; |
|
334 } |
|
335 |
|
336 ComplexNDArray |
|
337 NDArray::ifourier2d (void) const |
|
338 { |
|
339 dim_vector dv = dims(); |
|
340 dim_vector dv2 (dv(0), dv(1)); |
|
341 int rank = 2; |
|
342 ComplexNDArray retval (*this); |
5275
|
343 octave_idx_type stride = 1; |
4773
|
344 |
|
345 for (int i = 0; i < rank; i++) |
|
346 { |
5275
|
347 octave_idx_type npts = dv2(i); |
|
348 octave_idx_type nn = 4*npts+15; |
4773
|
349 Array<Complex> wsave (nn); |
|
350 Complex *pwsave = wsave.fortran_vec (); |
|
351 Array<Complex> row (npts); |
|
352 Complex *prow = row.fortran_vec (); |
|
353 |
5275
|
354 octave_idx_type howmany = numel () / npts; |
4773
|
355 howmany = (stride == 1 ? howmany : |
|
356 (howmany > stride ? stride : howmany)); |
5275
|
357 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
358 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
359 |
|
360 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
361 |
5275
|
362 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
363 { |
5275
|
364 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
365 { |
|
366 OCTAVE_QUIT; |
|
367 |
5275
|
368 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
369 prow[l] = retval ((l + k*npts)*stride + j*dist); |
|
370 |
|
371 F77_FUNC (cfftb, CFFTB) (npts, prow, pwsave); |
|
372 |
5275
|
373 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
374 retval ((l + k*npts)*stride + j*dist) = prow[l] / |
|
375 static_cast<double> (npts); |
|
376 } |
|
377 } |
|
378 |
|
379 stride *= dv2(i); |
|
380 } |
|
381 |
|
382 return retval; |
|
383 } |
|
384 |
|
385 ComplexNDArray |
|
386 NDArray::fourierNd (void) const |
|
387 { |
|
388 dim_vector dv = dims (); |
|
389 int rank = dv.length (); |
|
390 ComplexNDArray retval (*this); |
5275
|
391 octave_idx_type stride = 1; |
4773
|
392 |
|
393 for (int i = 0; i < rank; i++) |
|
394 { |
5275
|
395 octave_idx_type npts = dv(i); |
|
396 octave_idx_type nn = 4*npts+15; |
4773
|
397 Array<Complex> wsave (nn); |
|
398 Complex *pwsave = wsave.fortran_vec (); |
|
399 Array<Complex> row (npts); |
|
400 Complex *prow = row.fortran_vec (); |
|
401 |
5275
|
402 octave_idx_type howmany = numel () / npts; |
4773
|
403 howmany = (stride == 1 ? howmany : |
|
404 (howmany > stride ? stride : howmany)); |
5275
|
405 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
406 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
407 |
|
408 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
409 |
5275
|
410 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
411 { |
5275
|
412 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
413 { |
|
414 OCTAVE_QUIT; |
|
415 |
5275
|
416 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
417 prow[l] = retval ((l + k*npts)*stride + j*dist); |
|
418 |
|
419 F77_FUNC (cfftf, CFFTF) (npts, prow, pwsave); |
|
420 |
5275
|
421 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
422 retval ((l + k*npts)*stride + j*dist) = prow[l]; |
|
423 } |
|
424 } |
|
425 |
|
426 stride *= dv(i); |
|
427 } |
|
428 |
|
429 return retval; |
|
430 } |
|
431 |
|
432 ComplexNDArray |
|
433 NDArray::ifourierNd (void) const |
|
434 { |
|
435 dim_vector dv = dims (); |
|
436 int rank = dv.length (); |
|
437 ComplexNDArray retval (*this); |
5275
|
438 octave_idx_type stride = 1; |
4773
|
439 |
|
440 for (int i = 0; i < rank; i++) |
|
441 { |
5275
|
442 octave_idx_type npts = dv(i); |
|
443 octave_idx_type nn = 4*npts+15; |
4773
|
444 Array<Complex> wsave (nn); |
|
445 Complex *pwsave = wsave.fortran_vec (); |
|
446 Array<Complex> row (npts); |
|
447 Complex *prow = row.fortran_vec (); |
|
448 |
5275
|
449 octave_idx_type howmany = numel () / npts; |
4773
|
450 howmany = (stride == 1 ? howmany : |
|
451 (howmany > stride ? stride : howmany)); |
5275
|
452 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride); |
|
453 octave_idx_type dist = (stride == 1 ? npts : 1); |
4773
|
454 |
|
455 F77_FUNC (cffti, CFFTI) (npts, pwsave); |
|
456 |
5275
|
457 for (octave_idx_type k = 0; k < nloop; k++) |
4773
|
458 { |
5275
|
459 for (octave_idx_type j = 0; j < howmany; j++) |
4773
|
460 { |
|
461 OCTAVE_QUIT; |
|
462 |
5275
|
463 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
464 prow[l] = retval ((l + k*npts)*stride + j*dist); |
|
465 |
|
466 F77_FUNC (cfftb, CFFTB) (npts, prow, pwsave); |
|
467 |
5275
|
468 for (octave_idx_type l = 0; l < npts; l++) |
4773
|
469 retval ((l + k*npts)*stride + j*dist) = prow[l] / |
|
470 static_cast<double> (npts); |
|
471 } |
|
472 } |
|
473 |
|
474 stride *= dv(i); |
|
475 } |
|
476 |
|
477 return retval; |
|
478 } |
|
479 |
|
480 #endif |
|
481 |
4543
|
482 // unary operations |
|
483 |
|
484 boolNDArray |
|
485 NDArray::operator ! (void) const |
|
486 { |
|
487 boolNDArray b (dims ()); |
|
488 |
5275
|
489 for (octave_idx_type i = 0; i < length (); i++) |
4543
|
490 b.elem (i) = ! elem (i); |
|
491 |
|
492 return b; |
|
493 } |
|
494 |
4634
|
495 bool |
|
496 NDArray::any_element_is_negative (bool neg_zero) const |
|
497 { |
5275
|
498 octave_idx_type nel = nelem (); |
4634
|
499 |
|
500 if (neg_zero) |
|
501 { |
5275
|
502 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
503 if (lo_ieee_signbit (elem (i))) |
|
504 return true; |
|
505 } |
|
506 else |
|
507 { |
5275
|
508 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
509 if (elem (i) < 0) |
|
510 return true; |
|
511 } |
|
512 |
|
513 return false; |
|
514 } |
|
515 |
|
516 |
|
517 bool |
|
518 NDArray::any_element_is_inf_or_nan (void) const |
|
519 { |
5275
|
520 octave_idx_type nel = nelem (); |
4634
|
521 |
5275
|
522 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
523 { |
|
524 double val = elem (i); |
|
525 if (xisinf (val) || xisnan (val)) |
|
526 return true; |
|
527 } |
|
528 |
|
529 return false; |
|
530 } |
|
531 |
|
532 bool |
5943
|
533 NDArray::any_element_not_one_or_zero (void) const |
|
534 { |
|
535 octave_idx_type nel = nelem (); |
|
536 |
|
537 for (octave_idx_type i = 0; i < nel; i++) |
|
538 { |
|
539 double val = elem (i); |
|
540 if (val != 0 && val != 1) |
|
541 return true; |
|
542 } |
|
543 |
|
544 return false; |
|
545 } |
|
546 |
|
547 bool |
6989
|
548 NDArray::all_elements_are_zero (void) const |
|
549 { |
|
550 octave_idx_type nel = nelem (); |
|
551 |
|
552 for (octave_idx_type i = 0; i < nel; i++) |
|
553 if (elem (i) != 0) |
|
554 return false; |
|
555 |
|
556 return true; |
|
557 } |
|
558 |
|
559 bool |
4634
|
560 NDArray::all_elements_are_int_or_inf_or_nan (void) const |
|
561 { |
5275
|
562 octave_idx_type nel = nelem (); |
4634
|
563 |
5275
|
564 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
565 { |
|
566 double val = elem (i); |
|
567 if (xisnan (val) || D_NINT (val) == val) |
|
568 continue; |
|
569 else |
|
570 return false; |
|
571 } |
|
572 |
|
573 return true; |
|
574 } |
|
575 |
|
576 // Return nonzero if any element of M is not an integer. Also extract |
|
577 // the largest and smallest values and return them in MAX_VAL and MIN_VAL. |
|
578 |
|
579 bool |
|
580 NDArray::all_integers (double& max_val, double& min_val) const |
|
581 { |
5275
|
582 octave_idx_type nel = nelem (); |
4634
|
583 |
|
584 if (nel > 0) |
|
585 { |
|
586 max_val = elem (0); |
|
587 min_val = elem (0); |
|
588 } |
|
589 else |
|
590 return false; |
|
591 |
5275
|
592 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
593 { |
|
594 double val = elem (i); |
|
595 |
|
596 if (val > max_val) |
|
597 max_val = val; |
|
598 |
|
599 if (val < min_val) |
|
600 min_val = val; |
|
601 |
|
602 if (D_NINT (val) != val) |
|
603 return false; |
|
604 } |
|
605 |
|
606 return true; |
|
607 } |
|
608 |
|
609 bool |
|
610 NDArray::too_large_for_float (void) const |
|
611 { |
5275
|
612 octave_idx_type nel = nelem (); |
4634
|
613 |
5275
|
614 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
615 { |
|
616 double val = elem (i); |
|
617 |
5389
|
618 if (! (xisnan (val) || xisinf (val)) |
5387
|
619 && fabs (val) > FLT_MAX) |
4634
|
620 return true; |
|
621 } |
|
622 |
|
623 return false; |
|
624 } |
|
625 |
5775
|
626 // FIXME -- this is not quite the right thing. |
4513
|
627 |
4556
|
628 boolNDArray |
4513
|
629 NDArray::all (int dim) const |
|
630 { |
4569
|
631 MX_ND_ANY_ALL_REDUCTION (MX_ND_ALL_EVAL (MX_ND_ALL_EXPR), true); |
4513
|
632 } |
|
633 |
4556
|
634 boolNDArray |
4513
|
635 NDArray::any (int dim) const |
|
636 { |
5110
|
637 MX_ND_ANY_ALL_REDUCTION |
|
638 (MX_ND_ANY_EVAL (elem (iter_idx) != 0 |
|
639 && ! lo_ieee_isnan (elem (iter_idx))), false); |
4569
|
640 } |
|
641 |
4584
|
642 NDArray |
4569
|
643 NDArray::cumprod (int dim) const |
|
644 { |
4584
|
645 MX_ND_CUMULATIVE_OP (NDArray, double, 1, *); |
4569
|
646 } |
|
647 |
4584
|
648 NDArray |
4569
|
649 NDArray::cumsum (int dim) const |
|
650 { |
4584
|
651 MX_ND_CUMULATIVE_OP (NDArray, double, 0, +); |
4513
|
652 } |
|
653 |
4569
|
654 NDArray |
|
655 NDArray::prod (int dim) const |
|
656 { |
|
657 MX_ND_REAL_OP_REDUCTION (*= elem (iter_idx), 1); |
|
658 } |
|
659 |
|
660 NDArray |
|
661 NDArray::sumsq (int dim) const |
|
662 { |
|
663 MX_ND_REAL_OP_REDUCTION (+= std::pow (elem (iter_idx), 2), 0); |
|
664 } |
|
665 |
|
666 NDArray |
|
667 NDArray::sum (int dim) const |
|
668 { |
|
669 MX_ND_REAL_OP_REDUCTION (+= elem (iter_idx), 0); |
|
670 } |
|
671 |
4844
|
672 NDArray |
|
673 NDArray::max (int dim) const |
|
674 { |
5275
|
675 ArrayN<octave_idx_type> dummy_idx; |
4844
|
676 return max (dummy_idx, dim); |
|
677 } |
|
678 |
|
679 NDArray |
5275
|
680 NDArray::max (ArrayN<octave_idx_type>& idx_arg, int dim) const |
4844
|
681 { |
|
682 dim_vector dv = dims (); |
|
683 dim_vector dr = dims (); |
|
684 |
|
685 if (dv.numel () == 0 || dim > dv.length () || dim < 0) |
|
686 return NDArray (); |
|
687 |
|
688 dr(dim) = 1; |
|
689 |
|
690 NDArray result (dr); |
|
691 idx_arg.resize (dr); |
|
692 |
5275
|
693 octave_idx_type x_stride = 1; |
|
694 octave_idx_type x_len = dv(dim); |
4844
|
695 for (int i = 0; i < dim; i++) |
|
696 x_stride *= dv(i); |
|
697 |
5275
|
698 for (octave_idx_type i = 0; i < dr.numel (); i++) |
4844
|
699 { |
5275
|
700 octave_idx_type x_offset; |
4844
|
701 if (x_stride == 1) |
|
702 x_offset = i * x_len; |
|
703 else |
|
704 { |
5275
|
705 octave_idx_type x_offset2 = 0; |
4844
|
706 x_offset = i; |
|
707 while (x_offset >= x_stride) |
|
708 { |
|
709 x_offset -= x_stride; |
|
710 x_offset2++; |
|
711 } |
|
712 x_offset += x_offset2 * x_stride * x_len; |
|
713 } |
|
714 |
5275
|
715 octave_idx_type idx_j; |
4844
|
716 |
|
717 double tmp_max = octave_NaN; |
|
718 |
|
719 for (idx_j = 0; idx_j < x_len; idx_j++) |
|
720 { |
|
721 tmp_max = elem (idx_j * x_stride + x_offset); |
|
722 |
5389
|
723 if (! xisnan (tmp_max)) |
4844
|
724 break; |
|
725 } |
|
726 |
5275
|
727 for (octave_idx_type j = idx_j+1; j < x_len; j++) |
4844
|
728 { |
|
729 double tmp = elem (j * x_stride + x_offset); |
|
730 |
5389
|
731 if (xisnan (tmp)) |
4844
|
732 continue; |
|
733 else if (tmp > tmp_max) |
|
734 { |
|
735 idx_j = j; |
|
736 tmp_max = tmp; |
|
737 } |
|
738 } |
|
739 |
|
740 result.elem (i) = tmp_max; |
5389
|
741 idx_arg.elem (i) = xisnan (tmp_max) ? 0 : idx_j; |
4844
|
742 } |
|
743 |
|
744 return result; |
|
745 } |
|
746 |
|
747 NDArray |
|
748 NDArray::min (int dim) const |
|
749 { |
5275
|
750 ArrayN<octave_idx_type> dummy_idx; |
4844
|
751 return min (dummy_idx, dim); |
|
752 } |
|
753 |
|
754 NDArray |
5275
|
755 NDArray::min (ArrayN<octave_idx_type>& idx_arg, int dim) const |
4844
|
756 { |
|
757 dim_vector dv = dims (); |
|
758 dim_vector dr = dims (); |
|
759 |
|
760 if (dv.numel () == 0 || dim > dv.length () || dim < 0) |
|
761 return NDArray (); |
|
762 |
|
763 dr(dim) = 1; |
|
764 |
|
765 NDArray result (dr); |
|
766 idx_arg.resize (dr); |
|
767 |
5275
|
768 octave_idx_type x_stride = 1; |
|
769 octave_idx_type x_len = dv(dim); |
4844
|
770 for (int i = 0; i < dim; i++) |
|
771 x_stride *= dv(i); |
|
772 |
5275
|
773 for (octave_idx_type i = 0; i < dr.numel (); i++) |
4844
|
774 { |
5275
|
775 octave_idx_type x_offset; |
4844
|
776 if (x_stride == 1) |
|
777 x_offset = i * x_len; |
|
778 else |
|
779 { |
5275
|
780 octave_idx_type x_offset2 = 0; |
4844
|
781 x_offset = i; |
|
782 while (x_offset >= x_stride) |
|
783 { |
|
784 x_offset -= x_stride; |
|
785 x_offset2++; |
|
786 } |
|
787 x_offset += x_offset2 * x_stride * x_len; |
|
788 } |
|
789 |
5275
|
790 octave_idx_type idx_j; |
4844
|
791 |
|
792 double tmp_min = octave_NaN; |
|
793 |
|
794 for (idx_j = 0; idx_j < x_len; idx_j++) |
|
795 { |
|
796 tmp_min = elem (idx_j * x_stride + x_offset); |
|
797 |
5389
|
798 if (! xisnan (tmp_min)) |
4844
|
799 break; |
|
800 } |
|
801 |
5275
|
802 for (octave_idx_type j = idx_j+1; j < x_len; j++) |
4844
|
803 { |
|
804 double tmp = elem (j * x_stride + x_offset); |
|
805 |
5389
|
806 if (xisnan (tmp)) |
4844
|
807 continue; |
|
808 else if (tmp < tmp_min) |
|
809 { |
|
810 idx_j = j; |
|
811 tmp_min = tmp; |
|
812 } |
|
813 } |
|
814 |
|
815 result.elem (i) = tmp_min; |
5389
|
816 idx_arg.elem (i) = xisnan (tmp_min) ? 0 : idx_j; |
4844
|
817 } |
|
818 |
|
819 return result; |
|
820 } |
|
821 |
4915
|
822 NDArray |
5275
|
823 NDArray::concat (const NDArray& rb, const Array<octave_idx_type>& ra_idx) |
4758
|
824 { |
5073
|
825 if (rb.numel () > 0) |
|
826 insert (rb, ra_idx); |
|
827 return *this; |
|
828 } |
|
829 |
|
830 ComplexNDArray |
5275
|
831 NDArray::concat (const ComplexNDArray& rb, const Array<octave_idx_type>& ra_idx) |
5073
|
832 { |
|
833 ComplexNDArray retval (*this); |
4940
|
834 if (rb.numel () > 0) |
4915
|
835 retval.insert (rb, ra_idx); |
|
836 return retval; |
4758
|
837 } |
|
838 |
5073
|
839 charNDArray |
5275
|
840 NDArray::concat (const charNDArray& rb, const Array<octave_idx_type>& ra_idx) |
5073
|
841 { |
|
842 charNDArray retval (dims ()); |
5275
|
843 octave_idx_type nel = numel (); |
5073
|
844 |
5275
|
845 for (octave_idx_type i = 0; i < nel; i++) |
5073
|
846 { |
|
847 double d = elem (i); |
|
848 |
|
849 if (xisnan (d)) |
|
850 { |
|
851 (*current_liboctave_error_handler) |
|
852 ("invalid conversion from NaN to character"); |
|
853 return retval; |
|
854 } |
|
855 else |
|
856 { |
5275
|
857 octave_idx_type ival = NINTbig (d); |
5073
|
858 |
|
859 if (ival < 0 || ival > UCHAR_MAX) |
5775
|
860 // FIXME -- is there something |
5073
|
861 // better we could do? Should we warn the user? |
|
862 ival = 0; |
|
863 |
|
864 retval.elem (i) = static_cast<char>(ival); |
|
865 } |
|
866 } |
|
867 |
|
868 if (rb.numel () == 0) |
|
869 return retval; |
|
870 |
|
871 retval.insert (rb, ra_idx); |
|
872 return retval; |
|
873 } |
|
874 |
4634
|
875 NDArray |
|
876 real (const ComplexNDArray& a) |
|
877 { |
5275
|
878 octave_idx_type a_len = a.length (); |
4634
|
879 NDArray retval; |
|
880 if (a_len > 0) |
|
881 retval = NDArray (mx_inline_real_dup (a.data (), a_len), a.dims ()); |
|
882 return retval; |
|
883 } |
|
884 |
|
885 NDArray |
|
886 imag (const ComplexNDArray& a) |
|
887 { |
5275
|
888 octave_idx_type a_len = a.length (); |
4634
|
889 NDArray retval; |
|
890 if (a_len > 0) |
|
891 retval = NDArray (mx_inline_imag_dup (a.data (), a_len), a.dims ()); |
|
892 return retval; |
|
893 } |
|
894 |
4915
|
895 NDArray& |
5275
|
896 NDArray::insert (const NDArray& a, octave_idx_type r, octave_idx_type c) |
4915
|
897 { |
|
898 Array<double>::insert (a, r, c); |
|
899 return *this; |
|
900 } |
|
901 |
|
902 NDArray& |
5275
|
903 NDArray::insert (const NDArray& a, const Array<octave_idx_type>& ra_idx) |
4915
|
904 { |
|
905 Array<double>::insert (a, ra_idx); |
|
906 return *this; |
|
907 } |
|
908 |
4634
|
909 NDArray |
4569
|
910 NDArray::abs (void) const |
|
911 { |
4634
|
912 NDArray retval (dims ()); |
4569
|
913 |
5275
|
914 octave_idx_type nel = nelem (); |
4634
|
915 |
5275
|
916 for (octave_idx_type i = 0; i < nel; i++) |
4634
|
917 retval(i) = fabs (elem (i)); |
4569
|
918 |
|
919 return retval; |
|
920 } |
|
921 |
4532
|
922 Matrix |
|
923 NDArray::matrix_value (void) const |
|
924 { |
|
925 Matrix retval; |
|
926 |
|
927 int nd = ndims (); |
|
928 |
|
929 switch (nd) |
|
930 { |
|
931 case 1: |
|
932 retval = Matrix (Array2<double> (*this, dimensions(0), 1)); |
|
933 break; |
|
934 |
|
935 case 2: |
|
936 retval = Matrix (Array2<double> (*this, dimensions(0), dimensions(1))); |
|
937 break; |
|
938 |
|
939 default: |
|
940 (*current_liboctave_error_handler) |
4770
|
941 ("invalid conversion of NDArray to Matrix"); |
4532
|
942 break; |
|
943 } |
|
944 |
|
945 return retval; |
|
946 } |
|
947 |
|
948 void |
5275
|
949 NDArray::increment_index (Array<octave_idx_type>& ra_idx, |
4532
|
950 const dim_vector& dimensions, |
|
951 int start_dimension) |
|
952 { |
|
953 ::increment_index (ra_idx, dimensions, start_dimension); |
|
954 } |
|
955 |
5275
|
956 octave_idx_type |
|
957 NDArray::compute_index (Array<octave_idx_type>& ra_idx, |
4556
|
958 const dim_vector& dimensions) |
|
959 { |
|
960 return ::compute_index (ra_idx, dimensions); |
|
961 } |
|
962 |
4687
|
963 // This contains no information on the array structure !!! |
|
964 std::ostream& |
|
965 operator << (std::ostream& os, const NDArray& a) |
|
966 { |
5275
|
967 octave_idx_type nel = a.nelem (); |
4687
|
968 |
5275
|
969 for (octave_idx_type i = 0; i < nel; i++) |
4687
|
970 { |
|
971 os << " "; |
|
972 octave_write_double (os, a.elem (i)); |
|
973 os << "\n"; |
|
974 } |
|
975 return os; |
|
976 } |
|
977 |
|
978 std::istream& |
|
979 operator >> (std::istream& is, NDArray& a) |
|
980 { |
5275
|
981 octave_idx_type nel = a.nelem (); |
4687
|
982 |
|
983 if (nel < 1 ) |
|
984 is.clear (std::ios::badbit); |
|
985 else |
|
986 { |
|
987 double tmp; |
5275
|
988 for (octave_idx_type i = 0; i < nel; i++) |
4687
|
989 { |
|
990 tmp = octave_read_double (is); |
|
991 if (is) |
|
992 a.elem (i) = tmp; |
|
993 else |
|
994 goto done; |
|
995 } |
|
996 } |
|
997 |
|
998 done: |
|
999 |
|
1000 return is; |
|
1001 } |
|
1002 |
5775
|
1003 // FIXME -- it would be nice to share code among the min/max |
4844
|
1004 // functions below. |
|
1005 |
|
1006 #define EMPTY_RETURN_CHECK(T) \ |
|
1007 if (nel == 0) \ |
|
1008 return T (dv); |
|
1009 |
|
1010 NDArray |
|
1011 min (double d, const NDArray& m) |
|
1012 { |
|
1013 dim_vector dv = m.dims (); |
5275
|
1014 octave_idx_type nel = dv.numel (); |
4844
|
1015 |
|
1016 EMPTY_RETURN_CHECK (NDArray); |
|
1017 |
|
1018 NDArray result (dv); |
|
1019 |
5275
|
1020 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1021 { |
|
1022 OCTAVE_QUIT; |
|
1023 result (i) = xmin (d, m (i)); |
|
1024 } |
|
1025 |
|
1026 return result; |
|
1027 } |
|
1028 |
|
1029 NDArray |
|
1030 min (const NDArray& m, double d) |
|
1031 { |
|
1032 dim_vector dv = m.dims (); |
5275
|
1033 octave_idx_type nel = dv.numel (); |
4844
|
1034 |
|
1035 EMPTY_RETURN_CHECK (NDArray); |
|
1036 |
|
1037 NDArray result (dv); |
|
1038 |
5275
|
1039 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1040 { |
|
1041 OCTAVE_QUIT; |
|
1042 result (i) = xmin (d, m (i)); |
|
1043 } |
|
1044 |
|
1045 return result; |
|
1046 } |
|
1047 |
|
1048 NDArray |
|
1049 min (const NDArray& a, const NDArray& b) |
|
1050 { |
|
1051 dim_vector dv = a.dims (); |
5275
|
1052 octave_idx_type nel = dv.numel (); |
4844
|
1053 |
|
1054 if (dv != b.dims ()) |
|
1055 { |
|
1056 (*current_liboctave_error_handler) |
|
1057 ("two-arg min expecting args of same size"); |
|
1058 return NDArray (); |
|
1059 } |
|
1060 |
|
1061 EMPTY_RETURN_CHECK (NDArray); |
|
1062 |
|
1063 NDArray result (dv); |
|
1064 |
5275
|
1065 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1066 { |
|
1067 OCTAVE_QUIT; |
|
1068 result (i) = xmin (a (i), b (i)); |
|
1069 } |
|
1070 |
|
1071 return result; |
|
1072 } |
|
1073 |
|
1074 NDArray |
|
1075 max (double d, const NDArray& m) |
|
1076 { |
|
1077 dim_vector dv = m.dims (); |
5275
|
1078 octave_idx_type nel = dv.numel (); |
4844
|
1079 |
|
1080 EMPTY_RETURN_CHECK (NDArray); |
|
1081 |
|
1082 NDArray result (dv); |
|
1083 |
5275
|
1084 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1085 { |
|
1086 OCTAVE_QUIT; |
|
1087 result (i) = xmax (d, m (i)); |
|
1088 } |
|
1089 |
|
1090 return result; |
|
1091 } |
|
1092 |
|
1093 NDArray |
|
1094 max (const NDArray& m, double d) |
|
1095 { |
|
1096 dim_vector dv = m.dims (); |
5275
|
1097 octave_idx_type nel = dv.numel (); |
4844
|
1098 |
|
1099 EMPTY_RETURN_CHECK (NDArray); |
|
1100 |
|
1101 NDArray result (dv); |
|
1102 |
5275
|
1103 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1104 { |
|
1105 OCTAVE_QUIT; |
|
1106 result (i) = xmax (d, m (i)); |
|
1107 } |
|
1108 |
|
1109 return result; |
|
1110 } |
|
1111 |
|
1112 NDArray |
|
1113 max (const NDArray& a, const NDArray& b) |
|
1114 { |
|
1115 dim_vector dv = a.dims (); |
5275
|
1116 octave_idx_type nel = dv.numel (); |
4844
|
1117 |
|
1118 if (dv != b.dims ()) |
|
1119 { |
|
1120 (*current_liboctave_error_handler) |
|
1121 ("two-arg max expecting args of same size"); |
|
1122 return NDArray (); |
|
1123 } |
|
1124 |
|
1125 EMPTY_RETURN_CHECK (NDArray); |
|
1126 |
|
1127 NDArray result (dv); |
|
1128 |
5275
|
1129 for (octave_idx_type i = 0; i < nel; i++) |
4844
|
1130 { |
|
1131 OCTAVE_QUIT; |
|
1132 result (i) = xmax (a (i), b (i)); |
|
1133 } |
|
1134 |
|
1135 return result; |
|
1136 } |
|
1137 |
4543
|
1138 NDS_CMP_OPS(NDArray, , double, ) |
|
1139 NDS_BOOL_OPS(NDArray, double, 0.0) |
|
1140 |
|
1141 SND_CMP_OPS(double, , NDArray, ) |
|
1142 SND_BOOL_OPS(double, NDArray, 0.0) |
|
1143 |
|
1144 NDND_CMP_OPS(NDArray, , NDArray, ) |
|
1145 NDND_BOOL_OPS(NDArray, NDArray, 0.0) |
|
1146 |
4513
|
1147 /* |
|
1148 ;;; Local Variables: *** |
|
1149 ;;; mode: C++ *** |
|
1150 ;;; End: *** |
|
1151 */ |