Big 564: add hasNaN and isFinite members

This commit is contained in:
Gael Guennebaud 2013-04-16 15:10:40 +02:00
parent d4b0c19a46
commit 94e20f485c
4 changed files with 83 additions and 2 deletions

View File

@ -129,6 +129,26 @@ inline typename DenseBase<Derived>::Index DenseBase<Derived>::count() const
return derived().template cast<bool>().template cast<Index>().sum();
}
/** \returns true is \c *this contains at least one Not A Number (NaN).
*
* \sa isFinite()
*/
template<typename Derived>
inline bool DenseBase<Derived>::hasNaN() const
{
return !((derived().array()==derived().array()).all());
}
/** \returns true if \c *this contains only finite numbers, i.e., no NaN and no +/-INF values.
*
* \sa hasNaN()
*/
template<typename Derived>
inline bool DenseBase<Derived>::isFinite() const
{
return !((derived()-derived()).hasNaN());
}
} // end namespace Eigen
#endif // EIGEN_ALLANDANY_H

View File

@ -336,6 +336,9 @@ template<typename Derived> class DenseBase
bool isConstant(const Scalar& value, const RealScalar& prec = NumTraits<Scalar>::dummy_precision()) const;
bool isZero(const RealScalar& prec = NumTraits<Scalar>::dummy_precision()) const;
bool isOnes(const RealScalar& prec = NumTraits<Scalar>::dummy_precision()) const;
inline bool hasNaN() const;
inline bool isFinite() const;
inline Derived& operator*=(const Scalar& other);
inline Derived& operator/=(const Scalar& other);
@ -415,8 +418,6 @@ template<typename Derived> class DenseBase
return derived().coeff(0,0);
}
/////////// Array module ///////////
bool all(void) const;
bool any(void) const;
Index count() const;

View File

@ -222,6 +222,7 @@ ei_add_test(evaluators)
ei_add_test(sizeoverflow)
ei_add_test(prec_inverse_4x4)
ei_add_test(vectorwiseop)
ei_add_test(special_numbers)
ei_add_test(simplicial_cholesky)
ei_add_test(conjugate_gradient)

59
test/special_numbers.cpp Normal file
View File

@ -0,0 +1,59 @@
// This file is part of Eigen, a lightweight C++ template library
// for linear algebra.
//
// Copyright (C) 2013 Gael Guennebaud <gael.guennebaud@inria.fr>
//
// This Source Code Form is subject to the terms of the Mozilla
// Public License v. 2.0. If a copy of the MPL was not distributed
// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
#include "main.h"
template<typename Scalar> void special_numbers()
{
typedef typename NumTraits<Scalar>::Real RealScalar;
typedef Matrix<Scalar, Dynamic,Dynamic> MatType;
int rows = internal::random<int>(1,300);
int cols = internal::random<int>(1,300);
Scalar nan = Scalar(0)/Scalar(0);
Scalar inf = Scalar(1)/Scalar(0);
Scalar s1 = internal::random<Scalar>();
MatType m1 = MatType::Random(rows,cols),
mnan = MatType::Random(rows,cols),
minf = MatType::Random(rows,cols),
mboth = MatType::Random(rows,cols);
int n = internal::random<int>(1,10);
for(int k=0; k<n; ++k)
{
mnan(internal::random<int>(0,rows-1), internal::random<int>(0,cols-1)) = nan;
minf(internal::random<int>(0,rows-1), internal::random<int>(0,cols-1)) = inf;
}
mboth = mnan + minf;
VERIFY(!m1.hasNaN());
VERIFY(m1.isFinite());
VERIFY(mnan.hasNaN());
VERIFY((s1*mnan).hasNaN());
VERIFY(!minf.hasNaN());
VERIFY(!(2*minf).hasNaN());
VERIFY(mboth.hasNaN());
VERIFY(mboth.array().hasNaN());
VERIFY(!mnan.isFinite());
VERIFY(!minf.isFinite());
VERIFY(!(minf-mboth).isFinite());
VERIFY(!mboth.isFinite());
VERIFY(!mboth.array().isFinite());
}
void test_special_numbers()
{
for(int i = 0; i < 10*g_repeat; i++) {
CALL_SUBTEST_1( special_numbers<float>() );
CALL_SUBTEST_1( special_numbers<double>() );
}
}