AuroraSupport/AuROXTL
1
1
Fork 0
Code Issues 7 Pull Requests Releases Wiki Activity

[*] Updaterino my dumb atomics

Browse Source
This commit is contained in:
Reece Wilson 2023-08-28 15:25:46 +01:00
parent cda2c02de6
commit aac9d68c3c
1 changed files with 100 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

101
Include/auROXTL/auAtomic.hpp
View File

@ -11,6 +11,27 @@
#include <stdatomic.h>
#endif
// Defines:
// AuAtomicCompareExchange (no weak variants yet)
// -
// AuAtomicTestAndSet
// AuAtomicClearU8Lock
// AuAtomicSet
// AuAtomicUnset (ret: bool)
// AuAtomicAndUnsetBit (ret: T)
// -
// AuAtomicLoadWeak (tbd)
// (no weak store) (tbd)
// -
// AuAtomicLoad
// AuAtomicStore
// -
// AuAtomicOrSetBit
// AuAtomicOr
// AuAtomicAnd
// AuAtomicAdd
// AuAtomicSub
template <class T>
struct AuAtomicUtils
{
@ -77,6 +98,9 @@ struct AuAtomicUtils
//
static void ClearU8Lock(T *in);
//
static T LoadWeak(T *in);
};
#if defined(AURORA_COMPILER_MSVC)
@ -340,6 +364,12 @@ inline auline T AuAtomicUtils<T>::And(T *in, T value)
#endif
template <class T>
inline auline T AuAtomicUtils<T>::LoadWeak(T *in)
{
return AuAtomicUtils<T>::Load(in);
}
#if defined(AURORA_COMPILER_CLANG)
#define ATOMIC_PREFIX_HAX(name) __c11_ ## name
#else
@ -385,6 +415,14 @@ inline auline T AuAtomicUtils<T>::Load(T *in)
{
return ATOMIC_PREFIX_HAX(atomic_load)((atomic_int_fast32_t *)in, __ATOMIC_ACQUIRE);
}
else if constexpr (AuIsSame_v<AuUInt64, T>)
{
return ATOMIC_PREFIX_HAX(atomic_load)((atomic_uint_fast64_t *)in, __ATOMIC_ACQUIRE);
}
else if constexpr (AuIsSame_v<AuInt64, T>)
{
return ATOMIC_PREFIX_HAX(atomic_load)((atomic_int_fast64_t *)in, __ATOMIC_ACQUIRE);
}
else
{
static_assert(AuIsVoid_v<T>, "T");
@ -401,8 +439,47 @@ inline auline void AuAtomicUtils<T>::Store(T *in, T val)
#if defined(AURORA_COMPILER_MSVC) && (defined(AURORA_ARCH_X64) || defined(AURORA_ARCH_X86))
*in = val;
#elif defined(AURORA_COMPILER_MSVC)
*in = val;
#if 0
* in = val;
::MemoryBarrier();
#else
if constexpr (AuIsSame_v<AuUInt8, T>)
{
::InterlockedExchange8((CHAR volatile *)in, (CHAR)val);
}
else if constexpr (AuIsSame_v<AuInt8, T>)
{
::InterlockedExchange8((CHAR volatile *)in, (CHAR)val);
}
else if constexpr (AuIsSame_v<AuUInt16, T>)
{
::InterlockedExchange16((SHORT volatile *)in, (SHORT)val);
}
else if constexpr (AuIsSame_v<AuInt16, T>)
{
::InterlockedExchange16((SHORT volatile *)in, (SHORT)val);
}
else if constexpr (AuIsSame_v<AuUInt32, T>)
{
::InterlockedExchange32((LONG32 volatile *)in, (LONG32)val);
}
else if constexpr (AuIsSame_v<AuInt32, T>)
{
::InterlockedExchange32((LONG32 volatile *)in, (LONG32)val);
}
else if constexpr (AuIsSame_v<AuUInt64, T>)
{
::InterlockedExchange64((LONG64 volatile *)in, (LONG64)val);
}
else if constexpr (AuIsSame_v<AuInt64, T>)
{
::InterlockedExchange64((LONG64 volatile *)in, (LONG64)val);
}
else
{
static_assert(AuIsVoid_v<T>, "T");
}
#endif
#else
if constexpr (AuIsSame_v<AuUInt8, T>)
{
@ -428,6 +505,14 @@ inline auline void AuAtomicUtils<T>::Store(T *in, T val)
{
ATOMIC_PREFIX_HAX(atomic_store)((atomic_int_fast32_t *)in, val, __ATOMIC_RELEASE);
}
else if constexpr (AuIsSame_v<AuUInt64, T>)
{
ATOMIC_PREFIX_HAX(atomic_store)((atomic_uint_fast64_t *)in, val, __ATOMIC_RELEASE);
}
else if constexpr (AuIsSame_v<AuInt64, T>)
{
ATOMIC_PREFIX_HAX(atomic_store)((atomic_int_fast64_t *)in, val, __ATOMIC_RELEASE);
}
else
{
static_assert(AuIsVoid_v<T>, "T");
@ -529,6 +614,13 @@ T AuAtomicLoad(T *in)
return AuAtomicUtils<T>::Load(in);
}
template <class T>
auline
T AuAtomicLoadWeak(T *in)
{
return AuAtomicUtils<T>::LoadWeak(in);
}
template <class T>
auline
void AuAtomicClearU8Lock(T *in)
@ -621,6 +713,13 @@ T AuAtomicLoad(volatile T *in)
return AuAtomicUtils<T>::Load((T *)in);
}
template <class T>
auline
T AuAtomicLoadWeak(volatile T *in)
{
return AuAtomicUtils<T>::LoadWeak((T *)in);
}
template <class T>
auline
void AuAtomicClearU8Lock(volatile T *in)