11 #ifndef EIGEN_DIAGONAL_H 12 #define EIGEN_DIAGONAL_H 36 template<
typename MatrixType,
int DiagIndex>
37 struct traits<Diagonal<MatrixType,DiagIndex> >
40 typedef typename ref_selector<MatrixType>::type MatrixTypeNested;
41 typedef typename remove_reference<MatrixTypeNested>::type _MatrixTypeNested;
42 typedef typename MatrixType::StorageKind StorageKind;
45 : (EIGEN_PLAIN_ENUM_MIN(MatrixType::RowsAtCompileTime - EIGEN_PLAIN_ENUM_MAX(-DiagIndex, 0),
46 MatrixType::ColsAtCompileTime - EIGEN_PLAIN_ENUM_MAX( DiagIndex, 0))),
47 ColsAtCompileTime = 1,
48 MaxRowsAtCompileTime =
int(MatrixType::MaxSizeAtCompileTime) ==
Dynamic ?
Dynamic 49 : DiagIndex ==
DynamicIndex ? EIGEN_SIZE_MIN_PREFER_FIXED(MatrixType::MaxRowsAtCompileTime,
50 MatrixType::MaxColsAtCompileTime)
51 : (EIGEN_PLAIN_ENUM_MIN(MatrixType::MaxRowsAtCompileTime - EIGEN_PLAIN_ENUM_MAX(-DiagIndex, 0),
52 MatrixType::MaxColsAtCompileTime - EIGEN_PLAIN_ENUM_MAX( DiagIndex, 0))),
53 MaxColsAtCompileTime = 1,
54 MaskLvalueBit = is_lvalue<MatrixType>::value ?
LvalueBit : 0,
56 MatrixTypeOuterStride = outer_stride_at_compile_time<MatrixType>::ret,
57 InnerStrideAtCompileTime = MatrixTypeOuterStride ==
Dynamic ?
Dynamic : MatrixTypeOuterStride+1,
58 OuterStrideAtCompileTime = 0
63 template<
typename MatrixType,
int _DiagIndex>
class Diagonal 64 :
public internal::dense_xpr_base< Diagonal<MatrixType,_DiagIndex> >::type
68 enum { DiagIndex = _DiagIndex };
69 typedef typename internal::dense_xpr_base<Diagonal>::type Base;
70 EIGEN_DENSE_PUBLIC_INTERFACE(
Diagonal)
73 explicit inline Diagonal(MatrixType& matrix,
Index a_index = DiagIndex) : m_matrix(matrix), m_index(a_index)
75 eigen_assert( a_index <= m_matrix.cols() && -a_index <= m_matrix.rows() );
78 EIGEN_INHERIT_ASSIGNMENT_OPERATORS(Diagonal)
81 inline Index rows()
const 83 return m_index.value()<0 ? numext::mini<Index>(m_matrix.cols(),m_matrix.rows()+m_index.value())
84 : numext::mini<Index>(m_matrix.rows(),m_matrix.cols()-m_index.value());
87 EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR
88 inline Index cols()
const EIGEN_NOEXCEPT {
return 1; }
90 EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR
91 inline Index innerStride()
const EIGEN_NOEXCEPT {
92 return m_matrix.outerStride() + 1;
95 EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR
96 inline Index outerStride()
const EIGEN_NOEXCEPT {
return 0; }
98 typedef typename internal::conditional<
99 internal::is_lvalue<MatrixType>::value,
102 >::type ScalarWithConstIfNotLvalue;
105 inline ScalarWithConstIfNotLvalue* data() {
return &(m_matrix.coeffRef(rowOffset(), colOffset())); }
107 inline const Scalar* data()
const {
return &(m_matrix.coeffRef(rowOffset(), colOffset())); }
112 EIGEN_STATIC_ASSERT_LVALUE(MatrixType)
113 return m_matrix.coeffRef(row+rowOffset(), row+colOffset());
117 inline const Scalar& coeffRef(
Index row,
Index)
const 119 return m_matrix.coeffRef(row+rowOffset(), row+colOffset());
123 inline CoeffReturnType coeff(
Index row,
Index)
const 125 return m_matrix.coeff(row+rowOffset(), row+colOffset());
129 inline Scalar& coeffRef(
Index idx)
131 EIGEN_STATIC_ASSERT_LVALUE(MatrixType)
132 return m_matrix.coeffRef(idx+rowOffset(), idx+colOffset());
136 inline const Scalar& coeffRef(
Index idx)
const 138 return m_matrix.coeffRef(idx+rowOffset(), idx+colOffset());
142 inline CoeffReturnType coeff(
Index idx)
const 144 return m_matrix.coeff(idx+rowOffset(), idx+colOffset());
148 inline const typename internal::remove_all<typename MatrixType::Nested>::type&
149 nestedExpression()
const 155 inline Index index()
const 157 return m_index.value();
161 typename internal::ref_selector<MatrixType>::non_const_type m_matrix;
162 const internal::variable_if_dynamicindex<Index, DiagIndex> m_index;
166 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR
167 Index absDiagIndex()
const EIGEN_NOEXCEPT {
return m_index.value()>0 ? m_index.value() : -m_index.value(); }
168 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR
169 Index rowOffset()
const EIGEN_NOEXCEPT {
return m_index.value()>0 ? 0 : -m_index.value(); }
170 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE EIGEN_CONSTEXPR
171 Index colOffset()
const EIGEN_NOEXCEPT {
return m_index.value()>0 ? m_index.value() : 0; }
173 template<
int LoadMode>
typename MatrixType::PacketReturnType packet(
Index)
const;
174 template<
int LoadMode>
typename MatrixType::PacketReturnType packet(
Index,
Index)
const;
185 template<
typename Derived>
193 template<
typename Derived>
211 template<
typename Derived>
219 template<
typename Derived>
237 template<
typename Derived>
243 return typename DiagonalIndexReturnType<Index_>::Type(derived());
247 template<
typename Derived>
253 return typename ConstDiagonalIndexReturnType<Index_>::Type(derived());
258 #endif // EIGEN_DIAGONAL_H Expression of the transpose of a matrix.
Definition: Transpose.h:52
const unsigned int DirectAccessBit
Definition: Constants.h:155
const unsigned int LvalueBit
Definition: Constants.h:144
Namespace containing all symbols from the Eigen library.
Definition: Core:141
const int DynamicIndex
Definition: Constants.h:27
const unsigned int RowMajorBit
Definition: Constants.h:66
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74
Definition: Eigen_Colamd.h:50
DiagonalReturnType diagonal()
Definition: Diagonal.h:187
Expression of a diagonal/subdiagonal/superdiagonal in a matrix.
Definition: Diagonal.h:63
const int Dynamic
Definition: Constants.h:22
Base class for all dense matrices, vectors, and expressions.
Definition: MatrixBase.h:48