10 #ifndef EIGEN_VISITOR_H
11 #define EIGEN_VISITOR_H
17 template<
typename Visitor,
typename Derived,
int UnrollCount>
21 col = (UnrollCount-1) / Derived::RowsAtCompileTime,
22 row = (UnrollCount-1) % Derived::RowsAtCompileTime
25 static inline void run(
const Derived &mat, Visitor& visitor)
27 visitor_impl<Visitor, Derived, UnrollCount-1>::run(mat, visitor);
28 visitor(mat.coeff(row, col), row, col);
32 template<
typename Visitor,
typename Derived>
33 struct visitor_impl<Visitor, Derived, 1>
35 static inline void run(
const Derived &mat, Visitor& visitor)
37 return visitor.init(mat.coeff(0, 0), 0, 0);
41 template<
typename Visitor,
typename Derived>
42 struct visitor_impl<Visitor, Derived, Dynamic>
44 typedef typename Derived::Index Index;
45 static inline void run(
const Derived& mat, Visitor& visitor)
47 visitor.init(mat.coeff(0,0), 0, 0);
48 for(Index i = 1; i < mat.rows(); ++i)
49 visitor(mat.coeff(i, 0), i, 0);
50 for(Index j = 1; j < mat.cols(); ++j)
51 for(Index i = 0; i < mat.rows(); ++i)
52 visitor(mat.coeff(i, j), i, j);
75 template<
typename Derived>
76 template<
typename Visitor>
79 enum { unroll = SizeAtCompileTime != Dynamic
80 && CoeffReadCost != Dynamic
81 && (SizeAtCompileTime == 1 || internal::functor_traits<Visitor>::Cost != Dynamic)
82 && SizeAtCompileTime * CoeffReadCost + (SizeAtCompileTime-1) * internal::functor_traits<Visitor>::Cost
83 <= EIGEN_UNROLLING_LIMIT };
84 return internal::visitor_impl<Visitor, Derived,
85 unroll ? int(SizeAtCompileTime) : Dynamic
86 >::run(derived(), visitor);
94 template <
typename Derived>
97 typedef typename Derived::Index Index;
98 typedef typename Derived::Scalar Scalar;
101 inline void init(
const Scalar& value, Index i, Index j)
114 template <
typename Derived>
115 struct min_coeff_visitor : coeff_visitor<Derived>
117 typedef typename Derived::Index Index;
118 typedef typename Derived::Scalar Scalar;
119 void operator() (
const Scalar& value, Index i, Index j)
121 if(value < this->res)
130 template<
typename Scalar>
131 struct functor_traits<min_coeff_visitor<Scalar> > {
133 Cost = NumTraits<Scalar>::AddCost
142 template <
typename Derived>
143 struct max_coeff_visitor : coeff_visitor<Derived>
145 typedef typename Derived::Index Index;
146 typedef typename Derived::Scalar Scalar;
147 void operator() (
const Scalar& value, Index i, Index j)
149 if(value > this->res)
158 template<
typename Scalar>
159 struct functor_traits<max_coeff_visitor<Scalar> > {
161 Cost = NumTraits<Scalar>::AddCost
172 template<
typename Derived>
173 template<
typename IndexType>
174 typename internal::traits<Derived>::Scalar
177 internal::min_coeff_visitor<Derived> minVisitor;
178 this->visit(minVisitor);
179 *rowId = minVisitor.row;
180 if (colId) *colId = minVisitor.col;
181 return minVisitor.res;
189 template<
typename Derived>
190 template<
typename IndexType>
191 typename internal::traits<Derived>::Scalar
194 EIGEN_STATIC_ASSERT_VECTOR_ONLY(Derived)
195 internal::min_coeff_visitor<Derived> minVisitor;
196 this->visit(minVisitor);
197 *index = (RowsAtCompileTime==1) ? minVisitor.col : minVisitor.row;
198 return minVisitor.res;
206 template<
typename Derived>
207 template<
typename IndexType>
208 typename internal::traits<Derived>::Scalar
211 internal::max_coeff_visitor<Derived> maxVisitor;
212 this->visit(maxVisitor);
213 *rowPtr = maxVisitor.row;
214 if (colPtr) *colPtr = maxVisitor.col;
215 return maxVisitor.res;
223 template<
typename Derived>
224 template<
typename IndexType>
225 typename internal::traits<Derived>::Scalar
228 EIGEN_STATIC_ASSERT_VECTOR_ONLY(Derived)
229 internal::max_coeff_visitor<Derived> maxVisitor;
230 this->visit(maxVisitor);
231 *index = (RowsAtCompileTime==1) ? maxVisitor.col : maxVisitor.row;
232 return maxVisitor.res;
237 #endif // EIGEN_VISITOR_H
internal::traits< Derived >::Scalar minCoeff() const
Definition: Redux.h:338
void visit(Visitor &func) const
Definition: Visitor.h:77
internal::traits< Derived >::Scalar maxCoeff() const
Definition: Redux.h:348
Definition: Eigen_Colamd.h:54