11 #ifndef EIGEN_PERMUTATIONMATRIX_H
12 #define EIGEN_PERMUTATIONMATRIX_H
18 enum PermPermProduct_t {PermPermProduct};
45 template<
typename Derived>
48 typedef internal::traits<Derived> Traits;
52 #ifndef EIGEN_PARSED_BY_DOXYGEN
53 typedef typename Traits::IndicesType IndicesType;
55 Flags = Traits::Flags,
56 RowsAtCompileTime = Traits::RowsAtCompileTime,
57 ColsAtCompileTime = Traits::ColsAtCompileTime,
58 MaxRowsAtCompileTime = Traits::MaxRowsAtCompileTime,
59 MaxColsAtCompileTime = Traits::MaxColsAtCompileTime
61 typedef typename Traits::StorageIndex StorageIndex;
66 typedef PlainPermutationType PlainObject;
73 template<
typename OtherDerived>
81 template<
typename OtherDerived>
82 Derived&
operator=(
const TranspositionsBase<OtherDerived>& tr)
99 #ifndef EIGEN_PARSED_BY_DOXYGEN
100 template<
typename DenseDerived>
105 other.
coeffRef(
indices().coeff(i),i) =
typename DenseDerived::Scalar(1);
133 StorageIndex n = StorageIndex(
size());
134 for(StorageIndex i = 0; i < n; ++i)
157 eigen_assert(i>=0 && j>=0 && i<
size() && j<
size());
160 if(
indices().coeff(k) == i)
indices().coeffRef(k) = StorageIndex(j);
161 else if(
indices().coeff(k) == j)
indices().coeffRef(k) = StorageIndex(i);
176 eigen_assert(i>=0 && j>=0 && i<
size() && j<
size());
186 {
return InverseReturnType(
derived()); }
192 {
return InverseReturnType(
derived()); }
197 #ifndef EIGEN_PARSED_BY_DOXYGEN
199 template<
typename OtherDerived>
204 template<
typename Lhs,
typename Rhs>
205 void assignProduct(
const Lhs& lhs,
const Rhs& rhs)
207 eigen_assert(lhs.cols() == rhs.rows());
208 for (
Index i=0; i<
rows();++i)
indices().coeffRef(i) = lhs.indices().coeff(rhs.indices().coeff(i));
218 template<
typename Other>
220 {
return PlainPermutationType(internal::PermPermProduct,
derived(), other.
derived()); }
226 template<
typename Other>
227 inline PlainPermutationType
operator*(
const InverseImpl<Other,PermutationStorage>& other)
const
228 {
return PlainPermutationType(internal::PermPermProduct, *
this, other.eval()); }
234 template<
typename Other>
friend
236 {
return PlainPermutationType(internal::PermPermProduct, other.eval(), perm); }
252 while(r<n && mask[r]) r++;
272 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename _StorageIndex>
273 struct traits<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, _StorageIndex> >
274 : traits<Matrix<_StorageIndex,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
276 typedef PermutationStorage StorageKind;
277 typedef Matrix<_StorageIndex, SizeAtCompileTime, 1, 0, MaxSizeAtCompileTime, 1> IndicesType;
278 typedef _StorageIndex StorageIndex;
296 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename _StorageIndex>
300 typedef internal::traits<PermutationMatrix> Traits;
305 #ifndef EIGEN_PARSED_BY_DOXYGEN
306 typedef typename Traits::IndicesType IndicesType;
307 typedef typename Traits::StorageIndex StorageIndex;
321 template<
typename OtherDerived>
323 : m_indices(other.
indices()) {}
332 template<
typename Other>
337 template<
typename Other>
339 : m_indices(tr.
size())
345 template<
typename Other>
353 template<
typename Other>
360 const IndicesType&
indices()
const {
return m_indices; }
367 #ifndef EIGEN_PARSED_BY_DOXYGEN
368 template<
typename Other>
370 : m_indices(other.
derived().nestedExpression().
size())
373 StorageIndex end = StorageIndex(m_indices.size());
374 for (StorageIndex i=0; i<end;++i)
375 m_indices.coeffRef(other.derived().nestedExpression().indices().coeff(i)) = i;
377 template<
typename Lhs,
typename Rhs>
378 PermutationMatrix(internal::PermPermProduct_t,
const Lhs& lhs,
const Rhs& rhs)
381 Base::assignProduct(lhs,rhs);
387 IndicesType m_indices;
392 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename _StorageIndex,
int _PacketAccess>
393 struct traits<Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, _StorageIndex>,_PacketAccess> >
394 : traits<Matrix<_StorageIndex,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
396 typedef PermutationStorage StorageKind;
397 typedef Map<const Matrix<_StorageIndex, SizeAtCompileTime, 1, 0, MaxSizeAtCompileTime, 1>, _PacketAccess> IndicesType;
398 typedef _StorageIndex StorageIndex;
403 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename _StorageIndex,
int _PacketAccess>
404 class Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, _StorageIndex>,_PacketAccess>
405 :
public PermutationBase<Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, _StorageIndex>,_PacketAccess> >
407 typedef PermutationBase<Map> Base;
408 typedef internal::traits<Map> Traits;
411 #ifndef EIGEN_PARSED_BY_DOXYGEN
412 typedef typename Traits::IndicesType IndicesType;
413 typedef typename IndicesType::Scalar StorageIndex;
416 inline Map(
const StorageIndex* indicesPtr)
417 : m_indices(indicesPtr)
420 inline Map(
const StorageIndex* indicesPtr,
Index size)
421 : m_indices(indicesPtr,size)
425 template<
typename Other>
426 Map& operator=(
const PermutationBase<Other>& other)
427 {
return Base::operator=(other.derived()); }
430 template<
typename Other>
431 Map& operator=(
const TranspositionsBase<Other>& tr)
432 {
return Base::operator=(tr.derived()); }
434 #ifndef EIGEN_PARSED_BY_DOXYGEN
438 Map& operator=(
const Map& other)
440 m_indices = other.m_indices;
446 const IndicesType& indices()
const {
return m_indices; }
448 IndicesType& indices() {
return m_indices; }
452 IndicesType m_indices;
455 template<
typename _IndicesType>
class TranspositionsWrapper;
457 template<
typename _IndicesType>
458 struct traits<PermutationWrapper<_IndicesType> >
460 typedef PermutationStorage StorageKind;
462 typedef typename _IndicesType::Scalar StorageIndex;
463 typedef _IndicesType IndicesType;
465 RowsAtCompileTime = _IndicesType::SizeAtCompileTime,
466 ColsAtCompileTime = _IndicesType::SizeAtCompileTime,
467 MaxRowsAtCompileTime = IndicesType::MaxSizeAtCompileTime,
468 MaxColsAtCompileTime = IndicesType::MaxSizeAtCompileTime,
485 template<
typename _IndicesType>
489 typedef internal::traits<PermutationWrapper> Traits;
492 #ifndef EIGEN_PARSED_BY_DOXYGEN
493 typedef typename Traits::IndicesType IndicesType;
501 const typename internal::remove_all<typename IndicesType::Nested>::type&
506 typename IndicesType::Nested m_indices;
512 template<
typename MatrixDerived,
typename PermutationDerived>
514 const Product<MatrixDerived, PermutationDerived, AliasFreeProduct>
524 template<
typename PermutationDerived,
typename MatrixDerived>
526 const Product<PermutationDerived, MatrixDerived, AliasFreeProduct>
535 template<
typename PermutationType>
536 class InverseImpl<PermutationType, PermutationStorage>
537 :
public EigenBase<Inverse<PermutationType> >
539 typedef typename PermutationType::PlainPermutationType PlainPermutationType;
540 typedef internal::traits<PermutationType> PermTraits;
544 typedef Inverse<PermutationType> InverseType;
545 using EigenBase<Inverse<PermutationType> >::derived;
547 #ifndef EIGEN_PARSED_BY_DOXYGEN
548 typedef typename PermutationType::DenseMatrixType DenseMatrixType;
550 RowsAtCompileTime = PermTraits::RowsAtCompileTime,
551 ColsAtCompileTime = PermTraits::ColsAtCompileTime,
552 MaxRowsAtCompileTime = PermTraits::MaxRowsAtCompileTime,
553 MaxColsAtCompileTime = PermTraits::MaxColsAtCompileTime
557 #ifndef EIGEN_PARSED_BY_DOXYGEN
558 template<
typename DenseDerived>
559 void evalTo(MatrixBase<DenseDerived>& other)
const
562 for (
Index i=0; i<derived().rows();++i)
563 other.coeffRef(i, derived().nestedExpression().indices().coeff(i)) =
typename DenseDerived::Scalar(1);
568 PlainPermutationType eval()
const {
return derived(); }
570 DenseMatrixType toDenseMatrix()
const {
return derived(); }
574 template<
typename OtherDerived>
friend
575 const Product<OtherDerived, InverseType, AliasFreeProduct>
576 operator*(
const MatrixBase<OtherDerived>& matrix,
const InverseType& trPerm)
578 return Product<OtherDerived, InverseType, AliasFreeProduct>(matrix.derived(), trPerm.derived());
583 template<
typename OtherDerived>
584 const Product<InverseType, OtherDerived, AliasFreeProduct>
585 operator*(
const MatrixBase<OtherDerived>& matrix)
const
587 return Product<InverseType, OtherDerived, AliasFreeProduct>(derived(), matrix.derived());
591 template<
typename Derived>
592 const PermutationWrapper<const Derived> MatrixBase<Derived>::asPermutation()
const
599 template<>
struct AssignmentKind<DenseShape,PermutationShape> {
typedef EigenBase2EigenBase Kind; };
605 #endif // EIGEN_PERMUTATIONMATRIX_H