devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add CArg

Browse Source
This commit is contained in:
Charles Schlosser
2023-02-15 21:33:06 +00:00
parent 71a8e60a7a
commit 94b19dc5f2
6 changed files with 67 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
Eigen/src/Core/functors/UnaryFunctors.h
View File

@@ -146,6 +146,28 @@ struct functor_traits<scalar_arg_op<Scalar> >
PacketAccess = packet_traits<Scalar>::HasArg
};
};
/** \internal
* \brief Template functor to compute the complex argument, returned as a complex type
*
* \sa class CwiseUnaryOp, Cwise::carg
*/
template <typename Scalar>
struct scalar_carg_op {
using result_type = Scalar;
EIGEN_DEVICE_FUNC
EIGEN_STRONG_INLINE const Scalar operator()(const Scalar& a) const { return Scalar(numext::arg(a)); }
template <typename Packet>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const Packet packetOp(const Packet& a) const {
return pcarg(a);
}
};
template <typename Scalar>
struct functor_traits<scalar_carg_op<Scalar>> {
using RealScalar = typename NumTraits<Scalar>::Real;
enum { Cost = functor_traits<scalar_atan2_op<RealScalar>>::Cost, PacketAccess = packet_traits<RealScalar>::HasATan };
};
/** \internal
* \brief Template functor to cast a scalar to another type
*