1 | // This file is part of Eigen, a lightweight C++ template library
|
---|
2 | // for linear algebra.
|
---|
3 | //
|
---|
4 | // Copyright (C) 2008-2010 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_COREITERATORS_H
|
---|
11 | #define EIGEN_COREITERATORS_H
|
---|
12 |
|
---|
13 | namespace Eigen {
|
---|
14 |
|
---|
15 | /* This file contains the respective InnerIterator definition of the expressions defined in Eigen/Core
|
---|
16 | */
|
---|
17 |
|
---|
18 | /** \ingroup SparseCore_Module
|
---|
19 | * \class InnerIterator
|
---|
20 | * \brief An InnerIterator allows to loop over the element of a sparse (or dense) matrix or expression
|
---|
21 | *
|
---|
22 | * todo
|
---|
23 | */
|
---|
24 |
|
---|
25 | // generic version for dense matrix and expressions
|
---|
26 | template<typename Derived> class DenseBase<Derived>::InnerIterator
|
---|
27 | {
|
---|
28 | protected:
|
---|
29 | typedef typename Derived::Scalar Scalar;
|
---|
30 | typedef typename Derived::Index Index;
|
---|
31 |
|
---|
32 | enum { IsRowMajor = (Derived::Flags&RowMajorBit)==RowMajorBit };
|
---|
33 | public:
|
---|
34 | EIGEN_STRONG_INLINE InnerIterator(const Derived& expr, Index outer)
|
---|
35 | : m_expression(expr), m_inner(0), m_outer(outer), m_end(expr.innerSize())
|
---|
36 | {}
|
---|
37 |
|
---|
38 | EIGEN_STRONG_INLINE Scalar value() const
|
---|
39 | {
|
---|
40 | return (IsRowMajor) ? m_expression.coeff(m_outer, m_inner)
|
---|
41 | : m_expression.coeff(m_inner, m_outer);
|
---|
42 | }
|
---|
43 |
|
---|
44 | EIGEN_STRONG_INLINE InnerIterator& operator++() { m_inner++; return *this; }
|
---|
45 |
|
---|
46 | EIGEN_STRONG_INLINE Index index() const { return m_inner; }
|
---|
47 | inline Index row() const { return IsRowMajor ? m_outer : index(); }
|
---|
48 | inline Index col() const { return IsRowMajor ? index() : m_outer; }
|
---|
49 |
|
---|
50 | EIGEN_STRONG_INLINE operator bool() const { return m_inner < m_end && m_inner>=0; }
|
---|
51 |
|
---|
52 | protected:
|
---|
53 | const Derived& m_expression;
|
---|
54 | Index m_inner;
|
---|
55 | const Index m_outer;
|
---|
56 | const Index m_end;
|
---|
57 | };
|
---|
58 |
|
---|
59 | } // end namespace Eigen
|
---|
60 |
|
---|
61 | #endif // EIGEN_COREITERATORS_H
|
---|