11 #ifndef EIGEN_TRANSPOSE_H
12 #define EIGEN_TRANSPOSE_H
17 template<
typename MatrixType>
18 struct traits<Transpose<MatrixType> > :
public traits<MatrixType>
20 typedef typename ref_selector<MatrixType>::type MatrixTypeNested;
21 typedef typename remove_reference<MatrixTypeNested>::type MatrixTypeNestedPlain;
23 RowsAtCompileTime = MatrixType::ColsAtCompileTime,
24 ColsAtCompileTime = MatrixType::RowsAtCompileTime,
25 MaxRowsAtCompileTime = MatrixType::MaxColsAtCompileTime,
26 MaxColsAtCompileTime = MatrixType::MaxRowsAtCompileTime,
27 FlagsLvalueBit = is_lvalue<MatrixType>::value ?
LvalueBit : 0,
28 Flags0 = traits<MatrixTypeNestedPlain>::Flags & ~(
LvalueBit | NestByRefBit),
29 Flags1 = Flags0 | FlagsLvalueBit,
31 InnerStrideAtCompileTime = inner_stride_at_compile_time<MatrixType>::ret,
32 OuterStrideAtCompileTime = outer_stride_at_compile_time<MatrixType>::ret
37 template<
typename MatrixType,
typename StorageKind>
class TransposeImpl;
53 :
public TransposeImpl<MatrixType,typename internal::traits<MatrixType>::StorageKind>
57 typedef typename internal::ref_selector<MatrixType>::non_const_type MatrixTypeNested;
59 typedef typename TransposeImpl<MatrixType,typename internal::traits<MatrixType>::StorageKind>::Base Base;
61 typedef typename internal::remove_all<MatrixType>::type NestedExpression;
64 explicit inline Transpose(MatrixType& matrix) : m_matrix(matrix) {}
66 EIGEN_INHERIT_ASSIGNMENT_OPERATORS(
Transpose)
68 EIGEN_DEVICE_FUNC
inline Index rows()
const {
return m_matrix.cols(); }
69 EIGEN_DEVICE_FUNC
inline Index cols()
const {
return m_matrix.rows(); }
73 const typename internal::remove_all<MatrixTypeNested>::type&
78 typename internal::remove_reference<MatrixTypeNested>::type&
83 m_matrix.resize(ncols,nrows);
87 typename internal::ref_selector<MatrixType>::non_const_type m_matrix;
92 template<typename MatrixType, bool HasDirectAccess = has_direct_access<MatrixType>::ret>
93 struct TransposeImpl_base
95 typedef typename dense_xpr_base<Transpose<MatrixType> >::type type;
98 template<
typename MatrixType>
99 struct TransposeImpl_base<MatrixType, false>
101 typedef typename dense_xpr_base<Transpose<MatrixType> >::type type;
107 template<
typename XprType,
typename StorageKind>
109 :
public internal::generic_xpr_base<Transpose<XprType> >::type
112 typedef typename internal::generic_xpr_base<Transpose<XprType> >::type Base;
115 template<
typename MatrixType>
class TransposeImpl<MatrixType,Dense>
116 :
public internal::TransposeImpl_base<MatrixType>::type
120 typedef typename internal::TransposeImpl_base<MatrixType>::type Base;
121 using Base::coeffRef;
122 EIGEN_DENSE_PUBLIC_INTERFACE(Transpose<MatrixType>)
123 EIGEN_INHERIT_ASSIGNMENT_OPERATORS(TransposeImpl)
125 EIGEN_DEVICE_FUNC
inline Index innerStride()
const {
return derived().nestedExpression().innerStride(); }
126 EIGEN_DEVICE_FUNC
inline Index outerStride()
const {
return derived().nestedExpression().outerStride(); }
128 typedef typename internal::conditional<
129 internal::is_lvalue<MatrixType>::value,
132 >::type ScalarWithConstIfNotLvalue;
134 EIGEN_DEVICE_FUNC
inline ScalarWithConstIfNotLvalue* data() {
return derived().nestedExpression().data(); }
135 EIGEN_DEVICE_FUNC
inline const Scalar* data()
const {
return derived().nestedExpression().data(); }
139 inline const Scalar& coeffRef(
Index rowId,
Index colId)
const
141 return derived().nestedExpression().coeffRef(colId, rowId);
145 inline const Scalar& coeffRef(
Index index)
const
147 return derived().nestedExpression().coeffRef(index);
150 EIGEN_DEFAULT_EMPTY_CONSTRUCTOR_AND_DESTRUCTOR(TransposeImpl)
172 template<
typename Derived>
173 inline Transpose<Derived>
184 template<
typename Derived>
185 inline typename DenseBase<Derived>::ConstTransposeReturnType
188 return ConstTransposeReturnType(derived());
210 template<
typename Derived>
214 return AdjointReturnType(this->transpose());
223 template<
typename MatrixType,
224 bool IsSquare = (MatrixType::RowsAtCompileTime == MatrixType::ColsAtCompileTime) && MatrixType::RowsAtCompileTime!=
Dynamic,
225 bool MatchPacketSize =
226 (
int(MatrixType::RowsAtCompileTime) == int(internal::packet_traits<typename MatrixType::Scalar>::size))
228 struct inplace_transpose_selector;
230 template<
typename MatrixType>
231 struct inplace_transpose_selector<MatrixType,true,false> {
232 static void run(MatrixType& m) {
233 m.matrix().template triangularView<StrictlyUpper>().swap(m.matrix().transpose());
238 template<
typename MatrixType>
239 struct inplace_transpose_selector<MatrixType,true,true> {
240 static void run(MatrixType& m) {
241 typedef typename MatrixType::Scalar Scalar;
242 typedef typename internal::packet_traits<typename MatrixType::Scalar>::type Packet;
243 const Index PacketSize = internal::packet_traits<Scalar>::size;
244 const Index Alignment = internal::evaluator<MatrixType>::Alignment;
245 PacketBlock<Packet> A;
246 for (
Index i=0; i<PacketSize; ++i)
247 A.packet[i] = m.template packetByOuterInner<Alignment>(i,0);
248 internal::ptranspose(A);
249 for (
Index i=0; i<PacketSize; ++i)
250 m.template writePacket<Alignment>(m.rowIndexByOuterInner(i,0), m.colIndexByOuterInner(i,0), A.packet[i]);
254 template<
typename MatrixType,
bool MatchPacketSize>
255 struct inplace_transpose_selector<MatrixType,false,MatchPacketSize> {
256 static void run(MatrixType& m) {
257 if (m.rows()==m.cols())
258 m.matrix().template triangularView<StrictlyUpper>().swap(m.matrix().transpose());
260 m = m.transpose().eval();
285 template<
typename Derived>
288 eigen_assert((rows() == cols() || (RowsAtCompileTime ==
Dynamic && ColsAtCompileTime ==
Dynamic))
289 &&
"transposeInPlace() called on a non-square non-resizable matrix");
290 internal::inplace_transpose_selector<Derived>::run(derived());
316 template<
typename Derived>
319 derived() = adjoint().eval();
322 #ifndef EIGEN_NO_DEBUG
328 template<
bool DestIsTransposed,
typename OtherDerived>
329 struct check_transpose_aliasing_compile_time_selector
331 enum { ret = bool(blas_traits<OtherDerived>::IsTransposed) != DestIsTransposed };
334 template<
bool DestIsTransposed,
typename BinOp,
typename DerivedA,
typename DerivedB>
335 struct check_transpose_aliasing_compile_time_selector<DestIsTransposed,CwiseBinaryOp<BinOp,DerivedA,DerivedB> >
337 enum { ret = bool(blas_traits<DerivedA>::IsTransposed) != DestIsTransposed
338 || bool(blas_traits<DerivedB>::IsTransposed) != DestIsTransposed
342 template<
typename Scalar,
bool DestIsTransposed,
typename OtherDerived>
343 struct check_transpose_aliasing_run_time_selector
345 static bool run(
const Scalar* dest,
const OtherDerived& src)
347 return (
bool(blas_traits<OtherDerived>::IsTransposed) != DestIsTransposed) && (dest!=0 && dest==(
const Scalar*)extract_data(src));
351 template<
typename Scalar,
bool DestIsTransposed,
typename BinOp,
typename DerivedA,
typename DerivedB>
352 struct check_transpose_aliasing_run_time_selector<Scalar,DestIsTransposed,CwiseBinaryOp<BinOp,DerivedA,DerivedB> >
354 static bool run(
const Scalar* dest,
const CwiseBinaryOp<BinOp,DerivedA,DerivedB>& src)
356 return ((blas_traits<DerivedA>::IsTransposed != DestIsTransposed) && (dest!=0 && dest==(
const Scalar*)extract_data(src.lhs())))
357 || ((blas_traits<DerivedB>::IsTransposed != DestIsTransposed) && (dest!=0 && dest==(
const Scalar*)extract_data(src.rhs())));
367 template<
typename Derived,
typename OtherDerived,
368 bool MightHaveTransposeAliasing
369 = check_transpose_aliasing_compile_time_selector
370 <blas_traits<Derived>::IsTransposed,OtherDerived>::ret
372 struct checkTransposeAliasing_impl
374 static void run(
const Derived& dst,
const OtherDerived& other)
376 eigen_assert((!check_transpose_aliasing_run_time_selector
377 <
typename Derived::Scalar,blas_traits<Derived>::IsTransposed,OtherDerived>
378 ::run(extract_data(dst), other))
379 &&
"aliasing detected during transposition, use transposeInPlace() "
380 "or evaluate the rhs into a temporary using .eval()");
385 template<
typename Derived,
typename OtherDerived>
386 struct checkTransposeAliasing_impl<Derived, OtherDerived, false>
388 static void run(
const Derived&,
const OtherDerived&)
393 template<
typename Dst,
typename Src>
394 void check_for_aliasing(
const Dst &dst,
const Src &src)
396 internal::checkTransposeAliasing_impl<Dst, Src>::run(dst, src);
401 #endif // EIGEN_NO_DEBUG
405 #endif // EIGEN_TRANSPOSE_H