Chris@49
|
1 // Copyright (C) 2009-2011 NICTA (www.nicta.com.au)
|
Chris@49
|
2 // Copyright (C) 2009-2011 Conrad Sanderson
|
Chris@49
|
3 //
|
Chris@49
|
4 // This Source Code Form is subject to the terms of the Mozilla Public
|
Chris@49
|
5 // License, v. 2.0. If a copy of the MPL was not distributed with this
|
Chris@49
|
6 // file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
Chris@49
|
7
|
Chris@49
|
8
|
Chris@49
|
9 //! \addtogroup fn_svd
|
Chris@49
|
10 //! @{
|
Chris@49
|
11
|
Chris@49
|
12
|
Chris@49
|
13
|
Chris@49
|
14 template<typename T1>
|
Chris@49
|
15 inline
|
Chris@49
|
16 bool
|
Chris@49
|
17 svd
|
Chris@49
|
18 (
|
Chris@49
|
19 Col<typename T1::pod_type>& S,
|
Chris@49
|
20 const Base<typename T1::elem_type,T1>& X,
|
Chris@49
|
21 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
22 )
|
Chris@49
|
23 {
|
Chris@49
|
24 arma_extra_debug_sigprint();
|
Chris@49
|
25 arma_ignore(junk);
|
Chris@49
|
26
|
Chris@49
|
27 // it doesn't matter if X is an alias of S, as auxlib::svd() makes a copy of X
|
Chris@49
|
28
|
Chris@49
|
29 const bool status = auxlib::svd(S, X);
|
Chris@49
|
30
|
Chris@49
|
31 if(status == false)
|
Chris@49
|
32 {
|
Chris@49
|
33 S.reset();
|
Chris@49
|
34 arma_bad("svd(): failed to converge", false);
|
Chris@49
|
35 }
|
Chris@49
|
36
|
Chris@49
|
37 return status;
|
Chris@49
|
38 }
|
Chris@49
|
39
|
Chris@49
|
40
|
Chris@49
|
41
|
Chris@49
|
42 template<typename T1>
|
Chris@49
|
43 inline
|
Chris@49
|
44 Col<typename T1::pod_type>
|
Chris@49
|
45 svd
|
Chris@49
|
46 (
|
Chris@49
|
47 const Base<typename T1::elem_type,T1>& X,
|
Chris@49
|
48 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
49 )
|
Chris@49
|
50 {
|
Chris@49
|
51 arma_extra_debug_sigprint();
|
Chris@49
|
52 arma_ignore(junk);
|
Chris@49
|
53
|
Chris@49
|
54 Col<typename T1::pod_type> out;
|
Chris@49
|
55
|
Chris@49
|
56 const bool status = auxlib::svd(out, X);
|
Chris@49
|
57
|
Chris@49
|
58 if(status == false)
|
Chris@49
|
59 {
|
Chris@49
|
60 out.reset();
|
Chris@49
|
61 arma_bad("svd(): failed to converge");
|
Chris@49
|
62 }
|
Chris@49
|
63
|
Chris@49
|
64 return out;
|
Chris@49
|
65 }
|
Chris@49
|
66
|
Chris@49
|
67
|
Chris@49
|
68
|
Chris@49
|
69 template<typename T1>
|
Chris@49
|
70 inline
|
Chris@49
|
71 bool
|
Chris@49
|
72 svd
|
Chris@49
|
73 (
|
Chris@49
|
74 Mat<typename T1::elem_type>& U,
|
Chris@49
|
75 Col<typename T1::pod_type >& S,
|
Chris@49
|
76 Mat<typename T1::elem_type>& V,
|
Chris@49
|
77 const Base<typename T1::elem_type,T1>& X,
|
Chris@49
|
78 const char* method = "",
|
Chris@49
|
79 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
80 )
|
Chris@49
|
81 {
|
Chris@49
|
82 arma_extra_debug_sigprint();
|
Chris@49
|
83 arma_ignore(junk);
|
Chris@49
|
84
|
Chris@49
|
85 arma_debug_check
|
Chris@49
|
86 (
|
Chris@49
|
87 ( ((void*)(&U) == (void*)(&S)) || (&U == &V) || ((void*)(&S) == (void*)(&V)) ),
|
Chris@49
|
88 "svd(): two or more output objects are the same object"
|
Chris@49
|
89 );
|
Chris@49
|
90
|
Chris@49
|
91 bool use_divide_and_conquer = false;
|
Chris@49
|
92
|
Chris@49
|
93 const char sig = method[0];
|
Chris@49
|
94
|
Chris@49
|
95 switch(sig)
|
Chris@49
|
96 {
|
Chris@49
|
97 case '\0':
|
Chris@49
|
98 case 's':
|
Chris@49
|
99 break;
|
Chris@49
|
100
|
Chris@49
|
101 case 'd':
|
Chris@49
|
102 use_divide_and_conquer = true;
|
Chris@49
|
103 break;
|
Chris@49
|
104
|
Chris@49
|
105 default:
|
Chris@49
|
106 {
|
Chris@49
|
107 arma_stop("svd(): unknown method specified");
|
Chris@49
|
108 return false;
|
Chris@49
|
109 }
|
Chris@49
|
110 }
|
Chris@49
|
111
|
Chris@49
|
112 // auxlib::svd() makes an internal copy of X
|
Chris@49
|
113 const bool status = (use_divide_and_conquer == false) ? auxlib::svd(U, S, V, X) : auxlib::svd_dc(U, S, V, X);
|
Chris@49
|
114
|
Chris@49
|
115 if(status == false)
|
Chris@49
|
116 {
|
Chris@49
|
117 U.reset();
|
Chris@49
|
118 S.reset();
|
Chris@49
|
119 V.reset();
|
Chris@49
|
120 arma_bad("svd(): failed to converge", false);
|
Chris@49
|
121 }
|
Chris@49
|
122
|
Chris@49
|
123 return status;
|
Chris@49
|
124 }
|
Chris@49
|
125
|
Chris@49
|
126
|
Chris@49
|
127
|
Chris@49
|
128 template<typename T1>
|
Chris@49
|
129 inline
|
Chris@49
|
130 bool
|
Chris@49
|
131 svd_econ
|
Chris@49
|
132 (
|
Chris@49
|
133 Mat<typename T1::elem_type>& U,
|
Chris@49
|
134 Col<typename T1::pod_type >& S,
|
Chris@49
|
135 Mat<typename T1::elem_type>& V,
|
Chris@49
|
136 const Base<typename T1::elem_type,T1>& X,
|
Chris@49
|
137 const char mode = 'b',
|
Chris@49
|
138 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
139 )
|
Chris@49
|
140 {
|
Chris@49
|
141 arma_extra_debug_sigprint();
|
Chris@49
|
142 arma_ignore(junk);
|
Chris@49
|
143
|
Chris@49
|
144 arma_debug_check
|
Chris@49
|
145 (
|
Chris@49
|
146 ( ((void*)(&U) == (void*)(&S)) || (&U == &V) || ((void*)(&S) == (void*)(&V)) ),
|
Chris@49
|
147 "svd_econ(): two or more output objects are the same object"
|
Chris@49
|
148 );
|
Chris@49
|
149
|
Chris@49
|
150 arma_debug_check
|
Chris@49
|
151 (
|
Chris@49
|
152 ( (mode != 'l') && (mode != 'r') && (mode != 'b') ),
|
Chris@49
|
153 "svd_econ(): parameter 'mode' is incorrect"
|
Chris@49
|
154 );
|
Chris@49
|
155
|
Chris@49
|
156
|
Chris@49
|
157 // auxlib::svd_econ() makes an internal copy of X
|
Chris@49
|
158 const bool status = auxlib::svd_econ(U, S, V, X, mode);
|
Chris@49
|
159
|
Chris@49
|
160 if(status == false)
|
Chris@49
|
161 {
|
Chris@49
|
162 U.reset();
|
Chris@49
|
163 S.reset();
|
Chris@49
|
164 V.reset();
|
Chris@49
|
165 arma_bad("svd_econ(): failed to converge", false);
|
Chris@49
|
166 }
|
Chris@49
|
167
|
Chris@49
|
168 return status;
|
Chris@49
|
169 }
|
Chris@49
|
170
|
Chris@49
|
171
|
Chris@49
|
172
|
Chris@49
|
173 template<typename T1>
|
Chris@49
|
174 arma_deprecated
|
Chris@49
|
175 inline
|
Chris@49
|
176 bool
|
Chris@49
|
177 svd_thin
|
Chris@49
|
178 (
|
Chris@49
|
179 Mat<typename T1::elem_type>& U,
|
Chris@49
|
180 Col<typename T1::pod_type >& S,
|
Chris@49
|
181 Mat<typename T1::elem_type>& V,
|
Chris@49
|
182 const Base<typename T1::elem_type,T1>& X,
|
Chris@49
|
183 const char mode = 'b',
|
Chris@49
|
184 const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
|
Chris@49
|
185 )
|
Chris@49
|
186 {
|
Chris@49
|
187 arma_ignore(junk);
|
Chris@49
|
188
|
Chris@49
|
189 return svd_econ(U,S,V,X,mode);
|
Chris@49
|
190 }
|
Chris@49
|
191
|
Chris@49
|
192
|
Chris@49
|
193
|
Chris@49
|
194 //! @}
|