fn_roots.hpp 1.6 KB

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