|
4902
|
1 // N-D Array manipulations. |
|
|
2 /* |
|
|
3 |
|
|
4 Copyright (C) 2004 John W. Eaton |
|
|
5 |
|
|
6 This file is part of Octave. |
|
|
7 |
|
|
8 Octave is free software; you can redistribute it and/or modify it |
|
|
9 under the terms of the GNU General Public License as published by the |
|
|
10 Free Software Foundation; either version 2, or (at your option) any |
|
|
11 later version. |
|
|
12 |
|
|
13 Octave is distributed in the hope that it will be useful, but WITHOUT |
|
|
14 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
|
15 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
|
16 for more details. |
|
|
17 |
|
|
18 You should have received a copy of the GNU General Public License |
|
|
19 along with Octave; see the file COPYING. If not, write to the Free |
|
5307
|
20 Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
|
|
21 02110-1301, USA. |
|
4902
|
22 |
|
|
23 */ |
|
|
24 |
|
|
25 #ifdef HAVE_CONFIG_H |
|
|
26 #include <config.h> |
|
|
27 #endif |
|
|
28 |
|
|
29 #include "Array-util.h" |
|
|
30 #include "mx-base.h" |
|
|
31 #include "lo-ieee.h" |
|
|
32 |
|
|
33 // unary operations |
|
|
34 |
|
|
35 template <class T> |
|
|
36 boolNDArray |
|
|
37 intNDArray<T>::operator ! (void) const |
|
|
38 { |
|
4932
|
39 boolNDArray b (this->dims ()); |
|
4902
|
40 |
|
5275
|
41 for (octave_idx_type i = 0; i < this->length (); i++) |
|
4932
|
42 b.elem (i) = ! this->elem (i); |
|
4902
|
43 |
|
|
44 return b; |
|
|
45 } |
|
|
46 |
|
5943
|
47 template <class T> |
|
|
48 bool |
|
|
49 intNDArray<T>::any_element_not_one_or_zero (void) const |
|
|
50 { |
|
|
51 octave_idx_type nel = this->nelem (); |
|
|
52 |
|
|
53 for (octave_idx_type i = 0; i < nel; i++) |
|
|
54 { |
|
|
55 T val = this->elem (i); |
|
|
56 |
|
|
57 if (val != 0.0 && val != 1.0) |
|
|
58 return true; |
|
|
59 } |
|
|
60 |
|
|
61 return false; |
|
|
62 } |
|
|
63 |
|
6979
|
64 |
|
|
65 template <class T> |
|
|
66 intNDArray<T> |
|
|
67 intNDArray<T>::diag (void) const |
|
|
68 { |
|
|
69 return diag (0); |
|
|
70 } |
|
|
71 |
|
|
72 template <class T> |
|
|
73 intNDArray<T> |
|
|
74 intNDArray<T>::diag (octave_idx_type k) const |
|
|
75 { |
|
|
76 dim_vector dv = this->dims (); |
|
|
77 octave_idx_type nd = dv.length (); |
|
|
78 |
|
|
79 if (nd > 2) |
|
|
80 { |
|
|
81 (*current_liboctave_error_handler) ("Matrix must be 2-dimensional"); |
|
|
82 return intNDArray<T>(); |
|
|
83 } |
|
|
84 else |
|
|
85 { |
|
|
86 octave_idx_type nnr = dv (0); |
|
|
87 octave_idx_type nnc = dv (1); |
|
|
88 |
|
|
89 if (k > 0) |
|
|
90 nnc -= k; |
|
|
91 else if (k < 0) |
|
|
92 nnr += k; |
|
|
93 |
|
|
94 intNDArray<T> d; |
|
|
95 |
|
|
96 if (nnr > 0 && nnc > 0) |
|
|
97 { |
|
|
98 octave_idx_type ndiag = (nnr < nnc) ? nnr : nnc; |
|
|
99 |
|
|
100 d.resize (dim_vector (ndiag, 1)); |
|
|
101 |
|
|
102 if (k > 0) |
|
|
103 { |
|
|
104 for (octave_idx_type i = 0; i < ndiag; i++) |
|
|
105 d.xelem (i) = this->elem (i, i+k); |
|
|
106 } |
|
|
107 else if (k < 0) |
|
|
108 { |
|
|
109 for (octave_idx_type i = 0; i < ndiag; i++) |
|
|
110 d.xelem (i) = this->elem (i-k, i); |
|
|
111 } |
|
|
112 else |
|
|
113 { |
|
|
114 for (octave_idx_type i = 0; i < ndiag; i++) |
|
|
115 d.xelem (i) = this->elem (i, i); |
|
|
116 } |
|
|
117 } |
|
|
118 else |
|
|
119 (*current_liboctave_error_handler) |
|
|
120 ("diag: requested diagonal out of range"); |
|
|
121 |
|
|
122 return d; |
|
|
123 } |
|
|
124 } |
|
|
125 |
|
5775
|
126 // FIXME -- this is not quite the right thing. |
|
4902
|
127 |
|
|
128 template <class T> |
|
|
129 boolNDArray |
|
|
130 intNDArray<T>::all (int dim) const |
|
|
131 { |
|
4932
|
132 MX_ND_ANY_ALL_REDUCTION (MX_ND_ALL_EVAL (this->elem (iter_idx) == T (0)), true); |
|
4902
|
133 } |
|
|
134 |
|
|
135 template <class T> |
|
|
136 boolNDArray |
|
|
137 intNDArray<T>::any (int dim) const |
|
|
138 { |
|
4932
|
139 MX_ND_ANY_ALL_REDUCTION (MX_ND_ALL_EVAL (this->elem (iter_idx) == T (0)), false); |
|
4902
|
140 } |
|
|
141 |
|
|
142 template <class T> |
|
|
143 void |
|
5275
|
144 intNDArray<T>::increment_index (Array<octave_idx_type>& ra_idx, |
|
4902
|
145 const dim_vector& dimensions, |
|
|
146 int start_dimension) |
|
|
147 { |
|
|
148 ::increment_index (ra_idx, dimensions, start_dimension); |
|
|
149 } |
|
|
150 |
|
|
151 template <class T> |
|
5275
|
152 octave_idx_type |
|
|
153 intNDArray<T>::compute_index (Array<octave_idx_type>& ra_idx, |
|
4902
|
154 const dim_vector& dimensions) |
|
|
155 { |
|
|
156 return ::compute_index (ra_idx, dimensions); |
|
|
157 } |
|
|
158 |
|
4915
|
159 template <class T> |
|
5073
|
160 intNDArray<T> |
|
5275
|
161 intNDArray<T>::concat (const intNDArray<T>& rb, const Array<octave_idx_type>& ra_idx) |
|
5073
|
162 { |
|
6482
|
163 if (rb.numel () > 0) |
|
5073
|
164 insert (rb, ra_idx); |
|
|
165 return *this; |
|
|
166 } |
|
|
167 |
|
|
168 template <class T> |
|
4915
|
169 intNDArray<T>& |
|
5275
|
170 intNDArray<T>::insert (const intNDArray<T>& a, octave_idx_type r, octave_idx_type c) |
|
4915
|
171 { |
|
|
172 Array<T>::insert (a, r, c); |
|
|
173 return *this; |
|
|
174 } |
|
|
175 |
|
|
176 template <class T> |
|
|
177 intNDArray<T>& |
|
5275
|
178 intNDArray<T>::insert (const intNDArray<T>& a, const Array<octave_idx_type>& ra_idx) |
|
4915
|
179 { |
|
|
180 Array<T>::insert (a, ra_idx); |
|
|
181 return *this; |
|
|
182 } |
|
|
183 |
|
4902
|
184 // This contains no information on the array structure !!! |
|
|
185 |
|
|
186 template <class T> |
|
|
187 std::ostream& |
|
|
188 operator << (std::ostream& os, const intNDArray<T>& a) |
|
|
189 { |
|
5275
|
190 octave_idx_type nel = a.nelem (); |
|
4902
|
191 |
|
5275
|
192 for (octave_idx_type i = 0; i < nel; i++) |
|
4902
|
193 os << " " << a.elem (i) << "\n"; |
|
|
194 |
|
|
195 return os; |
|
|
196 } |
|
|
197 |
|
|
198 template <class T> |
|
|
199 std::istream& |
|
|
200 operator >> (std::istream& is, intNDArray<T>& a) |
|
|
201 { |
|
5275
|
202 octave_idx_type nel = a.nelem (); |
|
4902
|
203 |
|
|
204 if (nel < 1 ) |
|
|
205 is.clear (std::ios::badbit); |
|
|
206 else |
|
|
207 { |
|
|
208 T tmp; |
|
|
209 |
|
5275
|
210 for (octave_idx_type i = 0; i < nel; i++) |
|
4902
|
211 { |
|
|
212 is >> tmp; |
|
|
213 |
|
|
214 if (is) |
|
|
215 a.elem (i) = tmp; |
|
|
216 else |
|
|
217 goto done; |
|
|
218 } |
|
|
219 } |
|
|
220 |
|
|
221 done: |
|
|
222 |
|
|
223 return is; |
|
|
224 } |
|
|
225 |
|
|
226 /* |
|
|
227 ;;; Local Variables: *** |
|
|
228 ;;; mode: C++ *** |
|
|
229 ;;; End: *** |
|
|
230 */ |
