xref: /aosp_15_r20/external/eigen/doc/examples/class_CwiseUnaryOp.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1*bf2c3715SXin Li #include <Eigen/Core>
2*bf2c3715SXin Li #include <iostream>
3*bf2c3715SXin Li using namespace Eigen;
4*bf2c3715SXin Li using namespace std;
5*bf2c3715SXin Li 
6*bf2c3715SXin Li // define a custom template unary functor
7*bf2c3715SXin Li template<typename Scalar>
8*bf2c3715SXin Li struct CwiseClampOp {
CwiseClampOpCwiseClampOp9*bf2c3715SXin Li   CwiseClampOp(const Scalar& inf, const Scalar& sup) : m_inf(inf), m_sup(sup) {}
operator ()CwiseClampOp10*bf2c3715SXin Li   const Scalar operator()(const Scalar& x) const { return x<m_inf ? m_inf : (x>m_sup ? m_sup : x); }
11*bf2c3715SXin Li   Scalar m_inf, m_sup;
12*bf2c3715SXin Li };
13*bf2c3715SXin Li 
main(int,char **)14*bf2c3715SXin Li int main(int, char**)
15*bf2c3715SXin Li {
16*bf2c3715SXin Li   Matrix4d m1 = Matrix4d::Random();
17*bf2c3715SXin Li   cout << m1 << endl << "becomes: " << endl << m1.unaryExpr(CwiseClampOp<double>(-0.5,0.5)) << endl;
18*bf2c3715SXin Li   return 0;
19*bf2c3715SXin Li }
20