fn_numel.hpp 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  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_numel
  16. //! @{
  17. template<typename T1>
  18. arma_warn_unused
  19. inline
  20. typename enable_if2< is_arma_type<T1>::value, uword >::result
  21. numel(const T1& X)
  22. {
  23. arma_extra_debug_sigprint();
  24. const Proxy<T1> P(X);
  25. return P.get_n_elem();
  26. }
  27. template<typename T1>
  28. arma_warn_unused
  29. inline
  30. typename enable_if2< is_arma_cube_type<T1>::value, uword >::result
  31. numel(const T1& X)
  32. {
  33. arma_extra_debug_sigprint();
  34. const ProxyCube<T1> P(X);
  35. return P.get_n_elem();
  36. }
  37. template<typename T1>
  38. arma_warn_unused
  39. inline
  40. typename enable_if2< is_arma_sparse_type<T1>::value, uword >::result
  41. numel(const T1& X)
  42. {
  43. arma_extra_debug_sigprint();
  44. const SpProxy<T1> P(X);
  45. return P.get_n_elem();
  46. }
  47. template<typename oT>
  48. arma_warn_unused
  49. inline
  50. uword
  51. numel(const field<oT>& X)
  52. {
  53. arma_extra_debug_sigprint();
  54. return X.n_elem;
  55. }
  56. template<typename oT>
  57. arma_warn_unused
  58. inline
  59. uword
  60. numel(const subview_field<oT>& X)
  61. {
  62. arma_extra_debug_sigprint();
  63. return X.n_elem;
  64. }
  65. //! @}