Mercurial > octave
annotate liboctave/numeric/qrp.cc @ 23475:d691ed308237
maint: Clean up #includes in liboctave/numeric directory.
* build-aux/mk-opts.pl: Change Perl to generate "" around local include
libraries rather than <>. Include "lo-math.h" rather than <cmath>.
* CollocWt.cc, DAERTFunc.h, DASPK.cc, DASPK.h, DASRT.cc, DASRT.h, DASSL.cc,
DASSL.h, DET.h, EIG.cc, EIG.h, LSODE.cc, LSODE.h, ODE.h, ODES.cc, ODESFunc.h,
Quad.cc, aepbalance.cc, base-de.h, base-min.h, bsxfun-decl.h, bsxfun-defs.cc,
bsxfun.h, chol.cc, eigs-base.cc, fEIG.cc, fEIG.h, gepbalance.cc, gsvd.cc,
hess.cc, lo-blas-proto.h, lo-lapack-proto.h, lo-mappers.cc, lo-mappers.h,
lo-qrupdate-proto.h, lo-slatec-proto.h, lo-specfun.cc, lo-specfun.h, lu.cc,
lu.h, oct-convn.cc, oct-convn.h, oct-fftw.cc, oct-fftw.h, oct-norm.cc,
oct-rand.cc, oct-rand.h, oct-spparms.cc, oct-spparms.h, qr.cc, qr.h, qrp.cc,
randgamma.cc, randpoisson.cc, schur.cc, schur.h, sparse-chol.cc, sparse-chol.h,
sparse-dmsolve.cc, sparse-lu.cc, sparse-lu.h, sparse-qr.cc, sparse-qr.h,
svd.cc:
Rationalize #includes. Use forward declarations of just classes where
possible. Reformat some long lines < 80 characters. Reformat some comments
for readabliity.
* mx-inlines.cc: Rationalize #includes for this file in liboctave/operators
used by many in liboctave/numeric.
author | Rik <rik@octave.org> |
---|---|
date | Tue, 09 May 2017 08:46:07 -0700 |
parents | 855122b993da |
children | 08036a7f3660 |
rev | line source |
---|---|
538 | 1 /* |
2 | |
23219
3ac9f9ecfae5
maint: Update copyright dates.
John W. Eaton <jwe@octave.org>
parents:
23083
diff
changeset
|
3 Copyright (C) 1994-2017 John W. Eatonn |
10521
4d1fc073fbb7
add some missing copyright stmts
Jaroslav Hajek <highegg@gmail.com>
parents:
10350
diff
changeset
|
4 Copyright (C) 2009 VZLU Prague |
538 | 5 |
6 This file is part of Octave. | |
7 | |
8 Octave is free software; you can redistribute it and/or modify it | |
22755
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
9 under the terms of the GNU General Public License as published by |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
10 the Free Software Foundation; either version 3 of the License, or |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
11 (at your option) any later version. |
538 | 12 |
22755
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
13 Octave is distributed in the hope that it will be useful, but |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
14 WITHOUT ANY WARRANTY; without even the implied warranty of |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
3a2b891d0b33
maint: Standardize Copyright formatting.
Rik <rik@octave.org>
parents:
22402
diff
changeset
|
16 GNU General Public License for more details. |
538 | 17 |
18 You should have received a copy of the GNU General Public License | |
7016 | 19 along with Octave; see the file COPYING. If not, see |
20 <http://www.gnu.org/licenses/>. | |
538 | 21 |
22 */ | |
23 | |
21724
aba2e6293dd8
use "#if ..." consistently instead of "#ifdef" and "#ifndef"
John W. Eaton <jwe@octave.org>
parents:
21301
diff
changeset
|
24 #if defined (HAVE_CONFIG_H) |
21301
40de9f8f23a6
Use '#include "config.h"' rather than <config.h>.
Rik <rik@octave.org>
parents:
21280
diff
changeset
|
25 # include "config.h" |
538 | 26 #endif |
27 | |
1367 | 28 #include <cassert> |
538 | 29 |
23475
d691ed308237
maint: Clean up #includes in liboctave/numeric directory.
Rik <rik@octave.org>
parents:
23450
diff
changeset
|
30 #include <algorithm> |
d691ed308237
maint: Clean up #includes in liboctave/numeric directory.
Rik <rik@octave.org>
parents:
23450
diff
changeset
|
31 |
d691ed308237
maint: Clean up #includes in liboctave/numeric directory.
Rik <rik@octave.org>
parents:
23450
diff
changeset
|
32 #include "Array.h" |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
33 #include "CMatrix.h" |
23475
d691ed308237
maint: Clean up #includes in liboctave/numeric directory.
Rik <rik@octave.org>
parents:
23450
diff
changeset
|
34 #include "MArray.h" |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
35 #include "dMatrix.h" |
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
36 #include "dRowVector.h" |
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
37 #include "fCMatrix.h" |
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
38 #include "fMatrix.h" |
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
39 #include "fRowVector.h" |
22322
93b3cdd36854
move most f77 function decls to separate header files
John W. Eaton <jwe@octave.org>
parents:
22317
diff
changeset
|
40 #include "lo-lapack-proto.h" |
8597
c86718093c1b
improve & fix QR classes
Jaroslav Hajek <highegg@gmail.com>
parents:
8375
diff
changeset
|
41 #include "oct-locbuf.h" |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
42 #include "qrp.h" |
538 | 43 |
22317
6ca3acf5fad8
move some new numeric classes to namespace octave::math
John W. Eaton <jwe@octave.org>
parents:
22135
diff
changeset
|
44 namespace octave |
6ca3acf5fad8
move some new numeric classes to namespace octave::math
John W. Eaton <jwe@octave.org>
parents:
22135
diff
changeset
|
45 { |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
46 namespace math |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
47 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
48 // Specialization. |
538 | 49 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
50 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
51 void |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
52 qrp<Matrix>::init (const Matrix& a, type qr_type) |
8597
c86718093c1b
improve & fix QR classes
Jaroslav Hajek <highegg@gmail.com>
parents:
8375
diff
changeset
|
53 { |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
54 assert (qr_type != qr<Matrix>::raw); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
55 |
22988
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
56 F77_INT m = octave::to_f77_int (a.rows ()); |
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
57 F77_INT n = octave::to_f77_int (a.cols ()); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
58 |
23450
855122b993da
maint: Wrap tertiary operator in parentheses "(COND ? x : y)".
Rik <rik@octave.org>
parents:
23220
diff
changeset
|
59 F77_INT min_mn = (m < n ? m : n); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
60 OCTAVE_LOCAL_BUFFER (double, tau, min_mn); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
61 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
62 F77_INT info = 0; |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
63 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
64 Matrix afact = a; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
65 if (m > n && qr_type == qr<Matrix>::std) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
66 afact.resize (m, m); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
67 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
68 MArray<F77_INT> jpvt (dim_vector (n, 1), 0); |
8368
c72c1c9bccdc
call blocked permuted qr factorization routines from LAPACK
Jaroslav Hajek <highegg@gmail.com>
parents:
8367
diff
changeset
|
69 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
70 if (m > 0) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
71 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
72 // workspace query. |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
73 double rlwork; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
74 F77_XFCN (dgeqp3, DGEQP3, (m, n, afact.fortran_vec (), |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
75 m, jpvt.fortran_vec (), tau, |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
76 &rlwork, -1, info)); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
77 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
78 // allocate buffer and do the job. |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
79 F77_INT lwork = static_cast<F77_INT> (rlwork); |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
80 lwork = std::max (lwork, static_cast<F77_INT> (1)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
81 OCTAVE_LOCAL_BUFFER (double, work, lwork); |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
82 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
83 F77_XFCN (dgeqp3, DGEQP3, (m, n, afact.fortran_vec (), |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
84 m, jpvt.fortran_vec (), tau, |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
85 work, lwork, info)); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
86 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
87 else |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
88 { |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
89 for (F77_INT i = 0; i < n; i++) |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
90 jpvt(i) = i+1; |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
91 } |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
92 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
93 // Form Permutation matrix (if economy is requested, return the |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
94 // indices only!) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
95 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
96 jpvt -= static_cast<F77_INT> (1); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
97 p = PermMatrix (jpvt, true); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
98 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
99 form (n, afact, tau, qr_type); |
8597
c86718093c1b
improve & fix QR classes
Jaroslav Hajek <highegg@gmail.com>
parents:
8375
diff
changeset
|
100 } |
1922 | 101 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
102 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
103 qrp<Matrix>::qrp (const Matrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
104 : qr<Matrix> (), p () |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
105 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
106 init (a, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
107 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
108 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
109 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
110 RowVector |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
111 qrp<Matrix>::Pvec (void) const |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
112 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
113 Array<double> pa (p.col_perm_vec ()); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
114 RowVector pv (MArray<double> (pa) + 1.0); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
115 return pv; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
116 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
117 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
118 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
119 void |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
120 qrp<FloatMatrix>::init (const FloatMatrix& a, type qr_type) |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
121 { |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
122 assert (qr_type != qr<FloatMatrix>::raw); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
123 |
22988
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
124 F77_INT m = octave::to_f77_int (a.rows ()); |
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
125 F77_INT n = octave::to_f77_int (a.cols ()); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
126 |
23450
855122b993da
maint: Wrap tertiary operator in parentheses "(COND ? x : y)".
Rik <rik@octave.org>
parents:
23220
diff
changeset
|
127 F77_INT min_mn = (m < n ? m : n); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
128 OCTAVE_LOCAL_BUFFER (float, tau, min_mn); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
129 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
130 F77_INT info = 0; |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
131 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
132 FloatMatrix afact = a; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
133 if (m > n && qr_type == qr<FloatMatrix>::std) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
134 afact.resize (m, m); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
135 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
136 MArray<F77_INT> jpvt (dim_vector (n, 1), 0); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
137 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
138 if (m > 0) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
139 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
140 // workspace query. |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
141 float rlwork; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
142 F77_XFCN (sgeqp3, SGEQP3, (m, n, afact.fortran_vec (), |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
143 m, jpvt.fortran_vec (), tau, |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
144 &rlwork, -1, info)); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
145 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
146 // allocate buffer and do the job. |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
147 F77_INT lwork = static_cast<F77_INT> (rlwork); |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
148 lwork = std::max (lwork, static_cast<F77_INT> (1)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
149 OCTAVE_LOCAL_BUFFER (float, work, lwork); |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
150 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
151 F77_XFCN (sgeqp3, SGEQP3, (m, n, afact.fortran_vec (), |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
152 m, jpvt.fortran_vec (), tau, |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
153 work, lwork, info)); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
154 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
155 else |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
156 { |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
157 for (F77_INT i = 0; i < n; i++) |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
158 jpvt(i) = i+1; |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
159 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
160 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
161 // Form Permutation matrix (if economy is requested, return the |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
162 // indices only!) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
163 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
164 jpvt -= static_cast<F77_INT> (1); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
165 p = PermMatrix (jpvt, true); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
166 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
167 form (n, afact, tau, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
168 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
169 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
170 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
171 qrp<FloatMatrix>::qrp (const FloatMatrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
172 : qr<FloatMatrix> (), p () |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
173 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
174 init (a, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
175 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
176 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
177 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
178 FloatRowVector |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
179 qrp<FloatMatrix>::Pvec (void) const |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
180 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
181 Array<float> pa (p.col_perm_vec ()); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
182 FloatRowVector pv (MArray<float> (pa) + 1.0f); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
183 return pv; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
184 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
185 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
186 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
187 void |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
188 qrp<ComplexMatrix>::init (const ComplexMatrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
189 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
190 assert (qr_type != qr<ComplexMatrix>::raw); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
191 |
22988
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
192 F77_INT m = octave::to_f77_int (a.rows ()); |
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
193 F77_INT n = octave::to_f77_int (a.cols ()); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
194 |
23450
855122b993da
maint: Wrap tertiary operator in parentheses "(COND ? x : y)".
Rik <rik@octave.org>
parents:
23220
diff
changeset
|
195 F77_INT min_mn = (m < n ? m : n); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
196 OCTAVE_LOCAL_BUFFER (Complex, tau, min_mn); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
197 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
198 F77_INT info = 0; |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
199 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
200 ComplexMatrix afact = a; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
201 if (m > n && qr_type == qr<ComplexMatrix>::std) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
202 afact.resize (m, m); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
203 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
204 MArray<F77_INT> jpvt (dim_vector (n, 1), 0); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
205 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
206 if (m > 0) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
207 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
208 OCTAVE_LOCAL_BUFFER (double, rwork, 2*n); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
209 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
210 // workspace query. |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
211 Complex clwork; |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
212 F77_XFCN (zgeqp3, ZGEQP3, (m, n, |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
213 F77_DBLE_CMPLX_ARG (afact.fortran_vec ()), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
214 m, jpvt.fortran_vec (), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
215 F77_DBLE_CMPLX_ARG (tau), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
216 F77_DBLE_CMPLX_ARG (&clwork), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
217 -1, rwork, info)); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
218 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
219 // allocate buffer and do the job. |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
220 F77_INT lwork = static_cast<F77_INT> (clwork.real ()); |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
221 lwork = std::max (lwork, static_cast<F77_INT> (1)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
222 OCTAVE_LOCAL_BUFFER (Complex, work, lwork); |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
223 |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
224 F77_XFCN (zgeqp3, ZGEQP3, (m, n, |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
225 F77_DBLE_CMPLX_ARG (afact.fortran_vec ()), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
226 m, jpvt.fortran_vec (), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
227 F77_DBLE_CMPLX_ARG (tau), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
228 F77_DBLE_CMPLX_ARG (work), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
229 lwork, rwork, info)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
230 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
231 else |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
232 { |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
233 for (F77_INT i = 0; i < n; i++) |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
234 jpvt(i) = i+1; |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
235 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
236 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
237 // Form Permutation matrix (if economy is requested, return the |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
238 // indices only!) |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
239 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
240 jpvt -= static_cast<F77_INT> (1); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
241 p = PermMatrix (jpvt, true); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
242 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
243 form (n, afact, tau, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
244 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
245 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
246 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
247 qrp<ComplexMatrix>::qrp (const ComplexMatrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
248 : qr<ComplexMatrix> (), p () |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
249 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
250 init (a, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
251 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
252 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
253 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
254 RowVector |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
255 qrp<ComplexMatrix>::Pvec (void) const |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
256 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
257 Array<double> pa (p.col_perm_vec ()); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
258 RowVector pv (MArray<double> (pa) + 1.0); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
259 return pv; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
260 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
261 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
262 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
263 void |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
264 qrp<FloatComplexMatrix>::init (const FloatComplexMatrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
265 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
266 assert (qr_type != qr<FloatComplexMatrix>::raw); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
267 |
22988
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
268 F77_INT m = octave::to_f77_int (a.rows ()); |
cd33c785e80e
put to_f77_int inside octave namespace
John W. Eaton <jwe@octave.org>
parents:
22951
diff
changeset
|
269 F77_INT n = octave::to_f77_int (a.cols ()); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
270 |
23450
855122b993da
maint: Wrap tertiary operator in parentheses "(COND ? x : y)".
Rik <rik@octave.org>
parents:
23220
diff
changeset
|
271 F77_INT min_mn = (m < n ? m : n); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
272 OCTAVE_LOCAL_BUFFER (FloatComplex, tau, min_mn); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
273 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
274 F77_INT info = 0; |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
275 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
276 FloatComplexMatrix afact = a; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
277 if (m > n && qr_type == qr<FloatComplexMatrix>::std) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
278 afact.resize (m, m); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
279 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
280 MArray<F77_INT> jpvt (dim_vector (n, 1), 0); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
281 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
282 if (m > 0) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
283 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
284 OCTAVE_LOCAL_BUFFER (float, rwork, 2*n); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
285 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
286 // workspace query. |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
287 FloatComplex clwork; |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
288 F77_XFCN (cgeqp3, CGEQP3, (m, n, |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
289 F77_CMPLX_ARG (afact.fortran_vec ()), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
290 m, jpvt.fortran_vec (), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
291 F77_CMPLX_ARG (tau), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
292 F77_CMPLX_ARG (&clwork), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
293 -1, rwork, info)); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
294 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
295 // allocate buffer and do the job. |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
296 F77_INT lwork = static_cast<F77_INT> (clwork.real ()); |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
297 lwork = std::max (lwork, static_cast<F77_INT> (1)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
298 OCTAVE_LOCAL_BUFFER (FloatComplex, work, lwork); |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
299 |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
300 F77_XFCN (cgeqp3, CGEQP3, (m, n, |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
301 F77_CMPLX_ARG (afact.fortran_vec ()), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
302 m, jpvt.fortran_vec (), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
303 F77_CMPLX_ARG (tau), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
304 F77_CMPLX_ARG (work), |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
305 lwork, rwork, info)); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
306 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
307 else |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
308 { |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
309 for (F77_INT i = 0; i < n; i++) |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
310 jpvt(i) = i+1; |
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
311 } |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
312 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
313 // Form Permutation matrix (if economy is requested, return the |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
314 // indices only!) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
315 |
22951
2e6a327e7903
use F77_INT instead of octave_idx_type for liboctave QRP functions
John W. Eaton <jwe@octave.org>
parents:
22755
diff
changeset
|
316 jpvt -= static_cast<F77_INT> (1); |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
317 p = PermMatrix (jpvt, true); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
318 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
319 form (n, afact, tau, qr_type); |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
320 } |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
321 |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
322 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
323 qrp<FloatComplexMatrix>::qrp (const FloatComplexMatrix& a, type qr_type) |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
324 : qr<FloatComplexMatrix> (), p () |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
325 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
326 init (a, qr_type); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
327 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
328 |
22329
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
329 template <> |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
330 FloatRowVector |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
331 qrp<FloatComplexMatrix>::Pvec (void) const |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
332 { |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
333 Array<float> pa (p.col_perm_vec ()); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
334 FloatRowVector pv (MArray<float> (pa) + 1.0f); |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
335 return pv; |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
336 } |
7f3c7a8bd131
maint: Indent namespaces in liboctave/numeric files.
John W. Eaton <jwe@octave.org>
parents:
22323
diff
changeset
|
337 } |
21280
ebdf74c15722
better use of templates for qrp classes
John W. Eaton <jwe@octave.org>
parents:
21279
diff
changeset
|
338 } |