11 #ifndef EIGEN_PERMUTATIONMATRIX_H
12 #define EIGEN_PERMUTATIONMATRIX_H
16 template<
int RowCol,
typename IndicesType,
typename MatrixType,
typename StorageKind>
class PermutedImpl;
44 template<
typename PermutationType,
typename MatrixType,
int S
ide,
bool Transposed=false>
45 struct permut_matrix_product_retval;
46 template<
typename PermutationType,
typename MatrixType,
int S
ide,
bool Transposed=false>
47 struct permut_sparsematrix_product_retval;
48 enum PermPermProduct_t {PermPermProduct};
52 template<
typename Derived>
55 typedef internal::traits<Derived> Traits;
59 #ifndef EIGEN_PARSED_BY_DOXYGEN
60 typedef typename Traits::IndicesType IndicesType;
62 Flags = Traits::Flags,
63 CoeffReadCost = Traits::CoeffReadCost,
64 RowsAtCompileTime = Traits::RowsAtCompileTime,
65 ColsAtCompileTime = Traits::ColsAtCompileTime,
66 MaxRowsAtCompileTime = Traits::MaxRowsAtCompileTime,
67 MaxColsAtCompileTime = Traits::MaxColsAtCompileTime
69 typedef typename Traits::Scalar Scalar;
70 typedef typename Traits::Index Index;
79 template<
typename OtherDerived>
87 template<
typename OtherDerived>
88 Derived&
operator=(
const TranspositionsBase<OtherDerived>& tr)
91 for(Index k=
size()-1; k>=0; --k)
96 #ifndef EIGEN_PARSED_BY_DOXYGEN
116 #ifndef EIGEN_PARSED_BY_DOXYGEN
117 template<
typename DenseDerived>
121 for (
int i=0; i<
rows();++i)
122 other.coeffRef(
indices().coeff(i),i) =
typename DenseDerived::Scalar(1);
150 for(Index i = 0; i <
size(); ++i)
173 eigen_assert(i>=0 && j>=0 && i<
size() && j<
size());
174 for(Index k = 0; k <
size(); ++k)
192 eigen_assert(i>=0 && j>=0 && i<
size() && j<
size());
213 #ifndef EIGEN_PARSED_BY_DOXYGEN
215 template<
typename OtherDerived>
220 template<
typename Lhs,
typename Rhs>
221 void assignProduct(
const Lhs& lhs,
const Rhs& rhs)
223 eigen_assert(lhs.cols() == rhs.rows());
224 for (
int i=0; i<
rows();++i)
indices().coeffRef(i) = lhs.indices().coeff(rhs.indices().coeff(i));
234 template<
typename Other>
236 {
return PlainPermutationType(internal::PermPermProduct,
derived(), other.
derived()); }
242 template<
typename Other>
244 {
return PlainPermutationType(internal::PermPermProduct, *
this, other.eval()); }
250 template<
typename Other>
friend
252 {
return PlainPermutationType(internal::PermPermProduct, other.eval(), perm); }
273 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename IndexType>
274 struct traits<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, IndexType> >
275 : traits<Matrix<IndexType,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
277 typedef IndexType Index;
278 typedef Matrix<IndexType, SizeAtCompileTime, 1, 0, MaxSizeAtCompileTime, 1> IndicesType;
282 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename IndexType>
286 typedef internal::traits<PermutationMatrix> Traits;
289 #ifndef EIGEN_PARSED_BY_DOXYGEN
290 typedef typename Traits::IndicesType IndicesType;
302 template<
typename OtherDerived>
304 : m_indices(other.
indices()) {}
306 #ifndef EIGEN_PARSED_BY_DOXYGEN
319 template<
typename Other>
324 template<
typename Other>
326 : m_indices(tr.
size())
332 template<
typename Other>
340 template<
typename Other>
346 #ifndef EIGEN_PARSED_BY_DOXYGEN
352 m_indices = other.m_indices;
358 const IndicesType&
indices()
const {
return m_indices; }
365 #ifndef EIGEN_PARSED_BY_DOXYGEN
366 template<
typename Other>
368 : m_indices(other.nestedPermutation().
size())
370 for (
int i=0; i<m_indices.size();++i) m_indices.coeffRef(other.nestedPermutation().
indices().coeff(i)) = i;
372 template<
typename Lhs,
typename Rhs>
373 PermutationMatrix(internal::PermPermProduct_t,
const Lhs& lhs,
const Rhs& rhs)
376 Base::assignProduct(lhs,rhs);
382 IndicesType m_indices;
387 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename IndexType,
int _PacketAccess>
388 struct traits<Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, IndexType>,_PacketAccess> >
389 : traits<Matrix<IndexType,SizeAtCompileTime,SizeAtCompileTime,0,MaxSizeAtCompileTime,MaxSizeAtCompileTime> >
391 typedef IndexType Index;
392 typedef Map<const Matrix<IndexType, SizeAtCompileTime, 1, 0, MaxSizeAtCompileTime, 1>, _PacketAccess> IndicesType;
396 template<
int SizeAtCompileTime,
int MaxSizeAtCompileTime,
typename IndexType,
int _PacketAccess>
397 class Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, IndexType>,_PacketAccess>
398 :
public PermutationBase<Map<PermutationMatrix<SizeAtCompileTime, MaxSizeAtCompileTime, IndexType>,_PacketAccess> >
400 typedef PermutationBase<Map> Base;
401 typedef internal::traits<Map> Traits;
404 #ifndef EIGEN_PARSED_BY_DOXYGEN
405 typedef typename Traits::IndicesType IndicesType;
406 typedef typename IndicesType::Scalar Index;
409 inline Map(
const Index* indices)
413 inline Map(
const Index* indices, Index size)
414 : m_indices(indices,size)
418 template<
typename Other>
419 Map& operator=(
const PermutationBase<Other>& other)
420 {
return Base::operator=(other.derived()); }
423 template<
typename Other>
424 Map& operator=(
const TranspositionsBase<Other>& tr)
425 {
return Base::operator=(tr.derived()); }
427 #ifndef EIGEN_PARSED_BY_DOXYGEN
431 Map& operator=(
const Map& other)
433 m_indices = other.m_indices;
439 const IndicesType& indices()
const {
return m_indices; }
441 IndicesType& indices() {
return m_indices; }
445 IndicesType m_indices;
460 struct PermutationStorage {};
462 template<
typename _IndicesType>
class TranspositionsWrapper;
464 template<
typename _IndicesType>
465 struct traits<PermutationWrapper<_IndicesType> >
467 typedef PermutationStorage StorageKind;
468 typedef typename _IndicesType::Scalar Scalar;
469 typedef typename _IndicesType::Scalar Index;
470 typedef _IndicesType IndicesType;
472 RowsAtCompileTime = _IndicesType::SizeAtCompileTime,
473 ColsAtCompileTime = _IndicesType::SizeAtCompileTime,
474 MaxRowsAtCompileTime = IndicesType::MaxRowsAtCompileTime,
475 MaxColsAtCompileTime = IndicesType::MaxColsAtCompileTime,
477 CoeffReadCost = _IndicesType::CoeffReadCost
482 template<
typename _IndicesType>
486 typedef internal::traits<PermutationWrapper> Traits;
489 #ifndef EIGEN_PARSED_BY_DOXYGEN
490 typedef typename Traits::IndicesType IndicesType;
498 const typename internal::remove_all<typename IndicesType::Nested>::type&
503 typename IndicesType::Nested m_indices;
508 template<
typename Derived,
typename PermutationDerived>
509 inline const internal::permut_matrix_product_retval<PermutationDerived, Derived, OnTheRight>
513 return internal::permut_matrix_product_retval
515 (permutation.
derived(), matrix.derived());
520 template<
typename Derived,
typename PermutationDerived>
521 inline const internal::permut_matrix_product_retval
526 return internal::permut_matrix_product_retval
528 (permutation.
derived(), matrix.derived());
533 template<
typename PermutationType,
typename MatrixType,
int S
ide,
bool Transposed>
534 struct traits<permut_matrix_product_retval<PermutationType, MatrixType, Side, Transposed> >
536 typedef typename MatrixType::PlainObject ReturnType;
539 template<
typename PermutationType,
typename MatrixType,
int S
ide,
bool Transposed>
540 struct permut_matrix_product_retval
541 :
public ReturnByValue<permut_matrix_product_retval<PermutationType, MatrixType, Side, Transposed> >
543 typedef typename remove_all<typename MatrixType::Nested>::type MatrixTypeNestedCleaned;
545 permut_matrix_product_retval(
const PermutationType& perm,
const MatrixType& matrix)
546 : m_permutation(perm), m_matrix(matrix)
549 inline int rows()
const {
return m_matrix.rows(); }
550 inline int cols()
const {
return m_matrix.cols(); }
552 template<
typename Dest>
inline void evalTo(Dest& dst)
const
554 const int n = Side==
OnTheLeft ? rows() : cols();
556 if(is_same<MatrixTypeNestedCleaned,Dest>::value && extract_data(dst) == extract_data(m_matrix))
559 Matrix<bool,PermutationType::RowsAtCompileTime,1,0,PermutationType::MaxRowsAtCompileTime> mask(m_permutation.size());
562 while(r < m_permutation.size())
565 while(r<m_permutation.size() && mask[r]) r++;
566 if(r>=m_permutation.size())
571 mask.coeffRef(k0) =
true;
572 for(
int k=m_permutation.indices().coeff(k0); k!=k0; k=m_permutation.indices().coeff(k))
574 Block<Dest, Side==OnTheLeft ? 1 : Dest::RowsAtCompileTime, Side==OnTheRight ? 1 : Dest::ColsAtCompileTime>(dst, k)
575 .swap(Block<Dest, Side==OnTheLeft ? 1 : Dest::RowsAtCompileTime, Side==OnTheRight ? 1 : Dest::ColsAtCompileTime>
576 (dst,((Side==
OnTheLeft) ^ Transposed) ? k0 : kPrev));
578 mask.coeffRef(k) =
true;
585 for(
int i = 0; i < n; ++i)
587 Block<Dest, Side==OnTheLeft ? 1 : Dest::RowsAtCompileTime, Side==OnTheRight ? 1 : Dest::ColsAtCompileTime>
588 (dst, ((Side==
OnTheLeft) ^ Transposed) ? m_permutation.indices().coeff(i) : i)
592 Block<const MatrixTypeNestedCleaned,Side==OnTheLeft ? 1 : MatrixType::RowsAtCompileTime,Side==OnTheRight ? 1 : MatrixType::ColsAtCompileTime>
593 (m_matrix, ((Side==
OnTheRight) ^ Transposed) ? m_permutation.indices().coeff(i) : i);
599 const PermutationType& m_permutation;
600 typename MatrixType::Nested m_matrix;
605 template<
typename Derived>
606 struct traits<Transpose<PermutationBase<Derived> > >
612 template<
typename Derived>
613 class Transpose<PermutationBase<Derived> >
614 :
public EigenBase<Transpose<PermutationBase<Derived> > >
616 typedef Derived PermutationType;
617 typedef typename PermutationType::IndicesType IndicesType;
618 typedef typename PermutationType::PlainPermutationType PlainPermutationType;
621 #ifndef EIGEN_PARSED_BY_DOXYGEN
622 typedef internal::traits<PermutationType> Traits;
623 typedef typename Derived::DenseMatrixType DenseMatrixType;
625 Flags = Traits::Flags,
626 CoeffReadCost = Traits::CoeffReadCost,
627 RowsAtCompileTime = Traits::RowsAtCompileTime,
628 ColsAtCompileTime = Traits::ColsAtCompileTime,
629 MaxRowsAtCompileTime = Traits::MaxRowsAtCompileTime,
630 MaxColsAtCompileTime = Traits::MaxColsAtCompileTime
632 typedef typename Traits::Scalar Scalar;
635 Transpose(
const PermutationType& p) : m_permutation(p) {}
637 inline int rows()
const {
return m_permutation.rows(); }
638 inline int cols()
const {
return m_permutation.cols(); }
640 #ifndef EIGEN_PARSED_BY_DOXYGEN
641 template<
typename DenseDerived>
642 void evalTo(MatrixBase<DenseDerived>& other)
const
645 for (
int i=0; i<rows();++i)
646 other.coeffRef(i, m_permutation.indices().coeff(i)) =
typename DenseDerived::Scalar(1);
651 PlainPermutationType eval()
const {
return *
this; }
653 DenseMatrixType toDenseMatrix()
const {
return *
this; }
657 template<
typename OtherDerived>
friend
658 inline const internal::permut_matrix_product_retval<PermutationType, OtherDerived, OnTheRight, true>
659 operator*(
const MatrixBase<OtherDerived>& matrix,
const Transpose& trPerm)
661 return internal::permut_matrix_product_retval<PermutationType, OtherDerived, OnTheRight, true>(trPerm.m_permutation, matrix.derived());
666 template<
typename OtherDerived>
667 inline const internal::permut_matrix_product_retval<PermutationType, OtherDerived, OnTheLeft, true>
668 operator*(
const MatrixBase<OtherDerived>& matrix)
const
670 return internal::permut_matrix_product_retval<PermutationType, OtherDerived, OnTheLeft, true>(m_permutation, matrix.derived());
673 const PermutationType& nestedPermutation()
const {
return m_permutation; }
676 const PermutationType& m_permutation;
679 template<
typename Derived>
680 const PermutationWrapper<const Derived> MatrixBase<Derived>::asPermutation()
const
687 #endif // EIGEN_PERMUTATIONMATRIX_H