mirror of
https://gitlab.com/libeigen/eigen.git
synced 2024-12-09 07:00:27 +08:00
27 lines
967 B
C++
27 lines
967 B
C++
#include <Eigen/Core>
|
|
USING_PART_OF_NAMESPACE_EIGEN
|
|
using namespace std;
|
|
|
|
// define a custom template binary functor
|
|
struct CwiseMinOp EIGEN_EMPTY_STRUCT {
|
|
template<typename Scalar>
|
|
Scalar operator()(const Scalar& a, const Scalar& b) const { return std::min(a,b); }
|
|
};
|
|
|
|
// define a custom binary operator between two matrices
|
|
template<typename Derived1, typename Derived2>
|
|
const Eigen::CwiseBinaryOp<CwiseMinOp, Derived1, Derived2>
|
|
cwiseMin(const MatrixBase<Derived1> &mat1, const MatrixBase<Derived2> &mat2)
|
|
{
|
|
return Eigen::CwiseBinaryOp<CwiseMinOp, Derived1, Derived2>(mat1, mat2);
|
|
}
|
|
|
|
int main(int, char**)
|
|
{
|
|
Matrix4d m1 = Matrix4d::random(), m2 = Matrix4d::random();
|
|
cout << cwiseMin(m1,m2) << endl; // use our new global operator
|
|
cout << m1.cwise<CwiseMinOp>(m2) << endl; // directly use the generic expression member
|
|
cout << m1.cwise(m2, CwiseMinOp()) << endl; // directly use the generic expression member (variant)
|
|
return 0;
|
|
}
|