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());
88 inline Index cols()
const {
return 1; }
91 inline Index innerStride()
const 93 return m_matrix.outerStride() + 1;
97 inline Index outerStride()
const 102 typedef typename internal::conditional<
103 internal::is_lvalue<MatrixType>::value,
106 >::type ScalarWithConstIfNotLvalue;
109 inline ScalarWithConstIfNotLvalue* data() {
return &(m_matrix.coeffRef(rowOffset(), colOffset())); }
111 inline const Scalar* data()
const {
return &(m_matrix.coeffRef(rowOffset(), colOffset())); }
116 EIGEN_STATIC_ASSERT_LVALUE(MatrixType)
117 return m_matrix.coeffRef(row+rowOffset(), row+colOffset());
121 inline const Scalar& coeffRef(
Index row,
Index)
const 123 return m_matrix.coeffRef(row+rowOffset(), row+colOffset());
127 inline CoeffReturnType coeff(
Index row,
Index)
const 129 return m_matrix.coeff(row+rowOffset(), row+colOffset());
133 inline Scalar& coeffRef(
Index idx)
135 EIGEN_STATIC_ASSERT_LVALUE(MatrixType)
136 return m_matrix.coeffRef(idx+rowOffset(), idx+colOffset());
140 inline const Scalar& coeffRef(
Index idx)
const 142 return m_matrix.coeffRef(idx+rowOffset(), idx+colOffset());
146 inline CoeffReturnType coeff(
Index idx)
const 148 return m_matrix.coeff(idx+rowOffset(), idx+colOffset());
152 inline const typename internal::remove_all<typename MatrixType::Nested>::type&
153 nestedExpression()
const 159 inline Index index()
const 161 return m_index.value();
165 typename internal::ref_selector<MatrixType>::non_const_type m_matrix;
166 const internal::variable_if_dynamicindex<Index, DiagIndex> m_index;
171 EIGEN_STRONG_INLINE
Index absDiagIndex()
const {
return m_index.value()>0 ? m_index.value() : -m_index.value(); }
173 EIGEN_STRONG_INLINE
Index rowOffset()
const {
return m_index.value()>0 ? 0 : -m_index.value(); }
175 EIGEN_STRONG_INLINE
Index colOffset()
const {
return m_index.value()>0 ? m_index.value() : 0; }
177 template<
int LoadMode>
typename MatrixType::PacketReturnType packet(
Index)
const;
178 template<
int LoadMode>
typename MatrixType::PacketReturnType packet(
Index,
Index)
const;
189 template<
typename Derived>
197 template<
typename Derived>
215 template<
typename Derived>
223 template<
typename Derived>
241 template<
typename Derived>
246 return typename DiagonalIndexReturnType<Index_>::Type(derived());
250 template<
typename Derived>
252 inline typename MatrixBase<Derived>::template ConstDiagonalIndexReturnType<Index_>::Type
255 return typename ConstDiagonalIndexReturnType<Index_>::Type(derived());
260 #endif // EIGEN_DIAGONAL_H Expression of the transpose of a matrix.
Definition: Transpose.h:52
const unsigned int DirectAccessBit
Definition: Constants.h:150
const unsigned int LvalueBit
Definition: Constants.h:139
Namespace containing all symbols from the Eigen library.
Definition: Core:306
const int DynamicIndex
Definition: Constants.h:26
Eigen::Index Index
The interface type of indices.
Definition: EigenBase.h:38
const unsigned int RowMajorBit
Definition: Constants.h:61
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:33
DiagonalReturnType diagonal()
Definition: Diagonal.h:191
Expression of a diagonal/subdiagonal/superdiagonal in a matrix.
Definition: Diagonal.h:63
const int Dynamic
Definition: Constants.h:21
Base class for all dense matrices, vectors, and expressions.
Definition: MatrixBase.h:48