//===-- Utility class to test different flavors of float div --------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_LIBC_TEST_SRC_MATH_SMOKE_DIVTEST_H
#define LLVM_LIBC_TEST_SRC_MATH_SMOKE_DIVTEST_H

#include "hdr/errno_macros.h"
#include "hdr/fenv_macros.h"
#include "test/UnitTest/FEnvSafeTest.h"
#include "test/UnitTest/FPMatcher.h"
#include "test/UnitTest/RoundingModeUtils.h"
#include "test/UnitTest/Test.h"

using LIBC_NAMESPACE::Sign;

template <typename OutType, typename InType>
class DivTest : public LIBC_NAMESPACE::testing::FEnvSafeTest {

  DECLARE_SPECIAL_CONSTANTS(OutType)

  struct InConstants {
    DECLARE_SPECIAL_CONSTANTS(InType)
  };

  using InFPBits = typename InConstants::FPBits;
  using InStorageType = typename InConstants::StorageType;

  InConstants in;

public:
  using DivFunc = OutType (*)(InType, InType);

  void test_special_numbers(DivFunc func) {
    EXPECT_FP_IS_NAN(func(in.aNaN, in.aNaN));
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.sNaN, in.sNaN), FE_INVALID);

    InType qnan_42 = InFPBits::quiet_nan(Sign::POS, 0x42).get_val();
    EXPECT_FP_IS_NAN(func(qnan_42, in.zero));
    EXPECT_FP_IS_NAN(func(in.zero, qnan_42));

    EXPECT_FP_EQ(inf, func(in.inf, in.zero));
    EXPECT_FP_EQ(neg_inf, func(in.neg_inf, in.zero));
    EXPECT_FP_EQ(neg_inf, func(in.inf, in.neg_zero));
    EXPECT_FP_EQ(inf, func(in.neg_inf, in.neg_zero));
    EXPECT_FP_EQ(zero, func(in.min_normal, in.inf));
    EXPECT_FP_EQ(zero, func(in.zero, in.inf));
    EXPECT_FP_EQ(zero, func(in.neg_zero, in.neg_inf));
    EXPECT_FP_EQ(neg_zero, func(in.min_normal, in.neg_inf));
    EXPECT_FP_EQ(neg_zero, func(in.zero, in.neg_inf));
  }

  void test_division_by_zero(DivFunc func) {
    EXPECT_FP_EQ_WITH_EXCEPTION(inf, func(InType(1.0), in.zero), FE_DIVBYZERO);
    EXPECT_FP_EQ_WITH_EXCEPTION(neg_inf, func(InType(-1.0), in.zero),
                                FE_DIVBYZERO);
    EXPECT_FP_EQ_WITH_EXCEPTION(neg_inf, func(InType(1.0), in.neg_zero),
                                FE_DIVBYZERO);
    EXPECT_FP_EQ_WITH_EXCEPTION(inf, func(InType(1.0), in.zero), FE_DIVBYZERO);
  }

  void test_invalid_operations(DivFunc func) {
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.zero, in.zero), FE_INVALID);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.neg_zero, in.zero), FE_INVALID);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.zero, in.neg_zero), FE_INVALID);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.neg_zero, in.neg_zero), FE_INVALID);

    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.inf, in.inf), FE_INVALID);
    EXPECT_MATH_ERRNO(EDOM);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.neg_inf, in.inf), FE_INVALID);
    EXPECT_MATH_ERRNO(EDOM);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.inf, in.neg_inf), FE_INVALID);
    EXPECT_MATH_ERRNO(EDOM);
    EXPECT_FP_IS_NAN_WITH_EXCEPTION(func(in.neg_inf, in.neg_inf), FE_INVALID);
    EXPECT_MATH_ERRNO(EDOM);
  }

  void test_range_errors(DivFunc func) {
    using namespace LIBC_NAMESPACE::fputil::testing;

    if (ForceRoundingMode r(RoundingMode::Nearest); r.success) {
      EXPECT_FP_EQ_WITH_EXCEPTION(inf, func(in.max_normal, in.min_normal),
                                  FE_OVERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(
          -inf, func(in.neg_max_normal, in.min_denormal), FE_OVERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);

      EXPECT_FP_EQ_WITH_EXCEPTION(zero, func(in.min_denormal, in.max_normal),
                                  FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(
          neg_zero, func(in.neg_min_denormal, in.max_normal), FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
    }

    if (ForceRoundingMode r(RoundingMode::TowardZero); r.success) {
      EXPECT_FP_EQ_WITH_EXCEPTION(
          max_normal, func(in.max_normal, in.min_normal), FE_OVERFLOW);
      EXPECT_FP_EQ_WITH_EXCEPTION(neg_max_normal,
                                  func(in.neg_max_normal, in.min_denormal),
                                  FE_OVERFLOW);

      EXPECT_FP_EQ_WITH_EXCEPTION(zero, func(in.min_denormal, in.max_normal),
                                  FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(
          neg_zero, func(in.neg_min_denormal, in.max_normal), FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
    }

    if (ForceRoundingMode r(RoundingMode::Downward); r.success) {
      EXPECT_FP_EQ_WITH_EXCEPTION(
          max_normal, func(in.max_normal, in.min_normal), FE_OVERFLOW);
      EXPECT_FP_EQ_WITH_EXCEPTION(
          -inf, func(in.neg_max_normal, in.min_denormal), FE_OVERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);

      EXPECT_FP_EQ_WITH_EXCEPTION(zero, func(in.min_denormal, in.max_normal),
                                  FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(neg_min_denormal,
                                  func(in.neg_min_denormal, in.max_normal),
                                  FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
    }

    if (ForceRoundingMode r(RoundingMode::Upward); r.success) {
      EXPECT_FP_EQ_WITH_EXCEPTION(inf, func(in.max_normal, in.min_normal),
                                  FE_OVERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(neg_max_normal,
                                  func(in.neg_max_normal, in.min_denormal),
                                  FE_OVERFLOW);

      EXPECT_FP_EQ_WITH_EXCEPTION(
          min_denormal, func(in.min_denormal, in.max_normal), FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
      EXPECT_FP_EQ_WITH_EXCEPTION(
          neg_zero, func(in.neg_min_denormal, in.max_normal), FE_UNDERFLOW);
      EXPECT_MATH_ERRNO(ERANGE);
    }
  }

  void test_inexact_results(DivFunc func) {
    func(InType(1.0), InType(3.0));
    EXPECT_FP_EXCEPTION(FE_INEXACT);
  }
};

#define LIST_DIV_TESTS(Name, OutType, InType, func)                            \
  using LlvmLibc##Name##Test = DivTest<OutType, InType>;                       \
  TEST_F(LlvmLibc##Name##Test, SpecialNumbers) {                               \
    test_special_numbers(&func);                                               \
  }                                                                            \
  TEST_F(LlvmLibc##Name##Test, DivisionByZero) {                               \
    test_division_by_zero(&func);                                              \
  }                                                                            \
  TEST_F(LlvmLibc##Name##Test, InvalidOperations) {                            \
    test_invalid_operations(&func);                                            \
  }                                                                            \
  TEST_F(LlvmLibc##Name##Test, RangeErrors) { test_range_errors(&func); }      \
  TEST_F(LlvmLibc##Name##Test, InexactResults) { test_inexact_results(&func); }

#endif // LLVM_LIBC_TEST_SRC_MATH_SMOKE_DIVTEST_H