Chris@49: // Copyright (C) 2009-2012 NICTA (www.nicta.com.au)
Chris@49: // Copyright (C) 2009-2012 Conrad Sanderson
Chris@49: // 
Chris@49: // This Source Code Form is subject to the terms of the Mozilla Public
Chris@49: // License, v. 2.0. If a copy of the MPL was not distributed with this
Chris@49: // file, You can obtain one at http://mozilla.org/MPL/2.0/.
Chris@49: 
Chris@49: 
Chris@49: //! \addtogroup fn_mean
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_inline
Chris@49: const Op<T1, op_mean>
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const uword dim = 0,
Chris@49:   const typename enable_if< is_arma_type<T1>::value       == true  >::result* junk1 = 0,
Chris@49:   const typename enable_if< resolves_to_vector<T1>::value == false >::result* junk2 = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_ignore(junk1);
Chris@49:   arma_ignore(junk2);
Chris@49:   
Chris@49:   return Op<T1, op_mean>(X, dim, 0);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_inline
Chris@49: const Op<T1, op_mean>
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const uword dim,
Chris@49:   const typename enable_if<resolves_to_vector<T1>::value == true>::result* junk = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_ignore(junk);
Chris@49:   
Chris@49:   return Op<T1, op_mean>(X, dim, 0);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: typename T1::elem_type
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const arma_empty_class junk1 = arma_empty_class(),
Chris@49:   const typename enable_if<resolves_to_vector<T1>::value == true>::result* junk2 = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_ignore(junk1);
Chris@49:   arma_ignore(junk2);
Chris@49:   
Chris@49:   return op_mean::mean_all(X);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! \brief
Chris@49: //! Immediate 'find mean value' operation,
Chris@49: //! invoked, for example, by: mean(mean(A))
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: typename T1::elem_type
Chris@49: mean(const Op<T1, op_mean>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_extra_debug_print("mean(): two consecutive mean() calls detected");
Chris@49:   
Chris@49:   return op_mean::mean_all(in.m);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_inline
Chris@49: const Op< Op<T1, op_mean>, op_mean>
Chris@49: mean(const Op<T1, op_mean>& in, const uword dim)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   return Op< Op<T1, op_mean>, op_mean>(in, dim, 0);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T>
Chris@49: arma_inline
Chris@49: arma_warn_unused
Chris@49: const typename arma_scalar_only<T>::result &
Chris@49: mean(const T& x)
Chris@49:   {
Chris@49:   return x;
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: const SpOp<T1, spop_mean>
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const uword dim = 0,
Chris@49:   const typename enable_if< is_arma_sparse_type<T1>::value       == true  >::result* junk1 = 0,
Chris@49:   const typename enable_if< resolves_to_sparse_vector<T1>::value == false >::result* junk2 = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49: 
Chris@49:   arma_ignore(junk1);
Chris@49:   arma_ignore(junk2);
Chris@49: 
Chris@49:   return SpOp<T1, spop_mean>(X, dim, 0);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: const SpOp<T1, spop_mean>
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const uword dim,
Chris@49:   const typename enable_if< resolves_to_sparse_vector<T1>::value == true >::result* junk1 = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_ignore(junk1);
Chris@49: 
Chris@49:   return SpOp<T1, spop_mean>(X, dim, 0);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: typename T1::elem_type
Chris@49: mean
Chris@49:   (
Chris@49:   const T1& X,
Chris@49:   const arma_empty_class junk1 = arma_empty_class(),
Chris@49:   const typename enable_if< resolves_to_sparse_vector<T1>::value == true >::result* junk2 = 0
Chris@49:   )
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49: 
Chris@49:   arma_ignore(junk1);
Chris@49:   arma_ignore(junk2);
Chris@49: 
Chris@49:   return spop_mean::mean_all(X);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: inline
Chris@49: arma_warn_unused
Chris@49: typename T1::elem_type
Chris@49: mean(const SpOp<T1, spop_mean>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   arma_extra_debug_print("mean(): two consecutive mean() calls detected");
Chris@49: 
Chris@49:   return spop_mean::mean_all(in.m);
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}