1 // SPDX-License-Identifier: Apache-2.0
2 //
3 // Copyright 2008-2016 Conrad Sanderson (http://conradsanderson.id.au)
4 // Copyright 2008-2016 National ICT Australia (NICTA)
5 //
6 // Licensed under the Apache License, Version 2.0 (the "License");
7 // you may not use this file except in compliance with the License.
8 // You may obtain a copy of the License at
9 // http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 // ------------------------------------------------------------------------
17
18
19 //! \addtogroup fn_roots
20 //! @{
21
22
23
24 template<typename T1>
25 arma_warn_unused
26 inline
27 typename
28 enable_if2
29 <
30 is_supported_blas_type<typename T1::elem_type>::value,
31 const mtOp<std::complex<typename T1::pod_type>, T1, op_roots>
32 >::result
roots(const Base<typename T1::elem_type,T1> & X)33 roots(const Base<typename T1::elem_type, T1>& X)
34 {
35 arma_extra_debug_sigprint();
36
37 return mtOp<std::complex<typename T1::pod_type>, T1, op_roots>(X.get_ref());
38 }
39
40
41
42 template<typename T1>
43 inline
44 typename
45 enable_if2
46 <
47 is_supported_blas_type<typename T1::elem_type>::value,
48 bool
49 >::result
roots(Mat<std::complex<typename T1::pod_type>> & out,const Base<typename T1::elem_type,T1> & X)50 roots(Mat< std::complex<typename T1::pod_type> >& out, const Base<typename T1::elem_type, T1>& X)
51 {
52 arma_extra_debug_sigprint();
53
54 const bool status = op_roots::apply_direct(out, X.get_ref());
55
56 if(status == false)
57 {
58 out.soft_reset();
59 arma_debug_warn_level(3, "roots(): eigen decomposition failed");
60 }
61
62 return status;
63 }
64
65
66
67 //! @}
68