Chris@49: // Copyright (C) 2012 Ryan Curtin
Chris@49: // Copyright (C) 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 spop_sum
Chris@49: //! @{
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_hot
Chris@49: inline
Chris@49: void
Chris@49: spop_sum::apply(SpMat<typename T1::elem_type>& out, const SpOp<T1,spop_sum>& in)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   typedef typename T1::elem_type eT;
Chris@49:   
Chris@49:   const uword dim = in.aux_uword_a;
Chris@49:   arma_debug_check((dim > 1), "sum(): incorrect usage. dim must be 0 or 1");
Chris@49:   
Chris@49:   const SpProxy<T1> p(in.m);
Chris@49:   
Chris@49:   if(p.is_alias(out) == false)
Chris@49:     {
Chris@49:     spop_sum::apply_noalias(out, p, dim);
Chris@49:     }
Chris@49:   else
Chris@49:     {
Chris@49:     SpMat<eT> tmp;
Chris@49:     
Chris@49:     spop_sum::apply_noalias(tmp, p, dim);
Chris@49:     
Chris@49:     out.steal_mem(tmp);
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: template<typename T1>
Chris@49: arma_hot
Chris@49: inline
Chris@49: void
Chris@49: spop_sum::apply_noalias(SpMat<typename T1::elem_type>& out, const SpProxy<T1>& p, const uword dim)
Chris@49:   {
Chris@49:   arma_extra_debug_sigprint();
Chris@49:   
Chris@49:   if(dim == 0) // find the sum in each column
Chris@49:     {
Chris@49:     out.zeros(1, p.get_n_cols());
Chris@49:     
Chris@49:     typename SpProxy<T1>::const_iterator_type it     = p.begin();
Chris@49:     typename SpProxy<T1>::const_iterator_type it_end = p.end();
Chris@49:     
Chris@49:     while(it != it_end)
Chris@49:       {
Chris@49:       out.at(0, it.col()) += (*it);
Chris@49:       ++it;
Chris@49:       }
Chris@49:     }
Chris@49:   else // find the sum in each row
Chris@49:     {
Chris@49:     out.zeros(p.get_n_rows(), 1);
Chris@49:     
Chris@49:     typename SpProxy<T1>::const_iterator_type it     = p.begin();
Chris@49:     typename SpProxy<T1>::const_iterator_type it_end = p.end();
Chris@49:     
Chris@49:     while(it != it_end)
Chris@49:       {
Chris@49:       out.at(it.row(), 0) += (*it);
Chris@49:       ++it;
Chris@49:       }
Chris@49:     }
Chris@49:   }
Chris@49: 
Chris@49: 
Chris@49: 
Chris@49: //! @}