1// This file is part of Eigen, a lightweight C++ template library
2// for linear algebra.
3//
4// Copyright (C) 2008-2014 Gael Guennebaud <gael.guennebaud@inria.fr>
5//
6// This Source Code Form is subject to the terms of the Mozilla
7// Public License v. 2.0. If a copy of the MPL was not distributed
8// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9
10#ifndef EIGEN_SPARSEREDUX_H
11#define EIGEN_SPARSEREDUX_H
12
13namespace Eigen {
14
15template<typename Derived>
16typename internal::traits<Derived>::Scalar
17SparseMatrixBase<Derived>::sum() const
18{
19  eigen_assert(rows()>0 && cols()>0 && "you are using a non initialized matrix");
20  Scalar res(0);
21  internal::evaluator<Derived> thisEval(derived());
22  for (Index j=0; j<outerSize(); ++j)
23    for (typename internal::evaluator<Derived>::InnerIterator iter(thisEval,j); iter; ++iter)
24      res += iter.value();
25  return res;
26}
27
28template<typename _Scalar, int _Options, typename _Index>
29typename internal::traits<SparseMatrix<_Scalar,_Options,_Index> >::Scalar
30SparseMatrix<_Scalar,_Options,_Index>::sum() const
31{
32  eigen_assert(rows()>0 && cols()>0 && "you are using a non initialized matrix");
33  if(this->isCompressed())
34    return Matrix<Scalar,1,Dynamic>::Map(m_data.valuePtr(), m_data.size()).sum();
35  else
36    return Base::sum();
37}
38
39template<typename _Scalar, int _Options, typename _Index>
40typename internal::traits<SparseVector<_Scalar,_Options, _Index> >::Scalar
41SparseVector<_Scalar,_Options,_Index>::sum() const
42{
43  eigen_assert(rows()>0 && cols()>0 && "you are using a non initialized matrix");
44  return Matrix<Scalar,1,Dynamic>::Map(m_data.valuePtr(), m_data.size()).sum();
45}
46
47} // end namespace Eigen
48
49#endif // EIGEN_SPARSEREDUX_H
50