123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190 |
- // Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
- // Copyright 2008-2016 National ICT Australia (NICTA)
- //
- // Licensed under the Apache License, Version 2.0 (the "License");
- // you may not use this file except in compliance with the License.
- // You may obtain a copy of the License at
- // http://www.apache.org/licenses/LICENSE-2.0
- //
- // Unless required by applicable law or agreed to in writing, software
- // distributed under the License is distributed on an "AS IS" BASIS,
- // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- // See the License for the specific language governing permissions and
- // limitations under the License.
- // ------------------------------------------------------------------------
- //! \addtogroup fn_svd
- //! @{
- template<typename T1>
- inline
- bool
- svd
- (
- Col<typename T1::pod_type>& S,
- const Base<typename T1::elem_type,T1>& X,
- const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
- )
- {
- arma_extra_debug_sigprint();
- arma_ignore(junk);
-
- // it doesn't matter if X is an alias of S, as auxlib::svd() makes a copy of X
-
- const bool status = auxlib::svd_dc(S, X);
-
- if(status == false)
- {
- S.soft_reset();
- arma_debug_warn("svd(): decomposition failed");
- }
-
- return status;
- }
- template<typename T1>
- arma_warn_unused
- inline
- Col<typename T1::pod_type>
- svd
- (
- const Base<typename T1::elem_type,T1>& X,
- const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
- )
- {
- arma_extra_debug_sigprint();
- arma_ignore(junk);
-
- Col<typename T1::pod_type> out;
-
- const bool status = auxlib::svd_dc(out, X);
-
- if(status == false)
- {
- out.soft_reset();
- arma_stop_runtime_error("svd(): decomposition failed");
- }
-
- return out;
- }
- template<typename T1>
- inline
- bool
- svd
- (
- Mat<typename T1::elem_type>& U,
- Col<typename T1::pod_type >& S,
- Mat<typename T1::elem_type>& V,
- const Base<typename T1::elem_type,T1>& X,
- const char* method = "dc",
- const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
- )
- {
- arma_extra_debug_sigprint();
- arma_ignore(junk);
-
- arma_debug_check
- (
- ( ((void*)(&U) == (void*)(&S)) || (&U == &V) || ((void*)(&S) == (void*)(&V)) ),
- "svd(): two or more output objects are the same object"
- );
-
- const char sig = (method != NULL) ? method[0] : char(0);
-
- arma_debug_check( ((sig != 's') && (sig != 'd')), "svd(): unknown method specified" );
-
- // auxlib::svd() makes an internal copy of X
- const bool status = (sig == 'd') ? auxlib::svd_dc(U, S, V, X) : auxlib::svd(U, S, V, X);
-
- if(status == false)
- {
- U.soft_reset();
- S.soft_reset();
- V.soft_reset();
- arma_debug_warn("svd(): decomposition failed");
- }
-
- return status;
- }
- template<typename T1>
- inline
- bool
- svd_econ
- (
- Mat<typename T1::elem_type>& U,
- Col<typename T1::pod_type >& S,
- Mat<typename T1::elem_type>& V,
- const Base<typename T1::elem_type,T1>& X,
- const char mode,
- const char* method = "dc",
- const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
- )
- {
- arma_extra_debug_sigprint();
- arma_ignore(junk);
-
- arma_debug_check
- (
- ( ((void*)(&U) == (void*)(&S)) || (&U == &V) || ((void*)(&S) == (void*)(&V)) ),
- "svd_econ(): two or more output objects are the same object"
- );
-
- arma_debug_check
- (
- ( (mode != 'l') && (mode != 'r') && (mode != 'b') ),
- "svd_econ(): parameter 'mode' is incorrect"
- );
-
- const char sig = (method != NULL) ? method[0] : char(0);
-
- arma_debug_check( ((sig != 's') && (sig != 'd')), "svd_econ(): unknown method specified" );
-
- const bool status = ((mode == 'b') && (sig == 'd')) ? auxlib::svd_dc_econ(U, S, V, X) : auxlib::svd_econ(U, S, V, X, mode);
-
- if(status == false)
- {
- U.soft_reset();
- S.soft_reset();
- V.soft_reset();
- arma_debug_warn("svd_econ(): decomposition failed");
- }
-
- return status;
- }
- template<typename T1>
- inline
- bool
- svd_econ
- (
- Mat<typename T1::elem_type>& U,
- Col<typename T1::pod_type >& S,
- Mat<typename T1::elem_type>& V,
- const Base<typename T1::elem_type,T1>& X,
- const char* mode = "both",
- const char* method = "dc",
- const typename arma_blas_type_only<typename T1::elem_type>::result* junk = 0
- )
- {
- arma_extra_debug_sigprint();
- arma_ignore(junk);
-
- return svd_econ(U, S, V, X, ((mode != NULL) ? mode[0] : char(0)), method);
- }
- //! @}
|