11 #ifndef EIGEN_PARTIAL_REDUX_H
12 #define EIGEN_PARTIAL_REDUX_H
32 template<
typename MatrixType,
typename MemberOp,
int Direction>
33 class PartialReduxExpr;
36 template<
typename MatrixType,
typename MemberOp,
int Direction>
40 typedef typename MemberOp::result_type Scalar;
43 typedef typename MatrixType::Scalar InputScalar;
45 RowsAtCompileTime = Direction==
Vertical ? 1 : MatrixType::RowsAtCompileTime,
46 ColsAtCompileTime = Direction==
Horizontal ? 1 : MatrixType::ColsAtCompileTime,
47 MaxRowsAtCompileTime = Direction==
Vertical ? 1 : MatrixType::MaxRowsAtCompileTime,
48 MaxColsAtCompileTime = Direction==
Horizontal ? 1 : MatrixType::MaxColsAtCompileTime,
50 TraversalSize = Direction==
Vertical ? MatrixType::RowsAtCompileTime : MatrixType::ColsAtCompileTime
55 template<
typename MatrixType,
typename MemberOp,
int Direction>
65 explicit PartialReduxExpr(
const MatrixType& mat,
const MemberOp& func = MemberOp())
66 : m_matrix(mat), m_functor(func) {}
69 Index rows()
const {
return (Direction==
Vertical ? 1 : m_matrix.rows()); }
71 Index cols()
const {
return (Direction==
Horizontal ? 1 : m_matrix.cols()); }
74 typename MatrixType::Nested nestedExpression()
const {
return m_matrix; }
77 const MemberOp& functor()
const {
return m_functor; }
80 typename MatrixType::Nested m_matrix;
81 const MemberOp m_functor;
86 #define EIGEN_MAKE_PARTIAL_REDUX_FUNCTOR(MEMBER,COST,VECTORIZABLE,BINARYOP) \
87 template <typename ResultType,typename Scalar> \
88 struct member_##MEMBER { \
89 EIGEN_EMPTY_STRUCT_CTOR(member_##MEMBER) \
90 typedef ResultType result_type; \
91 typedef BINARYOP<Scalar,Scalar> BinaryOp; \
92 template<int Size> struct Cost { enum { value = COST }; }; \
93 enum { Vectorizable = VECTORIZABLE }; \
94 template<typename XprType> \
95 EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE \
96 ResultType operator()(const XprType& mat) const \
97 { return mat.MEMBER(); } \
98 BinaryOp binaryFunc() const { return BinaryOp(); } \
101 #define EIGEN_MEMBER_FUNCTOR(MEMBER,COST) \
102 EIGEN_MAKE_PARTIAL_REDUX_FUNCTOR(MEMBER,COST,0,partial_redux_dummy_func)
109 EIGEN_MEMBER_FUNCTOR(hypotNorm, (Size-1) * functor_traits<scalar_hypot_op<Scalar> >::Cost );
119 template <
int p,
typename ResultType,
typename Scalar>
121 typedef ResultType result_type;
122 enum { Vectorizable = 0 };
126 template<
typename XprType>
127 EIGEN_DEVICE_FUNC
inline ResultType operator()(
const XprType& mat)
const
128 {
return mat.template lpNorm<p>(); }
131 template <
typename BinaryOpT,
typename Scalar>
133 typedef BinaryOpT BinaryOp;
135 BinaryOp(
const Scalar&,
const Scalar&)
140 EIGEN_DEVICE_FUNC
explicit member_redux(
const BinaryOp func) : m_functor(func) {}
141 template<
typename Derived>
143 {
return mat.redux(m_functor); }
144 const BinaryOp& binaryFunc()
const {
return m_functor; }
145 const BinaryOp m_functor;
190 typedef typename ExpressionType::Scalar Scalar;
191 typedef typename ExpressionType::RealScalar RealScalar;
194 typedef typename internal::remove_all<ExpressionTypeNested>::type ExpressionTypeNestedCleaned;
196 template<
template<
typename OutScalar,
typename InputScalar>
class Functor,
200 Functor<ReturnScalar,Scalar>,
214 isVertical = (Direction==
Vertical) ? 1 : 0,
215 isHorizontal = (Direction==
Horizontal) ? 1 : 0
222 isVertical ? 1 : ExpressionType::RowsAtCompileTime,
223 isHorizontal ? 1 : ExpressionType::ColsAtCompileTime>
Type;
228 template<
typename OtherDerived>
233 EIGEN_STATIC_ASSERT(EIGEN_IMPLIES(isVertical, OtherDerived::MaxColsAtCompileTime==1),
234 YOU_PASSED_A_ROW_VECTOR_BUT_A_COLUMN_VECTOR_WAS_EXPECTED)
235 EIGEN_STATIC_ASSERT(EIGEN_IMPLIES(isHorizontal, OtherDerived::MaxRowsAtCompileTime==1),
236 YOU_PASSED_A_COLUMN_VECTOR_BUT_A_ROW_VECTOR_WAS_EXPECTED)
239 isVertical ? 1 : m_matrix.rows(),
240 isHorizontal ? 1 : m_matrix.cols());
245 isHorizontal ? 1 : ExpressionType::RowsAtCompileTime,
246 isVertical ? 1 : ExpressionType::ColsAtCompileTime>
Type;
251 template<
typename OtherDerived>
256 EIGEN_STATIC_ASSERT(EIGEN_IMPLIES(isHorizontal, OtherDerived::MaxColsAtCompileTime==1),
257 YOU_PASSED_A_ROW_VECTOR_BUT_A_COLUMN_VECTOR_WAS_EXPECTED)
258 EIGEN_STATIC_ASSERT(EIGEN_IMPLIES(isVertical, OtherDerived::MaxRowsAtCompileTime==1),
259 YOU_PASSED_A_COLUMN_VECTOR_BUT_A_ROW_VECTOR_WAS_EXPECTED)
262 isHorizontal ? 1 : m_matrix.rows(),
263 isVertical ? 1 : m_matrix.cols());
268 explicit inline VectorwiseOp(ExpressionType& matrix) : m_matrix(matrix) {}
272 inline const ExpressionType& _expression()
const {
return m_matrix; }
274 #ifdef EIGEN_PARSED_BY_DOXYGEN
278 random_access_iterator_type iterator;
280 random_access_iterator_type const_iterator;
282 typedef internal::subvector_stl_iterator<ExpressionType,
DirectionType(Direction)> iterator;
283 typedef internal::subvector_stl_iterator<
const ExpressionType,
DirectionType(Direction)> const_iterator;
284 typedef internal::subvector_stl_reverse_iterator<ExpressionType,
DirectionType(Direction)> reverse_iterator;
285 typedef internal::subvector_stl_reverse_iterator<
const ExpressionType,
DirectionType(Direction)> const_reverse_iterator;
334 template<
typename BinaryOp>
336 const typename ReduxReturnType<BinaryOp>::Type
337 redux(
const BinaryOp& func = BinaryOp())
const
339 eigen_assert(redux_length()>0 &&
"you are using an empty matrix");
343 typedef typename ReturnType<internal::member_minCoeff>::Type MinCoeffReturnType;
344 typedef typename ReturnType<internal::member_maxCoeff>::Type MaxCoeffReturnType;
347 typedef typename ReturnType<internal::member_blueNorm,RealScalar>::Type BlueNormReturnType;
348 typedef typename ReturnType<internal::member_stableNorm,RealScalar>::Type StableNormReturnType;
349 typedef typename ReturnType<internal::member_hypotNorm,RealScalar>::Type HypotNormReturnType;
350 typedef typename ReturnType<internal::member_sum>::Type SumReturnType;
351 typedef EIGEN_EXPR_BINARYOP_SCALAR_RETURN_TYPE(SumReturnType,Scalar,quotient) MeanReturnType;
352 typedef typename ReturnType<internal::member_all>::Type AllReturnType;
353 typedef typename ReturnType<internal::member_any>::Type AnyReturnType;
355 typedef typename ReturnType<internal::member_prod>::Type ProdReturnType;
378 eigen_assert(redux_length()>0 &&
"you are using an empty matrix");
397 eigen_assert(redux_length()>0 &&
"you are using an empty matrix");
487 const MeanReturnType
mean()
const
488 {
return sum() / Scalar(Direction==
Vertical?m_matrix.rows():m_matrix.cols()); }
571 (_expression(),isVertical?factor:1,isHorizontal?factor:1);
577 template<
typename OtherDerived>
581 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
582 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
584 return m_matrix = extendedTo(other.derived());
588 template<
typename OtherDerived>
592 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
593 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
594 return m_matrix += extendedTo(other.derived());
598 template<
typename OtherDerived>
602 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
603 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
604 return m_matrix -= extendedTo(other.derived());
608 template<
typename OtherDerived>
612 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
613 EIGEN_STATIC_ASSERT_ARRAYXPR(ExpressionType)
614 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
615 m_matrix *= extendedTo(other.derived());
620 template<
typename OtherDerived>
624 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
625 EIGEN_STATIC_ASSERT_ARRAYXPR(ExpressionType)
626 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
627 m_matrix /= extendedTo(other.derived());
632 template<
typename OtherDerived> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC
634 const ExpressionTypeNestedCleaned,
635 const typename ExtendedType<OtherDerived>::Type>
638 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
639 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
640 return m_matrix + extendedTo(other.derived());
644 template<
typename OtherDerived>
647 const ExpressionTypeNestedCleaned,
648 const typename ExtendedType<OtherDerived>::Type>
651 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
652 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
653 return m_matrix - extendedTo(other.derived());
658 template<
typename OtherDerived> EIGEN_STRONG_INLINE EIGEN_DEVICE_FUNC
660 const ExpressionTypeNestedCleaned,
661 const typename ExtendedType<OtherDerived>::Type>
665 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
666 EIGEN_STATIC_ASSERT_ARRAYXPR(ExpressionType)
667 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
668 return m_matrix * extendedTo(other.derived());
673 template<
typename OtherDerived>
676 const ExpressionTypeNestedCleaned,
677 const typename ExtendedType<OtherDerived>::Type>
680 EIGEN_STATIC_ASSERT_VECTOR_ONLY(OtherDerived)
681 EIGEN_STATIC_ASSERT_ARRAYXPR(ExpressionType)
682 EIGEN_STATIC_ASSERT_SAME_XPR_KIND(ExpressionType, OtherDerived)
683 return m_matrix / extendedTo(other.derived());
692 const ExpressionTypeNestedCleaned,
693 const typename OppositeExtendedType<NormReturnType>::Type>
694 normalized()
const {
return m_matrix.cwiseQuotient(extendedToOpposite(this->
norm())); }
712 typedef typename ExpressionType::PlainObject CrossReturnType;
713 template<
typename OtherDerived>
720 HNormalized_SizeMinusOne = HNormalized_Size==
Dynamic ?
Dynamic : HNormalized_Size-1
722 typedef Block<
const ExpressionType,
723 Direction==
Vertical ? int(HNormalized_SizeMinusOne)
724 : int(internal::traits<ExpressionType>::RowsAtCompileTime),
725 Direction==
Horizontal ? int(HNormalized_SizeMinusOne)
726 : int(internal::traits<ExpressionType>::ColsAtCompileTime)>
728 typedef Block<
const ExpressionType,
729 Direction==
Vertical ? 1 : int(internal::traits<ExpressionType>::RowsAtCompileTime),
730 Direction==
Horizontal ? 1 : int(internal::traits<ExpressionType>::ColsAtCompileTime)>
732 typedef CwiseBinaryOp<internal::scalar_quotient_op<typename internal::traits<ExpressionType>::Scalar>,
733 const HNormalized_Block,
734 const Replicate<HNormalized_Factors,
735 Direction==
Vertical ? HNormalized_SizeMinusOne : 1,
736 Direction==
Horizontal ? HNormalized_SizeMinusOne : 1> >
737 HNormalizedReturnType;
742 # ifdef EIGEN_VECTORWISEOP_PLUGIN
743 # include EIGEN_VECTORWISEOP_PLUGIN
747 Index redux_length()
const
749 return Direction==
Vertical ? m_matrix.rows() : m_matrix.cols();
751 ExpressionTypeNested m_matrix;
761 template<
typename Derived>
762 EIGEN_DEVICE_FUNC
inline typename DenseBase<Derived>::ColwiseReturnType
775 template<
typename Derived>
784 #endif // EIGEN_PARTIAL_REDUX_H