AnalysisSystemForRadionucli.../include/armadillo_bits/fn_svd.hpp
2024-06-04 15:25:02 +08:00

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);
}
//! @}