// SPDX-FileCopyrightText: 2006-2008 Benoit Jacob <jacob.benoit.1@gmail.com>
// SPDX-FileCopyrightText: 2008 Gael Guennebaud <gael.guennebaud@inria.fr>
// SPDX-FileCopyrightText: The Eigen Authors
// SPDX-License-Identifier: MPL-2.0

#ifndef EIGEN_TEST_NUMERICAL_TEST_HELPERS_H
#define EIGEN_TEST_NUMERICAL_TEST_HELPERS_H

#include <algorithm>
#include <complex>
#include <iostream>
#include <type_traits>

#include <Eigen/Core>

#define VERIFY_IS_EQUAL(a, b) VERIFY(test_is_equal(a, b, true))
#define VERIFY_IS_NOT_EQUAL(a, b) VERIFY(test_is_equal(a, b, false))
#define VERIFY_IS_APPROX(a, b) VERIFY(verifyIsApprox(a, b))
#define VERIFY_IS_NOT_APPROX(a, b) VERIFY(!test_isApprox(a, b))
#define VERIFY_IS_MUCH_SMALLER_THAN(a, b) VERIFY(test_isMuchSmallerThan(a, b))
#define VERIFY_IS_NOT_MUCH_SMALLER_THAN(a, b) VERIFY(!test_isMuchSmallerThan(a, b))
#define VERIFY_IS_APPROX_OR_LESS_THAN(a, b) VERIFY(test_isApproxOrLessThan(a, b))
#define VERIFY_IS_CWISE_EQUAL(a, b) VERIFY(verifyIsCwiseApprox(a, b, true))
#define VERIFY_IS_CWISE_APPROX(a, b) VERIFY(verifyIsCwiseApprox(a, b, false))
#define VERIFY_IS_APPROX_SCALED(a, b, scale) VERIFY(verifyIsApproxScaled(a, b, scale))

#define VERIFY_IS_UNITARY(a) VERIFY(test_isUnitary(a))

namespace Eigen {

#if EIGEN_COMP_ICC
template <typename T, typename U>
bool test_is_equal(const T& actual, const U& expected, bool expect_equal = true);
#endif

// Legacy defaults for broad consistency checks, not accuracy bounds. Numerical tests should use explicit
// bounds derived from epsilon, the operation count and conditioning, or a documented ULP budget.
template <typename T>
inline typename NumTraits<T>::Real test_precision() {
  return NumTraits<T>::dummy_precision();
}
template <>
inline float test_precision<float>() {
  return 1e-3f;
}
template <>
inline double test_precision<double>() {
  return 1e-6;
}
template <>
inline long double test_precision<long double>() {
  return 1e-6l;
}
template <>
inline float test_precision<std::complex<float>>() {
  return test_precision<float>();
}
template <>
inline double test_precision<std::complex<double>>() {
  return test_precision<double>();
}
template <>
inline long double test_precision<std::complex<long double>>() {
  return test_precision<long double>();
}

#define EIGEN_TEST_SCALAR_TEST_OVERLOAD(TYPE)                                                                    \
  inline bool test_isCwiseApprox(TYPE a, TYPE b, bool exact,                                                     \
                                 typename NumTraits<TYPE>::Real precision = test_precision<TYPE>()) {            \
    return numext::equal_strict(a, b) || ((numext::isnan)(a) && (numext::isnan)(b)) ||                           \
           (!exact && (numext::isfinite)(a) && (numext::isfinite)(b) && internal::isApprox(a, b, precision));    \
  }                                                                                                              \
  inline bool test_isApprox(TYPE a, TYPE b) { return test_isCwiseApprox(a, b, false); }                          \
  inline bool test_isMuchSmallerThan(TYPE a, TYPE b) {                                                           \
    return (numext::isfinite)(a) && internal::isMuchSmallerThan(a, b, test_precision<TYPE>());                   \
  }                                                                                                              \
  inline bool test_isApproxOrLessThan(TYPE a, TYPE b) {                                                          \
    return a <= b ||                                                                                             \
           ((numext::isfinite)(a) && (numext::isfinite)(b) && internal::isApprox(a, b, test_precision<TYPE>())); \
  }

EIGEN_TEST_SCALAR_TEST_OVERLOAD(short)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(bool)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(unsigned short)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(int)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(unsigned int)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(long)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(unsigned long)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(long long)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(unsigned long long)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(float)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(double)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(half)
EIGEN_TEST_SCALAR_TEST_OVERLOAD(bfloat16)

#ifndef EIGEN_TEST_NO_LONGDOUBLE
EIGEN_TEST_SCALAR_TEST_OVERLOAD(long double)
#endif

#undef EIGEN_TEST_SCALAR_TEST_OVERLOAD

#ifndef EIGEN_TEST_NO_COMPLEX
#define EIGEN_TEST_COMPLEX_TEST_OVERLOAD(REAL)                                                               \
  inline bool test_isApprox(const std::complex<REAL>& a, const std::complex<REAL>& b) {                      \
    using Vector = Matrix<std::complex<REAL>, 1, 1>;                                                         \
    return Vector::Constant(a).isApprox(Vector::Constant(b), test_precision<std::complex<REAL>>());          \
  }                                                                                                          \
  inline bool test_isMuchSmallerThan(const std::complex<REAL>& a, const std::complex<REAL>& b) {             \
    using Vector = Matrix<std::complex<REAL>, 1, 1>;                                                         \
    return Vector::Constant(a).isMuchSmallerThan(Vector::Constant(b), test_precision<std::complex<REAL>>()); \
  }                                                                                                          \
  inline bool test_isCwiseApprox(const std::complex<REAL>& a, const std::complex<REAL>& b, bool exact,       \
                                 REAL precision = test_precision<std::complex<REAL>>()) {                    \
    using Vector = Matrix<std::complex<REAL>, 1, 1>;                                                         \
    return numext::equal_strict(a, b) || ((numext::isnan)(a) && (numext::isnan)(b)) ||                       \
           (!exact && Vector::Constant(a).isApprox(Vector::Constant(b), precision));                         \
  }

EIGEN_TEST_COMPLEX_TEST_OVERLOAD(float)
EIGEN_TEST_COMPLEX_TEST_OVERLOAD(double)
#ifndef EIGEN_TEST_NO_LONGDOUBLE
EIGEN_TEST_COMPLEX_TEST_OVERLOAD(long double)
#endif
#undef EIGEN_TEST_COMPLEX_TEST_OVERLOAD
#endif

template <typename Scalar, bool Scaled>
struct test_relative_error_impl {
  template <typename X, typename Y>
  static typename NumTraits<Scalar>::Real run(const X& a, const Y& b) {
    return numext::sqrt((a.matrix() - b.matrix()).cwiseAbs2().sum() /
                        numext::mini(a.cwiseAbs2().sum(), b.cwiseAbs2().sum()));
  }
};

template <typename Scalar>
struct test_relative_error_impl<Scalar, true> {
  template <typename X, typename Y>
  static typename NumTraits<Scalar>::Real run(const X& a, const Y& b) {
    using Real = typename NumTraits<Scalar>::Real;
    const auto na = internal::scaled_comparison_norm(a);
    const auto nb = internal::scaled_comparison_norm(b);
    const auto denominator = na <= nb ? na : nb;
    const auto difference = internal::scaled_comparison_distance(a, b);
    if (difference.fraction == 0 && denominator.fraction == 0) return Real(0);
    return Real(numext::ldexp(difference.fraction / denominator.fraction, difference.exponent - denominator.exponent));
  }
};

// Promote integers before subtraction and squaring, avoiding Boolean subtraction and integer overflow. Other
// scalars keep their type: a custom complex scalar can inherit a real NonInteger from NumTraits<Real>.
template <typename Scalar>
using test_difference_scalar_t =
    std::conditional_t<NumTraits<Scalar>::IsInteger, typename NumTraits<Scalar>::NonInteger, Scalar>;

// test_relative_error returns the relative difference between a and b as a real scalar as used in isApprox.
template <typename T1, typename T2>
typename NumTraits<typename T1::RealScalar>::NonInteger test_relative_error(const EigenBase<T1>& a,
                                                                            const EigenBase<T2>& b) {
  using DiffScalar1 = test_difference_scalar_t<typename T1::Scalar>;
  using DiffScalar2 = test_difference_scalar_t<typename T2::Scalar>;
  typename internal::nested_eval<T1, 2>::type ea(a.derived());
  typename internal::nested_eval<T2, 2>::type eb(b.derived());
  // Exponent scaling needs binary floating-point on both sides.
  constexpr bool kScaled =
      internal::use_scaled_comparison<DiffScalar1>::value && internal::use_scaled_comparison<DiffScalar2>::value;
  return test_relative_error_impl<DiffScalar1, kScaled>::run(ea.template cast<DiffScalar1>(),
                                                             eb.template cast<DiffScalar2>());
}

template <typename T1, typename T2>
typename T1::RealScalar test_relative_error(const T1& a, const T2& b, const typename T1::Coefficients* = 0) {
  return test_relative_error(a.coeffs(), b.coeffs());
}

template <typename T1, typename T2>
typename T1::Scalar test_relative_error(const T1& a, const T2& b, const typename T1::MatrixType* = 0) {
  return test_relative_error(a.matrix(), b.matrix());
}

template <typename S, int D>
S test_relative_error(const Translation<S, D>& a, const Translation<S, D>& b) {
  return test_relative_error(a.vector(), b.vector());
}

template <typename S, int D, int O>
S test_relative_error(const ParametrizedLine<S, D, O>& a, const ParametrizedLine<S, D, O>& b) {
  return (std::max)(test_relative_error(a.origin(), b.origin()), test_relative_error(a.direction(), b.direction()));
}

template <typename S, int D>
S test_relative_error(const AlignedBox<S, D>& a, const AlignedBox<S, D>& b) {
  return (std::max)(test_relative_error((a.min)(), (b.min)()), test_relative_error((a.max)(), (b.max)()));
}

template <typename Derived>
class SparseMatrixBase;
template <typename T1, typename T2>
typename T1::RealScalar test_relative_error(const MatrixBase<T1>& a, const SparseMatrixBase<T2>& b) {
  return test_relative_error(a, b.toDense());
}

template <typename Derived>
class SparseMatrixBase;
template <typename T1, typename T2>
typename T1::RealScalar test_relative_error(const SparseMatrixBase<T1>& a, const MatrixBase<T2>& b) {
  return test_relative_error(a.toDense(), b);
}

template <typename Derived>
class SparseMatrixBase;
template <typename T1, typename T2>
typename T1::RealScalar test_relative_error(const SparseMatrixBase<T1>& a, const SparseMatrixBase<T2>& b) {
  return test_relative_error(a.toDense(), b.toDense());
}

template <typename T1, typename T2,
          std::enable_if_t<internal::is_arithmetic<typename NumTraits<T1>::Real>::value, int> = 0>
typename NumTraits<typename NumTraits<T1>::Real>::NonInteger test_relative_error(const T1& a, const T2& b) {
  using Scalar = decltype(a - b);
  using Real = typename NumTraits<typename NumTraits<T1>::Real>::NonInteger;
  using Vector = Matrix<Scalar, 1, 1>;
  return Real(test_relative_error(Vector::Constant(Scalar(a)), Vector::Constant(Scalar(b))));
}

template <typename T>
T test_relative_error(const Rotation2D<T>& a, const Rotation2D<T>& b) {
  return test_relative_error(a.angle(), b.angle());
}

template <typename T>
T test_relative_error(const AngleAxis<T>& a, const AngleAxis<T>& b) {
  return (std::max)(test_relative_error(a.angle(), b.angle()), test_relative_error(a.axis(), b.axis()));
}

template <typename Type1, typename Type2>
inline bool test_isApprox(const Type1& a, const Type2& b, typename Type1::Scalar* = 0)  // Enabled for Eigen's type only
{
  return a.isApprox(b, test_precision<typename Type1::Scalar>());
}

// get_test_precision is a small wrapper to test_precision allowing to return the scalar precision for either scalars or
// expressions
template <typename T>
typename NumTraits<typename T::Scalar>::Real get_test_precision(const T&, const typename T::Scalar* = 0) {
  return test_precision<typename NumTraits<typename T::Scalar>::Real>();
}

template <typename T, std::enable_if_t<internal::is_arithmetic<typename NumTraits<T>::Real>::value, int> = 0>
typename NumTraits<T>::Real get_test_precision(const T&) {
  return test_precision<typename NumTraits<T>::Real>();
}

// verifyIsApprox is a wrapper to test_isApprox that outputs the relative difference magnitude if the test fails.
template <typename Type1, typename Type2>
inline bool verifyIsApprox(const Type1& a, const Type2& b) {
  bool ret = test_isApprox(a, b);
  if (!ret) {
    std::cerr << "Difference too large wrt tolerance " << get_test_precision(a)
              << ", relative error is: " << test_relative_error(a, b) << std::endl;
  }
  return ret;
}

// verifyIsCwiseApprox is a wrapper to test_isCwiseApprox that outputs the relative difference magnitude if the test
// fails.
template <typename Type1, typename Type2>
inline bool verifyIsCwiseApprox(const Type1& a, const Type2& b, bool exact) {
  bool ret = test_isCwiseApprox(a, b, exact);
  if (!ret) {
    if (exact) {
      std::cerr << "Values are not an exact match";
    } else {
      std::cerr << "Difference too large wrt tolerance " << get_test_precision(a);
    }
    std::cerr << ", relative error is: " << test_relative_error(a, b) << std::endl;
  }
  return ret;
}

// Largest coefficient magnitude of a matrix or of an array expression.
template <typename Derived>
typename NumTraits<typename Derived::Scalar>::Real max_abs_coeff(const DenseBase<Derived>& m) {
  using Real = typename NumTraits<typename Derived::Scalar>::Real;
  return m.size() == 0 ? Real(0) : m.derived().matrix().cwiseAbs().template maxCoeff<PropagateNaN>();
}

// Compares two expressions that are mathematically equal but whose evaluations differ by rounding proportional to
// `scale` rather than to the result. verifyIsApprox measures the error relative to the result, which no
// implementation can meet once the result is formed by cancellation.
template <typename Type1, typename Type2>
inline bool verifyIsApproxScaled(const Type1& a, const Type2& b,
                                 const typename NumTraits<typename Type1::Scalar>::Real& scale) {
  typedef typename NumTraits<typename Type1::Scalar>::Real RealScalar;
  const RealScalar error = max_abs_coeff((a - b).eval());
  const RealScalar tolerance = test_precision<typename Type1::Scalar>() * scale;
  // An infinite or NaN bound would admit any result, so treat it as a failure of the caller rather
  // than as a passing comparison.
  if (!((numext::isfinite)(tolerance))) {
    std::cerr << "Invalid tolerance " << tolerance << " for scale " << scale << std::endl;
    return false;
  }
  if (!(error <= tolerance)) {
    std::cerr << "Difference " << error << " too large wrt tolerance " << tolerance << std::endl;
    return false;
  }
  return true;
}

template <typename Derived1, typename Derived2>
inline bool test_isMuchSmallerThan(const DenseBase<Derived1>& m1, const DenseBase<Derived2>& m2) {
  return m1.isMuchSmallerThan(m2, test_precision<typename internal::traits<Derived1>::Scalar>());
}

template <typename Derived>
inline bool test_isMuchSmallerThan(const DenseBase<Derived>& m,
                                   const typename NumTraits<typename internal::traits<Derived>::Scalar>::Real& s) {
  return m.isMuchSmallerThan(s, test_precision<typename internal::traits<Derived>::Scalar>());
}

template <typename Derived>
inline bool test_isUnitary(const MatrixBase<Derived>& m) {
  return m.isUnitary(test_precision<typename internal::traits<Derived>::Scalar>());
}

// Checks component-wise, works with infs and nans. A scalar without a dedicated overload above, such as a custom
// complex type over a built-in real, delegates to its own test_isApprox.
template <typename Scalar,
          std::enable_if_t<
              !std::is_base_of<EigenBase<Scalar>, Scalar>::value && !internal::is_arithmetic<Scalar>::value, int> = 0>
bool test_isCwiseApprox(const Scalar& a, const Scalar& b, bool exact) {
  return numext::equal_strict(a, b) || ((numext::isnan)(a) && (numext::isnan)(b)) || (!exact && test_isApprox(a, b));
}

template <typename Derived1, typename Derived2>
bool test_isCwiseApprox(const DenseBase<Derived1>& m1, const DenseBase<Derived2>& m2, bool exact) {
  if (m1.rows() != m2.rows()) {
    return false;
  }
  if (m1.cols() != m2.cols()) {
    return false;
  }
  for (Index r = 0; r < m1.rows(); ++r) {
    for (Index c = 0; c < m1.cols(); ++c) {
      if (!test_isCwiseApprox(m1(r, c), m2(r, c), exact)) {
        return false;
      }
    }
  }
  return true;
}

template <typename Derived1, typename Derived2>
bool test_isCwiseApprox(const SparseMatrixBase<Derived1>& m1, const SparseMatrixBase<Derived2>& m2, bool exact) {
  return test_isCwiseApprox(m1.toDense(), m2.toDense(), exact);
}

template <typename T, typename U>
bool test_is_equal(const T& actual, const U& expected, bool expect_equal) {
  if (numext::equal_strict(actual, expected) == expect_equal) return true;
  // false:
  std::cerr << "\n    actual   = " << actual << "\n    expected " << (expect_equal ? "= " : "!=") << expected << "\n\n";
  return false;
}

/**
 * Check if number is "not a number" (NaN).
 *
 * @tparam T input type
 * @param x input value
 * @return true, if input value is "not a number" (NaN)
 */
template <typename T>
bool isNotNaN(const T& x) {
  return x == x;
}

/**
 * Check if number is plus infinity.
 *
 * @tparam T input type
 * @param x input value
 * @return true, if input value is plus infinity
 */
template <typename T>
bool isPlusInf(const T& x) {
  return x > NumTraits<T>::highest();
}

/**
 * Check if number is minus infinity.
 *
 * @tparam T input type
 * @param x input value
 * @return true, if input value is minus infinity
 */
template <typename T>
bool isMinusInf(const T& x) {
  return x < NumTraits<T>::lowest();
}

}  // end namespace Eigen

#endif  // EIGEN_TEST_NUMERICAL_TEST_HELPERS_H