xref: /aosp_15_r20/external/eigen/test/eigen2support.cpp (revision bf2c37156dfe67e5dfebd6d394bad8b2ab5804d4)
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2009 Gael Guennebaud <[email protected]>
5 //
6 // This Source Code Form is subject to the terms of the Mozilla
7 // Public License v. 2.0. If a copy of the MPL was not distributed
8 // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9 
10 #define EIGEN2_SUPPORT
11 
12 #include "main.h"
13 
eigen2support(const MatrixType & m)14 template<typename MatrixType> void eigen2support(const MatrixType& m)
15 {
16   typedef typename MatrixType::Scalar Scalar;
17 
18   Index rows = m.rows();
19   Index cols = m.cols();
20 
21   MatrixType m1 = MatrixType::Random(rows, cols),
22              m3(rows, cols);
23 
24   Scalar  s1 = internal::random<Scalar>(),
25           s2 = internal::random<Scalar>();
26 
27   // scalar addition
28   VERIFY_IS_APPROX(m1.cwise() + s1, s1 + m1.cwise());
29   VERIFY_IS_APPROX(m1.cwise() + s1, MatrixType::Constant(rows,cols,s1) + m1);
30   VERIFY_IS_APPROX((m1*Scalar(2)).cwise() - s2, (m1+m1) - MatrixType::Constant(rows,cols,s2) );
31   m3 = m1;
32   m3.cwise() += s2;
33   VERIFY_IS_APPROX(m3, m1.cwise() + s2);
34   m3 = m1;
35   m3.cwise() -= s1;
36   VERIFY_IS_APPROX(m3, m1.cwise() - s1);
37 
38   VERIFY_IS_EQUAL((m1.corner(TopLeft,1,1)), (m1.block(0,0,1,1)));
39   VERIFY_IS_EQUAL((m1.template corner<1,1>(TopLeft)), (m1.template block<1,1>(0,0)));
40   VERIFY_IS_EQUAL((m1.col(0).start(1)), (m1.col(0).segment(0,1)));
41   VERIFY_IS_EQUAL((m1.col(0).template start<1>()), (m1.col(0).segment(0,1)));
42   VERIFY_IS_EQUAL((m1.col(0).end(1)), (m1.col(0).segment(rows-1,1)));
43   VERIFY_IS_EQUAL((m1.col(0).template end<1>()), (m1.col(0).segment(rows-1,1)));
44 
45   using std::cos;
46   using numext::real;
47   using numext::abs2;
48   VERIFY_IS_EQUAL(ei_cos(s1), cos(s1));
49   VERIFY_IS_EQUAL(ei_real(s1), real(s1));
50   VERIFY_IS_EQUAL(ei_abs2(s1), abs2(s1));
51 
52   m1.minor(0,0);
53 }
54 
EIGEN_DECLARE_TEST(eigen2support)55 EIGEN_DECLARE_TEST(eigen2support)
56 {
57   for(int i = 0; i < g_repeat; i++) {
58     CALL_SUBTEST_1( eigen2support(Matrix<double,1,1>()) );
59     CALL_SUBTEST_2( eigen2support(MatrixXd(1,1)) );
60     CALL_SUBTEST_4( eigen2support(Matrix3f()) );
61     CALL_SUBTEST_5( eigen2support(Matrix4d()) );
62     CALL_SUBTEST_2( eigen2support(MatrixXf(200,200)) );
63     CALL_SUBTEST_6( eigen2support(MatrixXcd(100,100)) );
64   }
65 }
66