Mercurial > octave-nkf
annotate liboctave/SparseQR.h @ 7948:af10baa63915 ss-3-1-50
3.1.50 snapshot
| author | John W. Eaton <jwe@octave.org> |
|---|---|
| date | Fri, 18 Jul 2008 17:42:48 -0400 |
| parents | 82be108cc558 |
| children | eb63fbe60fab |
| rev | line source |
|---|---|
| 5610 | 1 /* |
| 2 | |
| 7017 | 3 Copyright (C) 2005, 2006, 2007 David Bateman |
| 5610 | 4 |
| 7016 | 5 This file is part of Octave. |
| 6 | |
| 5610 | 7 Octave is free software; you can redistribute it and/or modify it |
| 8 under the terms of the GNU General Public License as published by the | |
| 7016 | 9 Free Software Foundation; either version 3 of the License, or (at your |
| 10 option) any later version. | |
| 5610 | 11 |
| 12 Octave is distributed in the hope that it will be useful, but WITHOUT | |
| 13 ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| 14 FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| 15 for more details. | |
| 16 | |
| 17 You should have received a copy of the GNU General Public License | |
| 7016 | 18 along with Octave; see the file COPYING. If not, see |
| 19 <http://www.gnu.org/licenses/>. | |
| 5610 | 20 |
| 21 */ | |
| 22 | |
| 23 #if !defined (sparse_QR_h) | |
| 24 #define sparse_QR_h 1 | |
| 25 | |
| 26 #include <iostream> | |
| 27 | |
| 28 #include "dMatrix.h" | |
| 29 #include "CMatrix.h" | |
| 30 #include "dSparse.h" | |
| 31 #include "CSparse.h" | |
| 32 #include "oct-sparse.h" | |
| 33 | |
| 34 #ifdef IDX_TYPE_LONG | |
| 5648 | 35 #define CXSPARSE_DNAME(name) cs_dl ## name |
| 5610 | 36 #else |
| 5648 | 37 #define CXSPARSE_DNAME(name) cs_di ## name |
| 5610 | 38 #endif |
| 39 | |
| 40 class | |
| 6108 | 41 OCTAVE_API |
| 5610 | 42 SparseQR |
| 43 { | |
| 44 protected: | |
| 45 class SparseQR_rep | |
| 46 { | |
| 47 public: | |
| 48 SparseQR_rep (const SparseMatrix& a, int order); | |
| 49 | |
| 50 ~SparseQR_rep (void); | |
| 51 #ifdef HAVE_CXSPARSE | |
| 52 bool ok (void) const { return (N && S); } | |
| 53 #else | |
| 54 bool ok (void) const { return false; } | |
| 55 #endif | |
| 56 SparseMatrix V (void) const; | |
| 57 | |
| 58 ColumnVector Pinv (void) const; | |
| 59 | |
| 60 ColumnVector P (void) const; | |
| 61 | |
| 62 SparseMatrix R (const bool econ) const; | |
| 63 | |
| 64 Matrix C (const Matrix &b) const; | |
| 65 | |
|
7505
f5005d9510f4
Remove dispatched sparse functions and treat in the generic versions of the functions
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
66 Matrix Q (void) const; |
|
f5005d9510f4
Remove dispatched sparse functions and treat in the generic versions of the functions
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
67 |
| 5610 | 68 int count; |
| 69 | |
| 70 octave_idx_type nrows; | |
| 71 #ifdef HAVE_CXSPARSE | |
| 5648 | 72 CXSPARSE_DNAME (s) *S; |
| 5610 | 73 |
| 5648 | 74 CXSPARSE_DNAME (n) *N; |
| 5610 | 75 #endif |
| 76 }; | |
| 77 private: | |
| 78 SparseQR_rep *rep; | |
| 79 | |
| 80 public: | |
| 5792 | 81 SparseQR (void) : rep (new SparseQR_rep (SparseMatrix(), 0)) { } |
| 5610 | 82 |
| 5792 | 83 SparseQR (const SparseMatrix& a, int order = 0) : |
| 5610 | 84 rep (new SparseQR_rep (a, order)) { } |
| 85 | |
| 86 SparseQR (const SparseQR& a) : rep (a.rep) { rep->count++; } | |
| 87 | |
| 88 ~SparseQR (void) | |
| 89 { | |
| 90 if (--rep->count <= 0) | |
| 91 delete rep; | |
| 92 } | |
| 93 | |
| 94 SparseQR& operator = (const SparseQR& a) | |
| 95 { | |
| 96 if (this != &a) | |
| 97 { | |
| 98 if (--rep->count <= 0) | |
| 99 delete rep; | |
| 100 | |
| 101 rep = a.rep; | |
| 102 rep->count++; | |
| 103 } | |
| 104 return *this; | |
| 105 } | |
| 106 | |
| 107 bool ok (void) const { return rep->ok(); } | |
| 108 | |
| 109 SparseMatrix V (void) const { return rep->V(); } | |
| 110 | |
| 111 ColumnVector Pinv (void) const { return rep->P(); } | |
| 112 | |
| 113 ColumnVector P (void) const { return rep->P(); } | |
| 114 | |
| 115 SparseMatrix R (const bool econ = false) const { return rep->R(econ); } | |
| 116 | |
| 117 Matrix C (const Matrix &b) const { return rep->C(b); } | |
| 118 | |
|
7505
f5005d9510f4
Remove dispatched sparse functions and treat in the generic versions of the functions
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
119 Matrix Q (void) const { return rep->Q(); } |
|
f5005d9510f4
Remove dispatched sparse functions and treat in the generic versions of the functions
David Bateman <dbateman@free.fr>
parents:
7017
diff
changeset
|
120 |
| 5610 | 121 friend Matrix qrsolve (const SparseMatrix &a, const Matrix &b, |
| 122 octave_idx_type &info); | |
| 123 | |
| 124 friend SparseMatrix qrsolve (const SparseMatrix &a, const SparseMatrix &b, | |
| 125 octave_idx_type &info); | |
| 126 | |
| 127 friend ComplexMatrix qrsolve (const SparseMatrix &a, const ComplexMatrix &b, | |
| 128 octave_idx_type &info); | |
| 129 | |
| 130 friend SparseComplexMatrix qrsolve (const SparseMatrix &a, | |
| 131 const SparseComplexMatrix &b, | |
| 132 octave_idx_type &info); | |
| 133 | |
| 134 protected: | |
| 135 #ifdef HAVE_CXSPARSE | |
| 5648 | 136 CXSPARSE_DNAME (s) * S (void) { return rep->S; } |
| 5610 | 137 |
| 5648 | 138 CXSPARSE_DNAME (n) * N (void) { return rep->N; } |
| 5610 | 139 #endif |
| 140 }; | |
| 141 | |
| 5713 | 142 |
| 143 // Publish externally used friend functions. | |
| 144 | |
| 145 extern Matrix qrsolve (const SparseMatrix &a, const Matrix &b, | |
| 146 octave_idx_type &info); | |
| 147 | |
|
7789
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
148 extern Matrix qrsolve (const SparseMatrix &a, const MArray2<double> &b, |
|
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
149 octave_idx_type &info); |
|
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
150 |
| 5713 | 151 extern SparseMatrix qrsolve (const SparseMatrix &a, const SparseMatrix &b, |
| 152 octave_idx_type &info); | |
| 153 | |
| 154 extern ComplexMatrix qrsolve (const SparseMatrix &a, const ComplexMatrix &b, | |
| 155 octave_idx_type &info); | |
| 156 | |
|
7789
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
157 extern ComplexMatrix qrsolve (const SparseMatrix &a, const MArray2<Complex> &b, |
|
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
158 octave_idx_type &info); |
|
82be108cc558
First attempt at single precision tyeps
David Bateman <dbateman@free.fr>
parents:
7505
diff
changeset
|
159 |
| 5713 | 160 extern SparseComplexMatrix qrsolve (const SparseMatrix &a, |
| 161 const SparseComplexMatrix &b, | |
| 162 octave_idx_type &info); | |
| 163 | |
| 5610 | 164 #endif |
| 165 | |
| 166 /* | |
| 167 ;;; Local Variables: *** | |
| 168 ;;; mode: C++ *** | |
| 169 ;;; End: *** | |
| 170 */ |