You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

19 lines
566 B

  1. #include <Eigen/Core>
  2. #include <iostream>
  3. using namespace StormEigen;
  4. using namespace std;
  5. // define a custom template unary functor
  6. template<typename Scalar>
  7. struct CwiseClampOp {
  8. CwiseClampOp(const Scalar& inf, const Scalar& sup) : m_inf(inf), m_sup(sup) {}
  9. const Scalar operator()(const Scalar& x) const { return x<m_inf ? m_inf : (x>m_sup ? m_sup : x); }
  10. Scalar m_inf, m_sup;
  11. };
  12. int main(int, char**)
  13. {
  14. Matrix4d m1 = Matrix4d::Random();
  15. cout << m1 << endl << "becomes: " << endl << m1.unaryExpr(CwiseClampOp<double>(-0.5,0.5)) << endl;
  16. return 0;
  17. }