Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 1 | // This file is part of Eigen, a lightweight C++ template library |
| 2 | // for linear algebra. |
| 3 | // |
| 4 | // Copyright (C) 2009 Hauke Heibel <hauke.heibel@gmail.com> |
| 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 | |
| 12 | #include <Eigen/Core> |
| 13 | |
| 14 | using namespace Eigen; |
| 15 | |
| 16 | template <typename Scalar, int Storage> |
| 17 | void run_matrix_tests() |
| 18 | { |
| 19 | typedef Matrix<Scalar, Eigen::Dynamic, Eigen::Dynamic, Storage> MatrixType; |
| 20 | typedef typename MatrixType::Index Index; |
| 21 | |
| 22 | MatrixType m, n; |
| 23 | |
| 24 | // boundary cases ... |
| 25 | m = n = MatrixType::Random(50,50); |
| 26 | m.conservativeResize(1,50); |
| 27 | VERIFY_IS_APPROX(m, n.block(0,0,1,50)); |
| 28 | |
| 29 | m = n = MatrixType::Random(50,50); |
| 30 | m.conservativeResize(50,1); |
| 31 | VERIFY_IS_APPROX(m, n.block(0,0,50,1)); |
| 32 | |
| 33 | m = n = MatrixType::Random(50,50); |
| 34 | m.conservativeResize(50,50); |
| 35 | VERIFY_IS_APPROX(m, n.block(0,0,50,50)); |
| 36 | |
| 37 | // random shrinking ... |
| 38 | for (int i=0; i<25; ++i) |
| 39 | { |
| 40 | const Index rows = internal::random<Index>(1,50); |
| 41 | const Index cols = internal::random<Index>(1,50); |
| 42 | m = n = MatrixType::Random(50,50); |
| 43 | m.conservativeResize(rows,cols); |
| 44 | VERIFY_IS_APPROX(m, n.block(0,0,rows,cols)); |
| 45 | } |
| 46 | |
| 47 | // random growing with zeroing ... |
| 48 | for (int i=0; i<25; ++i) |
| 49 | { |
| 50 | const Index rows = internal::random<Index>(50,75); |
| 51 | const Index cols = internal::random<Index>(50,75); |
| 52 | m = n = MatrixType::Random(50,50); |
| 53 | m.conservativeResizeLike(MatrixType::Zero(rows,cols)); |
| 54 | VERIFY_IS_APPROX(m.block(0,0,n.rows(),n.cols()), n); |
| 55 | VERIFY( rows<=50 || m.block(50,0,rows-50,cols).sum() == Scalar(0) ); |
| 56 | VERIFY( cols<=50 || m.block(0,50,rows,cols-50).sum() == Scalar(0) ); |
| 57 | } |
| 58 | } |
| 59 | |
| 60 | template <typename Scalar> |
| 61 | void run_vector_tests() |
| 62 | { |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 63 | typedef Matrix<Scalar, 1, Eigen::Dynamic> VectorType; |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 64 | |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 65 | VectorType m, n; |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 66 | |
| 67 | // boundary cases ... |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 68 | m = n = VectorType::Random(50); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 69 | m.conservativeResize(1); |
| 70 | VERIFY_IS_APPROX(m, n.segment(0,1)); |
| 71 | |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 72 | m = n = VectorType::Random(50); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 73 | m.conservativeResize(50); |
| 74 | VERIFY_IS_APPROX(m, n.segment(0,50)); |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 75 | |
| 76 | m = n = VectorType::Random(50); |
| 77 | m.conservativeResize(m.rows(),1); |
| 78 | VERIFY_IS_APPROX(m, n.segment(0,1)); |
| 79 | |
| 80 | m = n = VectorType::Random(50); |
| 81 | m.conservativeResize(m.rows(),50); |
| 82 | VERIFY_IS_APPROX(m, n.segment(0,50)); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 83 | |
| 84 | // random shrinking ... |
| 85 | for (int i=0; i<50; ++i) |
| 86 | { |
| 87 | const int size = internal::random<int>(1,50); |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 88 | m = n = VectorType::Random(50); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 89 | m.conservativeResize(size); |
| 90 | VERIFY_IS_APPROX(m, n.segment(0,size)); |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 91 | |
| 92 | m = n = VectorType::Random(50); |
| 93 | m.conservativeResize(m.rows(), size); |
| 94 | VERIFY_IS_APPROX(m, n.segment(0,size)); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 95 | } |
| 96 | |
| 97 | // random growing with zeroing ... |
| 98 | for (int i=0; i<50; ++i) |
| 99 | { |
| 100 | const int size = internal::random<int>(50,100); |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 101 | m = n = VectorType::Random(50); |
| 102 | m.conservativeResizeLike(VectorType::Zero(size)); |
| 103 | VERIFY_IS_APPROX(m.segment(0,50), n); |
| 104 | VERIFY( size<=50 || m.segment(50,size-50).sum() == Scalar(0) ); |
| 105 | |
| 106 | m = n = VectorType::Random(50); |
| 107 | m.conservativeResizeLike(Matrix<Scalar,Dynamic,Dynamic>::Zero(1,size)); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 108 | VERIFY_IS_APPROX(m.segment(0,50), n); |
| 109 | VERIFY( size<=50 || m.segment(50,size-50).sum() == Scalar(0) ); |
| 110 | } |
| 111 | } |
| 112 | |
| 113 | void test_conservative_resize() |
| 114 | { |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 115 | for(int i=0; i<g_repeat; ++i) |
| 116 | { |
| 117 | CALL_SUBTEST_1((run_matrix_tests<int, Eigen::RowMajor>())); |
| 118 | CALL_SUBTEST_1((run_matrix_tests<int, Eigen::ColMajor>())); |
| 119 | CALL_SUBTEST_2((run_matrix_tests<float, Eigen::RowMajor>())); |
| 120 | CALL_SUBTEST_2((run_matrix_tests<float, Eigen::ColMajor>())); |
| 121 | CALL_SUBTEST_3((run_matrix_tests<double, Eigen::RowMajor>())); |
| 122 | CALL_SUBTEST_3((run_matrix_tests<double, Eigen::ColMajor>())); |
| 123 | CALL_SUBTEST_4((run_matrix_tests<std::complex<float>, Eigen::RowMajor>())); |
| 124 | CALL_SUBTEST_4((run_matrix_tests<std::complex<float>, Eigen::ColMajor>())); |
| 125 | CALL_SUBTEST_5((run_matrix_tests<std::complex<double>, Eigen::RowMajor>())); |
| 126 | CALL_SUBTEST_6((run_matrix_tests<std::complex<double>, Eigen::ColMajor>())); |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 127 | |
Carlos Hernandez | 7faaa9f | 2014-08-05 17:53:32 -0700 | [diff] [blame] | 128 | CALL_SUBTEST_1((run_vector_tests<int>())); |
| 129 | CALL_SUBTEST_2((run_vector_tests<float>())); |
| 130 | CALL_SUBTEST_3((run_vector_tests<double>())); |
| 131 | CALL_SUBTEST_4((run_vector_tests<std::complex<float> >())); |
| 132 | CALL_SUBTEST_5((run_vector_tests<std::complex<double> >())); |
| 133 | } |
Narayan Kamath | c981c48 | 2012-11-02 10:59:05 +0000 | [diff] [blame] | 134 | } |