annotate liboctave/dNDArray.cc @ 10329:83fa590b8a09

simplify min/max definitions in liboctave
author Jaroslav Hajek <highegg@gmail.com>
date Wed, 17 Feb 2010 10:40:36 +0100
parents 07ebe522dac2
children 12884915a8e4
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
1 // N-D Array manipulations.
4511
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
2 /*
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
3
8920
eb63fbe60fab update copyright notices
John W. Eaton <jwe@octave.org>
parents: 8777
diff changeset
4 Copyright (C) 1996, 1997, 2003, 2004, 2005, 2006, 2007, 2008,
eb63fbe60fab update copyright notices
John W. Eaton <jwe@octave.org>
parents: 8777
diff changeset
5 2009 John W. Eaton
9601
a9b37bae1802 add a couple of missing copyright statements
Jaroslav Hajek <highegg@gmail.com>
parents: 9578
diff changeset
6 Copyright (C) 2009 VZLU Prague, a.s.
4511
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
7
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
8 This file is part of Octave.
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
9
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
10 Octave is free software; you can redistribute it and/or modify it
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
11 under the terms of the GNU General Public License as published by the
7016
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6989
diff changeset
12 Free Software Foundation; either version 3 of the License, or (at your
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6989
diff changeset
13 option) any later version.
4511
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
14
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
15 Octave is distributed in the hope that it will be useful, but WITHOUT
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
16 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
17 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
18 for more details.
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
19
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
20 You should have received a copy of the GNU General Public License
7016
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6989
diff changeset
21 along with Octave; see the file COPYING. If not, see
93c65f2a5668 [project @ 2007-10-12 06:40:56 by jwe]
jwe
parents: 6989
diff changeset
22 <http://www.gnu.org/licenses/>.
4511
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
23
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
24 */
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
25
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
26 #ifdef HAVE_CONFIG_H
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
27 #include <config.h>
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
28 #endif
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
29
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
30 #include <cfloat>
5164
57077d0ddc8e [project @ 2005-02-25 19:55:24 by jwe]
jwe
parents: 5110
diff changeset
31
4780
5eb5b8aaed8a [project @ 2004-02-17 15:32:13 by jwe]
jwe
parents: 4773
diff changeset
32 #include <vector>
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
33
4588
19bfd295f400 [project @ 2003-11-10 17:18:48 by jwe]
jwe
parents: 4584
diff changeset
34 #include "Array-util.h"
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
35 #include "dNDArray.h"
9523
0ce82753dd72 more configure changes for libraries
John W. Eaton <jwe@octave.org>
parents: 9513
diff changeset
36 #include "f77-fcn.h"
7503
8c32f95c2639 convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
37 #include "functor.h"
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
38 #include "lo-error.h"
4511
24af46b4ce84 [project @ 2003-09-12 16:46:04 by jwe]
jwe
parents:
diff changeset
39 #include "lo-ieee.h"
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
40 #include "lo-mappers.h"
9523
0ce82753dd72 more configure changes for libraries
John W. Eaton <jwe@octave.org>
parents: 9513
diff changeset
41 #include "mx-base.h"
8774
b756ce0002db split implementation and interface in mx-op-defs and MArray-defs
Jaroslav Hajek <highegg@gmail.com>
parents: 8751
diff changeset
42 #include "mx-op-defs.h"
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
43 #include "oct-fftw.h"
9523
0ce82753dd72 more configure changes for libraries
John W. Eaton <jwe@octave.org>
parents: 9513
diff changeset
44 #include "oct-locbuf.h"
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
45
9743
26abff55f6fe optimize bsxfun for common built-in operations
Jaroslav Hajek <highegg@gmail.com>
parents: 9732
diff changeset
46 #include "bsxfun-defs.cc"
26abff55f6fe optimize bsxfun for common built-in operations
Jaroslav Hajek <highegg@gmail.com>
parents: 9732
diff changeset
47
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
48 NDArray::NDArray (const Array<octave_idx_type>& a, bool zero_based,
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
49 bool negative_to_nan)
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
50 {
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
51 const octave_idx_type *pa = a.fortran_vec ();
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
52 resize (a.dims ());
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
53 double *ptmp = fortran_vec ();
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
54 if (negative_to_nan)
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
55 {
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
56 double nan_val = lo_ieee_nan_value ();
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
57
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
58 if (zero_based)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
59 for (octave_idx_type i = 0; i < a.numel (); i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
60 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
61 double val = static_cast<double>
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
62 (pa[i] + static_cast<octave_idx_type> (1));
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
63 if (val <= 0)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
64 ptmp[i] = nan_val;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
65 else
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
66 ptmp[i] = val;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
67 }
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
68 else
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
69 for (octave_idx_type i = 0; i < a.numel (); i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
70 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
71 double val = static_cast<double> (pa[i]);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
72 if (val <= 0)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
73 ptmp[i] = nan_val;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
74 else
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
75 ptmp[i] = val;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
76 }
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
77 }
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
78 else
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
79 {
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
80 if (zero_based)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
81 for (octave_idx_type i = 0; i < a.numel (); i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
82 ptmp[i] = static_cast<double>
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
83 (pa[i] + static_cast<octave_idx_type> (1));
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
84 else
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
85 for (octave_idx_type i = 0; i < a.numel (); i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
86 ptmp[i] = static_cast<double> (pa[i]);
7919
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
87 }
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
88 }
9d080df0c843 new NDArray constructor for ArrayN<octave_idx_type>
David Bateman <dbateman@free.fr>
parents: 7789
diff changeset
89
8956
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
90 NDArray::NDArray (const charNDArray& a)
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
91 : MArrayN<double> (a.dims ())
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
92 {
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
93 octave_idx_type n = a.numel ();
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
94 for (octave_idx_type i = 0; i < n; i++)
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
95 xelem (i) = static_cast<unsigned char> (a(i));
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
96 }
d91fa4b20bbb ensure nonnegative char -> real conversion
Jaroslav Hajek <highegg@gmail.com>
parents: 8920
diff changeset
97
9523
0ce82753dd72 more configure changes for libraries
John W. Eaton <jwe@octave.org>
parents: 9513
diff changeset
98 #if defined (HAVE_FFTW)
7941
f8cab9eeb128 Fix NDArray compilation/export
John W. Eaton <jwe@octave.org>
parents: 7922
diff changeset
99
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
100 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
101 NDArray::fourier (int dim) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
102 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
103 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
104
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
105 if (dim > dv.length () || dim < 0)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
106 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
107
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
108 octave_idx_type stride = 1;
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
109 octave_idx_type n = dv(dim);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
110
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
111 for (int i = 0; i < dim; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
112 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
113
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
114 octave_idx_type howmany = numel () / dv (dim);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
115 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
116 octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv (dim) / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
117 octave_idx_type dist = (stride == 1 ? n : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
118
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
119 const double *in (fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
120 ComplexNDArray retval (dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
121 Complex *out (retval.fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
122
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
123 // Need to be careful here about the distance between fft's
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
124 for (octave_idx_type k = 0; k < nloop; k++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
125 octave_fftw::fft (in + k * stride * n, out + k * stride * n,
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
126 n, howmany, stride, dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
127
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
128 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
129 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
130
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
131 ComplexNDArray
4816
4908c82cd1a1 [project @ 2004-03-04 18:13:46 by jwe]
jwe
parents: 4806
diff changeset
132 NDArray::ifourier (int dim) const
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
133 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
134 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
135
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
136 if (dim > dv.length () || dim < 0)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
137 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
138
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
139 octave_idx_type stride = 1;
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
140 octave_idx_type n = dv(dim);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
141
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
142 for (int i = 0; i < dim; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
143 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
144
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
145 octave_idx_type howmany = numel () / dv (dim);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
146 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
147 octave_idx_type nloop = (stride == 1 ? 1 : numel () / dv (dim) / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
148 octave_idx_type dist = (stride == 1 ? n : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
149
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
150 ComplexNDArray retval (*this);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
151 Complex *out (retval.fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
152
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
153 // Need to be careful here about the distance between fft's
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
154 for (octave_idx_type k = 0; k < nloop; k++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
155 octave_fftw::ifft (out + k * stride * n, out + k * stride * n,
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
156 n, howmany, stride, dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
157
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
158 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
159 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
160
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
161 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
162 NDArray::fourier2d (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
163 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
164 dim_vector dv = dims();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
165 if (dv.length () < 2)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
166 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
167
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
168 dim_vector dv2(dv(0), dv(1));
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
169 const double *in = fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
170 ComplexNDArray retval (dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
171 Complex *out = retval.fortran_vec ();
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
172 octave_idx_type howmany = numel() / dv(0) / dv(1);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
173 octave_idx_type dist = dv(0) * dv(1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
174
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
175 for (octave_idx_type i=0; i < howmany; i++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
176 octave_fftw::fftNd (in + i*dist, out + i*dist, 2, dv2);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
177
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
178 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
179 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
180
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
181 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
182 NDArray::ifourier2d (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
183 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
184 dim_vector dv = dims();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
185 if (dv.length () < 2)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
186 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
187
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
188 dim_vector dv2(dv(0), dv(1));
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
189 ComplexNDArray retval (*this);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
190 Complex *out = retval.fortran_vec ();
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
191 octave_idx_type howmany = numel() / dv(0) / dv(1);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
192 octave_idx_type dist = dv(0) * dv(1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
193
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
194 for (octave_idx_type i=0; i < howmany; i++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
195 octave_fftw::ifftNd (out + i*dist, out + i*dist, 2, dv2);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
196
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
197 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
198 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
199
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
200 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
201 NDArray::fourierNd (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
202 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
203 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
204 int rank = dv.length ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
205
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
206 const double *in (fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
207 ComplexNDArray retval (dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
208 Complex *out (retval.fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
209
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
210 octave_fftw::fftNd (in, out, rank, dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
211
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
212 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
213 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
214
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
215 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
216 NDArray::ifourierNd (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
217 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
218 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
219 int rank = dv.length ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
220
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
221 ComplexNDArray tmp (*this);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
222 Complex *in (tmp.fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
223 ComplexNDArray retval (dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
224 Complex *out (retval.fortran_vec ());
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
225
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
226 octave_fftw::ifftNd (in, out, rank, dv);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
227
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
228 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
229 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
230
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
231 #else
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
232
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
233 extern "C"
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
234 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
235 // Note that the original complex fft routines were not written for
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
236 // double complex arguments. They have been modified by adding an
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
237 // implicit double precision (a-h,o-z) statement at the beginning of
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
238 // each subroutine.
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
239
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
240 F77_RET_T
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
241 F77_FUNC (zffti, ZFFTI) (const octave_idx_type&, Complex*);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
242
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
243 F77_RET_T
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
244 F77_FUNC (zfftf, ZFFTF) (const octave_idx_type&, Complex*, Complex*);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
245
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
246 F77_RET_T
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
247 F77_FUNC (zfftb, ZFFTB) (const octave_idx_type&, Complex*, Complex*);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
248 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
249
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
250 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
251 NDArray::fourier (int dim) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
252 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
253 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
254
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
255 if (dim > dv.length () || dim < 0)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
256 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
257
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
258 ComplexNDArray retval (dv);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
259 octave_idx_type npts = dv(dim);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
260 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
261 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
262 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
263
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
264 OCTAVE_LOCAL_BUFFER (Complex, tmp, npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
265
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
266 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
267
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
268 for (int i = 0; i < dim; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
269 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
270
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
271 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
272 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
273 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
274 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
275
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
276 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
277
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
278 for (octave_idx_type k = 0; k < nloop; k++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
279 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
280 for (octave_idx_type j = 0; j < howmany; j++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
281 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
282 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
283
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
284 for (octave_idx_type i = 0; i < npts; i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
285 tmp[i] = elem((i + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
286
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
287 F77_FUNC (zfftf, ZFFTF) (npts, tmp, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
288
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
289 for (octave_idx_type i = 0; i < npts; i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
290 retval ((i + k*npts)*stride + j*dist) = tmp[i];
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
291 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
292 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
293
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
294 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
295 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
296
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
297 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
298 NDArray::ifourier (int dim) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
299 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
300 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
301
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
302 if (dim > dv.length () || dim < 0)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
303 return ComplexNDArray ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
304
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
305 ComplexNDArray retval (dv);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
306 octave_idx_type npts = dv(dim);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
307 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
308 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
309 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
310
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
311 OCTAVE_LOCAL_BUFFER (Complex, tmp, npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
312
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
313 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
314
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
315 for (int i = 0; i < dim; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
316 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
317
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
318 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
319 howmany = (stride == 1 ? howmany : (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
320 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
321 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
322
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
323 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
324
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
325 for (octave_idx_type k = 0; k < nloop; k++)
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
326 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
327 for (octave_idx_type j = 0; j < howmany; j++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
328 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
329 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
330
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
331 for (octave_idx_type i = 0; i < npts; i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
332 tmp[i] = elem((i + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
333
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
334 F77_FUNC (zfftb, ZFFTB) (npts, tmp, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
335
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
336 for (octave_idx_type i = 0; i < npts; i++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
337 retval ((i + k*npts)*stride + j*dist) = tmp[i] /
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
338 static_cast<double> (npts);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
339 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
340 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
341
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
342 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
343 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
344
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
345 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
346 NDArray::fourier2d (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
347 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
348 dim_vector dv = dims();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
349 dim_vector dv2 (dv(0), dv(1));
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
350 int rank = 2;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
351 ComplexNDArray retval (*this);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
352 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
353
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
354 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
355 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
356 octave_idx_type npts = dv2(i);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
357 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
358 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
359 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
360 Array<Complex> row (npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
361 Complex *prow = row.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
362
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
363 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
364 howmany = (stride == 1 ? howmany :
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
365 (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
366 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
367 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
368
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
369 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
370
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
371 for (octave_idx_type k = 0; k < nloop; k++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
372 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
373 for (octave_idx_type j = 0; j < howmany; j++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
374 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
375 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
376
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
377 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
378 prow[l] = retval ((l + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
379
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
380 F77_FUNC (zfftf, ZFFTF) (npts, prow, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
381
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
382 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
383 retval ((l + k*npts)*stride + j*dist) = prow[l];
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
384 }
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
385 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
386
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
387 stride *= dv2(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
388 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
389
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
390 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
391 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
392
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
393 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
394 NDArray::ifourier2d (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
395 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
396 dim_vector dv = dims();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
397 dim_vector dv2 (dv(0), dv(1));
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
398 int rank = 2;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
399 ComplexNDArray retval (*this);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
400 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
401
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
402 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
403 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
404 octave_idx_type npts = dv2(i);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
405 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
406 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
407 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
408 Array<Complex> row (npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
409 Complex *prow = row.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
410
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
411 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
412 howmany = (stride == 1 ? howmany :
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
413 (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
414 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
415 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
416
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
417 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
418
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
419 for (octave_idx_type k = 0; k < nloop; k++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
420 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
421 for (octave_idx_type j = 0; j < howmany; j++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
422 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
423 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
424
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
425 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
426 prow[l] = retval ((l + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
427
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
428 F77_FUNC (zfftb, ZFFTB) (npts, prow, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
429
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
430 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
431 retval ((l + k*npts)*stride + j*dist) = prow[l] /
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
432 static_cast<double> (npts);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
433 }
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
434 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
435
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
436 stride *= dv2(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
437 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
438
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
439 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
440 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
441
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
442 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
443 NDArray::fourierNd (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
444 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
445 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
446 int rank = dv.length ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
447 ComplexNDArray retval (*this);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
448 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
449
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
450 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
451 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
452 octave_idx_type npts = dv(i);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
453 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
454 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
455 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
456 Array<Complex> row (npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
457 Complex *prow = row.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
458
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
459 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
460 howmany = (stride == 1 ? howmany :
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
461 (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
462 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
463 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
464
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
465 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
466
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
467 for (octave_idx_type k = 0; k < nloop; k++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
468 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
469 for (octave_idx_type j = 0; j < howmany; j++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
470 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
471 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
472
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
473 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
474 prow[l] = retval ((l + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
475
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
476 F77_FUNC (zfftf, ZFFTF) (npts, prow, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
477
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
478 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
479 retval ((l + k*npts)*stride + j*dist) = prow[l];
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
480 }
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
481 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
482
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
483 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
484 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
485
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
486 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
487 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
488
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
489 ComplexNDArray
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
490 NDArray::ifourierNd (void) const
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
491 {
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
492 dim_vector dv = dims ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
493 int rank = dv.length ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
494 ComplexNDArray retval (*this);
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
495 octave_idx_type stride = 1;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
496
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
497 for (int i = 0; i < rank; i++)
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
498 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
499 octave_idx_type npts = dv(i);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
500 octave_idx_type nn = 4*npts+15;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
501 Array<Complex> wsave (nn);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
502 Complex *pwsave = wsave.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
503 Array<Complex> row (npts);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
504 Complex *prow = row.fortran_vec ();
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
505
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
506 octave_idx_type howmany = numel () / npts;
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
507 howmany = (stride == 1 ? howmany :
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
508 (howmany > stride ? stride : howmany));
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
509 octave_idx_type nloop = (stride == 1 ? 1 : numel () / npts / stride);
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
510 octave_idx_type dist = (stride == 1 ? npts : 1);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
511
7789
82be108cc558 First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents: 7620
diff changeset
512 F77_FUNC (zffti, ZFFTI) (npts, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
513
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
514 for (octave_idx_type k = 0; k < nloop; k++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
515 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
516 for (octave_idx_type j = 0; j < howmany; j++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
517 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
518 octave_quit ();
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
519
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
520 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
521 prow[l] = retval ((l + k*npts)*stride + j*dist);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
522
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
523 F77_FUNC (zfftb, ZFFTB) (npts, prow, pwsave);
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
524
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
525 for (octave_idx_type l = 0; l < npts; l++)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
526 retval ((l + k*npts)*stride + j*dist) = prow[l] /
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
527 static_cast<double> (npts);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
528 }
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
529 }
4773
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
530
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
531 stride *= dv(i);
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
532 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
533
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
534 return retval;
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
535 }
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
536
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
537 #endif
ccfbd6047a54 [project @ 2004-02-16 19:02:32 by jwe]
jwe
parents: 4770
diff changeset
538
4543
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
539 // unary operations
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
540
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
541 boolNDArray
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
542 NDArray::operator ! (void) const
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
543 {
9553
0c72d9284087 further bool ops tweaks
Jaroslav Hajek <highegg@gmail.com>
parents: 9551
diff changeset
544 return do_mx_unary_op<boolNDArray, NDArray> (*this, mx_inline_not);
4543
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
545 }
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
546
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
547 bool
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
548 NDArray::any_element_is_negative (bool neg_zero) const
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
549 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
550 octave_idx_type nel = nelem ();
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
551
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
552 if (neg_zero)
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
553 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
554 for (octave_idx_type i = 0; i < nel; i++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
555 if (lo_ieee_signbit (elem (i)))
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
556 return true;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
557 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
558 else
9827
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
559 return mx_inline_any_negative (numel (), data ());
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
560
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
561 return false;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
562 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
563
7922
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
564 bool
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
565 NDArray::any_element_is_nan (void) const
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
566 {
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
567 octave_idx_type nel = nelem ();
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
568
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
569 for (octave_idx_type i = 0; i < nel; i++)
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
570 {
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
571 double val = elem (i);
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
572 if (xisnan (val))
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
573 return true;
7922
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
574 }
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
575
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
576 return false;
935be827eaf8 error for NaN values in & and | expressions
John W. Eaton <jwe@octave.org>
parents: 7919
diff changeset
577 }
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
578
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
579 bool
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
580 NDArray::any_element_is_inf_or_nan (void) const
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
581 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
582 octave_idx_type nel = nelem ();
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
583
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
584 for (octave_idx_type i = 0; i < nel; i++)
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
585 {
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
586 double val = elem (i);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
587 if (xisinf (val) || xisnan (val))
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
588 return true;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
589 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
590
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
591 return false;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
592 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
593
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
594 bool
5943
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
595 NDArray::any_element_not_one_or_zero (void) const
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
596 {
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
597 octave_idx_type nel = nelem ();
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
598
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
599 for (octave_idx_type i = 0; i < nel; i++)
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
600 {
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
601 double val = elem (i);
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
602 if (val != 0 && val != 1)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
603 return true;
5943
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
604 }
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
605
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
606 return false;
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
607 }
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
608
c038c2947ee1 [project @ 2006-08-18 21:27:03 by jwe]
jwe
parents: 5775
diff changeset
609 bool
6989
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
610 NDArray::all_elements_are_zero (void) const
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
611 {
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
612 octave_idx_type nel = nelem ();
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
613
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
614 for (octave_idx_type i = 0; i < nel; i++)
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
615 if (elem (i) != 0)
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
616 return false;
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
617
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
618 return true;
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
619 }
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
620
2d326000e09b [project @ 2007-10-09 20:32:42 by jwe]
jwe
parents: 5943
diff changeset
621 bool
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
622 NDArray::all_elements_are_int_or_inf_or_nan (void) const
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
623 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
624 octave_idx_type nel = nelem ();
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
625
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
626 for (octave_idx_type i = 0; i < nel; i++)
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
627 {
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
628 double val = elem (i);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
629 if (xisnan (val) || D_NINT (val) == val)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
630 continue;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
631 else
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
632 return false;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
633 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
635 return true;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
636 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
637
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
638 // Return nonzero if any element of M is not an integer. Also extract
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
639 // the largest and smallest values and return them in MAX_VAL and MIN_VAL.
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
640
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
641 bool
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
642 NDArray::all_integers (double& max_val, double& min_val) const
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
643 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
644 octave_idx_type nel = nelem ();
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
645
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
646 if (nel > 0)
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
647 {
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
648 max_val = elem (0);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
649 min_val = elem (0);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
650 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
651 else
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
652 return false;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
653
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
654 for (octave_idx_type i = 0; i < nel; i++)
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
655 {
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
656 double val = elem (i);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
657
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
658 if (val > max_val)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
659 max_val = val;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
660
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
661 if (val < min_val)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
662 min_val = val;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
663
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
664 if (D_NINT (val) != val)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
665 return false;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
666 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
667
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
668 return true;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
669 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
670
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
671 bool
9827
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
672 NDArray::all_integers (void) const
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
673 {
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
674 octave_idx_type nel = nelem ();
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
675
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
676 for (octave_idx_type i = 0; i < nel; i++)
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
677 {
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
678 double val = elem (i);
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
679
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
680 if (D_NINT (val) != val)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
681 return false;
9827
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
682 }
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
683
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
684 return true;
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
685 }
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
686
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
687 bool
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
688 NDArray::too_large_for_float (void) const
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
689 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
690 octave_idx_type nel = nelem ();
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
691
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
692 for (octave_idx_type i = 0; i < nel; i++)
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
693 {
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
694 double val = elem (i);
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
695
5389
25c8956d2204 [project @ 2005-06-15 03:45:46 by jwe]
jwe
parents: 5387
diff changeset
696 if (! (xisnan (val) || xisinf (val))
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
697 && fabs (val) > FLT_MAX)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
698 return true;
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
699 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
700
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
701 return false;
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
702 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
703
5775
ace8d8d26933 [project @ 2006-04-24 19:13:06 by jwe]
jwe
parents: 5389
diff changeset
704 // FIXME -- this is not quite the right thing.
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
705
4556
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
706 boolNDArray
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
707 NDArray::all (int dim) const
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
708 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
709 return do_mx_red_op<boolNDArray, double> (*this, dim, mx_inline_all);
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
710 }
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
711
4556
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
712 boolNDArray
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
713 NDArray::any (int dim) const
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
714 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
715 return do_mx_red_op<boolNDArray, double> (*this, dim, mx_inline_any);
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
716 }
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
717
4584
f7697d703481 [project @ 2003-10-31 19:47:35 by jwe]
jwe
parents: 4569
diff changeset
718 NDArray
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
719 NDArray::cumprod (int dim) const
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
720 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
721 return do_mx_cum_op<NDArray, double> (*this, dim, mx_inline_cumprod);
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
722 }
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
723
4584
f7697d703481 [project @ 2003-10-31 19:47:35 by jwe]
jwe
parents: 4569
diff changeset
724 NDArray
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
725 NDArray::cumsum (int dim) const
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
726 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
727 return do_mx_cum_op<NDArray, double> (*this, dim, mx_inline_cumsum);
4513
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
728 }
508238e65af7 [project @ 2003-09-19 21:40:57 by jwe]
jwe
parents: 4511
diff changeset
729
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
730 NDArray
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
731 NDArray::prod (int dim) const
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
732 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
733 return do_mx_red_op<NDArray, double> (*this, dim, mx_inline_prod);
8736
53b4fdeacc2e improve reduction functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8650
diff changeset
734 }
53b4fdeacc2e improve reduction functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8650
diff changeset
735
53b4fdeacc2e improve reduction functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8650
diff changeset
736 NDArray
53b4fdeacc2e improve reduction functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8650
diff changeset
737 NDArray::sum (int dim) const
53b4fdeacc2e improve reduction functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8650
diff changeset
738 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
739 return do_mx_red_op<NDArray, double> (*this, dim, mx_inline_sum);
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
740 }
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
741
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
742 NDArray
9721
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
743 NDArray::xsum (int dim) const
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
744 {
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
745 return do_mx_red_op<NDArray, double> (*this, dim, mx_inline_xsum);
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
746 }
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
747
192d94cff6c1 improve sum & implement the 'extra' option, refactor some code
Jaroslav Hajek <highegg@gmail.com>
parents: 9601
diff changeset
748 NDArray
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
749 NDArray::sumsq (int dim) const
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
750 {
9227
8145f2255276 use explicit template qualifs to please Intel C++ and MSVC++
Jaroslav Hajek <highegg@gmail.com>
parents: 8999
diff changeset
751 return do_mx_red_op<NDArray, double> (*this, dim, mx_inline_sumsq);
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
752 }
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
753
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
754 NDArray
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
755 NDArray::max (int dim) const
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
756 {
8751
9f7ce4bf7650 optimize min/max functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8743
diff changeset
757 return do_mx_minmax_op<NDArray> (*this, dim, mx_inline_max);
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
758 }
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
759
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
760 NDArray
9732
b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops
Jaroslav Hajek <highegg@gmail.com>
parents: 9721
diff changeset
761 NDArray::max (Array<octave_idx_type>& idx_arg, int dim) const
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
762 {
8751
9f7ce4bf7650 optimize min/max functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8743
diff changeset
763 return do_mx_minmax_op<NDArray> (*this, idx_arg, dim, mx_inline_max);
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
764 }
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
765
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
766 NDArray
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
767 NDArray::min (int dim) const
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
768 {
8751
9f7ce4bf7650 optimize min/max functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8743
diff changeset
769 return do_mx_minmax_op<NDArray> (*this, dim, mx_inline_min);
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
770 }
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
771
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
772 NDArray
9732
b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops
Jaroslav Hajek <highegg@gmail.com>
parents: 9721
diff changeset
773 NDArray::min (Array<octave_idx_type>& idx_arg, int dim) const
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
774 {
8751
9f7ce4bf7650 optimize min/max functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8743
diff changeset
775 return do_mx_minmax_op<NDArray> (*this, idx_arg, dim, mx_inline_min);
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
776 }
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
777
4915
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
778 NDArray
8777
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
779 NDArray::cummax (int dim) const
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
780 {
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
781 return do_mx_cumminmax_op<NDArray> (*this, dim, mx_inline_cummax);
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
782 }
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
783
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
784 NDArray
9732
b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops
Jaroslav Hajek <highegg@gmail.com>
parents: 9721
diff changeset
785 NDArray::cummax (Array<octave_idx_type>& idx_arg, int dim) const
8777
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
786 {
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
787 return do_mx_cumminmax_op<NDArray> (*this, idx_arg, dim, mx_inline_cummax);
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
788 }
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
789
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
790 NDArray
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
791 NDArray::cummin (int dim) const
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
792 {
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
793 return do_mx_cumminmax_op<NDArray> (*this, dim, mx_inline_cummin);
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
794 }
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
795
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
796 NDArray
9732
b4fdfee405b5 remove ArrayN<T> + fix nonhom. diag-scalar ops
Jaroslav Hajek <highegg@gmail.com>
parents: 9721
diff changeset
797 NDArray::cummin (Array<octave_idx_type>& idx_arg, int dim) const
8777
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
798 {
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
799 return do_mx_cumminmax_op<NDArray> (*this, idx_arg, dim, mx_inline_cummin);
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
800 }
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
801
724c0f46d9d4 implement cummin/cummax functions
Jaroslav Hajek <highegg@gmail.com>
parents: 8774
diff changeset
802 NDArray
9513
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
803 NDArray::diff (octave_idx_type order, int dim) const
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
804 {
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
805 return do_mx_diff_op<NDArray> (*this, dim, order, mx_inline_diff);
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
806 }
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
807
9f870f73ab7d implement built-in diff
Jaroslav Hajek <highegg@gmail.com>
parents: 9469
diff changeset
808 NDArray
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
809 NDArray::concat (const NDArray& rb, const Array<octave_idx_type>& ra_idx)
4758
a308566c8b42 [project @ 2004-02-13 23:01:34 by jwe]
jwe
parents: 4687
diff changeset
810 {
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
811 if (rb.numel () > 0)
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
812 insert (rb, ra_idx);
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
813 return *this;
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
814 }
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
815
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
816 ComplexNDArray
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
817 NDArray::concat (const ComplexNDArray& rb, const Array<octave_idx_type>& ra_idx)
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
818 {
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
819 ComplexNDArray retval (*this);
4940
954cc2ba6a49 [project @ 2004-08-24 19:30:31 by jwe]
jwe
parents: 4915
diff changeset
820 if (rb.numel () > 0)
4915
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
821 retval.insert (rb, ra_idx);
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
822 return retval;
4758
a308566c8b42 [project @ 2004-02-13 23:01:34 by jwe]
jwe
parents: 4687
diff changeset
823 }
a308566c8b42 [project @ 2004-02-13 23:01:34 by jwe]
jwe
parents: 4687
diff changeset
824
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
825 charNDArray
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
826 NDArray::concat (const charNDArray& rb, const Array<octave_idx_type>& ra_idx)
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
827 {
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
828 charNDArray retval (dims ());
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
829 octave_idx_type nel = numel ();
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
830
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
831 for (octave_idx_type i = 0; i < nel; i++)
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
832 {
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
833 double d = elem (i);
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
834
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
835 if (xisnan (d))
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
836 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
837 (*current_liboctave_error_handler)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
838 ("invalid conversion from NaN to character");
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
839 return retval;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
840 }
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
841 else
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
842 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
843 octave_idx_type ival = NINTbig (d);
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
844
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
845 if (ival < 0 || ival > UCHAR_MAX)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
846 // FIXME -- is there something
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
847 // better we could do? Should we warn the user?
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
848 ival = 0;
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
849
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
850 retval.elem (i) = static_cast<char>(ival);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
851 }
5073
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
852 }
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
853
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
854 if (rb.numel () == 0)
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
855 return retval;
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
856
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
857 retval.insert (rb, ra_idx);
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
858 return retval;
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
859 }
e71be9c548f2 [project @ 2004-11-09 18:31:25 by jwe]
jwe
parents: 4940
diff changeset
860
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
861 NDArray
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
862 real (const ComplexNDArray& a)
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
863 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
864 return do_mx_unary_op<NDArray, ComplexNDArray> (a, mx_inline_real);
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
865 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
866
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
867 NDArray
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
868 imag (const ComplexNDArray& a)
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
869 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
870 return do_mx_unary_op<NDArray, ComplexNDArray> (a, mx_inline_imag);
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
871 }
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
872
4915
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
873 NDArray&
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
874 NDArray::insert (const NDArray& a, octave_idx_type r, octave_idx_type c)
4915
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
875 {
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
876 Array<double>::insert (a, r, c);
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
877 return *this;
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
878 }
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
879
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
880 NDArray&
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
881 NDArray::insert (const NDArray& a, const Array<octave_idx_type>& ra_idx)
4915
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
882 {
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
883 Array<double>::insert (a, ra_idx);
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
884 return *this;
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
885 }
c638c144d4da [project @ 2004-07-23 19:01:22 by jwe]
jwe
parents: 4902
diff changeset
886
4634
79fe96966ca0 [project @ 2003-11-19 21:22:39 by jwe]
jwe
parents: 4588
diff changeset
887 NDArray
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
888 NDArray::abs (void) const
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
889 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
890 return do_mx_unary_map<NDArray, NDArray, std::abs> (*this);
4569
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
891 }
01e4957409a4 [project @ 2003-10-30 05:48:17 by jwe]
jwe
parents: 4563
diff changeset
892
8998
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
893 boolNDArray
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
894 NDArray::isnan (void) const
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
895 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
896 return do_mx_unary_map<boolNDArray, NDArray, xisnan> (*this);
8998
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
897 }
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
898
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
899 boolNDArray
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
900 NDArray::isinf (void) const
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
901 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
902 return do_mx_unary_map<boolNDArray, NDArray, xisinf> (*this);
8998
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
903 }
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
904
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
905 boolNDArray
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
906 NDArray::isfinite (void) const
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
907 {
9800
ef4c4186cb47 improve some mx_inline loops
Jaroslav Hajek <highegg@gmail.com>
parents: 9743
diff changeset
908 return do_mx_unary_map<boolNDArray, NDArray, xfinite> (*this);
8998
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
909 }
a48fba01e4ac optimize isnan/isinf/isfinite mappers
Jaroslav Hajek <highegg@gmail.com>
parents: 8981
diff changeset
910
4532
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
911 Matrix
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
912 NDArray::matrix_value (void) const
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
913 {
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
914 Matrix retval;
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
915
8981
ed5055b0a476 fix & simplify ndarray->matrix conversions
Jaroslav Hajek <highegg@gmail.com>
parents: 8956
diff changeset
916 if (ndims () == 2)
ed5055b0a476 fix & simplify ndarray->matrix conversions
Jaroslav Hajek <highegg@gmail.com>
parents: 8956
diff changeset
917 retval = Matrix (Array2<double> (*this));
ed5055b0a476 fix & simplify ndarray->matrix conversions
Jaroslav Hajek <highegg@gmail.com>
parents: 8956
diff changeset
918 else
ed5055b0a476 fix & simplify ndarray->matrix conversions
Jaroslav Hajek <highegg@gmail.com>
parents: 8956
diff changeset
919 (*current_liboctave_error_handler)
ed5055b0a476 fix & simplify ndarray->matrix conversions
Jaroslav Hajek <highegg@gmail.com>
parents: 8956
diff changeset
920 ("invalid conversion of NDArray to Matrix");
4532
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
921
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
922 return retval;
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
923 }
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
924
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
925 void
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
926 NDArray::increment_index (Array<octave_idx_type>& ra_idx,
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
927 const dim_vector& dimensions,
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
928 int start_dimension)
4532
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
929 {
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
930 ::increment_index (ra_idx, dimensions, start_dimension);
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
931 }
01ee68d18069 [project @ 2003-10-10 00:49:54 by jwe]
jwe
parents: 4513
diff changeset
932
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
933 octave_idx_type
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
934 NDArray::compute_index (Array<octave_idx_type>& ra_idx,
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
935 const dim_vector& dimensions)
4556
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
936 {
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
937 return ::compute_index (ra_idx, dimensions);
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
938 }
773a21e4fce8 [project @ 2003-10-28 05:02:10 by jwe]
jwe
parents: 4543
diff changeset
939
7503
8c32f95c2639 convert mapper functions to new format
David Bateman <dbateman@free.fr>
parents: 7017
diff changeset
940 NDArray
7620
36594d5bbe13 Move diag function into the octave_value class
David Bateman <dbateman@free.fr>
parents: 7600
diff changeset
941 NDArray::diag (octave_idx_type k) const
36594d5bbe13 Move diag function into the octave_value class
David Bateman <dbateman@free.fr>
parents: 7600
diff changeset
942 {
36594d5bbe13 Move diag function into the octave_value class
David Bateman <dbateman@free.fr>
parents: 7600
diff changeset
943 return MArrayN<double>::diag (k);
36594d5bbe13 Move diag function into the octave_value class
David Bateman <dbateman@free.fr>
parents: 7600
diff changeset
944 }
36594d5bbe13 Move diag function into the octave_value class
David Bateman <dbateman@free.fr>
parents: 7600
diff changeset
945
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
946 // This contains no information on the array structure !!!
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
947 std::ostream&
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
948 operator << (std::ostream& os, const NDArray& a)
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
949 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
950 octave_idx_type nel = a.nelem ();
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
951
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
952 for (octave_idx_type i = 0; i < nel; i++)
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
953 {
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
954 os << " ";
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
955 octave_write_double (os, a.elem (i));
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
956 os << "\n";
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
957 }
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
958 return os;
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
959 }
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
960
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
961 std::istream&
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
962 operator >> (std::istream& is, NDArray& a)
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
963 {
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
964 octave_idx_type nel = a.nelem ();
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
965
8999
dc07bc4157b8 allow empty matrices in stream input operators
Jaroslav Hajek <highegg@gmail.com>
parents: 8998
diff changeset
966 if (nel > 0)
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
967 {
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
968 double tmp;
5275
23b37da9fd5b [project @ 2005-04-08 16:07:35 by jwe]
jwe
parents: 5164
diff changeset
969 for (octave_idx_type i = 0; i < nel; i++)
10314
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
970 {
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
971 tmp = octave_read_value<double> (is);
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
972 if (is)
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
973 a.elem (i) = tmp;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
974 else
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
975 goto done;
07ebe522dac2 untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents: 10158
diff changeset
976 }
4687
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
977 }
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
978
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
979 done:
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
980
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
981 return is;
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
982 }
e95c86d48732 [project @ 2004-01-06 21:53:34 by jwe]
jwe
parents: 4634
diff changeset
983
10329
83fa590b8a09 simplify min/max definitions in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents: 10314
diff changeset
984 MINMAX_FCNS (NDArray, double)
4844
9f7ef92b50b0 [project @ 2004-04-02 17:26:53 by jwe]
jwe
parents: 4816
diff changeset
985
9578
7dafdb8b062f refactor comparison ops implementations
Jaroslav Hajek <highegg@gmail.com>
parents: 9553
diff changeset
986 NDS_CMP_OPS (NDArray, double)
9550
3d6a9aea2aea refactor binary & bool ops in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents: 9523
diff changeset
987 NDS_BOOL_OPS (NDArray, double)
4543
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
988
9578
7dafdb8b062f refactor comparison ops implementations
Jaroslav Hajek <highegg@gmail.com>
parents: 9553
diff changeset
989 SND_CMP_OPS (double, NDArray)
9550
3d6a9aea2aea refactor binary & bool ops in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents: 9523
diff changeset
990 SND_BOOL_OPS (double, NDArray)
4543
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
991
9578
7dafdb8b062f refactor comparison ops implementations
Jaroslav Hajek <highegg@gmail.com>
parents: 9553
diff changeset
992 NDND_CMP_OPS (NDArray, NDArray)
9550
3d6a9aea2aea refactor binary & bool ops in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents: 9523
diff changeset
993 NDND_BOOL_OPS (NDArray, NDArray)
4543
79df15d4470c [project @ 2003-10-18 03:53:52 by jwe]
jwe
parents: 4532
diff changeset
994
9743
26abff55f6fe optimize bsxfun for common built-in operations
Jaroslav Hajek <highegg@gmail.com>
parents: 9732
diff changeset
995 BSXFUN_STDOP_DEFS_MXLOOP (NDArray)
26abff55f6fe optimize bsxfun for common built-in operations
Jaroslav Hajek <highegg@gmail.com>
parents: 9732
diff changeset
996 BSXFUN_STDREL_DEFS_MXLOOP (NDArray)
26abff55f6fe optimize bsxfun for common built-in operations
Jaroslav Hajek <highegg@gmail.com>
parents: 9732
diff changeset
997
9827
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
998 BSXFUN_OP_DEF_MXLOOP (pow, NDArray, mx_inline_pow)
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
999 BSXFUN_OP2_DEF_MXLOOP (pow, ComplexNDArray, ComplexNDArray,
c15a5ed0da58 optimize bsxfun (@power, ...)
Jaroslav Hajek <highegg@gmail.com>
parents: 9812
diff changeset
1000 NDArray, mx_inline_pow)