|
458
|
1 // -*- C++ -*- |
|
|
2 /* |
|
|
3 |
|
|
4 Copyright (C) 1992, 1993, 1994 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 |
|
|
20 Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA. |
|
|
21 |
|
|
22 */ |
|
|
23 |
|
|
24 #if !defined (octave_ComplexMatrix_h) |
|
|
25 #define octave_ComplexMatrix_h 1 |
|
|
26 |
|
|
27 #if defined (__GNUG__) |
|
|
28 #pragma interface |
|
|
29 #endif |
|
|
30 |
|
|
31 #include <Complex.h> |
|
|
32 |
|
|
33 #include "Array.h" |
|
|
34 |
|
|
35 #include "mx-defs.h" |
|
|
36 |
|
|
37 extern "C++" { |
|
|
38 |
|
|
39 class ComplexMatrix : public Array2<Complex> |
|
|
40 { |
|
|
41 friend class ComplexLU; |
|
|
42 friend class ComplexSVD; |
|
|
43 |
|
|
44 public: |
|
|
45 |
|
|
46 ComplexMatrix (void) : Array2<Complex> () { } |
|
|
47 ComplexMatrix (int r, int c) : Array2<Complex> (r, c) { } |
|
|
48 ComplexMatrix (int r, int c, const Complex& val) |
|
|
49 : Array2<Complex> (r, c, val) { } |
|
|
50 ComplexMatrix (const Matrix& a); |
|
|
51 ComplexMatrix (const Array2<Complex>& a) : Array2<Complex> (a) { } |
|
|
52 ComplexMatrix (const ComplexMatrix& a) : Array2<Complex> (a) { } |
|
|
53 ComplexMatrix (const DiagMatrix& a); |
|
|
54 ComplexMatrix (const DiagArray<Complex>& a) : Array2<Complex> (a) { } |
|
|
55 ComplexMatrix (const ComplexDiagMatrix& a); |
|
|
56 // ComplexMatrix (double a) : Array2<Complex> (1, 1, a) { } |
|
|
57 // ComplexMatrix (const Complex& a) : Array2<Complex> (1, 1, a) { } |
|
|
58 |
|
|
59 ComplexMatrix& operator = (const ComplexMatrix& a) |
|
|
60 { |
|
|
61 Array2<Complex>::operator = (a); |
|
|
62 return *this; |
|
|
63 } |
|
|
64 |
|
|
65 // operator Array2<Complex>& () const { return *this; } |
|
|
66 |
|
|
67 int operator == (const ComplexMatrix& a) const; |
|
|
68 int operator != (const ComplexMatrix& a) const; |
|
|
69 |
|
|
70 // destructive insert/delete/reorder operations |
|
|
71 |
|
|
72 ComplexMatrix& insert (const Matrix& a, int r, int c); |
|
|
73 ComplexMatrix& insert (const RowVector& a, int r, int c); |
|
|
74 ComplexMatrix& insert (const ColumnVector& a, int r, int c); |
|
|
75 ComplexMatrix& insert (const DiagMatrix& a, int r, int c); |
|
|
76 |
|
|
77 ComplexMatrix& insert (const ComplexMatrix& a, int r, int c); |
|
|
78 ComplexMatrix& insert (const ComplexRowVector& a, int r, int c); |
|
|
79 ComplexMatrix& insert (const ComplexColumnVector& a, int r, int c); |
|
|
80 ComplexMatrix& insert (const ComplexDiagMatrix& a, int r, int c); |
|
|
81 |
|
|
82 ComplexMatrix& fill (double val); |
|
|
83 ComplexMatrix& fill (const Complex& val); |
|
|
84 ComplexMatrix& fill (double val, int r1, int c1, int r2, int c2); |
|
|
85 ComplexMatrix& fill (const Complex& val, int r1, int c1, int r2, int c2); |
|
|
86 |
|
|
87 ComplexMatrix append (const Matrix& a) const; |
|
|
88 ComplexMatrix append (const RowVector& a) const; |
|
|
89 ComplexMatrix append (const ColumnVector& a) const; |
|
|
90 ComplexMatrix append (const DiagMatrix& a) const; |
|
|
91 |
|
|
92 ComplexMatrix append (const ComplexMatrix& a) const; |
|
|
93 ComplexMatrix append (const ComplexRowVector& a) const; |
|
|
94 ComplexMatrix append (const ComplexColumnVector& a) const; |
|
|
95 ComplexMatrix append (const ComplexDiagMatrix& a) const; |
|
|
96 |
|
|
97 ComplexMatrix stack (const Matrix& a) const; |
|
|
98 ComplexMatrix stack (const RowVector& a) const; |
|
|
99 ComplexMatrix stack (const ColumnVector& a) const; |
|
|
100 ComplexMatrix stack (const DiagMatrix& a) const; |
|
|
101 |
|
|
102 ComplexMatrix stack (const ComplexMatrix& a) const; |
|
|
103 ComplexMatrix stack (const ComplexRowVector& a) const; |
|
|
104 ComplexMatrix stack (const ComplexColumnVector& a) const; |
|
|
105 ComplexMatrix stack (const ComplexDiagMatrix& a) const; |
|
|
106 |
|
|
107 ComplexMatrix hermitian (void) const; // complex conjugate transpose |
|
|
108 ComplexMatrix transpose (void) const; |
|
|
109 |
|
|
110 friend Matrix real (const ComplexMatrix& a); |
|
|
111 friend Matrix imag (const ComplexMatrix& a); |
|
|
112 friend ComplexMatrix conj (const ComplexMatrix& a); |
|
|
113 |
|
|
114 // resize is the destructive equivalent for this one |
|
|
115 |
|
|
116 ComplexMatrix extract (int r1, int c1, int r2, int c2) const; |
|
|
117 |
|
|
118 // extract row or column i. |
|
|
119 |
|
|
120 ComplexRowVector row (int i) const; |
|
|
121 ComplexRowVector row (char *s) const; |
|
|
122 |
|
|
123 ComplexColumnVector column (int i) const; |
|
|
124 ComplexColumnVector column (char *s) const; |
|
|
125 |
|
|
126 ComplexMatrix inverse (void) const; |
|
|
127 ComplexMatrix inverse (int& info) const; |
|
|
128 ComplexMatrix inverse (int& info, double& rcond) const; |
|
|
129 |
|
|
130 ComplexMatrix fourier (void) const; |
|
|
131 ComplexMatrix ifourier (void) const; |
|
|
132 |
|
|
133 ComplexDET determinant (void) const; |
|
|
134 ComplexDET determinant (int& info) const; |
|
|
135 ComplexDET determinant (int& info, double& rcond) const; |
|
|
136 |
|
|
137 ComplexMatrix solve (const Matrix& b) const; |
|
|
138 ComplexMatrix solve (const Matrix& b, int& info) const; |
|
|
139 ComplexMatrix solve (const Matrix& b, int& info, double& rcond) const; |
|
|
140 |
|
|
141 ComplexMatrix solve (const ComplexMatrix& b) const; |
|
|
142 ComplexMatrix solve (const ComplexMatrix& b, int& info) const; |
|
|
143 ComplexMatrix solve (const ComplexMatrix& b, int& info, double& rcond) const; |
|
|
144 |
|
|
145 ComplexColumnVector solve (const ComplexColumnVector& b) const; |
|
|
146 ComplexColumnVector solve (const ComplexColumnVector& b, int& info) const; |
|
|
147 ComplexColumnVector solve (const ComplexColumnVector& b, int& info, |
|
|
148 double& rcond) const; |
|
|
149 |
|
|
150 ComplexMatrix lssolve (const ComplexMatrix& b) const; |
|
|
151 ComplexMatrix lssolve (const ComplexMatrix& b, int& info) const; |
|
|
152 ComplexMatrix lssolve (const ComplexMatrix& b, int& info, |
|
|
153 int& rank) const; |
|
|
154 |
|
|
155 ComplexColumnVector lssolve (const ComplexColumnVector& b) const; |
|
|
156 ComplexColumnVector lssolve (const ComplexColumnVector& b, int& info) const; |
|
|
157 ComplexColumnVector lssolve (const ComplexColumnVector& b, int& info, |
|
|
158 int& rank) const; |
|
|
159 |
|
|
160 // matrix by diagonal matrix -> matrix operations |
|
|
161 |
|
|
162 ComplexMatrix& operator += (const DiagMatrix& a); |
|
|
163 ComplexMatrix& operator -= (const DiagMatrix& a); |
|
|
164 |
|
|
165 ComplexMatrix& operator += (const ComplexDiagMatrix& a); |
|
|
166 ComplexMatrix& operator -= (const ComplexDiagMatrix& a); |
|
|
167 |
|
|
168 // matrix by matrix -> matrix operations |
|
|
169 |
|
|
170 ComplexMatrix& operator += (const Matrix& a); |
|
|
171 ComplexMatrix& operator -= (const Matrix& a); |
|
|
172 |
|
|
173 ComplexMatrix& operator += (const ComplexMatrix& a); |
|
|
174 ComplexMatrix& operator -= (const ComplexMatrix& a); |
|
|
175 |
|
|
176 // unary operations |
|
|
177 |
|
|
178 Matrix operator ! (void) const; |
|
|
179 |
|
|
180 // matrix by scalar -> matrix operations |
|
|
181 |
|
|
182 friend ComplexMatrix operator + (const ComplexMatrix& a, double s); |
|
|
183 friend ComplexMatrix operator - (const ComplexMatrix& a, double s); |
|
|
184 friend ComplexMatrix operator * (const ComplexMatrix& a, double s); |
|
|
185 friend ComplexMatrix operator / (const ComplexMatrix& a, double s); |
|
|
186 |
|
|
187 // scalar by matrix -> matrix operations |
|
|
188 |
|
|
189 friend ComplexMatrix operator + (double s, const ComplexMatrix& a); |
|
|
190 friend ComplexMatrix operator - (double s, const ComplexMatrix& a); |
|
|
191 friend ComplexMatrix operator * (double s, const ComplexMatrix& a); |
|
|
192 friend ComplexMatrix operator / (double s, const ComplexMatrix& a); |
|
|
193 |
|
|
194 // matrix by column vector -> column vector operations |
|
|
195 |
|
|
196 friend ComplexColumnVector operator * (const ComplexMatrix& a, |
|
|
197 const ColumnVector& b); |
|
|
198 |
|
|
199 friend ComplexColumnVector operator * (const ComplexMatrix& a, |
|
|
200 const ComplexColumnVector& b); |
|
|
201 |
|
|
202 // matrix by diagonal matrix -> matrix operations |
|
|
203 |
|
|
204 friend ComplexMatrix operator + (const ComplexMatrix& a, |
|
|
205 const DiagMatrix& b); |
|
|
206 friend ComplexMatrix operator - (const ComplexMatrix& a, |
|
|
207 const DiagMatrix& b); |
|
|
208 friend ComplexMatrix operator * (const ComplexMatrix& a, |
|
|
209 const DiagMatrix& b); |
|
|
210 |
|
|
211 friend ComplexMatrix operator + (const ComplexMatrix& a, |
|
|
212 const ComplexDiagMatrix& b); |
|
|
213 friend ComplexMatrix operator - (const ComplexMatrix& a, |
|
|
214 const ComplexDiagMatrix& b); |
|
|
215 friend ComplexMatrix operator * (const ComplexMatrix& a, |
|
|
216 const ComplexDiagMatrix& b); |
|
|
217 |
|
|
218 // matrix by matrix -> matrix operations |
|
|
219 |
|
|
220 friend ComplexMatrix operator + (const ComplexMatrix& a, const Matrix& b); |
|
|
221 friend ComplexMatrix operator - (const ComplexMatrix& a, const Matrix& b); |
|
|
222 |
|
|
223 friend ComplexMatrix operator * (const ComplexMatrix& a, const Matrix& b); |
|
|
224 friend ComplexMatrix operator * (const ComplexMatrix& a, |
|
|
225 const ComplexMatrix& b); |
|
|
226 |
|
|
227 friend ComplexMatrix product (const ComplexMatrix& a, const Matrix& b); |
|
|
228 friend ComplexMatrix quotient (const ComplexMatrix& a, const Matrix& b); |
|
|
229 |
|
|
230 // other operations |
|
|
231 |
|
|
232 friend ComplexMatrix map (c_c_Mapper f, const ComplexMatrix& a); |
|
|
233 friend Matrix map (d_c_Mapper f, const ComplexMatrix& a); |
|
|
234 void map (c_c_Mapper f); |
|
|
235 |
|
|
236 Matrix all (void) const; |
|
|
237 Matrix any (void) const; |
|
|
238 |
|
|
239 ComplexMatrix cumprod (void) const; |
|
|
240 ComplexMatrix cumsum (void) const; |
|
|
241 ComplexMatrix prod (void) const; |
|
|
242 ComplexMatrix sum (void) const; |
|
|
243 ComplexMatrix sumsq (void) const; |
|
|
244 |
|
|
245 ComplexColumnVector diag (void) const; |
|
|
246 ComplexColumnVector diag (int k) const; |
|
|
247 |
|
|
248 ComplexColumnVector row_min (void) const; |
|
|
249 ComplexColumnVector row_min_loc (void) const; |
|
|
250 |
|
|
251 ComplexColumnVector row_max (void) const; |
|
|
252 ComplexColumnVector row_max_loc (void) const; |
|
|
253 |
|
|
254 ComplexRowVector column_min (void) const; |
|
|
255 ComplexRowVector column_min_loc (void) const; |
|
|
256 |
|
|
257 ComplexRowVector column_max (void) const; |
|
|
258 ComplexRowVector column_max_loc (void) const; |
|
|
259 |
|
|
260 // i/o |
|
|
261 |
|
|
262 friend ostream& operator << (ostream& os, const ComplexMatrix& a); |
|
|
263 friend istream& operator >> (istream& is, ComplexMatrix& a); |
|
|
264 |
|
|
265 #define KLUDGE_MATRICES |
|
|
266 #define TYPE Complex |
|
|
267 #define KL_MAT_TYPE ComplexMatrix |
|
|
268 #include "mx-kludge.h" |
|
|
269 #undef KLUDGE_MATRICES |
|
|
270 #undef TYPE |
|
|
271 #undef KL_MAT_TYPE |
|
|
272 |
|
|
273 private: |
|
|
274 |
|
|
275 ComplexMatrix (Complex *d, int r, int c) : Array2<Complex> (d, r, c) { } |
|
|
276 }; |
|
|
277 |
|
|
278 } // extern "C++" |
|
|
279 |
|
|
280 #endif |
|
|
281 |
|
|
282 /* |
|
|
283 ;;; Local Variables: *** |
|
|
284 ;;; mode: C++ *** |
|
|
285 ;;; page-delimiter: "^/\\*" *** |
|
|
286 ;;; End: *** |
|
|
287 */ |
