fn_chol.hpp 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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_chol
  16. //! @{
  17. template<typename T1>
  18. arma_warn_unused
  19. inline
  20. typename enable_if2< is_supported_blas_type<typename T1::elem_type>::value, const Op<T1, op_chol> >::result
  21. chol
  22. (
  23. const Base<typename T1::elem_type,T1>& X,
  24. const char* layout = "upper"
  25. )
  26. {
  27. arma_extra_debug_sigprint();
  28. const char sig = (layout != NULL) ? layout[0] : char(0);
  29. arma_debug_check( ((sig != 'u') && (sig != 'l')), "chol(): layout must be \"upper\" or \"lower\"" );
  30. return Op<T1, op_chol>(X.get_ref(), ((sig == 'u') ? 0 : 1), 0 );
  31. }
  32. template<typename T1>
  33. inline
  34. typename enable_if2< is_supported_blas_type<typename T1::elem_type>::value, bool >::result
  35. chol
  36. (
  37. Mat<typename T1::elem_type>& out,
  38. const Base<typename T1::elem_type,T1>& X,
  39. const char* layout = "upper"
  40. )
  41. {
  42. arma_extra_debug_sigprint();
  43. const char sig = (layout != NULL) ? layout[0] : char(0);
  44. arma_debug_check( ((sig != 'u') && (sig != 'l')), "chol(): layout must be \"upper\" or \"lower\"" );
  45. const bool status = op_chol::apply_direct(out, X.get_ref(), ((sig == 'u') ? 0 : 1));
  46. if(status == false)
  47. {
  48. out.soft_reset();
  49. arma_debug_warn("chol(): decomposition failed");
  50. }
  51. return status;
  52. }
  53. //! @}