10 #ifndef EIGEN_FUNCTORS_H
11 #define EIGEN_FUNCTORS_H
24 template<
typename Scalar>
struct scalar_sum_op {
25 EIGEN_EMPTY_STRUCT_CTOR(scalar_sum_op)
26 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a, const Scalar& b)
const {
return a + b; }
27 template<
typename Packet>
28 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
29 {
return internal::padd(a,b); }
30 template<
typename Packet>
31 EIGEN_STRONG_INLINE
const Scalar predux(
const Packet& a)
const
32 {
return internal::predux(a); }
34 template<
typename Scalar>
35 struct functor_traits<scalar_sum_op<Scalar> > {
37 Cost = NumTraits<Scalar>::AddCost,
38 PacketAccess = packet_traits<Scalar>::HasAdd
47 template<
typename LhsScalar,
typename RhsScalar>
struct scalar_product_op {
50 Vectorizable = is_same<LhsScalar,RhsScalar>::value && packet_traits<LhsScalar>::HasMul && packet_traits<RhsScalar>::HasMul
52 typedef typename scalar_product_traits<LhsScalar,RhsScalar>::ReturnType result_type;
53 EIGEN_EMPTY_STRUCT_CTOR(scalar_product_op)
54 EIGEN_STRONG_INLINE const result_type operator() (const LhsScalar& a, const RhsScalar& b)
const {
return a * b; }
55 template<
typename Packet>
56 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
57 {
return internal::pmul(a,b); }
58 template<
typename Packet>
59 EIGEN_STRONG_INLINE
const result_type predux(
const Packet& a)
const
60 {
return internal::predux_mul(a); }
62 template<
typename LhsScalar,
typename RhsScalar>
63 struct functor_traits<scalar_product_op<LhsScalar,RhsScalar> > {
65 Cost = (NumTraits<LhsScalar>::MulCost + NumTraits<RhsScalar>::MulCost)/2,
66 PacketAccess = scalar_product_op<LhsScalar,RhsScalar>::Vectorizable
75 template<
typename LhsScalar,
typename RhsScalar>
struct scalar_conj_product_op {
78 Conj = NumTraits<LhsScalar>::IsComplex
81 typedef typename scalar_product_traits<LhsScalar,RhsScalar>::ReturnType result_type;
83 EIGEN_EMPTY_STRUCT_CTOR(scalar_conj_product_op)
84 EIGEN_STRONG_INLINE const result_type operator() (const LhsScalar& a, const RhsScalar& b)
const
85 {
return conj_helper<LhsScalar,RhsScalar,Conj,false>().pmul(a,b); }
87 template<
typename Packet>
88 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
89 {
return conj_helper<Packet,Packet,Conj,false>().pmul(a,b); }
91 template<
typename LhsScalar,
typename RhsScalar>
92 struct functor_traits<scalar_conj_product_op<LhsScalar,RhsScalar> > {
94 Cost = NumTraits<LhsScalar>::MulCost,
95 PacketAccess = internal::is_same<LhsScalar, RhsScalar>::value && packet_traits<LhsScalar>::HasMul
104 template<
typename Scalar>
struct scalar_min_op {
105 EIGEN_EMPTY_STRUCT_CTOR(scalar_min_op)
106 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a, const Scalar& b)
const {
using std::min;
return (min)(a, b); }
107 template<
typename Packet>
108 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
109 {
return internal::pmin(a,b); }
110 template<
typename Packet>
111 EIGEN_STRONG_INLINE
const Scalar predux(
const Packet& a)
const
112 {
return internal::predux_min(a); }
114 template<
typename Scalar>
115 struct functor_traits<scalar_min_op<Scalar> > {
117 Cost = NumTraits<Scalar>::AddCost,
118 PacketAccess = packet_traits<Scalar>::HasMin
127 template<
typename Scalar>
struct scalar_max_op {
128 EIGEN_EMPTY_STRUCT_CTOR(scalar_max_op)
129 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a, const Scalar& b)
const {
using std::max;
return (max)(a, b); }
130 template<
typename Packet>
131 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
132 {
return internal::pmax(a,b); }
133 template<
typename Packet>
134 EIGEN_STRONG_INLINE
const Scalar predux(
const Packet& a)
const
135 {
return internal::predux_max(a); }
137 template<
typename Scalar>
138 struct functor_traits<scalar_max_op<Scalar> > {
140 Cost = NumTraits<Scalar>::AddCost,
141 PacketAccess = packet_traits<Scalar>::HasMax
150 template<
typename Scalar>
struct scalar_hypot_op {
151 EIGEN_EMPTY_STRUCT_CTOR(scalar_hypot_op)
153 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& _x, const Scalar& _y)
const
157 Scalar p = (max)(_x, _y);
158 Scalar q = (min)(_x, _y);
160 return p * sqrt(Scalar(1) + qp*qp);
163 template<
typename Scalar>
164 struct functor_traits<scalar_hypot_op<Scalar> > {
165 enum { Cost = 5 * NumTraits<Scalar>::MulCost, PacketAccess=0 };
171 template<
typename Scalar,
typename OtherScalar>
struct scalar_binary_pow_op {
172 EIGEN_EMPTY_STRUCT_CTOR(scalar_binary_pow_op)
173 inline Scalar operator() (const Scalar& a, const OtherScalar& b)
const {
return internal::pow(a, b); }
175 template<
typename Scalar,
typename OtherScalar>
176 struct functor_traits<scalar_binary_pow_op<Scalar,OtherScalar> > {
177 enum { Cost = 5 * NumTraits<Scalar>::MulCost, PacketAccess =
false };
187 template<
typename Scalar>
struct scalar_difference_op {
188 EIGEN_EMPTY_STRUCT_CTOR(scalar_difference_op)
189 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a, const Scalar& b)
const {
return a - b; }
190 template<
typename Packet>
191 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
192 {
return internal::psub(a,b); }
194 template<
typename Scalar>
195 struct functor_traits<scalar_difference_op<Scalar> > {
197 Cost = NumTraits<Scalar>::AddCost,
198 PacketAccess = packet_traits<Scalar>::HasSub
207 template<
typename Scalar>
struct scalar_quotient_op {
208 EIGEN_EMPTY_STRUCT_CTOR(scalar_quotient_op)
209 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a, const Scalar& b)
const {
return a / b; }
210 template<
typename Packet>
211 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a,
const Packet& b)
const
212 {
return internal::pdiv(a,b); }
214 template<
typename Scalar>
215 struct functor_traits<scalar_quotient_op<Scalar> > {
217 Cost = 2 * NumTraits<Scalar>::MulCost,
218 PacketAccess = packet_traits<Scalar>::HasDiv
227 struct scalar_boolean_and_op {
228 EIGEN_EMPTY_STRUCT_CTOR(scalar_boolean_and_op)
229 EIGEN_STRONG_INLINE
bool operator() (const
bool& a, const
bool& b)
const {
return a && b; }
231 template<>
struct functor_traits<scalar_boolean_and_op> {
233 Cost = NumTraits<bool>::AddCost,
243 struct scalar_boolean_or_op {
244 EIGEN_EMPTY_STRUCT_CTOR(scalar_boolean_or_op)
245 EIGEN_STRONG_INLINE
bool operator() (const
bool& a, const
bool& b)
const {
return a || b; }
247 template<>
struct functor_traits<scalar_boolean_or_op> {
249 Cost = NumTraits<bool>::AddCost,
261 template<
typename Scalar>
struct scalar_opposite_op {
262 EIGEN_EMPTY_STRUCT_CTOR(scalar_opposite_op)
263 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a)
const {
return -a; }
264 template<
typename Packet>
265 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const
266 {
return internal::pnegate(a); }
268 template<
typename Scalar>
269 struct functor_traits<scalar_opposite_op<Scalar> >
271 Cost = NumTraits<Scalar>::AddCost,
272 PacketAccess = packet_traits<Scalar>::HasNegate };
280 template<
typename Scalar>
struct scalar_abs_op {
281 EIGEN_EMPTY_STRUCT_CTOR(scalar_abs_op)
282 typedef typename NumTraits<Scalar>::Real result_type;
283 EIGEN_STRONG_INLINE const result_type operator() (const Scalar& a)
const {
return internal::abs(a); }
284 template<
typename Packet>
285 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const
286 {
return internal::pabs(a); }
288 template<
typename Scalar>
289 struct functor_traits<scalar_abs_op<Scalar> >
292 Cost = NumTraits<Scalar>::AddCost,
293 PacketAccess = packet_traits<Scalar>::HasAbs
302 template<
typename Scalar>
struct scalar_abs2_op {
303 EIGEN_EMPTY_STRUCT_CTOR(scalar_abs2_op)
304 typedef typename NumTraits<Scalar>::Real result_type;
305 EIGEN_STRONG_INLINE const result_type operator() (const Scalar& a)
const {
return internal::abs2(a); }
306 template<
typename Packet>
307 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const
308 {
return internal::pmul(a,a); }
310 template<
typename Scalar>
311 struct functor_traits<scalar_abs2_op<Scalar> >
312 {
enum { Cost = NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasAbs2 }; };
319 template<
typename Scalar>
struct scalar_conjugate_op {
320 EIGEN_EMPTY_STRUCT_CTOR(scalar_conjugate_op)
321 EIGEN_STRONG_INLINE const Scalar operator() (const Scalar& a)
const {
return internal::conj(a); }
322 template<
typename Packet>
323 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const {
return internal::pconj(a); }
325 template<
typename Scalar>
326 struct functor_traits<scalar_conjugate_op<Scalar> >
329 Cost = NumTraits<Scalar>::IsComplex ? NumTraits<Scalar>::AddCost : 0,
330 PacketAccess = packet_traits<Scalar>::HasConj
339 template<
typename Scalar,
typename NewType>
340 struct scalar_cast_op {
341 EIGEN_EMPTY_STRUCT_CTOR(scalar_cast_op)
342 typedef NewType result_type;
343 EIGEN_STRONG_INLINE const NewType operator() (const Scalar& a)
const {
return cast<Scalar, NewType>(a); }
345 template<
typename Scalar,
typename NewType>
346 struct functor_traits<scalar_cast_op<Scalar,NewType> >
347 {
enum { Cost = is_same<Scalar, NewType>::value ? 0 : NumTraits<NewType>::AddCost, PacketAccess =
false }; };
354 template<
typename Scalar>
355 struct scalar_real_op {
356 EIGEN_EMPTY_STRUCT_CTOR(scalar_real_op)
357 typedef typename NumTraits<Scalar>::Real result_type;
358 EIGEN_STRONG_INLINE result_type operator() (const Scalar& a)
const {
return internal::real(a); }
360 template<
typename Scalar>
361 struct functor_traits<scalar_real_op<Scalar> >
362 {
enum { Cost = 0, PacketAccess =
false }; };
369 template<
typename Scalar>
370 struct scalar_imag_op {
371 EIGEN_EMPTY_STRUCT_CTOR(scalar_imag_op)
372 typedef typename NumTraits<Scalar>::Real result_type;
373 EIGEN_STRONG_INLINE result_type operator() (const Scalar& a)
const {
return internal::imag(a); }
375 template<
typename Scalar>
376 struct functor_traits<scalar_imag_op<Scalar> >
377 {
enum { Cost = 0, PacketAccess =
false }; };
384 template<
typename Scalar>
385 struct scalar_real_ref_op {
386 EIGEN_EMPTY_STRUCT_CTOR(scalar_real_ref_op)
387 typedef typename NumTraits<Scalar>::Real result_type;
388 EIGEN_STRONG_INLINE result_type& operator() (const Scalar& a)
const {
return internal::real_ref(*const_cast<Scalar*>(&a)); }
390 template<
typename Scalar>
391 struct functor_traits<scalar_real_ref_op<Scalar> >
392 {
enum { Cost = 0, PacketAccess =
false }; };
399 template<
typename Scalar>
400 struct scalar_imag_ref_op {
401 EIGEN_EMPTY_STRUCT_CTOR(scalar_imag_ref_op)
402 typedef typename NumTraits<Scalar>::Real result_type;
403 EIGEN_STRONG_INLINE result_type& operator() (const Scalar& a)
const {
return internal::imag_ref(*const_cast<Scalar*>(&a)); }
405 template<
typename Scalar>
406 struct functor_traits<scalar_imag_ref_op<Scalar> >
407 {
enum { Cost = 0, PacketAccess =
false }; };
415 template<
typename Scalar>
struct scalar_exp_op {
416 EIGEN_EMPTY_STRUCT_CTOR(scalar_exp_op)
417 inline const Scalar operator() (const Scalar& a)
const {
return internal::exp(a); }
418 typedef typename packet_traits<Scalar>::type Packet;
419 inline Packet packetOp(
const Packet& a)
const {
return internal::pexp(a); }
421 template<
typename Scalar>
422 struct functor_traits<scalar_exp_op<Scalar> >
423 {
enum { Cost = 5 * NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasExp }; };
431 template<
typename Scalar>
struct scalar_log_op {
432 EIGEN_EMPTY_STRUCT_CTOR(scalar_log_op)
433 inline const Scalar operator() (const Scalar& a)
const {
return internal::log(a); }
434 typedef typename packet_traits<Scalar>::type Packet;
435 inline Packet packetOp(
const Packet& a)
const {
return internal::plog(a); }
437 template<
typename Scalar>
438 struct functor_traits<scalar_log_op<Scalar> >
439 {
enum { Cost = 5 * NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasLog }; };
454 template<
typename Scalar>
455 struct scalar_multiple_op {
456 typedef typename packet_traits<Scalar>::type Packet;
458 EIGEN_STRONG_INLINE scalar_multiple_op(
const scalar_multiple_op& other) : m_other(other.m_other) { }
459 EIGEN_STRONG_INLINE scalar_multiple_op(
const Scalar& other) : m_other(other) { }
460 EIGEN_STRONG_INLINE Scalar operator() (
const Scalar& a)
const {
return a * m_other; }
461 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const
462 {
return internal::pmul(a, pset1<Packet>(m_other)); }
463 typename add_const_on_value_type<typename NumTraits<Scalar>::Nested>::type m_other;
465 template<
typename Scalar>
466 struct functor_traits<scalar_multiple_op<Scalar> >
467 {
enum { Cost = NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasMul }; };
469 template<
typename Scalar1,
typename Scalar2>
470 struct scalar_multiple2_op {
471 typedef typename scalar_product_traits<Scalar1,Scalar2>::ReturnType result_type;
472 EIGEN_STRONG_INLINE scalar_multiple2_op(
const scalar_multiple2_op& other) : m_other(other.m_other) { }
473 EIGEN_STRONG_INLINE scalar_multiple2_op(
const Scalar2& other) : m_other(other) { }
474 EIGEN_STRONG_INLINE result_type operator() (
const Scalar1& a)
const {
return a * m_other; }
475 typename add_const_on_value_type<typename NumTraits<Scalar2>::Nested>::type m_other;
477 template<
typename Scalar1,
typename Scalar2>
478 struct functor_traits<scalar_multiple2_op<Scalar1,Scalar2> >
479 {
enum { Cost = NumTraits<Scalar1>::MulCost, PacketAccess =
false }; };
489 template<
typename Scalar>
490 struct scalar_quotient1_op {
491 typedef typename packet_traits<Scalar>::type Packet;
493 EIGEN_STRONG_INLINE scalar_quotient1_op(
const scalar_quotient1_op& other) : m_other(other.m_other) { }
494 EIGEN_STRONG_INLINE scalar_quotient1_op(
const Scalar& other) : m_other(other) {}
495 EIGEN_STRONG_INLINE Scalar operator() (
const Scalar& a)
const {
return a / m_other; }
496 EIGEN_STRONG_INLINE
const Packet packetOp(
const Packet& a)
const
497 {
return internal::pdiv(a, pset1<Packet>(m_other)); }
498 typename add_const_on_value_type<typename NumTraits<Scalar>::Nested>::type m_other;
500 template<
typename Scalar>
501 struct functor_traits<scalar_quotient1_op<Scalar> >
502 {
enum { Cost = 2 * NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasDiv }; };
506 template<
typename Scalar>
507 struct scalar_constant_op {
508 typedef typename packet_traits<Scalar>::type Packet;
509 EIGEN_STRONG_INLINE scalar_constant_op(
const scalar_constant_op& other) : m_other(other.m_other) { }
510 EIGEN_STRONG_INLINE scalar_constant_op(
const Scalar& other) : m_other(other) { }
511 template<
typename Index>
512 EIGEN_STRONG_INLINE
const Scalar operator() (Index, Index = 0)
const {
return m_other; }
513 template<
typename Index>
514 EIGEN_STRONG_INLINE
const Packet packetOp(Index, Index = 0)
const {
return internal::pset1<Packet>(m_other); }
515 const Scalar m_other;
517 template<
typename Scalar>
518 struct functor_traits<scalar_constant_op<Scalar> >
520 {
enum { Cost = 1, PacketAccess = packet_traits<Scalar>::Vectorizable, IsRepeatable =
true }; };
522 template<
typename Scalar>
struct scalar_identity_op {
523 EIGEN_EMPTY_STRUCT_CTOR(scalar_identity_op)
524 template<typename Index>
525 EIGEN_STRONG_INLINE const Scalar operator() (Index row, Index col)
const {
return row==col ? Scalar(1) : Scalar(0); }
527 template<
typename Scalar>
528 struct functor_traits<scalar_identity_op<Scalar> >
529 {
enum { Cost = NumTraits<Scalar>::AddCost, PacketAccess =
false, IsRepeatable =
true }; };
531 template <
typename Scalar,
bool RandomAccess>
struct linspaced_op_impl;
541 template <
typename Scalar>
542 struct linspaced_op_impl<Scalar,false>
544 typedef typename packet_traits<Scalar>::type Packet;
546 linspaced_op_impl(Scalar low, Scalar step) :
547 m_low(low), m_step(step),
548 m_packetStep(pset1<Packet>(packet_traits<Scalar>::size*step)),
549 m_base(padd(pset1<Packet>(low), pmul(pset1<Packet>(step),plset<Scalar>(-packet_traits<Scalar>::size)))) {}
551 template<
typename Index>
552 EIGEN_STRONG_INLINE
const Scalar operator() (Index i)
const
554 m_base = padd(m_base, pset1<Packet>(m_step));
555 return m_low+i*m_step;
558 template<
typename Index>
559 EIGEN_STRONG_INLINE
const Packet packetOp(Index)
const {
return m_base = padd(m_base,m_packetStep); }
563 const Packet m_packetStep;
564 mutable Packet m_base;
570 template <
typename Scalar>
571 struct linspaced_op_impl<Scalar,true>
573 typedef typename packet_traits<Scalar>::type Packet;
575 linspaced_op_impl(Scalar low, Scalar step) :
576 m_low(low), m_step(step),
577 m_lowPacket(pset1<Packet>(m_low)), m_stepPacket(pset1<Packet>(m_step)), m_interPacket(plset<Scalar>(0)) {}
579 template<
typename Index>
580 EIGEN_STRONG_INLINE
const Scalar operator() (Index i)
const {
return m_low+i*m_step; }
582 template<
typename Index>
583 EIGEN_STRONG_INLINE
const Packet packetOp(Index i)
const
584 {
return internal::padd(m_lowPacket, pmul(m_stepPacket, padd(pset1<Packet>(i),m_interPacket))); }
588 const Packet m_lowPacket;
589 const Packet m_stepPacket;
590 const Packet m_interPacket;
598 template <
typename Scalar,
bool RandomAccess = true>
struct linspaced_op;
599 template <
typename Scalar,
bool RandomAccess>
struct functor_traits< linspaced_op<Scalar,RandomAccess> >
600 {
enum { Cost = 1, PacketAccess = packet_traits<Scalar>::HasSetLinear, IsRepeatable =
true }; };
601 template <
typename Scalar,
bool RandomAccess>
struct linspaced_op
603 typedef typename packet_traits<Scalar>::type Packet;
604 linspaced_op(Scalar low, Scalar high,
int num_steps) : impl((num_steps==1 ? high : low), (num_steps==1 ? Scalar() : (high-low)/(num_steps-1))) {}
606 template<
typename Index>
607 EIGEN_STRONG_INLINE
const Scalar operator() (Index i)
const {
return impl(i); }
611 template<
typename Index>
612 EIGEN_STRONG_INLINE
const Scalar operator() (Index row, Index col)
const
614 eigen_assert(col==0 || row==0);
615 return impl(col + row);
618 template<
typename Index>
619 EIGEN_STRONG_INLINE
const Packet packetOp(Index i)
const {
return impl.packetOp(i); }
623 template<
typename Index>
624 EIGEN_STRONG_INLINE
const Packet packetOp(Index row, Index col)
const
626 eigen_assert(col==0 || row==0);
627 return impl.packetOp(col + row);
633 const linspaced_op_impl<Scalar,RandomAccess> impl;
640 template<
typename Functor>
struct functor_has_linear_access {
enum { ret = 1 }; };
641 template<
typename Scalar>
struct functor_has_linear_access<scalar_identity_op<Scalar> > {
enum { ret = 0 }; };
646 template<
typename Functor>
struct functor_allows_mixing_real_and_complex {
enum { ret = 0 }; };
647 template<
typename LhsScalar,
typename RhsScalar>
struct functor_allows_mixing_real_and_complex<scalar_product_op<LhsScalar,RhsScalar> > {
enum { ret = 1 }; };
648 template<
typename LhsScalar,
typename RhsScalar>
struct functor_allows_mixing_real_and_complex<scalar_conj_product_op<LhsScalar,RhsScalar> > {
enum { ret = 1 }; };
656 template<
typename Scalar>
657 struct scalar_add_op {
658 typedef typename packet_traits<Scalar>::type Packet;
660 inline scalar_add_op(
const scalar_add_op& other) : m_other(other.m_other) { }
661 inline scalar_add_op(
const Scalar& other) : m_other(other) { }
662 inline Scalar operator() (
const Scalar& a)
const {
return a + m_other; }
663 inline const Packet packetOp(
const Packet& a)
const
664 {
return internal::padd(a, pset1<Packet>(m_other)); }
665 const Scalar m_other;
667 template<
typename Scalar>
668 struct functor_traits<scalar_add_op<Scalar> >
669 {
enum { Cost = NumTraits<Scalar>::AddCost, PacketAccess = packet_traits<Scalar>::HasAdd }; };
675 template<
typename Scalar>
struct scalar_sqrt_op {
676 EIGEN_EMPTY_STRUCT_CTOR(scalar_sqrt_op)
677 inline const Scalar operator() (const Scalar& a)
const {
return internal::sqrt(a); }
678 typedef typename packet_traits<Scalar>::type Packet;
679 inline Packet packetOp(
const Packet& a)
const {
return internal::psqrt(a); }
681 template<
typename Scalar>
682 struct functor_traits<scalar_sqrt_op<Scalar> >
684 Cost = 5 * NumTraits<Scalar>::MulCost,
685 PacketAccess = packet_traits<Scalar>::HasSqrt
693 template<
typename Scalar>
struct scalar_cos_op {
694 EIGEN_EMPTY_STRUCT_CTOR(scalar_cos_op)
695 inline Scalar operator() (const Scalar& a)
const {
return internal::cos(a); }
696 typedef typename packet_traits<Scalar>::type Packet;
697 inline Packet packetOp(
const Packet& a)
const {
return internal::pcos(a); }
699 template<
typename Scalar>
700 struct functor_traits<scalar_cos_op<Scalar> >
703 Cost = 5 * NumTraits<Scalar>::MulCost,
704 PacketAccess = packet_traits<Scalar>::HasCos
712 template<
typename Scalar>
struct scalar_sin_op {
713 EIGEN_EMPTY_STRUCT_CTOR(scalar_sin_op)
714 inline const Scalar operator() (const Scalar& a)
const {
return internal::sin(a); }
715 typedef typename packet_traits<Scalar>::type Packet;
716 inline Packet packetOp(
const Packet& a)
const {
return internal::psin(a); }
718 template<
typename Scalar>
719 struct functor_traits<scalar_sin_op<Scalar> >
722 Cost = 5 * NumTraits<Scalar>::MulCost,
723 PacketAccess = packet_traits<Scalar>::HasSin
732 template<
typename Scalar>
struct scalar_tan_op {
733 EIGEN_EMPTY_STRUCT_CTOR(scalar_tan_op)
734 inline const Scalar operator() (const Scalar& a)
const {
return internal::tan(a); }
735 typedef typename packet_traits<Scalar>::type Packet;
736 inline Packet packetOp(
const Packet& a)
const {
return internal::ptan(a); }
738 template<
typename Scalar>
739 struct functor_traits<scalar_tan_op<Scalar> >
742 Cost = 5 * NumTraits<Scalar>::MulCost,
743 PacketAccess = packet_traits<Scalar>::HasTan
751 template<
typename Scalar>
struct scalar_acos_op {
752 EIGEN_EMPTY_STRUCT_CTOR(scalar_acos_op)
753 inline const Scalar operator() (const Scalar& a)
const {
return internal::acos(a); }
754 typedef typename packet_traits<Scalar>::type Packet;
755 inline Packet packetOp(
const Packet& a)
const {
return internal::pacos(a); }
757 template<
typename Scalar>
758 struct functor_traits<scalar_acos_op<Scalar> >
761 Cost = 5 * NumTraits<Scalar>::MulCost,
762 PacketAccess = packet_traits<Scalar>::HasACos
770 template<
typename Scalar>
struct scalar_asin_op {
771 EIGEN_EMPTY_STRUCT_CTOR(scalar_asin_op)
772 inline const Scalar operator() (const Scalar& a)
const {
return internal::asin(a); }
773 typedef typename packet_traits<Scalar>::type Packet;
774 inline Packet packetOp(
const Packet& a)
const {
return internal::pasin(a); }
776 template<
typename Scalar>
777 struct functor_traits<scalar_asin_op<Scalar> >
780 Cost = 5 * NumTraits<Scalar>::MulCost,
781 PacketAccess = packet_traits<Scalar>::HasASin
789 template<
typename Scalar>
790 struct scalar_pow_op {
792 inline scalar_pow_op(
const scalar_pow_op& other) : m_exponent(other.m_exponent) { }
793 inline scalar_pow_op(
const Scalar& exponent) : m_exponent(exponent) {}
794 inline Scalar operator() (
const Scalar& a)
const {
return internal::pow(a, m_exponent); }
795 const Scalar m_exponent;
797 template<
typename Scalar>
798 struct functor_traits<scalar_pow_op<Scalar> >
799 {
enum { Cost = 5 * NumTraits<Scalar>::MulCost, PacketAccess =
false }; };
805 template<
typename Scalar>
806 struct scalar_inverse_mult_op {
807 scalar_inverse_mult_op(
const Scalar& other) : m_other(other) {}
808 inline Scalar operator() (
const Scalar& a)
const {
return m_other / a; }
809 template<
typename Packet>
810 inline const Packet packetOp(
const Packet& a)
const
811 {
return internal::pdiv(pset1<Packet>(m_other),a); }
819 template<
typename Scalar>
820 struct scalar_inverse_op {
821 EIGEN_EMPTY_STRUCT_CTOR(scalar_inverse_op)
822 inline Scalar operator() (const Scalar& a)
const {
return Scalar(1)/a; }
823 template<
typename Packet>
824 inline const Packet packetOp(
const Packet& a)
const
825 {
return internal::pdiv(pset1<Packet>(Scalar(1)),a); }
827 template<
typename Scalar>
828 struct functor_traits<scalar_inverse_op<Scalar> >
829 {
enum { Cost = NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasDiv }; };
835 template<
typename Scalar>
836 struct scalar_square_op {
837 EIGEN_EMPTY_STRUCT_CTOR(scalar_square_op)
838 inline Scalar operator() (const Scalar& a)
const {
return a*a; }
839 template<
typename Packet>
840 inline const Packet packetOp(
const Packet& a)
const
841 {
return internal::pmul(a,a); }
843 template<
typename Scalar>
844 struct functor_traits<scalar_square_op<Scalar> >
845 {
enum { Cost = NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasMul }; };
851 template<
typename Scalar>
852 struct scalar_cube_op {
853 EIGEN_EMPTY_STRUCT_CTOR(scalar_cube_op)
854 inline Scalar operator() (const Scalar& a)
const {
return a*a*a; }
855 template<
typename Packet>
856 inline const Packet packetOp(
const Packet& a)
const
857 {
return internal::pmul(a,pmul(a,a)); }
859 template<
typename Scalar>
860 struct functor_traits<scalar_cube_op<Scalar> >
861 {
enum { Cost = 2*NumTraits<Scalar>::MulCost, PacketAccess = packet_traits<Scalar>::HasMul }; };
866 struct functor_traits<std::multiplies<T> >
867 {
enum { Cost = NumTraits<T>::MulCost, PacketAccess =
false }; };
870 struct functor_traits<std::divides<T> >
871 {
enum { Cost = NumTraits<T>::MulCost, PacketAccess =
false }; };
874 struct functor_traits<std::plus<T> >
875 {
enum { Cost = NumTraits<T>::AddCost, PacketAccess =
false }; };
878 struct functor_traits<std::minus<T> >
879 {
enum { Cost = NumTraits<T>::AddCost, PacketAccess =
false }; };
882 struct functor_traits<std::negate<T> >
883 {
enum { Cost = NumTraits<T>::AddCost, PacketAccess =
false }; };
886 struct functor_traits<std::logical_or<T> >
887 {
enum { Cost = 1, PacketAccess =
false }; };
890 struct functor_traits<std::logical_and<T> >
891 {
enum { Cost = 1, PacketAccess =
false }; };
894 struct functor_traits<std::logical_not<T> >
895 {
enum { Cost = 1, PacketAccess =
false }; };
898 struct functor_traits<std::greater<T> >
899 {
enum { Cost = 1, PacketAccess =
false }; };
902 struct functor_traits<std::less<T> >
903 {
enum { Cost = 1, PacketAccess =
false }; };
906 struct functor_traits<std::greater_equal<T> >
907 {
enum { Cost = 1, PacketAccess =
false }; };
910 struct functor_traits<std::less_equal<T> >
911 {
enum { Cost = 1, PacketAccess =
false }; };
914 struct functor_traits<std::equal_to<T> >
915 {
enum { Cost = 1, PacketAccess =
false }; };
918 struct functor_traits<std::not_equal_to<T> >
919 {
enum { Cost = 1, PacketAccess =
false }; };
922 struct functor_traits<std::binder2nd<T> >
923 {
enum { Cost = functor_traits<T>::Cost, PacketAccess =
false }; };
926 struct functor_traits<std::binder1st<T> >
927 {
enum { Cost = functor_traits<T>::Cost, PacketAccess =
false }; };
930 struct functor_traits<std::unary_negate<T> >
931 {
enum { Cost = 1 + functor_traits<T>::Cost, PacketAccess =
false }; };
934 struct functor_traits<std::binary_negate<T> >
935 {
enum { Cost = 1 + functor_traits<T>::Cost, PacketAccess =
false }; };
937 #ifdef EIGEN_STDEXT_SUPPORT
939 template<
typename T0,
typename T1>
940 struct functor_traits<std::project1st<T0,T1> >
941 {
enum { Cost = 0, PacketAccess =
false }; };
943 template<
typename T0,
typename T1>
944 struct functor_traits<std::project2nd<T0,T1> >
945 {
enum { Cost = 0, PacketAccess =
false }; };
947 template<
typename T0,
typename T1>
948 struct functor_traits<std::select2nd<std::pair<T0,T1> > >
949 {
enum { Cost = 0, PacketAccess =
false }; };
951 template<
typename T0,
typename T1>
952 struct functor_traits<std::select1st<std::pair<T0,T1> > >
953 {
enum { Cost = 0, PacketAccess =
false }; };
955 template<
typename T0,
typename T1>
956 struct functor_traits<std::unary_compose<T0,T1> >
957 {
enum { Cost = functor_traits<T0>::Cost + functor_traits<T1>::Cost, PacketAccess =
false }; };
959 template<
typename T0,
typename T1,
typename T2>
960 struct functor_traits<std::binary_compose<T0,T1,T2> >
961 {
enum { Cost = functor_traits<T0>::Cost + functor_traits<T1>::Cost + functor_traits<T2>::Cost, PacketAccess =
false }; };
963 #endif // EIGEN_STDEXT_SUPPORT
967 #ifdef EIGEN_FUNCTORS_PLUGIN
968 #include EIGEN_FUNCTORS_PLUGIN
975 #endif // EIGEN_FUNCTORS_H