Mercurial > hg > octave-nkf
annotate liboctave/array/CDiagMatrix.cc @ 17296:3a9efb68272d ss-3-7-6
snapshot 3.7.6
* configure.ac (OCTAVE_VERSION): Bump to 3.7.6.
| author | John W. Eaton <jwe@octave.org> |
|---|---|
| date | Tue, 20 Aug 2013 15:17:54 -0400 |
| parents | 0a0912a9ab6e |
| children | 7975d75f933c |
| rev | line source |
|---|---|
| 1993 | 1 // DiagMatrix manipulations. |
| 458 | 2 /* |
| 3 | |
|
14138
72c96de7a403
maint: update copyright notices for 2012
John W. Eaton <jwe@octave.org>
parents:
11586
diff
changeset
|
4 Copyright (C) 1994-2012 John W. Eaton |
|
9601
a9b37bae1802
add a couple of missing copyright statements
Jaroslav Hajek <highegg@gmail.com>
parents:
9550
diff
changeset
|
5 Copyright (C) 2009 VZLU Prague |
| 458 | 6 |
| 7 This file is part of Octave. | |
| 8 | |
| 9 Octave is free software; you can redistribute it and/or modify it | |
| 10 under the terms of the GNU General Public License as published by the | |
| 7016 | 11 Free Software Foundation; either version 3 of the License, or (at your |
| 12 option) any later version. | |
| 458 | 13 |
| 14 Octave is distributed in the hope that it will be useful, but WITHOUT | |
| 15 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 16 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 17 for more details. | |
| 18 | |
| 19 You should have received a copy of the GNU General Public License | |
| 7016 | 20 along with Octave; see the file COPYING. If not, see |
| 21 <http://www.gnu.org/licenses/>. | |
| 458 | 22 |
| 23 */ | |
| 24 | |
| 25 #ifdef HAVE_CONFIG_H | |
| 1192 | 26 #include <config.h> |
| 458 | 27 #endif |
| 28 | |
| 3503 | 29 #include <iostream> |
| 458 | 30 |
| 4669 | 31 #include "Array-util.h" |
| 1368 | 32 #include "lo-error.h" |
|
9003
0631d397fbe0
replace lo_ieee_isnan by xisnan, add missing includes
Jaroslav Hajek <highegg@gmail.com>
parents:
8920
diff
changeset
|
33 #include "lo-ieee.h" |
| 458 | 34 #include "mx-base.h" |
| 35 #include "mx-inlines.cc" | |
| 1650 | 36 #include "oct-cmplx.h" |
| 458 | 37 |
| 1360 | 38 // Complex Diagonal Matrix class |
| 458 | 39 |
| 40 ComplexDiagMatrix::ComplexDiagMatrix (const DiagMatrix& a) | |
| 1989 | 41 : MDiagArray2<Complex> (a.rows (), a.cols ()) |
| 458 | 42 { |
| 5275 | 43 for (octave_idx_type i = 0; i < length (); i++) |
| 458 | 44 elem (i, i) = a.elem (i, i); |
| 45 } | |
| 46 | |
| 2386 | 47 bool |
| 458 | 48 ComplexDiagMatrix::operator == (const ComplexDiagMatrix& a) const |
| 49 { | |
| 50 if (rows () != a.rows () || cols () != a.cols ()) | |
| 51 return 0; | |
| 52 | |
|
9550
3d6a9aea2aea
refactor binary & bool ops in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents:
9003
diff
changeset
|
53 return mx_inline_equal (length (), data (), a.data ()); |
| 458 | 54 } |
| 55 | |
| 2386 | 56 bool |
| 458 | 57 ComplexDiagMatrix::operator != (const ComplexDiagMatrix& a) const |
| 58 { | |
| 59 return !(*this == a); | |
| 60 } | |
| 61 | |
| 62 ComplexDiagMatrix& | |
| 63 ComplexDiagMatrix::fill (double val) | |
| 64 { | |
| 5275 | 65 for (octave_idx_type i = 0; i < length (); i++) |
| 458 | 66 elem (i, i) = val; |
| 67 return *this; | |
| 68 } | |
| 69 | |
| 70 ComplexDiagMatrix& | |
| 71 ComplexDiagMatrix::fill (const Complex& val) | |
| 72 { | |
| 5275 | 73 for (octave_idx_type i = 0; i < length (); i++) |
| 458 | 74 elem (i, i) = val; |
| 75 return *this; | |
| 76 } | |
| 77 | |
| 78 ComplexDiagMatrix& | |
| 5275 | 79 ComplexDiagMatrix::fill (double val, octave_idx_type beg, octave_idx_type end) |
| 458 | 80 { |
| 81 if (beg < 0 || end >= length () || end < beg) | |
| 82 { | |
| 83 (*current_liboctave_error_handler) ("range error for fill"); | |
| 84 return *this; | |
| 85 } | |
| 86 | |
| 5275 | 87 for (octave_idx_type i = beg; i <= end; i++) |
| 458 | 88 elem (i, i) = val; |
| 89 | |
| 90 return *this; | |
| 91 } | |
| 92 | |
| 93 ComplexDiagMatrix& | |
| 5275 | 94 ComplexDiagMatrix::fill (const Complex& val, octave_idx_type beg, octave_idx_type end) |
| 458 | 95 { |
| 96 if (beg < 0 || end >= length () || end < beg) | |
| 97 { | |
| 98 (*current_liboctave_error_handler) ("range error for fill"); | |
| 99 return *this; | |
| 100 } | |
| 101 | |
| 5275 | 102 for (octave_idx_type i = beg; i <= end; i++) |
| 458 | 103 elem (i, i) = val; |
| 104 | |
| 105 return *this; | |
| 106 } | |
| 107 | |
| 108 ComplexDiagMatrix& | |
| 109 ComplexDiagMatrix::fill (const ColumnVector& a) | |
| 110 { | |
| 5275 | 111 octave_idx_type len = length (); |
| 458 | 112 if (a.length () != len) |
| 113 { | |
| 114 (*current_liboctave_error_handler) ("range error for fill"); | |
| 115 return *this; | |
| 116 } | |
| 117 | |
| 5275 | 118 for (octave_idx_type i = 0; i < len; i++) |
| 458 | 119 elem (i, i) = a.elem (i); |
| 120 | |
| 121 return *this; | |
| 122 } | |
| 123 | |
| 124 ComplexDiagMatrix& | |
| 125 ComplexDiagMatrix::fill (const ComplexColumnVector& a) | |
| 126 { | |
| 5275 | 127 octave_idx_type len = length (); |
| 458 | 128 if (a.length () != len) |
| 129 { | |
| 130 (*current_liboctave_error_handler) ("range error for fill"); | |
| 131 return *this; | |
| 132 } | |
| 133 | |
| 5275 | 134 for (octave_idx_type i = 0; i < len; i++) |
| 458 | 135 elem (i, i) = a.elem (i); |
| 136 | |
| 137 return *this; | |
| 138 } | |
| 139 | |
| 140 ComplexDiagMatrix& | |
| 141 ComplexDiagMatrix::fill (const RowVector& a) | |
| 142 { | |
| 5275 | 143 octave_idx_type len = length (); |
| 458 | 144 if (a.length () != len) |
| 145 { | |
| 146 (*current_liboctave_error_handler) ("range error for fill"); | |
| 147 return *this; | |
| 148 } | |
| 149 | |
| 5275 | 150 for (octave_idx_type i = 0; i < len; i++) |
| 458 | 151 elem (i, i) = a.elem (i); |
| 152 | |
| 153 return *this; | |
| 154 } | |
| 155 | |
| 156 ComplexDiagMatrix& | |
| 157 ComplexDiagMatrix::fill (const ComplexRowVector& a) | |
| 158 { | |
| 5275 | 159 octave_idx_type len = length (); |
| 458 | 160 if (a.length () != len) |
| 161 { | |
| 162 (*current_liboctave_error_handler) ("range error for fill"); | |
| 163 return *this; | |
| 164 } | |
| 165 | |
| 5275 | 166 for (octave_idx_type i = 0; i < len; i++) |
| 458 | 167 elem (i, i) = a.elem (i); |
| 168 | |
| 169 return *this; | |
| 170 } | |
| 171 | |
| 172 ComplexDiagMatrix& | |
| 5275 | 173 ComplexDiagMatrix::fill (const ColumnVector& a, octave_idx_type beg) |
| 458 | 174 { |
| 5275 | 175 octave_idx_type a_len = a.length (); |
| 458 | 176 if (beg < 0 || beg + a_len >= length ()) |
| 177 { | |
| 178 (*current_liboctave_error_handler) ("range error for fill"); | |
| 179 return *this; | |
| 180 } | |
| 181 | |
| 5275 | 182 for (octave_idx_type i = 0; i < a_len; i++) |
| 458 | 183 elem (i+beg, i+beg) = a.elem (i); |
| 184 | |
| 185 return *this; | |
| 186 } | |
| 187 | |
| 188 ComplexDiagMatrix& | |
| 5275 | 189 ComplexDiagMatrix::fill (const ComplexColumnVector& a, octave_idx_type beg) |
| 458 | 190 { |
| 5275 | 191 octave_idx_type a_len = a.length (); |
| 458 | 192 if (beg < 0 || beg + a_len >= length ()) |
| 193 { | |
| 194 (*current_liboctave_error_handler) ("range error for fill"); | |
| 195 return *this; | |
| 196 } | |
| 197 | |
| 5275 | 198 for (octave_idx_type i = 0; i < a_len; i++) |
| 458 | 199 elem (i+beg, i+beg) = a.elem (i); |
| 200 | |
| 201 return *this; | |
| 202 } | |
| 203 | |
| 204 ComplexDiagMatrix& | |
| 5275 | 205 ComplexDiagMatrix::fill (const RowVector& a, octave_idx_type beg) |
| 458 | 206 { |
| 5275 | 207 octave_idx_type a_len = a.length (); |
| 458 | 208 if (beg < 0 || beg + a_len >= length ()) |
| 209 { | |
| 210 (*current_liboctave_error_handler) ("range error for fill"); | |
| 211 return *this; | |
| 212 } | |
| 213 | |
| 5275 | 214 for (octave_idx_type i = 0; i < a_len; i++) |
| 458 | 215 elem (i+beg, i+beg) = a.elem (i); |
| 216 | |
| 217 return *this; | |
| 218 } | |
| 219 | |
| 220 ComplexDiagMatrix& | |
| 5275 | 221 ComplexDiagMatrix::fill (const ComplexRowVector& a, octave_idx_type beg) |
| 458 | 222 { |
| 5275 | 223 octave_idx_type a_len = a.length (); |
| 458 | 224 if (beg < 0 || beg + a_len >= length ()) |
| 225 { | |
| 226 (*current_liboctave_error_handler) ("range error for fill"); | |
| 227 return *this; | |
| 228 } | |
| 229 | |
| 5275 | 230 for (octave_idx_type i = 0; i < a_len; i++) |
| 458 | 231 elem (i+beg, i+beg) = a.elem (i); |
| 232 | |
| 233 return *this; | |
| 234 } | |
| 235 | |
|
8366
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
236 DiagMatrix |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
237 ComplexDiagMatrix::abs (void) const |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
238 { |
|
15448
0a0912a9ab6e
Replace deprecated DiagArray2<T>::diag calls with DiagArray2<T>::extract_diag
Jordi Gutiérrez Hermoso <jordigh@octave.org>
parents:
15271
diff
changeset
|
239 return DiagMatrix (extract_diag ().abs (), rows (), columns ()); |
|
8366
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
240 } |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
241 |
| 458 | 242 ComplexDiagMatrix |
| 243 conj (const ComplexDiagMatrix& a) | |
| 244 { | |
|
15448
0a0912a9ab6e
Replace deprecated DiagArray2<T>::diag calls with DiagArray2<T>::extract_diag
Jordi Gutiérrez Hermoso <jordigh@octave.org>
parents:
15271
diff
changeset
|
245 return ComplexDiagMatrix (conj (a.extract_diag ()), a.rows (), a.columns ()); |
| 458 | 246 } |
| 247 | |
| 248 // resize is the destructive analog for this one | |
| 249 | |
| 250 ComplexMatrix | |
| 5275 | 251 ComplexDiagMatrix::extract (octave_idx_type r1, octave_idx_type c1, octave_idx_type r2, octave_idx_type c2) const |
| 458 | 252 { |
| 5275 | 253 if (r1 > r2) { octave_idx_type tmp = r1; r1 = r2; r2 = tmp; } |
| 254 if (c1 > c2) { octave_idx_type tmp = c1; c1 = c2; c2 = tmp; } | |
| 458 | 255 |
| 5275 | 256 octave_idx_type new_r = r2 - r1 + 1; |
| 257 octave_idx_type new_c = c2 - c1 + 1; | |
| 458 | 258 |
| 259 ComplexMatrix result (new_r, new_c); | |
| 260 | |
| 5275 | 261 for (octave_idx_type j = 0; j < new_c; j++) |
| 262 for (octave_idx_type i = 0; i < new_r; i++) | |
| 458 | 263 result.elem (i, j) = elem (r1+i, c1+j); |
| 264 | |
| 265 return result; | |
| 266 } | |
| 267 | |
| 268 // extract row or column i. | |
| 269 | |
| 270 ComplexRowVector | |
| 5275 | 271 ComplexDiagMatrix::row (octave_idx_type i) const |
| 458 | 272 { |
| 5275 | 273 octave_idx_type r = rows (); |
| 274 octave_idx_type c = cols (); | |
| 3504 | 275 if (i < 0 || i >= r) |
| 458 | 276 { |
| 277 (*current_liboctave_error_handler) ("invalid row selection"); | |
|
11586
12df7854fa7c
strip trailing whitespace from source files
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
278 return ComplexRowVector (); |
| 458 | 279 } |
| 280 | |
| 3504 | 281 ComplexRowVector retval (c, 0.0); |
| 282 if (r <= c || (r > c && i < c)) | |
| 458 | 283 retval.elem (i) = elem (i, i); |
| 284 | |
| 285 return retval; | |
| 286 } | |
| 287 | |
| 288 ComplexRowVector | |
| 289 ComplexDiagMatrix::row (char *s) const | |
| 290 { | |
| 533 | 291 if (! s) |
| 458 | 292 { |
| 293 (*current_liboctave_error_handler) ("invalid row selection"); | |
|
11586
12df7854fa7c
strip trailing whitespace from source files
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
294 return ComplexRowVector (); |
| 458 | 295 } |
| 296 | |
| 297 char c = *s; | |
| 298 if (c == 'f' || c == 'F') | |
| 5275 | 299 return row (static_cast<octave_idx_type>(0)); |
| 458 | 300 else if (c == 'l' || c == 'L') |
| 301 return row (rows () - 1); | |
| 302 else | |
| 303 { | |
| 304 (*current_liboctave_error_handler) ("invalid row selection"); | |
| 305 return ComplexRowVector (); | |
| 306 } | |
| 307 } | |
| 308 | |
| 309 ComplexColumnVector | |
| 5275 | 310 ComplexDiagMatrix::column (octave_idx_type i) const |
| 458 | 311 { |
| 5275 | 312 octave_idx_type r = rows (); |
| 313 octave_idx_type c = cols (); | |
| 3504 | 314 if (i < 0 || i >= c) |
| 458 | 315 { |
| 316 (*current_liboctave_error_handler) ("invalid column selection"); | |
|
11586
12df7854fa7c
strip trailing whitespace from source files
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
317 return ComplexColumnVector (); |
| 458 | 318 } |
| 319 | |
| 3504 | 320 ComplexColumnVector retval (r, 0.0); |
| 321 if (r >= c || (r < c && i < r)) | |
| 458 | 322 retval.elem (i) = elem (i, i); |
| 323 | |
| 324 return retval; | |
| 325 } | |
| 326 | |
| 327 ComplexColumnVector | |
| 328 ComplexDiagMatrix::column (char *s) const | |
| 329 { | |
| 533 | 330 if (! s) |
| 458 | 331 { |
| 332 (*current_liboctave_error_handler) ("invalid column selection"); | |
|
11586
12df7854fa7c
strip trailing whitespace from source files
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
333 return ComplexColumnVector (); |
| 458 | 334 } |
| 335 | |
| 336 char c = *s; | |
| 337 if (c == 'f' || c == 'F') | |
| 5275 | 338 return column (static_cast<octave_idx_type>(0)); |
| 458 | 339 else if (c == 'l' || c == 'L') |
| 340 return column (cols () - 1); | |
| 341 else | |
| 342 { | |
| 343 (*current_liboctave_error_handler) ("invalid column selection"); | |
|
11586
12df7854fa7c
strip trailing whitespace from source files
John W. Eaton <jwe@octave.org>
parents:
11523
diff
changeset
|
344 return ComplexColumnVector (); |
| 458 | 345 } |
| 346 } | |
| 347 | |
| 348 ComplexDiagMatrix | |
| 349 ComplexDiagMatrix::inverse (void) const | |
| 350 { | |
| 8811 | 351 octave_idx_type info; |
| 458 | 352 return inverse (info); |
| 353 } | |
| 354 | |
| 355 ComplexDiagMatrix | |
| 8811 | 356 ComplexDiagMatrix::inverse (octave_idx_type& info) const |
| 458 | 357 { |
| 5275 | 358 octave_idx_type r = rows (); |
| 359 octave_idx_type c = cols (); | |
| 3504 | 360 if (r != c) |
| 458 | 361 { |
| 362 (*current_liboctave_error_handler) ("inverse requires square matrix"); | |
| 3585 | 363 return ComplexDiagMatrix (); |
| 458 | 364 } |
| 365 | |
| 3504 | 366 ComplexDiagMatrix retval (r, c); |
| 458 | 367 |
| 368 info = 0; | |
| 5275 | 369 for (octave_idx_type i = 0; i < length (); i++) |
| 458 | 370 { |
| 371 if (elem (i, i) == 0.0) | |
|
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 info = -1; |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
374 return *this; |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
375 } |
| 458 | 376 else |
|
10314
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
377 retval.elem (i, i) = 1.0 / elem (i, i); |
| 458 | 378 } |
| 379 | |
| 1627 | 380 return retval; |
| 458 | 381 } |
| 382 | |
|
8840
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
383 ComplexDiagMatrix |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
384 ComplexDiagMatrix::pseudo_inverse (void) const |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
385 { |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
386 octave_idx_type r = rows (); |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
387 octave_idx_type c = cols (); |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
388 octave_idx_type len = length (); |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
389 |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
390 ComplexDiagMatrix retval (c, r); |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
391 |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
392 for (octave_idx_type i = 0; i < len; i++) |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
393 { |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
394 if (elem (i, i) != 0.0) |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
395 retval.elem (i, i) = 1.0 / elem (i, i); |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
396 else |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
397 retval.elem (i, i) = 0.0; |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
398 } |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
399 |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
400 return retval; |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
401 } |
|
c690e3772583
support diagonal matrices in pinv
Jaroslav Hajek <highegg@gmail.com>
parents:
8811
diff
changeset
|
402 |
|
8366
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
403 bool |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
404 ComplexDiagMatrix::all_elements_are_real (void) const |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
405 { |
|
9825
7483fe200fab
narrow complex values with negative zero imaginary parts
Jaroslav Hajek <highegg@gmail.com>
parents:
9812
diff
changeset
|
406 return mx_inline_all_real (length (), data ()); |
|
8366
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
407 } |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
408 |
| 458 | 409 // diagonal matrix by diagonal matrix -> diagonal matrix operations |
| 410 | |
| 411 ComplexDiagMatrix& | |
| 412 ComplexDiagMatrix::operator += (const DiagMatrix& a) | |
| 413 { | |
| 5275 | 414 octave_idx_type r = rows (); |
| 415 octave_idx_type c = cols (); | |
| 2386 | 416 |
| 5275 | 417 octave_idx_type a_nr = a.rows (); |
| 418 octave_idx_type a_nc = a.cols (); | |
| 2386 | 419 |
| 3504 | 420 if (r != a_nr || c != a_nc) |
| 458 | 421 { |
| 3504 | 422 gripe_nonconformant ("operator +=", r, c, a_nr, a_nc); |
| 458 | 423 return *this; |
| 424 } | |
| 425 | |
| 3504 | 426 if (r == 0 || c == 0) |
| 458 | 427 return *this; |
| 428 | |
| 429 Complex *d = fortran_vec (); // Ensures only one reference to my privates! | |
| 430 | |
|
9550
3d6a9aea2aea
refactor binary & bool ops in liboctave
Jaroslav Hajek <highegg@gmail.com>
parents:
9003
diff
changeset
|
431 mx_inline_add2 (length (), d, a.data ()); |
| 458 | 432 return *this; |
| 433 } | |
| 434 | |
| 435 ComplexDiagMatrix | |
| 436 operator * (const ComplexDiagMatrix& a, const DiagMatrix& b) | |
| 437 { | |
| 5275 | 438 octave_idx_type a_nr = a.rows (); |
| 439 octave_idx_type a_nc = a.cols (); | |
| 2386 | 440 |
| 5275 | 441 octave_idx_type b_nr = b.rows (); |
| 442 octave_idx_type b_nc = b.cols (); | |
| 2386 | 443 |
| 3504 | 444 if (a_nc != b_nr) |
|
9698
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
445 gripe_nonconformant ("operator *", a_nr, a_nc, b_nr, b_nc); |
| 458 | 446 |
| 3504 | 447 ComplexDiagMatrix c (a_nr, b_nc); |
| 458 | 448 |
|
9698
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
449 octave_idx_type len = c.length (), lenm = len < a_nc ? len : a_nc; |
| 458 | 450 |
|
9698
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
451 for (octave_idx_type i = 0; i < lenm; i++) |
|
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
452 c.dgxelem (i) = a.dgelem (i) * b.dgelem (i); |
|
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
453 for (octave_idx_type i = lenm; i < len; i++) |
|
7c6d5d8c8d37
fix diag*diag multiplication
Jaroslav Hajek <highegg@gmail.com>
parents:
9601
diff
changeset
|
454 c.dgxelem (i) = 0.0; |
| 458 | 455 |
| 456 return c; | |
| 457 } | |
| 458 | |
| 459 ComplexDiagMatrix | |
| 1205 | 460 operator * (const DiagMatrix& a, const ComplexDiagMatrix& b) |
| 461 { | |
| 5275 | 462 octave_idx_type a_nr = a.rows (); |
| 463 octave_idx_type a_nc = a.cols (); | |
| 2386 | 464 |
| 5275 | 465 octave_idx_type b_nr = b.rows (); |
| 466 octave_idx_type b_nc = b.cols (); | |
| 2386 | 467 |
| 3504 | 468 if (a_nc != b_nr) |
| 1205 | 469 { |
| 3504 | 470 gripe_nonconformant ("operator *", a_nr, a_nc, b_nr, b_nc); |
| 1205 | 471 return ComplexDiagMatrix (); |
| 472 } | |
| 473 | |
| 3504 | 474 if (a_nr == 0 || a_nc == 0 || b_nc == 0) |
| 475 return ComplexDiagMatrix (a_nr, a_nc, 0.0); | |
| 1205 | 476 |
| 3504 | 477 ComplexDiagMatrix c (a_nr, b_nc); |
| 1205 | 478 |
| 5275 | 479 octave_idx_type len = a_nr < b_nc ? a_nr : b_nc; |
| 1205 | 480 |
| 5275 | 481 for (octave_idx_type i = 0; i < len; i++) |
| 1205 | 482 { |
| 483 double a_element = a.elem (i, i); | |
| 484 Complex b_element = b.elem (i, i); | |
| 485 | |
|
8366
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
486 c.elem (i, i) = a_element * b_element; |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
487 } |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
488 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
489 return c; |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
490 } |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
491 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
492 ComplexDiagMatrix |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
493 operator * (const ComplexDiagMatrix& a, const ComplexDiagMatrix& b) |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
494 { |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
495 octave_idx_type a_nr = a.rows (); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
496 octave_idx_type a_nc = a.cols (); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
497 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
498 octave_idx_type b_nr = b.rows (); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
499 octave_idx_type b_nc = b.cols (); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
500 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
501 if (a_nc != b_nr) |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
502 { |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
503 gripe_nonconformant ("operator *", a_nr, a_nc, b_nr, b_nc); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
504 return ComplexDiagMatrix (); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
505 } |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
506 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
507 if (a_nr == 0 || a_nc == 0 || b_nc == 0) |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
508 return ComplexDiagMatrix (a_nr, a_nc, 0.0); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
509 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
510 ComplexDiagMatrix c (a_nr, b_nc); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
511 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
512 octave_idx_type len = a_nr < b_nc ? a_nr : b_nc; |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
513 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
514 for (octave_idx_type i = 0; i < len; i++) |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
515 { |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
516 Complex a_element = a.elem (i, i); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
517 Complex b_element = b.elem (i, i); |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
518 |
|
8b1a2555c4e2
implement diagonal matrix objects
Jaroslav Hajek <highegg@gmail.com>
parents:
7789
diff
changeset
|
519 c.elem (i, i) = a_element * b_element; |
| 1205 | 520 } |
| 521 | |
| 522 return c; | |
| 523 } | |
| 524 | |
| 458 | 525 // other operations |
| 526 | |
|
8371
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
527 ComplexDET |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
528 ComplexDiagMatrix::determinant (void) const |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
529 { |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
530 ComplexDET det (1.0); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
531 if (rows () != cols ()) |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
532 { |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
533 (*current_liboctave_error_handler) ("determinant requires square matrix"); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
534 det = ComplexDET (0.0); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
535 } |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
536 else |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
537 { |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
538 octave_idx_type len = length (); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
539 for (octave_idx_type i = 0; i < len; i++) |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
540 det *= elem (i, i); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
541 } |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
542 |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
543 return det; |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
544 } |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
545 |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
546 double |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
547 ComplexDiagMatrix::rcond (void) const |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
548 { |
|
15448
0a0912a9ab6e
Replace deprecated DiagArray2<T>::diag calls with DiagArray2<T>::extract_diag
Jordi Gutiérrez Hermoso <jordigh@octave.org>
parents:
15271
diff
changeset
|
549 ColumnVector av = extract_diag (0).map<double> (std::abs); |
|
8371
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
550 double amx = av.max (), amn = av.min (); |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
551 return amx == 0 ? 0.0 : amn / amx; |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
552 } |
|
c3f7e2549abb
make det & inv aware of diagonal & permutation matrices
Jaroslav Hajek <highegg@gmail.com>
parents:
8366
diff
changeset
|
553 |
| 458 | 554 // i/o |
| 555 | |
| 3504 | 556 std::ostream& |
| 557 operator << (std::ostream& os, const ComplexDiagMatrix& a) | |
| 458 | 558 { |
| 559 Complex ZERO (0.0); | |
| 560 // int field_width = os.precision () + 7; | |
| 5275 | 561 for (octave_idx_type i = 0; i < a.rows (); i++) |
| 458 | 562 { |
| 5275 | 563 for (octave_idx_type j = 0; j < a.cols (); j++) |
|
10314
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
564 { |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
565 if (i == j) |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
566 os << " " /* setw (field_width) */ << a.elem (i, i); |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
567 else |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
568 os << " " /* setw (field_width) */ << ZERO; |
|
07ebe522dac2
untabify liboctave C++ sources
John W. Eaton <jwe@octave.org>
parents:
10158
diff
changeset
|
569 } |
| 458 | 570 os << "\n"; |
| 571 } | |
| 572 return os; | |
| 573 } |