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(); }
59 template<
typename MatrixDerived>
60 const DiagonalProduct<MatrixDerived, Derived, OnTheLeft>
61 operator*(
const MatrixBase<MatrixDerived> &matrix)
const;
63 inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_inverse_op<Scalar>,
const DiagonalVectorType> >
66 return diagonal().cwiseInverse();
69 inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_multiple_op<Scalar>,
const DiagonalVectorType> >
72 return diagonal() * scalar;
74 friend inline const DiagonalWrapper<const CwiseUnaryOp<internal::scalar_multiple_op<Scalar>,
const DiagonalVectorType> >
75 operator*(
const Scalar& scalar,
const DiagonalBase& other)
77 return other.diagonal() * scalar;
81 template<
typename OtherDerived>
82 bool isApprox(
const DiagonalBase<OtherDerived>& other,
typename NumTraits<Scalar>::Real precision = NumTraits<Scalar>::dummy_precision())
const
84 return diagonal().isApprox(other.diagonal(), precision);
86 template<
typename OtherDerived>
87 bool isApprox(
const MatrixBase<OtherDerived>& other,
typename NumTraits<Scalar>::Real precision = NumTraits<Scalar>::dummy_precision())
const
89 return toDenseMatrix().isApprox(other, precision);
94 template<
typename Derived>
95 template<
typename DenseDerived>
96 void DiagonalBase<Derived>::evalTo(MatrixBase<DenseDerived> &other)
const
99 other.diagonal() = diagonal();
117 template<
typename _Scalar,
int SizeAtCompileTime,
int MaxSizeAtCompileTime>
118 struct traits<DiagonalMatrix<_Scalar,SizeAtCompileTime,MaxSizeAtCompileTime> >
119 : traits<Matrix<_Scalar,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
121 typedef Matrix<_Scalar,SizeAtCompileTime,1,0,MaxSizeAtCompileTime,1> DiagonalVectorType;
122 typedef Dense StorageKind;
123 typedef DenseIndex Index;
129 template<
typename _Scalar,
int SizeAtCompileTime,
int MaxSizeAtCompileTime>
131 :
public DiagonalBase<DiagonalMatrix<_Scalar,SizeAtCompileTime,MaxSizeAtCompileTime> >
134 #ifndef EIGEN_PARSED_BY_DOXYGEN
135 typedef typename internal::traits<DiagonalMatrix>::DiagonalVectorType DiagonalVectorType;
137 typedef _Scalar Scalar;
138 typedef typename internal::traits<DiagonalMatrix>::StorageKind StorageKind;
139 typedef typename internal::traits<DiagonalMatrix>::Index Index;
144 DiagonalVectorType m_diagonal;
149 inline const DiagonalVectorType&
diagonal()
const {
return m_diagonal; }
151 inline DiagonalVectorType&
diagonal() {
return m_diagonal; }
163 inline DiagonalMatrix(
const Scalar& x,
const Scalar& y,
const Scalar& z) : m_diagonal(x,y,z) {}
166 template<
typename OtherDerived>
169 #ifndef EIGEN_PARSED_BY_DOXYGEN
175 template<
typename OtherDerived>
180 template<
typename OtherDerived>
183 m_diagonal = other.diagonal();
187 #ifndef EIGEN_PARSED_BY_DOXYGEN
199 inline void resize(Index size) { m_diagonal.resize(size); }
201 inline void setZero() { m_diagonal.setZero(); }
203 inline void setZero(Index size) { m_diagonal.setZero(size); }
225 template<
typename _DiagonalVectorType>
226 struct traits<DiagonalWrapper<_DiagonalVectorType> >
228 typedef _DiagonalVectorType DiagonalVectorType;
229 typedef typename DiagonalVectorType::Scalar Scalar;
230 typedef typename DiagonalVectorType::Index Index;
231 typedef typename DiagonalVectorType::StorageKind StorageKind;
233 RowsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
234 ColsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
235 MaxRowsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
236 MaxColsAtCompileTime = DiagonalVectorType::SizeAtCompileTime,
237 Flags = traits<DiagonalVectorType>::Flags &
LvalueBit
242 template<
typename _DiagonalVectorType>
244 :
public DiagonalBase<DiagonalWrapper<_DiagonalVectorType> >, internal::no_assignment_operator
247 #ifndef EIGEN_PARSED_BY_DOXYGEN
248 typedef _DiagonalVectorType DiagonalVectorType;
256 const DiagonalVectorType&
diagonal()
const {
return m_diagonal; }
259 typename DiagonalVectorType::Nested m_diagonal;
271 template<
typename Derived>
272 inline const DiagonalWrapper<const Derived>
286 template<
typename Derived>
289 if(cols() != rows())
return false;
290 RealScalar maxAbsOnDiagonal =
static_cast<RealScalar
>(-1);
291 for(
Index j = 0; j < cols(); ++j)
293 RealScalar absOnDiagonal = internal::abs(coeff(j,j));
294 if(absOnDiagonal > maxAbsOnDiagonal) maxAbsOnDiagonal = absOnDiagonal;
296 for(
Index j = 0; j < cols(); ++j)
297 for(
Index i = 0; i < j; ++i)
299 if(!internal::isMuchSmallerThan(coeff(i, j), maxAbsOnDiagonal, prec))
return false;
300 if(!internal::isMuchSmallerThan(coeff(j, i), maxAbsOnDiagonal, prec))
return false;
307 #endif // EIGEN_DIAGONALMATRIX_H