1// This file is part of Eigen, a lightweight C++ template library
2// for linear algebra.
3//
4// Copyright (C) 2008-2014 Gael Guennebaud <[email protected]>
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_SPARSEUTIL_H
11#define EIGEN_SPARSEUTIL_H
12
13namespace Eigen {
14
15#ifdef NDEBUG
16#define EIGEN_DBG_SPARSE(X)
17#else
18#define EIGEN_DBG_SPARSE(X) X
19#endif
20
21#define EIGEN_SPARSE_INHERIT_ASSIGNMENT_OPERATOR(Derived, Op) \
22template<typename OtherDerived> \
23EIGEN_STRONG_INLINE Derived& operator Op(const Eigen::SparseMatrixBase<OtherDerived>& other) \
24{ \
25 return Base::operator Op(other.derived()); \
26} \
27EIGEN_STRONG_INLINE Derived& operator Op(const Derived& other) \
28{ \
29 return Base::operator Op(other); \
30}
31
32#define EIGEN_SPARSE_INHERIT_SCALAR_ASSIGNMENT_OPERATOR(Derived, Op) \
33template<typename Other> \
34EIGEN_STRONG_INLINE Derived& operator Op(const Other& scalar) \
35{ \
36 return Base::operator Op(scalar); \
37}
38
39#define EIGEN_SPARSE_INHERIT_ASSIGNMENT_OPERATORS(Derived) \
40EIGEN_SPARSE_INHERIT_ASSIGNMENT_OPERATOR(Derived, =)
41
42
43#define EIGEN_SPARSE_PUBLIC_INTERFACE(Derived) \
44 EIGEN_GENERIC_PUBLIC_INTERFACE(Derived)
45
46
47const int CoherentAccessPattern = 0x1;
48const int InnerRandomAccessPattern = 0x2 | CoherentAccessPattern;
49const int OuterRandomAccessPattern = 0x4 | CoherentAccessPattern;
50const int RandomAccessPattern = 0x8 | OuterRandomAccessPattern | InnerRandomAccessPattern;
51
52template<typename _Scalar, int _Flags = 0, typename _StorageIndex = int> class SparseMatrix;
53template<typename _Scalar, int _Flags = 0, typename _StorageIndex = int> class DynamicSparseMatrix;
54template<typename _Scalar, int _Flags = 0, typename _StorageIndex = int> class SparseVector;
55template<typename _Scalar, int _Flags = 0, typename _StorageIndex = int> class MappedSparseMatrix;
56
57template<typename MatrixType, unsigned int UpLo> class SparseSelfAdjointView;
58template<typename Lhs, typename Rhs> class SparseDiagonalProduct;
59template<typename MatrixType> class SparseView;
60
61template<typename Lhs, typename Rhs> class SparseSparseProduct;
62template<typename Lhs, typename Rhs> class SparseTimeDenseProduct;
63template<typename Lhs, typename Rhs> class DenseTimeSparseProduct;
64template<typename Lhs, typename Rhs, bool Transpose> class SparseDenseOuterProduct;
65
66template<typename Lhs, typename Rhs> struct SparseSparseProductReturnType;
67template<typename Lhs, typename Rhs,
68 int InnerSize = EIGEN_SIZE_MIN_PREFER_FIXED(internal::traits<Lhs>::ColsAtCompileTime,internal::traits<Rhs>::RowsAtCompileTime)> struct DenseSparseProductReturnType;
69
70template<typename Lhs, typename Rhs,
71 int InnerSize = EIGEN_SIZE_MIN_PREFER_FIXED(internal::traits<Lhs>::ColsAtCompileTime,internal::traits<Rhs>::RowsAtCompileTime)> struct SparseDenseProductReturnType;
72template<typename MatrixType,int UpLo> class SparseSymmetricPermutationProduct;
73
74namespace internal {
75
76template<typename T,int Rows,int Cols,int Flags> struct sparse_eval;
77
78template<typename T> struct eval<T,Sparse>
79 : sparse_eval<T, traits<T>::RowsAtCompileTime,traits<T>::ColsAtCompileTime,traits<T>::Flags>
80{};
81
82template<typename T,int Cols,int Flags> struct sparse_eval<T,1,Cols,Flags> {
83 typedef typename traits<T>::Scalar _Scalar;
84 typedef typename traits<T>::StorageIndex _StorageIndex;
85 public:
86 typedef SparseVector<_Scalar, RowMajor, _StorageIndex> type;
87};
88
89template<typename T,int Rows,int Flags> struct sparse_eval<T,Rows,1,Flags> {
90 typedef typename traits<T>::Scalar _Scalar;
91 typedef typename traits<T>::StorageIndex _StorageIndex;
92 public:
93 typedef SparseVector<_Scalar, ColMajor, _StorageIndex> type;
94};
95
96// TODO this seems almost identical to plain_matrix_type<T, Sparse>
97template<typename T,int Rows,int Cols,int Flags> struct sparse_eval {
98 typedef typename traits<T>::Scalar _Scalar;
99 typedef typename traits<T>::StorageIndex _StorageIndex;
100 enum { _Options = ((Flags&RowMajorBit)==RowMajorBit) ? RowMajor : ColMajor };
101 public:
102 typedef SparseMatrix<_Scalar, _Options, _StorageIndex> type;
103};
104
105template<typename T,int Flags> struct sparse_eval<T,1,1,Flags> {
106 typedef typename traits<T>::Scalar _Scalar;
107 public:
108 typedef Matrix<_Scalar, 1, 1> type;
109};
110
111template<typename T> struct plain_matrix_type<T,Sparse>
112{
113 typedef typename traits<T>::Scalar _Scalar;
114 typedef typename traits<T>::StorageIndex _StorageIndex;
115 enum { _Options = ((evaluator<T>::Flags&RowMajorBit)==RowMajorBit) ? RowMajor : ColMajor };
116 public:
117 typedef SparseMatrix<_Scalar, _Options, _StorageIndex> type;
118};
119
120template<typename T>
121struct plain_object_eval<T,Sparse>
122 : sparse_eval<T, traits<T>::RowsAtCompileTime,traits<T>::ColsAtCompileTime, evaluator<T>::Flags>
123{};
124
125template<typename Decomposition, typename RhsType>
126struct solve_traits<Decomposition,RhsType,Sparse>
127{
128 typedef typename sparse_eval<RhsType, RhsType::RowsAtCompileTime, RhsType::ColsAtCompileTime,traits<RhsType>::Flags>::type PlainObject;
129};
130
131template<typename Derived>
132struct generic_xpr_base<Derived, MatrixXpr, Sparse>
133{
134 typedef SparseMatrixBase<Derived> type;
135};
136
137struct SparseTriangularShape { static std::string debugName() { return "SparseTriangularShape"; } };
138struct SparseSelfAdjointShape { static std::string debugName() { return "SparseSelfAdjointShape"; } };
139
140template<> struct glue_shapes<SparseShape,SelfAdjointShape> { typedef SparseSelfAdjointShape type; };
141template<> struct glue_shapes<SparseShape,TriangularShape > { typedef SparseTriangularShape type; };
142
143} // end namespace internal
144
145/** \ingroup SparseCore_Module
146 *
147 * \class Triplet
148 *
149 * \brief A small structure to hold a non zero as a triplet (i,j,value).
150 *
151 * \sa SparseMatrix::setFromTriplets()
152 */
153template<typename Scalar, typename StorageIndex=typename SparseMatrix<Scalar>::StorageIndex >
154class Triplet
155{
156public:
157 Triplet() : m_row(0), m_col(0), m_value(0) {}
158
159 Triplet(const StorageIndex& i, const StorageIndex& j, const Scalar& v = Scalar(0))
160 : m_row(i), m_col(j), m_value(v)
161 {}
162
163 /** \returns the row index of the element */
164 const StorageIndex& row() const { return m_row; }
165
166 /** \returns the column index of the element */
167 const StorageIndex& col() const { return m_col; }
168
169 /** \returns the value of the element */
170 const Scalar& value() const { return m_value; }
171protected:
172 StorageIndex m_row, m_col;
173 Scalar m_value;
174};
175
176} // end namespace Eigen
177
178#endif // EIGEN_SPARSEUTIL_H
179