mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-12-21 07:19:46 +08:00
21 lines
649 B
C++
21 lines
649 B
C++
#include <Eigen/Core>
|
|
#include <Eigen/Array>
|
|
USING_PART_OF_NAMESPACE_EIGEN
|
|
using namespace std;
|
|
|
|
// define a custom template unary functor
|
|
template<typename Scalar>
|
|
struct CwiseClampOp {
|
|
CwiseClampOp(const Scalar& inf, const Scalar& sup) : m_inf(inf), m_sup(sup) {}
|
|
const Scalar operator()(const Scalar& x) const { return x<m_inf ? m_inf : (x>m_sup ? m_sup : x); }
|
|
Scalar m_inf, m_sup;
|
|
enum { Cost = Eigen::ConditionalJumpCost + Eigen::NumTraits<Scalar>::AddCost };
|
|
};
|
|
|
|
int main(int, char**)
|
|
{
|
|
Matrix4d m1 = Matrix4d::random();
|
|
cout << m1 << endl << "becomes: " << endl << m1.cwise(CwiseClampOp<double>(-0.5,0.5)) << endl;
|
|
return 0;
|
|
}
|