max@0: // Copyright (C) 2008-2011 NICTA (www.nicta.com.au)
max@0: // Copyright (C) 2008-2011 Conrad Sanderson
max@0: // 
max@0: // This file is part of the Armadillo C++ library.
max@0: // It is provided without any warranty of fitness
max@0: // for any purpose. You can redistribute this file
max@0: // and/or modify it under the terms of the GNU
max@0: // Lesser General Public License (LGPL) as published
max@0: // by the Free Software Foundation, either version 3
max@0: // of the License or (at your option) any later version.
max@0: // (see http://www.opensource.org/licenses for more info)
max@0: 
max@0: 
max@0: //! \addtogroup fn_accu
max@0: //! @{
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a matrix
max@0: template<typename T1>
max@0: arma_hot
max@0: inline
max@0: typename T1::elem_type
max@0: accu(const Base<typename T1::elem_type,T1>& X)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   typedef typename T1::elem_type      eT;
max@0:   typedef typename Proxy<T1>::ea_type ea_type;
max@0:   
max@0:   const Proxy<T1> A(X.get_ref());
max@0:   
max@0:   if(Proxy<T1>::prefer_at_accessor == false)
max@0:     {
max@0:           ea_type P      = A.get_ea();
max@0:     const uword   n_elem = A.get_n_elem();
max@0:     
max@0:     eT val1 = eT(0);
max@0:     eT val2 = eT(0);
max@0:     
max@0:     uword i,j;
max@0:     
max@0:     for(i=0, j=1; j<n_elem; i+=2, j+=2)
max@0:       {
max@0:       val1 += P[i];
max@0:       val2 += P[j];
max@0:       }
max@0:     
max@0:     if(i < n_elem)
max@0:       {
max@0:       val1 += P[i];
max@0:       }
max@0:     
max@0:     return val1 + val2;
max@0:     }
max@0:   else
max@0:     {
max@0:     const uword n_rows = A.get_n_rows();
max@0:     const uword n_cols = A.get_n_cols();
max@0:     
max@0:     eT val = eT(0);
max@0:     
max@0:     for(uword col=0; col<n_cols; ++col)
max@0:       {
max@0:       for(uword row=0; row<n_rows; ++row)
max@0:         {
max@0:         val += A.at(row,col);
max@0:         }
max@0:       }
max@0:     
max@0:     return val;
max@0:     }
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! explicit handling of Hamming norm (also known as zero norm)
max@0: template<typename T1>
max@0: arma_inline
max@0: arma_warn_unused
max@0: uword
max@0: accu(const mtOp<uword,T1,op_rel_noteq>& X)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   typedef typename T1::elem_type eT;
max@0:   
max@0:   const Proxy<T1> A(X.m);
max@0:   
max@0:   const uword n_elem = A.get_n_elem();
max@0:   const eT    val    = X.aux;
max@0:   
max@0:   uword n_nonzero = 0;
max@0:   for(uword i=0; i<n_elem; ++i)
max@0:     {
max@0:     if(A[i] != val)
max@0:       {
max@0:       ++n_nonzero;
max@0:       }
max@0:     }
max@0:   
max@0:   return n_nonzero;
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a cube
max@0: template<typename T1>
max@0: arma_hot
max@0: arma_warn_unused
max@0: inline
max@0: typename T1::elem_type
max@0: accu(const BaseCube<typename T1::elem_type,T1>& X)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   typedef typename T1::elem_type          eT;
max@0:   typedef typename ProxyCube<T1>::ea_type ea_type;
max@0:   
max@0:   const ProxyCube<T1> A(X.get_ref());
max@0:   
max@0:   if(ProxyCube<T1>::prefer_at_accessor == false)
max@0:     {
max@0:     
max@0:           ea_type P      = A.get_ea();
max@0:     const uword   n_elem = A.get_n_elem();
max@0:     
max@0:     eT val1 = eT(0);
max@0:     eT val2 = eT(0);
max@0:     
max@0:     uword i,j;
max@0:     
max@0:     for(i=0, j=1; j<n_elem; i+=2, j+=2)
max@0:       {
max@0:       val1 += P[i];
max@0:       val2 += P[j];
max@0:       }
max@0:     
max@0:     if(i < n_elem)
max@0:       {
max@0:       val1 += P[i];
max@0:       }
max@0:     
max@0:     return val1 + val2;
max@0:     }
max@0:   else
max@0:     {
max@0:     const uword n_rows   = A.get_n_rows();
max@0:     const uword n_cols   = A.get_n_cols();
max@0:     const uword n_slices = A.get_n_slices();
max@0:     
max@0:     eT val = eT(0);
max@0:     
max@0:     for(uword slice=0; slice<n_slices; ++slice)
max@0:       {
max@0:       for(uword col=0; col<n_cols; ++col)
max@0:         {
max@0:         for(uword row=0; row<n_rows; ++row)
max@0:           {
max@0:           val += A.at(row,col,slice);
max@0:           }
max@0:         }
max@0:       }
max@0:     
max@0:     return val;
max@0:     }
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a diagview
max@0: template<typename eT>
max@0: arma_pure
max@0: arma_warn_unused
max@0: inline
max@0: eT
max@0: accu(const diagview<eT>& X)
max@0:   {
max@0:   arma_extra_debug_sigprint();  
max@0:   
max@0:   const uword n_elem = X.n_elem;
max@0:   
max@0:   eT val = eT(0);
max@0:   
max@0:   for(uword i=0; i<n_elem; ++i)
max@0:     {
max@0:     val += X[i];
max@0:     }
max@0:   
max@0:   return val;
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a subview (submatrix)
max@0: template<typename eT>
max@0: arma_pure
max@0: arma_warn_unused
max@0: inline
max@0: eT
max@0: accu(const subview<eT>& S)
max@0:   {
max@0:   arma_extra_debug_sigprint();  
max@0:   
max@0:   const uword S_n_rows = S.n_rows;
max@0:   const uword S_n_cols = S.n_cols;
max@0:   const uword S_n_elem = S.n_elem;
max@0:   
max@0:   eT val = eT(0);
max@0:   
max@0:   if(S_n_elem > 0)
max@0:     {
max@0:     for(uword col=0; col<S_n_cols; ++col)
max@0:       {
max@0:       val += arrayops::accumulate( S.colptr(col), S_n_rows );
max@0:       }
max@0:     }
max@0:   
max@0:   return val;
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a subview_row
max@0: template<typename eT>
max@0: arma_pure
max@0: arma_warn_unused
max@0: inline
max@0: eT
max@0: accu(const subview_row<eT>& S)
max@0:   {
max@0:   arma_extra_debug_sigprint();  
max@0:   
max@0:   const Mat<eT>& X = S.m;
max@0:   
max@0:   const uword n_elem     = S.n_elem;
max@0:   const uword row        = S.aux_row1;
max@0:   const uword start_col  = S.aux_col1;
max@0:   const uword end_col_p1 = start_col + S.n_cols;
max@0:   
max@0:   eT val = eT(0);
max@0:     
max@0:   if(n_elem > 0)
max@0:     {
max@0:     uword i,j;
max@0:     
max@0:     for(i=start_col, j=start_col+1; j < end_col_p1; i+=2, j+=2)
max@0:       {
max@0:       val += X.at(row,i);
max@0:       val += X.at(row,j);
max@0:       }
max@0:     
max@0:     if(i < end_col_p1)
max@0:       {
max@0:       val += X.at(row,i);
max@0:       }
max@0:     }
max@0:   
max@0:   return val;
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! accumulate the elements of a subview_col
max@0: template<typename eT>
max@0: arma_pure
max@0: arma_warn_unused
max@0: inline
max@0: eT
max@0: accu(const subview_col<eT>& S)
max@0:   {
max@0:   arma_extra_debug_sigprint();
max@0:   
max@0:   return (S.n_elem > 0) ? arrayops::accumulate( S.colptr(0), S.n_rows ) : eT(0);
max@0:   }
max@0: 
max@0: 
max@0: 
max@0: //! @}