• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // This file is part of Eigen, a lightweight C++ template library
2 // for linear algebra.
3 //
4 // Copyright (C) 2009 Gael Guennebaud <gael.guennebaud@inria.fr>
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 #include "main.h"
11 
replicate(const MatrixType & m)12 template<typename MatrixType> void replicate(const MatrixType& m)
13 {
14   /* this test covers the following files:
15      Replicate.cpp
16   */
17   typedef typename MatrixType::Scalar Scalar;
18   typedef Matrix<Scalar, MatrixType::RowsAtCompileTime, 1> VectorType;
19   typedef Matrix<Scalar, Dynamic, Dynamic> MatrixX;
20   typedef Matrix<Scalar, Dynamic, 1> VectorX;
21 
22   Index rows = m.rows();
23   Index cols = m.cols();
24 
25   MatrixType m1 = MatrixType::Random(rows, cols),
26              m2 = MatrixType::Random(rows, cols);
27 
28   VectorType v1 = VectorType::Random(rows);
29 
30   MatrixX x1, x2;
31   VectorX vx1;
32 
33   int  f1 = internal::random<int>(1,10),
34        f2 = internal::random<int>(1,10);
35 
36   x1.resize(rows*f1,cols*f2);
37   for(int j=0; j<f2; j++)
38   for(int i=0; i<f1; i++)
39     x1.block(i*rows,j*cols,rows,cols) = m1;
40   VERIFY_IS_APPROX(x1, m1.replicate(f1,f2));
41 
42   x2.resize(2*rows,3*cols);
43   x2 << m2, m2, m2,
44         m2, m2, m2;
45   VERIFY_IS_APPROX(x2, (m2.template replicate<2,3>()));
46 
47   x2.resize(rows,3*cols);
48   x2 << m2, m2, m2;
49   VERIFY_IS_APPROX(x2, (m2.template replicate<1,3>()));
50 
51   vx1.resize(3*rows,cols);
52   vx1 << m2, m2, m2;
53   VERIFY_IS_APPROX(vx1+vx1, vx1+(m2.template replicate<3,1>()));
54 
55   vx1=m2+(m2.colwise().replicate(1));
56 
57   if(m2.cols()==1)
58     VERIFY_IS_APPROX(m2.coeff(0), (m2.template replicate<3,1>().coeff(m2.rows())));
59 
60   x2.resize(rows,f1);
61   for (int j=0; j<f1; ++j)
62     x2.col(j) = v1;
63   VERIFY_IS_APPROX(x2, v1.rowwise().replicate(f1));
64 
65   vx1.resize(rows*f2);
66   for (int j=0; j<f2; ++j)
67     vx1.segment(j*rows,rows) = v1;
68   VERIFY_IS_APPROX(vx1, v1.colwise().replicate(f2));
69 }
70 
EIGEN_DECLARE_TEST(array_replicate)71 EIGEN_DECLARE_TEST(array_replicate)
72 {
73   for(int i = 0; i < g_repeat; i++) {
74     CALL_SUBTEST_1( replicate(Matrix<float, 1, 1>()) );
75     CALL_SUBTEST_2( replicate(Vector2f()) );
76     CALL_SUBTEST_3( replicate(Vector3d()) );
77     CALL_SUBTEST_4( replicate(Vector4f()) );
78     CALL_SUBTEST_5( replicate(VectorXf(16)) );
79     CALL_SUBTEST_6( replicate(VectorXcd(10)) );
80   }
81 }
82