| 315 | } |
| 316 | |
| 317 | template<typename ArrayType> void array_complex(const ArrayType& m) |
| 318 | { |
| 319 | typedef typename ArrayType::Scalar Scalar; |
| 320 | typedef typename NumTraits<Scalar>::Real RealScalar; |
| 321 | |
| 322 | Index rows = m.rows(); |
| 323 | Index cols = m.cols(); |
| 324 | |
| 325 | ArrayType m1 = ArrayType::Random(rows, cols), |
| 326 | m2(rows, cols), |
| 327 | m4 = m1; |
| 328 | |
| 329 | m4.real() = (m4.real().abs()==RealScalar(0)).select(RealScalar(1),m4.real()); |
| 330 | m4.imag() = (m4.imag().abs()==RealScalar(0)).select(RealScalar(1),m4.imag()); |
| 331 | |
| 332 | Array<RealScalar, -1, -1> m3(rows, cols); |
| 333 | |
| 334 | for (Index i = 0; i < m.rows(); ++i) |
| 335 | for (Index j = 0; j < m.cols(); ++j) |
| 336 | m2(i,j) = sqrt(m1(i,j)); |
| 337 | |
| 338 | // these tests are mostly to check possible compilation issues with free-functions. |
| 339 | VERIFY_IS_APPROX(m1.sin(), sin(m1)); |
| 340 | VERIFY_IS_APPROX(m1.cos(), cos(m1)); |
| 341 | VERIFY_IS_APPROX(m1.tan(), tan(m1)); |
| 342 | VERIFY_IS_APPROX(m1.sinh(), sinh(m1)); |
| 343 | VERIFY_IS_APPROX(m1.cosh(), cosh(m1)); |
| 344 | VERIFY_IS_APPROX(m1.tanh(), tanh(m1)); |
| 345 | VERIFY_IS_APPROX(m1.arg(), arg(m1)); |
| 346 | VERIFY((m1.isNaN() == (Eigen::isnan)(m1)).all()); |
| 347 | VERIFY((m1.isInf() == (Eigen::isinf)(m1)).all()); |
| 348 | VERIFY((m1.isFinite() == (Eigen::isfinite)(m1)).all()); |
| 349 | VERIFY_IS_APPROX(m1.inverse(), inverse(m1)); |
| 350 | VERIFY_IS_APPROX(m1.log(), log(m1)); |
| 351 | VERIFY_IS_APPROX(m1.log10(), log10(m1)); |
| 352 | VERIFY_IS_APPROX(m1.abs(), abs(m1)); |
| 353 | VERIFY_IS_APPROX(m1.abs2(), abs2(m1)); |
| 354 | VERIFY_IS_APPROX(m1.sqrt(), sqrt(m1)); |
| 355 | VERIFY_IS_APPROX(m1.square(), square(m1)); |
| 356 | VERIFY_IS_APPROX(m1.cube(), cube(m1)); |
| 357 | VERIFY_IS_APPROX(cos(m1+RealScalar(3)*m2), cos((m1+RealScalar(3)*m2).eval())); |
| 358 | VERIFY_IS_APPROX(m1.sign(), sign(m1)); |
| 359 | |
| 360 | |
| 361 | VERIFY_IS_APPROX(m1.exp() * m2.exp(), exp(m1+m2)); |
| 362 | VERIFY_IS_APPROX(m1.exp(), exp(m1)); |
| 363 | VERIFY_IS_APPROX(m1.exp() / m2.exp(),(m1-m2).exp()); |
| 364 | |
| 365 | VERIFY_IS_APPROX(sinh(m1), 0.5*(exp(m1)-exp(-m1))); |
| 366 | VERIFY_IS_APPROX(cosh(m1), 0.5*(exp(m1)+exp(-m1))); |
| 367 | VERIFY_IS_APPROX(tanh(m1), (0.5*(exp(m1)-exp(-m1)))/(0.5*(exp(m1)+exp(-m1)))); |
| 368 | |
| 369 | for (Index i = 0; i < m.rows(); ++i) |
| 370 | for (Index j = 0; j < m.cols(); ++j) |
| 371 | m3(i,j) = std::atan2(imag(m1(i,j)), real(m1(i,j))); |
| 372 | VERIFY_IS_APPROX(arg(m1), m3); |
| 373 | |
| 374 | std::complex<RealScalar> zero(0.0,0.0); |