Skip to content

Commit

Permalink
AK: Add atomic free functions
Browse files Browse the repository at this point in the history
This allows for using atomic operations on any variables,
not only those wrapped in AK::Atomic<T>
  • Loading branch information
tomuta authored and awesomekling committed Jun 4, 2020
1 parent 841364b commit 93b9832
Showing 1 changed file with 67 additions and 0 deletions.
67 changes: 67 additions & 0 deletions AK/Atomic.h
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,63 @@ enum MemoryOrder {
memory_order_seq_cst = __ATOMIC_SEQ_CST
};

template <typename T>
static inline T atomic_exchange(volatile T* var, T desired, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_exchange_n(var, desired, order);
}

template <typename T>
static inline bool atomic_compare_exchange_strong(volatile T* var, T& expected, T desired, MemoryOrder order = memory_order_seq_cst) noexcept
{
if (order == memory_order_acq_rel || order == memory_order_release)
return __atomic_compare_exchange_n(var, &expected, desired, false, memory_order_release, memory_order_acquire);
else
return __atomic_compare_exchange_n(var, &expected, desired, false, order, order);
}

template <typename T>
static inline T atomic_fetch_add(volatile T* var, T val, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_fetch_add(var, val, order);
}

template <typename T>
static inline T atomic_fetch_sub(volatile T* var, T val, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_fetch_sub(var, val, order);
}

template <typename T>
static inline T atomic_fetch_and(volatile T* var, T val, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_fetch_and(var, val, order);
}

template <typename T>
static inline T atomic_fetch_or(volatile T* var, T val, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_fetch_or(var, val, order);
}

template <typename T>
static inline T atomic_fetch_xor(volatile T* var, T val, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_fetch_xor(var, val, order);
}

template <typename T>
static inline T atomic_load(volatile T* var, MemoryOrder order = memory_order_seq_cst) noexcept
{
return __atomic_load_n(var, order);
}

template <typename T>
static inline void atomic_store(volatile T* var, T desired, MemoryOrder order = memory_order_seq_cst) noexcept
{
__atomic_store_n(var, desired, order);
}

template<typename T>
class Atomic {
T m_value { 0 };
Expand All @@ -53,6 +110,11 @@ class Atomic {
{
}

volatile T* ptr()
{
return &m_value;
}

T exchange(T desired, MemoryOrder order = memory_order_seq_cst) volatile noexcept
{
return __atomic_exchange_n(&m_value, desired, order);
Expand Down Expand Up @@ -177,6 +239,11 @@ class Atomic<T*> {
{
}

volatile T** ptr()
{
return &m_value;
}

T* exchange(T* desired, MemoryOrder order = memory_order_seq_cst) volatile noexcept
{
return __atomic_exchange_n(&m_value, desired, order);
Expand Down

0 comments on commit 93b9832

Please sign in to comment.