184 lines
4.2 KiB
C++
184 lines
4.2 KiB
C++
// Copyright (C) 2009-2013 National ICT Australia (NICTA)
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
// -------------------------------------------------------------------
|
|
//
|
|
// Written by Conrad Sanderson - http://conradsanderson.id.au
|
|
|
|
|
|
//! \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.reset();
|
|
arma_debug_warn("svd(): decomposition failed");
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
|
|
|
|
template<typename T1>
|
|
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.reset();
|
|
arma_bad("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.reset();
|
|
S.reset();
|
|
V.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.reset();
|
|
S.reset();
|
|
V.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 = "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);
|
|
}
|
|
|
|
|
|
|
|
//! @}
|