From eda5bc26f44ee9a6f83dcf8c91f17296d7fc509d Mon Sep 17 00:00:00 2001 From: Nao Pross Date: Mon, 12 Feb 2024 14:52:43 +0100 Subject: Move into version control --- .../include/armadillo_bits/fn_eig_sym.hpp | 161 +++++++++++++++++++++ 1 file changed, 161 insertions(+) create mode 100644 src/armadillo/include/armadillo_bits/fn_eig_sym.hpp (limited to 'src/armadillo/include/armadillo_bits/fn_eig_sym.hpp') diff --git a/src/armadillo/include/armadillo_bits/fn_eig_sym.hpp b/src/armadillo/include/armadillo_bits/fn_eig_sym.hpp new file mode 100644 index 0000000..12043b6 --- /dev/null +++ b/src/armadillo/include/armadillo_bits/fn_eig_sym.hpp @@ -0,0 +1,161 @@ +// SPDX-License-Identifier: Apache-2.0 +// +// 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_eig_sym +//! @{ + + +//! Eigenvalues of real/complex symmetric/hermitian matrix X +template +inline +typename enable_if2< is_supported_blas_type::value, bool >::result +eig_sym + ( + Col& eigval, + const Base& X + ) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + Mat A(X.get_ref()); + + const bool status = auxlib::eig_sym(eigval, A); + + if(status == false) + { + eigval.soft_reset(); + arma_debug_warn_level(3, "eig_sym(): decomposition failed"); + } + + return status; + } + + + +//! Eigenvalues of real/complex symmetric/hermitian matrix X +template +arma_warn_unused +inline +typename enable_if2< is_supported_blas_type::value, Col >::result +eig_sym + ( + const Base& X + ) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + typedef typename T1::pod_type T; + + Col< T> eigval; + Mat A(X.get_ref()); + + const bool status = auxlib::eig_sym(eigval, A); + + if(status == false) + { + eigval.reset(); + arma_stop_runtime_error("eig_sym(): decomposition failed"); + } + + return eigval; + } + + + +//! internal helper function +template +inline +bool +eig_sym_helper + ( + Col::result>& eigval, + Mat& eigvec, + const Mat& X, + const char method_sig, + const char* caller_sig + ) + { + arma_extra_debug_sigprint(); + + if((arma_config::debug) && (auxlib::rudimentary_sym_check(X) == false)) + { + if(is_cx::no ) { arma_debug_warn_level(1, caller_sig, ": given matrix is not symmetric"); } + if(is_cx::yes) { arma_debug_warn_level(1, caller_sig, ": given matrix is not hermitian"); } + } + + bool status = false; + + if(method_sig == 'd') { status = auxlib::eig_sym_dc(eigval, eigvec, X); } + + if(status == false) { status = auxlib::eig_sym(eigval, eigvec, X); } + + return status; + } + + + +//! Eigenvalues and eigenvectors of real/complex symmetric/hermitian matrix X +template +inline +typename enable_if2< is_supported_blas_type::value, bool >::result +eig_sym + ( + Col& eigval, + Mat& eigvec, + const Base& expr, + const char* method = "dc" + ) + { + arma_extra_debug_sigprint(); + + typedef typename T1::elem_type eT; + + const char sig = (method != nullptr) ? method[0] : char(0); + + arma_debug_check( ((sig != 's') && (sig != 'd')), "eig_sym(): unknown method specified" ); + arma_debug_check( void_ptr(&eigval) == void_ptr(&eigvec), "eig_sym(): parameter 'eigval' is an alias of parameter 'eigvec'" ); + + const quasi_unwrap U(expr.get_ref()); + + const bool is_alias = U.is_alias(eigvec); + + Mat eigvec_tmp; + Mat& eigvec_out = (is_alias == false) ? eigvec : eigvec_tmp; + + const bool status = eig_sym_helper(eigval, eigvec_out, U.M, sig, "eig_sym()"); + + if(status == false) + { + eigval.soft_reset(); + eigvec.soft_reset(); + arma_debug_warn_level(3, "eig_sym(): decomposition failed"); + } + else + { + if(is_alias) { eigvec.steal_mem(eigvec_tmp); } + } + + return status; + } + + + +//! @} -- cgit v1.2.1