source: pacpussensors/trunk/Vislab/lib3dv/eigen/test/vectorwiseop.cpp@ 138

Last change on this file since 138 was 136, checked in by ldecherf, 8 years ago

Doc

File size: 6.9 KB
Line 
1// This file is part of Eigen, a lightweight C++ template library
2// for linear algebra.
3//
4// Copyright (C) 2011 Benoit Jacob <jacob.benoit.1@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#define EIGEN_NO_STATIC_ASSERT
11
12#include "main.h"
13
14template<typename ArrayType> void vectorwiseop_array(const ArrayType& m)
15{
16 typedef typename ArrayType::Index Index;
17 typedef typename ArrayType::Scalar Scalar;
18 typedef Array<Scalar, ArrayType::RowsAtCompileTime, 1> ColVectorType;
19 typedef Array<Scalar, 1, ArrayType::ColsAtCompileTime> RowVectorType;
20
21 Index rows = m.rows();
22 Index cols = m.cols();
23 Index r = internal::random<Index>(0, rows-1),
24 c = internal::random<Index>(0, cols-1);
25
26 ArrayType m1 = ArrayType::Random(rows, cols),
27 m2(rows, cols),
28 m3(rows, cols);
29
30 ColVectorType colvec = ColVectorType::Random(rows);
31 RowVectorType rowvec = RowVectorType::Random(cols);
32
33 // test addition
34
35 m2 = m1;
36 m2.colwise() += colvec;
37 VERIFY_IS_APPROX(m2, m1.colwise() + colvec);
38 VERIFY_IS_APPROX(m2.col(c), m1.col(c) + colvec);
39
40 VERIFY_RAISES_ASSERT(m2.colwise() += colvec.transpose());
41 VERIFY_RAISES_ASSERT(m1.colwise() + colvec.transpose());
42
43 m2 = m1;
44 m2.rowwise() += rowvec;
45 VERIFY_IS_APPROX(m2, m1.rowwise() + rowvec);
46 VERIFY_IS_APPROX(m2.row(r), m1.row(r) + rowvec);
47
48 VERIFY_RAISES_ASSERT(m2.rowwise() += rowvec.transpose());
49 VERIFY_RAISES_ASSERT(m1.rowwise() + rowvec.transpose());
50
51 // test substraction
52
53 m2 = m1;
54 m2.colwise() -= colvec;
55 VERIFY_IS_APPROX(m2, m1.colwise() - colvec);
56 VERIFY_IS_APPROX(m2.col(c), m1.col(c) - colvec);
57
58 VERIFY_RAISES_ASSERT(m2.colwise() -= colvec.transpose());
59 VERIFY_RAISES_ASSERT(m1.colwise() - colvec.transpose());
60
61 m2 = m1;
62 m2.rowwise() -= rowvec;
63 VERIFY_IS_APPROX(m2, m1.rowwise() - rowvec);
64 VERIFY_IS_APPROX(m2.row(r), m1.row(r) - rowvec);
65
66 VERIFY_RAISES_ASSERT(m2.rowwise() -= rowvec.transpose());
67 VERIFY_RAISES_ASSERT(m1.rowwise() - rowvec.transpose());
68
69 // test multiplication
70
71 m2 = m1;
72 m2.colwise() *= colvec;
73 VERIFY_IS_APPROX(m2, m1.colwise() * colvec);
74 VERIFY_IS_APPROX(m2.col(c), m1.col(c) * colvec);
75
76 VERIFY_RAISES_ASSERT(m2.colwise() *= colvec.transpose());
77 VERIFY_RAISES_ASSERT(m1.colwise() * colvec.transpose());
78
79 m2 = m1;
80 m2.rowwise() *= rowvec;
81 VERIFY_IS_APPROX(m2, m1.rowwise() * rowvec);
82 VERIFY_IS_APPROX(m2.row(r), m1.row(r) * rowvec);
83
84 VERIFY_RAISES_ASSERT(m2.rowwise() *= rowvec.transpose());
85 VERIFY_RAISES_ASSERT(m1.rowwise() * rowvec.transpose());
86
87 // test quotient
88
89 m2 = m1;
90 m2.colwise() /= colvec;
91 VERIFY_IS_APPROX(m2, m1.colwise() / colvec);
92 VERIFY_IS_APPROX(m2.col(c), m1.col(c) / colvec);
93
94 VERIFY_RAISES_ASSERT(m2.colwise() /= colvec.transpose());
95 VERIFY_RAISES_ASSERT(m1.colwise() / colvec.transpose());
96
97 m2 = m1;
98 m2.rowwise() /= rowvec;
99 VERIFY_IS_APPROX(m2, m1.rowwise() / rowvec);
100 VERIFY_IS_APPROX(m2.row(r), m1.row(r) / rowvec);
101
102 VERIFY_RAISES_ASSERT(m2.rowwise() /= rowvec.transpose());
103 VERIFY_RAISES_ASSERT(m1.rowwise() / rowvec.transpose());
104
105 m2 = m1;
106 // yes, there might be an aliasing issue there but ".rowwise() /="
107 // is suppposed to evaluate " m2.colwise().sum()" into to temporary to avoid
108 // evaluating the reducions multiple times
109 if(ArrayType::RowsAtCompileTime>2 || ArrayType::RowsAtCompileTime==Dynamic)
110 {
111 m2.rowwise() /= m2.colwise().sum();
112 VERIFY_IS_APPROX(m2, m1.rowwise() / m1.colwise().sum());
113 }
114
115 // all/any
116 Array<bool,Dynamic,Dynamic> mb(rows,cols);
117 mb = (m1.real()<=0.7).colwise().all();
118 VERIFY( (mb.col(c) == (m1.real().col(c)<=0.7).all()).all() );
119 mb = (m1.real()<=0.7).rowwise().all();
120 VERIFY( (mb.row(r) == (m1.real().row(r)<=0.7).all()).all() );
121
122 mb = (m1.real()>=0.7).colwise().any();
123 VERIFY( (mb.col(c) == (m1.real().col(c)>=0.7).any()).all() );
124 mb = (m1.real()>=0.7).rowwise().any();
125 VERIFY( (mb.row(r) == (m1.real().row(r)>=0.7).any()).all() );
126}
127
128template<typename MatrixType> void vectorwiseop_matrix(const MatrixType& m)
129{
130 typedef typename MatrixType::Index Index;
131 typedef typename MatrixType::Scalar Scalar;
132 typedef typename NumTraits<Scalar>::Real RealScalar;
133 typedef Matrix<Scalar, MatrixType::RowsAtCompileTime, 1> ColVectorType;
134 typedef Matrix<Scalar, 1, MatrixType::ColsAtCompileTime> RowVectorType;
135 typedef Matrix<RealScalar, MatrixType::RowsAtCompileTime, 1> RealColVectorType;
136 typedef Matrix<RealScalar, 1, MatrixType::ColsAtCompileTime> RealRowVectorType;
137
138 Index rows = m.rows();
139 Index cols = m.cols();
140 Index r = internal::random<Index>(0, rows-1),
141 c = internal::random<Index>(0, cols-1);
142
143 MatrixType m1 = MatrixType::Random(rows, cols),
144 m2(rows, cols),
145 m3(rows, cols);
146
147 ColVectorType colvec = ColVectorType::Random(rows);
148 RowVectorType rowvec = RowVectorType::Random(cols);
149 RealColVectorType rcres;
150 RealRowVectorType rrres;
151
152 // test addition
153
154 m2 = m1;
155 m2.colwise() += colvec;
156 VERIFY_IS_APPROX(m2, m1.colwise() + colvec);
157 VERIFY_IS_APPROX(m2.col(c), m1.col(c) + colvec);
158
159 VERIFY_RAISES_ASSERT(m2.colwise() += colvec.transpose());
160 VERIFY_RAISES_ASSERT(m1.colwise() + colvec.transpose());
161
162 m2 = m1;
163 m2.rowwise() += rowvec;
164 VERIFY_IS_APPROX(m2, m1.rowwise() + rowvec);
165 VERIFY_IS_APPROX(m2.row(r), m1.row(r) + rowvec);
166
167 VERIFY_RAISES_ASSERT(m2.rowwise() += rowvec.transpose());
168 VERIFY_RAISES_ASSERT(m1.rowwise() + rowvec.transpose());
169
170 // test substraction
171
172 m2 = m1;
173 m2.colwise() -= colvec;
174 VERIFY_IS_APPROX(m2, m1.colwise() - colvec);
175 VERIFY_IS_APPROX(m2.col(c), m1.col(c) - colvec);
176
177 VERIFY_RAISES_ASSERT(m2.colwise() -= colvec.transpose());
178 VERIFY_RAISES_ASSERT(m1.colwise() - colvec.transpose());
179
180 m2 = m1;
181 m2.rowwise() -= rowvec;
182 VERIFY_IS_APPROX(m2, m1.rowwise() - rowvec);
183 VERIFY_IS_APPROX(m2.row(r), m1.row(r) - rowvec);
184
185 VERIFY_RAISES_ASSERT(m2.rowwise() -= rowvec.transpose());
186 VERIFY_RAISES_ASSERT(m1.rowwise() - rowvec.transpose());
187
188 // test norm
189 rrres = m1.colwise().norm();
190 VERIFY_IS_APPROX(rrres(c), m1.col(c).norm());
191 rcres = m1.rowwise().norm();
192 VERIFY_IS_APPROX(rcres(r), m1.row(r).norm());
193
194 // test normalized
195 m2 = m1.colwise().normalized();
196 VERIFY_IS_APPROX(m2.col(c), m1.col(c).normalized());
197 m2 = m1.rowwise().normalized();
198 VERIFY_IS_APPROX(m2.row(r), m1.row(r).normalized());
199
200 // test normalize
201 m2 = m1;
202 m2.colwise().normalize();
203 VERIFY_IS_APPROX(m2.col(c), m1.col(c).normalized());
204 m2 = m1;
205 m2.rowwise().normalize();
206 VERIFY_IS_APPROX(m2.row(r), m1.row(r).normalized());
207}
208
209void test_vectorwiseop()
210{
211 CALL_SUBTEST_1(vectorwiseop_array(Array22cd()));
212 CALL_SUBTEST_2(vectorwiseop_array(Array<double, 3, 2>()));
213 CALL_SUBTEST_3(vectorwiseop_array(ArrayXXf(3, 4)));
214 CALL_SUBTEST_4(vectorwiseop_matrix(Matrix4cf()));
215 CALL_SUBTEST_5(vectorwiseop_matrix(Matrix<float,4,5>()));
216 CALL_SUBTEST_6(vectorwiseop_matrix(MatrixXd(7,2)));
217}
Note: See TracBrowser for help on using the repository browser.