xref: /aosp_15_r20/external/eigen/test/evaluators.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1*bf2c3715SXin Li 
2*bf2c3715SXin Li #include "main.h"
3*bf2c3715SXin Li 
4*bf2c3715SXin Li namespace Eigen {
5*bf2c3715SXin Li 
6*bf2c3715SXin Li   template<typename Lhs,typename Rhs>
7*bf2c3715SXin Li   const Product<Lhs,Rhs>
prod(const Lhs & lhs,const Rhs & rhs)8*bf2c3715SXin Li   prod(const Lhs& lhs, const Rhs& rhs)
9*bf2c3715SXin Li   {
10*bf2c3715SXin Li     return Product<Lhs,Rhs>(lhs,rhs);
11*bf2c3715SXin Li   }
12*bf2c3715SXin Li 
13*bf2c3715SXin Li   template<typename Lhs,typename Rhs>
14*bf2c3715SXin Li   const Product<Lhs,Rhs,LazyProduct>
lazyprod(const Lhs & lhs,const Rhs & rhs)15*bf2c3715SXin Li   lazyprod(const Lhs& lhs, const Rhs& rhs)
16*bf2c3715SXin Li   {
17*bf2c3715SXin Li     return Product<Lhs,Rhs,LazyProduct>(lhs,rhs);
18*bf2c3715SXin Li   }
19*bf2c3715SXin Li 
20*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
21*bf2c3715SXin Li   EIGEN_STRONG_INLINE
copy_using_evaluator(const EigenBase<DstXprType> & dst,const SrcXprType & src)22*bf2c3715SXin Li   DstXprType& copy_using_evaluator(const EigenBase<DstXprType> &dst, const SrcXprType &src)
23*bf2c3715SXin Li   {
24*bf2c3715SXin Li     call_assignment(dst.const_cast_derived(), src.derived(), internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar>());
25*bf2c3715SXin Li     return dst.const_cast_derived();
26*bf2c3715SXin Li   }
27*bf2c3715SXin Li 
28*bf2c3715SXin Li   template<typename DstXprType, template <typename> class StorageBase, typename SrcXprType>
29*bf2c3715SXin Li   EIGEN_STRONG_INLINE
copy_using_evaluator(const NoAlias<DstXprType,StorageBase> & dst,const SrcXprType & src)30*bf2c3715SXin Li   const DstXprType& copy_using_evaluator(const NoAlias<DstXprType, StorageBase>& dst, const SrcXprType &src)
31*bf2c3715SXin Li   {
32*bf2c3715SXin Li     call_assignment(dst, src.derived(), internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar>());
33*bf2c3715SXin Li     return dst.expression();
34*bf2c3715SXin Li   }
35*bf2c3715SXin Li 
36*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
37*bf2c3715SXin Li   EIGEN_STRONG_INLINE
copy_using_evaluator(const PlainObjectBase<DstXprType> & dst,const SrcXprType & src)38*bf2c3715SXin Li   DstXprType& copy_using_evaluator(const PlainObjectBase<DstXprType> &dst, const SrcXprType &src)
39*bf2c3715SXin Li   {
40*bf2c3715SXin Li     #ifdef EIGEN_NO_AUTOMATIC_RESIZING
41*bf2c3715SXin Li     eigen_assert((dst.size()==0 || (IsVectorAtCompileTime ? (dst.size() == src.size())
42*bf2c3715SXin Li                                                           : (dst.rows() == src.rows() && dst.cols() == src.cols())))
43*bf2c3715SXin Li                 && "Size mismatch. Automatic resizing is disabled because EIGEN_NO_AUTOMATIC_RESIZING is defined");
44*bf2c3715SXin Li   #else
45*bf2c3715SXin Li     dst.const_cast_derived().resizeLike(src.derived());
46*bf2c3715SXin Li   #endif
47*bf2c3715SXin Li 
48*bf2c3715SXin Li     call_assignment(dst.const_cast_derived(), src.derived(), internal::assign_op<typename DstXprType::Scalar,typename SrcXprType::Scalar>());
49*bf2c3715SXin Li     return dst.const_cast_derived();
50*bf2c3715SXin Li   }
51*bf2c3715SXin Li 
52*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
add_assign_using_evaluator(const DstXprType & dst,const SrcXprType & src)53*bf2c3715SXin Li   void add_assign_using_evaluator(const DstXprType& dst, const SrcXprType& src)
54*bf2c3715SXin Li   {
55*bf2c3715SXin Li     typedef typename DstXprType::Scalar Scalar;
56*bf2c3715SXin Li     call_assignment(const_cast<DstXprType&>(dst), src.derived(), internal::add_assign_op<Scalar,typename SrcXprType::Scalar>());
57*bf2c3715SXin Li   }
58*bf2c3715SXin Li 
59*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
subtract_assign_using_evaluator(const DstXprType & dst,const SrcXprType & src)60*bf2c3715SXin Li   void subtract_assign_using_evaluator(const DstXprType& dst, const SrcXprType& src)
61*bf2c3715SXin Li   {
62*bf2c3715SXin Li     typedef typename DstXprType::Scalar Scalar;
63*bf2c3715SXin Li     call_assignment(const_cast<DstXprType&>(dst), src.derived(), internal::sub_assign_op<Scalar,typename SrcXprType::Scalar>());
64*bf2c3715SXin Li   }
65*bf2c3715SXin Li 
66*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
multiply_assign_using_evaluator(const DstXprType & dst,const SrcXprType & src)67*bf2c3715SXin Li   void multiply_assign_using_evaluator(const DstXprType& dst, const SrcXprType& src)
68*bf2c3715SXin Li   {
69*bf2c3715SXin Li     typedef typename DstXprType::Scalar Scalar;
70*bf2c3715SXin Li     call_assignment(dst.const_cast_derived(), src.derived(), internal::mul_assign_op<Scalar,typename SrcXprType::Scalar>());
71*bf2c3715SXin Li   }
72*bf2c3715SXin Li 
73*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
divide_assign_using_evaluator(const DstXprType & dst,const SrcXprType & src)74*bf2c3715SXin Li   void divide_assign_using_evaluator(const DstXprType& dst, const SrcXprType& src)
75*bf2c3715SXin Li   {
76*bf2c3715SXin Li     typedef typename DstXprType::Scalar Scalar;
77*bf2c3715SXin Li     call_assignment(dst.const_cast_derived(), src.derived(), internal::div_assign_op<Scalar,typename SrcXprType::Scalar>());
78*bf2c3715SXin Li   }
79*bf2c3715SXin Li 
80*bf2c3715SXin Li   template<typename DstXprType, typename SrcXprType>
swap_using_evaluator(const DstXprType & dst,const SrcXprType & src)81*bf2c3715SXin Li   void swap_using_evaluator(const DstXprType& dst, const SrcXprType& src)
82*bf2c3715SXin Li   {
83*bf2c3715SXin Li     typedef typename DstXprType::Scalar Scalar;
84*bf2c3715SXin Li     call_assignment(dst.const_cast_derived(), src.const_cast_derived(), internal::swap_assign_op<Scalar>());
85*bf2c3715SXin Li   }
86*bf2c3715SXin Li 
87*bf2c3715SXin Li   namespace internal {
88*bf2c3715SXin Li     template<typename Dst, template <typename> class StorageBase, typename Src, typename Func>
call_assignment(const NoAlias<Dst,StorageBase> & dst,const Src & src,const Func & func)89*bf2c3715SXin Li     EIGEN_DEVICE_FUNC void call_assignment(const NoAlias<Dst,StorageBase>& dst, const Src& src, const Func& func)
90*bf2c3715SXin Li     {
91*bf2c3715SXin Li       call_assignment_no_alias(dst.expression(), src, func);
92*bf2c3715SXin Li     }
93*bf2c3715SXin Li 
94*bf2c3715SXin Li     template<typename Dst, template <typename> class StorageBase, typename Src, typename Func>
call_restricted_packet_assignment(const NoAlias<Dst,StorageBase> & dst,const Src & src,const Func & func)95*bf2c3715SXin Li     EIGEN_DEVICE_FUNC void call_restricted_packet_assignment(const NoAlias<Dst,StorageBase>& dst, const Src& src, const Func& func)
96*bf2c3715SXin Li     {
97*bf2c3715SXin Li       call_restricted_packet_assignment_no_alias(dst.expression(), src, func);
98*bf2c3715SXin Li     }
99*bf2c3715SXin Li   }
100*bf2c3715SXin Li 
101*bf2c3715SXin Li }
102*bf2c3715SXin Li 
get_cost(const XprType &)103*bf2c3715SXin Li template<typename XprType> long get_cost(const XprType& ) { return Eigen::internal::evaluator<XprType>::CoeffReadCost; }
104*bf2c3715SXin Li 
105*bf2c3715SXin Li using namespace std;
106*bf2c3715SXin Li 
107*bf2c3715SXin Li #define VERIFY_IS_APPROX_EVALUATOR(DEST,EXPR) VERIFY_IS_APPROX(copy_using_evaluator(DEST,(EXPR)), (EXPR).eval());
108*bf2c3715SXin Li #define VERIFY_IS_APPROX_EVALUATOR2(DEST,EXPR,REF) VERIFY_IS_APPROX(copy_using_evaluator(DEST,(EXPR)), (REF).eval());
109*bf2c3715SXin Li 
EIGEN_DECLARE_TEST(evaluators)110*bf2c3715SXin Li EIGEN_DECLARE_TEST(evaluators)
111*bf2c3715SXin Li {
112*bf2c3715SXin Li   // Testing Matrix evaluator and Transpose
113*bf2c3715SXin Li   Vector2d v = Vector2d::Random();
114*bf2c3715SXin Li   const Vector2d v_const(v);
115*bf2c3715SXin Li   Vector2d v2;
116*bf2c3715SXin Li   RowVector2d w;
117*bf2c3715SXin Li 
118*bf2c3715SXin Li   VERIFY_IS_APPROX_EVALUATOR(v2, v);
119*bf2c3715SXin Li   VERIFY_IS_APPROX_EVALUATOR(v2, v_const);
120*bf2c3715SXin Li 
121*bf2c3715SXin Li   // Testing Transpose
122*bf2c3715SXin Li   VERIFY_IS_APPROX_EVALUATOR(w, v.transpose()); // Transpose as rvalue
123*bf2c3715SXin Li   VERIFY_IS_APPROX_EVALUATOR(w, v_const.transpose());
124*bf2c3715SXin Li 
125*bf2c3715SXin Li   copy_using_evaluator(w.transpose(), v); // Transpose as lvalue
126*bf2c3715SXin Li   VERIFY_IS_APPROX(w,v.transpose().eval());
127*bf2c3715SXin Li 
128*bf2c3715SXin Li   copy_using_evaluator(w.transpose(), v_const);
129*bf2c3715SXin Li   VERIFY_IS_APPROX(w,v_const.transpose().eval());
130*bf2c3715SXin Li 
131*bf2c3715SXin Li   // Testing Array evaluator
132*bf2c3715SXin Li   {
133*bf2c3715SXin Li     ArrayXXf a(2,3);
134*bf2c3715SXin Li     ArrayXXf b(3,2);
135*bf2c3715SXin Li     a << 1,2,3, 4,5,6;
136*bf2c3715SXin Li     const ArrayXXf a_const(a);
137*bf2c3715SXin Li 
138*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(b, a.transpose());
139*bf2c3715SXin Li 
140*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(b, a_const.transpose());
141*bf2c3715SXin Li 
142*bf2c3715SXin Li     // Testing CwiseNullaryOp evaluator
143*bf2c3715SXin Li     copy_using_evaluator(w, RowVector2d::Random());
144*bf2c3715SXin Li     VERIFY((w.array() >= -1).all() && (w.array() <= 1).all()); // not easy to test ...
145*bf2c3715SXin Li 
146*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(w, RowVector2d::Zero());
147*bf2c3715SXin Li 
148*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(w, RowVector2d::Constant(3));
149*bf2c3715SXin Li 
150*bf2c3715SXin Li     // mix CwiseNullaryOp and transpose
151*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(w, Vector2d::Zero().transpose());
152*bf2c3715SXin Li   }
153*bf2c3715SXin Li 
154*bf2c3715SXin Li   {
155*bf2c3715SXin Li     // test product expressions
156*bf2c3715SXin Li     int s = internal::random<int>(1,100);
157*bf2c3715SXin Li     MatrixXf a(s,s), b(s,s), c(s,s), d(s,s);
158*bf2c3715SXin Li     a.setRandom();
159*bf2c3715SXin Li     b.setRandom();
160*bf2c3715SXin Li     c.setRandom();
161*bf2c3715SXin Li     d.setRandom();
162*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(d, (a + b));
163*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(d, (a + b).transpose());
164*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d, prod(a,b), a*b);
165*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d.noalias(), prod(a,b), a*b);
166*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d, prod(a,b) + c, a*b + c);
167*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d, s * prod(a,b), s * a*b);
168*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d, prod(a,b).transpose(), (a*b).transpose());
169*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(d, prod(a,b) + prod(b,c), a*b + b*c);
170*bf2c3715SXin Li 
171*bf2c3715SXin Li     // check that prod works even with aliasing present
172*bf2c3715SXin Li     c = a*a;
173*bf2c3715SXin Li     copy_using_evaluator(a, prod(a,a));
174*bf2c3715SXin Li     VERIFY_IS_APPROX(a,c);
175*bf2c3715SXin Li 
176*bf2c3715SXin Li     // check compound assignment of products
177*bf2c3715SXin Li     d = c;
178*bf2c3715SXin Li     add_assign_using_evaluator(c.noalias(), prod(a,b));
179*bf2c3715SXin Li     d.noalias() += a*b;
180*bf2c3715SXin Li     VERIFY_IS_APPROX(c, d);
181*bf2c3715SXin Li 
182*bf2c3715SXin Li     d = c;
183*bf2c3715SXin Li     subtract_assign_using_evaluator(c.noalias(), prod(a,b));
184*bf2c3715SXin Li     d.noalias() -= a*b;
185*bf2c3715SXin Li     VERIFY_IS_APPROX(c, d);
186*bf2c3715SXin Li   }
187*bf2c3715SXin Li 
188*bf2c3715SXin Li   {
189*bf2c3715SXin Li     // test product with all possible sizes
190*bf2c3715SXin Li     int s = internal::random<int>(1,100);
191*bf2c3715SXin Li     Matrix<float,      1,      1> m11, res11;  m11.setRandom(1,1);
192*bf2c3715SXin Li     Matrix<float,      1,      4> m14, res14;  m14.setRandom(1,4);
193*bf2c3715SXin Li     Matrix<float,      1,Dynamic> m1X, res1X;  m1X.setRandom(1,s);
194*bf2c3715SXin Li     Matrix<float,      4,      1> m41, res41;  m41.setRandom(4,1);
195*bf2c3715SXin Li     Matrix<float,      4,      4> m44, res44;  m44.setRandom(4,4);
196*bf2c3715SXin Li     Matrix<float,      4,Dynamic> m4X, res4X;  m4X.setRandom(4,s);
197*bf2c3715SXin Li     Matrix<float,Dynamic,      1> mX1, resX1;  mX1.setRandom(s,1);
198*bf2c3715SXin Li     Matrix<float,Dynamic,      4> mX4, resX4;  mX4.setRandom(s,4);
199*bf2c3715SXin Li     Matrix<float,Dynamic,Dynamic> mXX, resXX;  mXX.setRandom(s,s);
200*bf2c3715SXin Li 
201*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res11, prod(m11,m11), m11*m11);
202*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res11, prod(m14,m41), m14*m41);
203*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res11, prod(m1X,mX1), m1X*mX1);
204*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res14, prod(m11,m14), m11*m14);
205*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res14, prod(m14,m44), m14*m44);
206*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res14, prod(m1X,mX4), m1X*mX4);
207*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res1X, prod(m11,m1X), m11*m1X);
208*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res1X, prod(m14,m4X), m14*m4X);
209*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res1X, prod(m1X,mXX), m1X*mXX);
210*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res41, prod(m41,m11), m41*m11);
211*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res41, prod(m44,m41), m44*m41);
212*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res41, prod(m4X,mX1), m4X*mX1);
213*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res44, prod(m41,m14), m41*m14);
214*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res44, prod(m44,m44), m44*m44);
215*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res44, prod(m4X,mX4), m4X*mX4);
216*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res4X, prod(m41,m1X), m41*m1X);
217*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res4X, prod(m44,m4X), m44*m4X);
218*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(res4X, prod(m4X,mXX), m4X*mXX);
219*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX1, prod(mX1,m11), mX1*m11);
220*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX1, prod(mX4,m41), mX4*m41);
221*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX1, prod(mXX,mX1), mXX*mX1);
222*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX4, prod(mX1,m14), mX1*m14);
223*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX4, prod(mX4,m44), mX4*m44);
224*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resX4, prod(mXX,mX4), mXX*mX4);
225*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resXX, prod(mX1,m1X), mX1*m1X);
226*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resXX, prod(mX4,m4X), mX4*m4X);
227*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(resXX, prod(mXX,mXX), mXX*mXX);
228*bf2c3715SXin Li   }
229*bf2c3715SXin Li 
230*bf2c3715SXin Li   {
231*bf2c3715SXin Li     ArrayXXf a(2,3);
232*bf2c3715SXin Li     ArrayXXf b(3,2);
233*bf2c3715SXin Li     a << 1,2,3, 4,5,6;
234*bf2c3715SXin Li     const ArrayXXf a_const(a);
235*bf2c3715SXin Li 
236*bf2c3715SXin Li     // this does not work because Random is eval-before-nested:
237*bf2c3715SXin Li     // copy_using_evaluator(w, Vector2d::Random().transpose());
238*bf2c3715SXin Li 
239*bf2c3715SXin Li     // test CwiseUnaryOp
240*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(v2, 3 * v);
241*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(w, (3 * v).transpose());
242*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(b, (a + 3).transpose());
243*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(b, (2 * a_const + 3).transpose());
244*bf2c3715SXin Li 
245*bf2c3715SXin Li     // test CwiseBinaryOp
246*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(v2, v + Vector2d::Ones());
247*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(w, (v + Vector2d::Ones()).transpose().cwiseProduct(RowVector2d::Constant(3)));
248*bf2c3715SXin Li 
249*bf2c3715SXin Li     // dynamic matrices and arrays
250*bf2c3715SXin Li     MatrixXd mat1(6,6), mat2(6,6);
251*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mat1, MatrixXd::Identity(6,6));
252*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mat2, mat1);
253*bf2c3715SXin Li     copy_using_evaluator(mat2.transpose(), mat1);
254*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2.transpose(), mat1);
255*bf2c3715SXin Li 
256*bf2c3715SXin Li     ArrayXXd arr1(6,6), arr2(6,6);
257*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr1, ArrayXXd::Constant(6,6, 3.0));
258*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, arr1);
259*bf2c3715SXin Li 
260*bf2c3715SXin Li     // test automatic resizing
261*bf2c3715SXin Li     mat2.resize(3,3);
262*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mat2, mat1);
263*bf2c3715SXin Li     arr2.resize(9,9);
264*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, arr1);
265*bf2c3715SXin Li 
266*bf2c3715SXin Li     // test direct traversal
267*bf2c3715SXin Li     Matrix3f m3;
268*bf2c3715SXin Li     Array33f a3;
269*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, Matrix3f::Identity());  // matrix, nullary
270*bf2c3715SXin Li     // TODO: find a way to test direct traversal with array
271*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3.transpose(), Matrix3f::Identity().transpose());  // transpose
272*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, 2 * Matrix3f::Identity());  // unary
273*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, Matrix3f::Identity() + Matrix3f::Zero());  // binary
274*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3.block(0,0,2,2), Matrix3f::Identity().block(1,1,2,2));  // block
275*bf2c3715SXin Li 
276*bf2c3715SXin Li     // test linear traversal
277*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, Matrix3f::Zero());  // matrix, nullary
278*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(a3, Array33f::Zero());  // array
279*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3.transpose(), Matrix3f::Zero().transpose());  // transpose
280*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, 2 * Matrix3f::Zero());  // unary
281*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m3, Matrix3f::Zero() + m3);  // binary
282*bf2c3715SXin Li 
283*bf2c3715SXin Li     // test inner vectorization
284*bf2c3715SXin Li     Matrix4f m4, m4src = Matrix4f::Random();
285*bf2c3715SXin Li     Array44f a4, a4src = Matrix4f::Random();
286*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m4, m4src);  // matrix
287*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(a4, a4src);  // array
288*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m4.transpose(), m4src.transpose());  // transpose
289*bf2c3715SXin Li     // TODO: find out why Matrix4f::Zero() does not allow inner vectorization
290*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m4, 2 * m4src);  // unary
291*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m4, m4src + m4src);  // binary
292*bf2c3715SXin Li 
293*bf2c3715SXin Li     // test linear vectorization
294*bf2c3715SXin Li     MatrixXf mX(6,6), mXsrc = MatrixXf::Random(6,6);
295*bf2c3715SXin Li     ArrayXXf aX(6,6), aXsrc = ArrayXXf::Random(6,6);
296*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX, mXsrc);  // matrix
297*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(aX, aXsrc);  // array
298*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX.transpose(), mXsrc.transpose());  // transpose
299*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX, MatrixXf::Zero(6,6));  // nullary
300*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX, 2 * mXsrc);  // unary
301*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX, mXsrc + mXsrc);  // binary
302*bf2c3715SXin Li 
303*bf2c3715SXin Li     // test blocks and slice vectorization
304*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(m4, (mXsrc.block<4,4>(1,0)));
305*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(aX, ArrayXXf::Constant(10, 10, 3.0).block(2, 3, 6, 6));
306*bf2c3715SXin Li 
307*bf2c3715SXin Li     Matrix4f m4ref = m4;
308*bf2c3715SXin Li     copy_using_evaluator(m4.block(1, 1, 2, 3), m3.bottomRows(2));
309*bf2c3715SXin Li     m4ref.block(1, 1, 2, 3) = m3.bottomRows(2);
310*bf2c3715SXin Li     VERIFY_IS_APPROX(m4, m4ref);
311*bf2c3715SXin Li 
312*bf2c3715SXin Li     mX.setIdentity(20,20);
313*bf2c3715SXin Li     MatrixXf mXref = MatrixXf::Identity(20,20);
314*bf2c3715SXin Li     mXsrc = MatrixXf::Random(9,12);
315*bf2c3715SXin Li     copy_using_evaluator(mX.block(4, 4, 9, 12), mXsrc);
316*bf2c3715SXin Li     mXref.block(4, 4, 9, 12) = mXsrc;
317*bf2c3715SXin Li     VERIFY_IS_APPROX(mX, mXref);
318*bf2c3715SXin Li 
319*bf2c3715SXin Li     // test Map
320*bf2c3715SXin Li     const float raw[3] = {1,2,3};
321*bf2c3715SXin Li     float buffer[3] = {0,0,0};
322*bf2c3715SXin Li     Vector3f v3;
323*bf2c3715SXin Li     Array3f a3f;
324*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(v3, Map<const Vector3f>(raw));
325*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(a3f, Map<const Array3f>(raw));
326*bf2c3715SXin Li     Vector3f::Map(buffer) = 2*v3;
327*bf2c3715SXin Li     VERIFY(buffer[0] == 2);
328*bf2c3715SXin Li     VERIFY(buffer[1] == 4);
329*bf2c3715SXin Li     VERIFY(buffer[2] == 6);
330*bf2c3715SXin Li 
331*bf2c3715SXin Li     // test CwiseUnaryView
332*bf2c3715SXin Li     mat1.setRandom();
333*bf2c3715SXin Li     mat2.setIdentity();
334*bf2c3715SXin Li     MatrixXcd matXcd(6,6), matXcd_ref(6,6);
335*bf2c3715SXin Li     copy_using_evaluator(matXcd.real(), mat1);
336*bf2c3715SXin Li     copy_using_evaluator(matXcd.imag(), mat2);
337*bf2c3715SXin Li     matXcd_ref.real() = mat1;
338*bf2c3715SXin Li     matXcd_ref.imag() = mat2;
339*bf2c3715SXin Li     VERIFY_IS_APPROX(matXcd, matXcd_ref);
340*bf2c3715SXin Li 
341*bf2c3715SXin Li     // test Select
342*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(aX, (aXsrc > 0).select(aXsrc, -aXsrc));
343*bf2c3715SXin Li 
344*bf2c3715SXin Li     // test Replicate
345*bf2c3715SXin Li     mXsrc = MatrixXf::Random(6, 6);
346*bf2c3715SXin Li     VectorXf vX = VectorXf::Random(6);
347*bf2c3715SXin Li     mX.resize(6, 6);
348*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mX, mXsrc.colwise() + vX);
349*bf2c3715SXin Li     matXcd.resize(12, 12);
350*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(matXcd, matXcd_ref.replicate(2,2));
351*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(matXcd, (matXcd_ref.replicate<2,2>()));
352*bf2c3715SXin Li 
353*bf2c3715SXin Li     // test partial reductions
354*bf2c3715SXin Li     VectorXd vec1(6);
355*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(vec1, mat1.rowwise().sum());
356*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(vec1, mat1.colwise().sum().transpose());
357*bf2c3715SXin Li 
358*bf2c3715SXin Li     // test MatrixWrapper and ArrayWrapper
359*bf2c3715SXin Li     mat1.setRandom(6,6);
360*bf2c3715SXin Li     arr1.setRandom(6,6);
361*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mat2, arr1.matrix());
362*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, mat1.array());
363*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(mat2, (arr1 + 2).matrix());
364*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, mat1.array() + 2);
365*bf2c3715SXin Li     mat2.array() = arr1 * arr1;
366*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2, (arr1 * arr1).matrix());
367*bf2c3715SXin Li     arr2.matrix() = MatrixXd::Identity(6,6);
368*bf2c3715SXin Li     VERIFY_IS_APPROX(arr2, MatrixXd::Identity(6,6).array());
369*bf2c3715SXin Li 
370*bf2c3715SXin Li     // test Reverse
371*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, arr1.reverse());
372*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, arr1.colwise().reverse());
373*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(arr2, arr1.rowwise().reverse());
374*bf2c3715SXin Li     arr2.reverse() = arr1;
375*bf2c3715SXin Li     VERIFY_IS_APPROX(arr2, arr1.reverse());
376*bf2c3715SXin Li     mat2.array() = mat1.array().reverse();
377*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2.array(), mat1.array().reverse());
378*bf2c3715SXin Li 
379*bf2c3715SXin Li     // test Diagonal
380*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(vec1, mat1.diagonal());
381*bf2c3715SXin Li     vec1.resize(5);
382*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(vec1, mat1.diagonal(1));
383*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR(vec1, mat1.diagonal<-1>());
384*bf2c3715SXin Li     vec1.setRandom();
385*bf2c3715SXin Li 
386*bf2c3715SXin Li     mat2 = mat1;
387*bf2c3715SXin Li     copy_using_evaluator(mat1.diagonal(1), vec1);
388*bf2c3715SXin Li     mat2.diagonal(1) = vec1;
389*bf2c3715SXin Li     VERIFY_IS_APPROX(mat1, mat2);
390*bf2c3715SXin Li 
391*bf2c3715SXin Li     copy_using_evaluator(mat1.diagonal<-1>(), mat1.diagonal(1));
392*bf2c3715SXin Li     mat2.diagonal<-1>() = mat2.diagonal(1);
393*bf2c3715SXin Li     VERIFY_IS_APPROX(mat1, mat2);
394*bf2c3715SXin Li   }
395*bf2c3715SXin Li 
396*bf2c3715SXin Li   {
397*bf2c3715SXin Li     // test swapping
398*bf2c3715SXin Li     MatrixXd mat1, mat2, mat1ref, mat2ref;
399*bf2c3715SXin Li     mat1ref = mat1 = MatrixXd::Random(6, 6);
400*bf2c3715SXin Li     mat2ref = mat2 = 2 * mat1 + MatrixXd::Identity(6, 6);
401*bf2c3715SXin Li     swap_using_evaluator(mat1, mat2);
402*bf2c3715SXin Li     mat1ref.swap(mat2ref);
403*bf2c3715SXin Li     VERIFY_IS_APPROX(mat1, mat1ref);
404*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2, mat2ref);
405*bf2c3715SXin Li 
406*bf2c3715SXin Li     swap_using_evaluator(mat1.block(0, 0, 3, 3), mat2.block(3, 3, 3, 3));
407*bf2c3715SXin Li     mat1ref.block(0, 0, 3, 3).swap(mat2ref.block(3, 3, 3, 3));
408*bf2c3715SXin Li     VERIFY_IS_APPROX(mat1, mat1ref);
409*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2, mat2ref);
410*bf2c3715SXin Li 
411*bf2c3715SXin Li     swap_using_evaluator(mat1.row(2), mat2.col(3).transpose());
412*bf2c3715SXin Li     mat1.row(2).swap(mat2.col(3).transpose());
413*bf2c3715SXin Li     VERIFY_IS_APPROX(mat1, mat1ref);
414*bf2c3715SXin Li     VERIFY_IS_APPROX(mat2, mat2ref);
415*bf2c3715SXin Li   }
416*bf2c3715SXin Li 
417*bf2c3715SXin Li   {
418*bf2c3715SXin Li     // test compound assignment
419*bf2c3715SXin Li     const Matrix4d mat_const = Matrix4d::Random();
420*bf2c3715SXin Li     Matrix4d mat, mat_ref;
421*bf2c3715SXin Li     mat = mat_ref = Matrix4d::Identity();
422*bf2c3715SXin Li     add_assign_using_evaluator(mat, mat_const);
423*bf2c3715SXin Li     mat_ref += mat_const;
424*bf2c3715SXin Li     VERIFY_IS_APPROX(mat, mat_ref);
425*bf2c3715SXin Li 
426*bf2c3715SXin Li     subtract_assign_using_evaluator(mat.row(1), 2*mat.row(2));
427*bf2c3715SXin Li     mat_ref.row(1) -= 2*mat_ref.row(2);
428*bf2c3715SXin Li     VERIFY_IS_APPROX(mat, mat_ref);
429*bf2c3715SXin Li 
430*bf2c3715SXin Li     const ArrayXXf arr_const = ArrayXXf::Random(5,3);
431*bf2c3715SXin Li     ArrayXXf arr, arr_ref;
432*bf2c3715SXin Li     arr = arr_ref = ArrayXXf::Constant(5, 3, 0.5);
433*bf2c3715SXin Li     multiply_assign_using_evaluator(arr, arr_const);
434*bf2c3715SXin Li     arr_ref *= arr_const;
435*bf2c3715SXin Li     VERIFY_IS_APPROX(arr, arr_ref);
436*bf2c3715SXin Li 
437*bf2c3715SXin Li     divide_assign_using_evaluator(arr.row(1), arr.row(2) + 1);
438*bf2c3715SXin Li     arr_ref.row(1) /= (arr_ref.row(2) + 1);
439*bf2c3715SXin Li     VERIFY_IS_APPROX(arr, arr_ref);
440*bf2c3715SXin Li   }
441*bf2c3715SXin Li 
442*bf2c3715SXin Li   {
443*bf2c3715SXin Li     // test triangular shapes
444*bf2c3715SXin Li     MatrixXd A = MatrixXd::Random(6,6), B(6,6), C(6,6), D(6,6);
445*bf2c3715SXin Li     A.setRandom();B.setRandom();
446*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, A.triangularView<Upper>(), MatrixXd(A.triangularView<Upper>()));
447*bf2c3715SXin Li 
448*bf2c3715SXin Li     A.setRandom();B.setRandom();
449*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, A.triangularView<UnitLower>(), MatrixXd(A.triangularView<UnitLower>()));
450*bf2c3715SXin Li 
451*bf2c3715SXin Li     A.setRandom();B.setRandom();
452*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, A.triangularView<UnitUpper>(), MatrixXd(A.triangularView<UnitUpper>()));
453*bf2c3715SXin Li 
454*bf2c3715SXin Li     A.setRandom();B.setRandom();
455*bf2c3715SXin Li     C = B; C.triangularView<Upper>() = A;
456*bf2c3715SXin Li     copy_using_evaluator(B.triangularView<Upper>(), A);
457*bf2c3715SXin Li     VERIFY(B.isApprox(C) && "copy_using_evaluator(B.triangularView<Upper>(), A)");
458*bf2c3715SXin Li 
459*bf2c3715SXin Li     A.setRandom();B.setRandom();
460*bf2c3715SXin Li     C = B; C.triangularView<Lower>() = A.triangularView<Lower>();
461*bf2c3715SXin Li     copy_using_evaluator(B.triangularView<Lower>(), A.triangularView<Lower>());
462*bf2c3715SXin Li     VERIFY(B.isApprox(C) && "copy_using_evaluator(B.triangularView<Lower>(), A.triangularView<Lower>())");
463*bf2c3715SXin Li 
464*bf2c3715SXin Li 
465*bf2c3715SXin Li     A.setRandom();B.setRandom();
466*bf2c3715SXin Li     C = B; C.triangularView<Lower>() = A.triangularView<Upper>().transpose();
467*bf2c3715SXin Li     copy_using_evaluator(B.triangularView<Lower>(), A.triangularView<Upper>().transpose());
468*bf2c3715SXin Li     VERIFY(B.isApprox(C) && "copy_using_evaluator(B.triangularView<Lower>(), A.triangularView<Lower>().transpose())");
469*bf2c3715SXin Li 
470*bf2c3715SXin Li 
471*bf2c3715SXin Li     A.setRandom();B.setRandom(); C = B; D = A;
472*bf2c3715SXin Li     C.triangularView<Upper>().swap(D.triangularView<Upper>());
473*bf2c3715SXin Li     swap_using_evaluator(B.triangularView<Upper>(), A.triangularView<Upper>());
474*bf2c3715SXin Li     VERIFY(B.isApprox(C) && "swap_using_evaluator(B.triangularView<Upper>(), A.triangularView<Upper>())");
475*bf2c3715SXin Li 
476*bf2c3715SXin Li 
477*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, prod(A.triangularView<Upper>(),A), MatrixXd(A.triangularView<Upper>()*A));
478*bf2c3715SXin Li 
479*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, prod(A.selfadjointView<Upper>(),A), MatrixXd(A.selfadjointView<Upper>()*A));
480*bf2c3715SXin Li   }
481*bf2c3715SXin Li 
482*bf2c3715SXin Li   {
483*bf2c3715SXin Li     // test diagonal shapes
484*bf2c3715SXin Li     VectorXd d = VectorXd::Random(6);
485*bf2c3715SXin Li     MatrixXd A = MatrixXd::Random(6,6), B(6,6);
486*bf2c3715SXin Li     A.setRandom();B.setRandom();
487*bf2c3715SXin Li 
488*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, lazyprod(d.asDiagonal(),A), MatrixXd(d.asDiagonal()*A));
489*bf2c3715SXin Li     VERIFY_IS_APPROX_EVALUATOR2(B, lazyprod(A,d.asDiagonal()), MatrixXd(A*d.asDiagonal()));
490*bf2c3715SXin Li   }
491*bf2c3715SXin Li 
492*bf2c3715SXin Li   {
493*bf2c3715SXin Li     // test CoeffReadCost
494*bf2c3715SXin Li     Matrix4d a, b;
495*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a), 1 );
496*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a+b), 3);
497*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(2*a+b), 4);
498*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a*b), 1);
499*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a.lazyProduct(b)), 15);
500*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a*(a*b)), 1);
501*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a.lazyProduct(a*b)), 15);
502*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a*(a+b)), 1);
503*bf2c3715SXin Li     VERIFY_IS_EQUAL( get_cost(a.lazyProduct(a+b)), 15);
504*bf2c3715SXin Li   }
505*bf2c3715SXin Li 
506*bf2c3715SXin Li   // regression test for PR 544 and bug 1622 (introduced in #71609c4)
507*bf2c3715SXin Li   {
508*bf2c3715SXin Li     // test restricted_packet_assignment with an unaligned destination
509*bf2c3715SXin Li     const size_t M = 2;
510*bf2c3715SXin Li     const size_t K = 2;
511*bf2c3715SXin Li     const size_t N = 5;
512*bf2c3715SXin Li     float *destMem = new float[(M*N) + 1];
513*bf2c3715SXin Li     float *dest = (internal::UIntPtr(destMem)%EIGEN_MAX_ALIGN_BYTES) == 0 ? destMem+1 : destMem;
514*bf2c3715SXin Li 
515*bf2c3715SXin Li     const Matrix<float, Dynamic, Dynamic, RowMajor> a = Matrix<float, Dynamic, Dynamic, RowMajor>::Random(M, K);
516*bf2c3715SXin Li     const Matrix<float, Dynamic, Dynamic, RowMajor> b = Matrix<float, Dynamic, Dynamic, RowMajor>::Random(K, N);
517*bf2c3715SXin Li 
518*bf2c3715SXin Li     Map<Matrix<float, Dynamic, Dynamic, RowMajor> > z(dest, M, N);;
519*bf2c3715SXin Li     Product<Matrix<float, Dynamic, Dynamic, RowMajor>, Matrix<float, Dynamic, Dynamic, RowMajor>, LazyProduct> tmp(a,b);
520*bf2c3715SXin Li     internal::call_restricted_packet_assignment(z.noalias(), tmp.derived(), internal::assign_op<float, float>());
521*bf2c3715SXin Li 
522*bf2c3715SXin Li     VERIFY_IS_APPROX(z, a*b);
523*bf2c3715SXin Li     delete[] destMem;
524*bf2c3715SXin Li   }
525*bf2c3715SXin Li }
526