//
// 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
//
//===----------------------------------------------------------------------===//

// UNSUPPORTED: c++03, c++11, c++14, c++17
// XFAIL: !has-64-bit-atomics
// XFAIL: target={{x86_64-.*}} && tsan

// integral-type fetch_sub(integral-type, memory_order = memory_order::seq_cst) const noexcept;
// floating-point-type fetch_sub(floating-point-type, memory_order = memory_order::seq_cst) const noexcept;
// T* fetch_sub(difference_type, memory_order = memory_order::seq_cst) const noexcept;

#include <atomic>
#include <cassert>
#include <concepts>
#include <type_traits>
#include <utility>

#include "atomic_helpers.h"
#include "test_helper.h"
#include "test_macros.h"

template <typename T>
concept has_fetch_sub = requires {
  std::declval<T const>().fetch_sub(std::declval<T>());
  std::declval<T const>().fetch_sub(std::declval<T>(), std::declval<std::memory_order>());
};

template <typename T>
struct TestDoesNotHaveFetchSub {
  void operator()() const { static_assert(!has_fetch_sub<std::atomic_ref<T>>); }
};

template <typename T>
struct TestFetchSub {
  void operator()() const {
    if constexpr (std::is_arithmetic_v<T>) {
      T x(T(7));
      std::atomic_ref<T> const a(x);

      {
        std::same_as<T> decltype(auto) y = a.fetch_sub(T(4));
        assert(y == T(7));
        assert(x == T(3));
        ASSERT_NOEXCEPT(a.fetch_sub(T(0)));
      }

      {
        std::same_as<T> decltype(auto) y = a.fetch_sub(T(2), std::memory_order_relaxed);
        assert(y == T(3));
        assert(x == T(1));
        ASSERT_NOEXCEPT(a.fetch_sub(T(0), std::memory_order_relaxed));
      }
    } else if constexpr (std::is_pointer_v<T>) {
      using U = std::remove_pointer_t<T>;
      U t[9]  = {};
      T p{&t[7]};
      std::atomic_ref<T> const a(p);

      {
        std::same_as<T> decltype(auto) y = a.fetch_sub(4);
        assert(y == &t[7]);
        assert(a == &t[3]);
        ASSERT_NOEXCEPT(a.fetch_sub(0));
      }

      {
        std::same_as<T> decltype(auto) y = a.fetch_sub(2, std::memory_order_relaxed);
        assert(y == &t[3]);
        assert(a == &t[1]);
        ASSERT_NOEXCEPT(a.fetch_sub(0, std::memory_order_relaxed));
      }
    } else {
      static_assert(std::is_void_v<T>);
    }

    // memory_order::release
    {
      auto fetch_sub = [](std::atomic_ref<T> const& x, T old_val, T new_val) {
        x.fetch_sub(old_val - new_val, std::memory_order::release);
      };
      auto load = [](std::atomic_ref<T> const& x) { return x.load(std::memory_order::acquire); };
      test_acquire_release<T>(fetch_sub, load);
    }

    // memory_order::seq_cst
    {
      auto fetch_sub_no_arg = [](std::atomic_ref<T> const& x, T old_val, T new_val) { x.fetch_sub(old_val - new_val); };
      auto fetch_sub_with_order = [](std::atomic_ref<T> const& x, T old_val, T new_val) {
        x.fetch_sub(old_val - new_val, std::memory_order::seq_cst);
      };
      auto load = [](std::atomic_ref<T> const& x) { return x.load(); };
      test_seq_cst<T>(fetch_sub_no_arg, load);
      test_seq_cst<T>(fetch_sub_with_order, load);
    }
  }
};

int main(int, char**) {
  TestEachIntegralType<TestFetchSub>()();

  TestFetchSub<float>()();
  TestFetchSub<double>()();

  TestEachPointerType<TestFetchSub>()();

  TestDoesNotHaveFetchSub<bool>()();
  TestDoesNotHaveFetchSub<UserAtomicType>()();
  TestDoesNotHaveFetchSub<LargeUserAtomicType>()();

  return 0;
}
