11 #ifndef EIGEN_DIAGONALMATRIX_H
12 #define EIGEN_DIAGONALMATRIX_H
16 #ifndef EIGEN_PARSED_BY_DOXYGEN
17 template<
typename Derived>
18 class DiagonalBase :
public EigenBase<Derived>
21 typedef typename internal::traits<Derived>::DiagonalVectorType DiagonalVectorType;
22 typedef typename DiagonalVectorType::Scalar Scalar;
23 typedef typename DiagonalVectorType::RealScalar RealScalar;
24 typedef typename internal::traits<Derived>::StorageKind StorageKind;
25 typedef typename internal::traits<Derived>::Index Index;
28 RowsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
29 ColsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
30 MaxRowsAtCompileTime = DiagonalVectorType::MaxSizeAtCompileTime,
31 MaxColsAtCompileTime = DiagonalVectorType::MaxSizeAtCompileTime,
32 IsVectorAtCompileTime = 0,
36 typedef Matrix<Scalar, RowsAtCompileTime, ColsAtCompileTime, 0, MaxRowsAtCompileTime, MaxColsAtCompileTime> DenseMatrixType;
37 typedef DenseMatrixType DenseType;
38 typedef DiagonalMatrix<Scalar,DiagonalVectorType::SizeAtCompileTime,DiagonalVectorType::MaxSizeAtCompileTime> PlainObject;
40 inline const Derived& derived()
const {
return *
static_cast<const Derived*
>(
this); }
41 inline Derived& derived() {
return *
static_cast<Derived*
>(
this); }
43 DenseMatrixType toDenseMatrix()
const {
return derived(); }
44 template<
typename DenseDerived>
45 void evalTo(MatrixBase<DenseDerived> &other)
const;
46 template<
typename DenseDerived>
47 void addTo(MatrixBase<DenseDerived> &other)
const
48 { other.diagonal() += diagonal(); }
49 template<
typename DenseDerived>
50 void subTo(MatrixBase<DenseDerived> &other)
const
51 { other.diagonal() -= diagonal(); }
53 inline const DiagonalVectorType& diagonal()
const {
return derived().diagonal(); }
54 inline DiagonalVectorType& diagonal() {
return derived().diagonal(); }
56 inline Index rows()
const {
return diagonal().size(); }
57 inline Index cols()
const {
return diagonal().size(); }
61 template<
typename MatrixDerived>
62 const DiagonalProduct<MatrixDerived, Derived, OnTheLeft>
63 operator*(
const MatrixBase<MatrixDerived> &matrix)
const
65 return DiagonalProduct<MatrixDerived, Derived, OnTheLeft>(matrix.derived(), derived());
68 inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_inverse_op<Scalar>,
const DiagonalVectorType> >
71 return diagonal().cwiseInverse();
74 inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_multiple_op<Scalar>,
const DiagonalVectorType> >
77 return diagonal() * scalar;
79 friend inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_multiple_op<Scalar>,
const DiagonalVectorType> >
80 operator*(
const Scalar& scalar,
const DiagonalBase& other)
82 return other.diagonal() * scalar;
86 template<
typename OtherDerived>
87 bool isApprox(
const DiagonalBase<OtherDerived>& other,
typename NumTraits<Scalar>::Real precision = NumTraits<Scalar>::dummy_precision())
const
89 return diagonal().isApprox(other.diagonal(), precision);
91 template<
typename OtherDerived>
92 bool isApprox(
const MatrixBase<OtherDerived>& other,
typename NumTraits<Scalar>::Real precision = NumTraits<Scalar>::dummy_precision())
const
94 return toDenseMatrix().isApprox(other, precision);
99 template<
typename Derived>
100 template<
typename DenseDerived>
101 void DiagonalBase<Derived>::evalTo(MatrixBase<DenseDerived> &other)
const
104 other.diagonal() = diagonal();
122 template<
typename _Scalar,
int SizeAtCompileTime,
int MaxSizeAtCompileTime>
123 struct traits<DiagonalMatrix<_Scalar,SizeAtCompileTime,MaxSizeAtCompileTime> >
124 : traits<Matrix<_Scalar,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
126 typedef Matrix<_Scalar,SizeAtCompileTime,1,0,MaxSizeAtCompileTime,1> DiagonalVectorType;
127 typedef Dense StorageKind;
128 typedef DenseIndex Index;
134 template<
typename _Scalar,
int SizeAtCompileTime,
int MaxSizeAtCompileTime>
136 :
public DiagonalBase<DiagonalMatrix<_Scalar,SizeAtCompileTime,MaxSizeAtCompileTime> >
139 #ifndef EIGEN_PARSED_BY_DOXYGEN
140 typedef typename internal::traits<DiagonalMatrix>::DiagonalVectorType DiagonalVectorType;
142 typedef _Scalar Scalar;
143 typedef typename internal::traits<DiagonalMatrix>::StorageKind StorageKind;
144 typedef typename internal::traits<DiagonalMatrix>::Index Index;
149 DiagonalVectorType m_diagonal;
154 inline const DiagonalVectorType&
diagonal()
const {
return m_diagonal; }
156 inline DiagonalVectorType&
diagonal() {
return m_diagonal; }
168 inline DiagonalMatrix(
const Scalar& x,
const Scalar& y,
const Scalar& z) : m_diagonal(x,y,z) {}
171 template<
typename OtherDerived>
174 #ifndef EIGEN_PARSED_BY_DOXYGEN
180 template<
typename OtherDerived>
185 template<
typename OtherDerived>
188 m_diagonal = other.diagonal();
192 #ifndef EIGEN_PARSED_BY_DOXYGEN
204 inline void resize(Index size) { m_diagonal.resize(size); }
206 inline void setZero() { m_diagonal.setZero(); }
208 inline void setZero(Index size) { m_diagonal.setZero(size); }
230 template<
typename _DiagonalVectorType>
231 struct traits<DiagonalWrapper<_DiagonalVectorType> >
233 typedef _DiagonalVectorType DiagonalVectorType;
234 typedef typename DiagonalVectorType::Scalar Scalar;
235 typedef typename DiagonalVectorType::Index Index;
236 typedef typename DiagonalVectorType::StorageKind StorageKind;
238 RowsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
239 ColsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
240 MaxRowsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
241 MaxColsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
242 Flags = traits<DiagonalVectorType>::Flags &
LvalueBit
247 template<
typename _DiagonalVectorType>
249 :
public DiagonalBase<DiagonalWrapper<_DiagonalVectorType> >, internal::no_assignment_operator
252 #ifndef EIGEN_PARSED_BY_DOXYGEN
253 typedef _DiagonalVectorType DiagonalVectorType;
261 const DiagonalVectorType&
diagonal()
const {
return m_diagonal; }
264 typename DiagonalVectorType::Nested m_diagonal;
276 template<
typename Derived>
277 inline const DiagonalWrapper<const Derived>
291 template<
typename Derived>
295 if(cols() != rows())
return false;
296 RealScalar maxAbsOnDiagonal =
static_cast<RealScalar
>(-1);
297 for(
Index j = 0; j < cols(); ++j)
299 RealScalar absOnDiagonal = abs(coeff(j,j));
300 if(absOnDiagonal > maxAbsOnDiagonal) maxAbsOnDiagonal = absOnDiagonal;
302 for(
Index j = 0; j < cols(); ++j)
303 for(
Index i = 0; i < j; ++i)
305 if(!internal::isMuchSmallerThan(coeff(i, j), maxAbsOnDiagonal, prec))
return false;
306 if(!internal::isMuchSmallerThan(coeff(j, i), maxAbsOnDiagonal, prec))
return false;
313 #endif // EIGEN_DIAGONALMATRIX_H