#pragma once
#include <array>
#include <atomic>
#include <cassert>
#include <cstdint>
#include <optional>
namespace ableton
{
namespace util
{
template <typename T>
struct TripleBuffer
{
public:
TripleBuffer()
: mBuffers{{{}, {}, {}}}
{
assert(mState.is_lock_free());
}
explicit TripleBuffer(const T& initial)
: mBuffers{{initial, initial, initial}}
{
assert(mState.is_lock_free());
}
TripleBuffer(const TripleBuffer&) = delete;
TripleBuffer& operator=(const TripleBuffer&) = delete;
T read() noexcept
{
loadReadBuffer();
return mBuffers[mReadIndex];
}
std::optional<T> readNew()
{
if (loadReadBuffer())
{
return std::optional<T>(mBuffers[mReadIndex]);
}
return std::nullopt;
}
template <typename U>
void write(U&& value)
{
mBuffers[mWriteIndex] = std::forward<U>(value);
const auto prevState =
mState.exchange(makeState(mWriteIndex, true), std::memory_order_acq_rel);
mWriteIndex = backIndex(prevState);
}
private:
bool loadReadBuffer()
{
auto state = mState.load(std::memory_order_acquire);
auto isNew = isNewWrite(state);
if (isNew)
{
const auto prevState =
mState.exchange(makeState(mReadIndex, false), std::memory_order_acq_rel);
mReadIndex = backIndex(prevState);
}
return isNew;
}
using BackingState = uint32_t;
static constexpr bool isNewWrite(const BackingState state)
{
return (state & 0x0000FFFFu) != 0;
}
static constexpr uint32_t backIndex(const BackingState state) { return state >> 16; }
static constexpr BackingState makeState(const uint32_t backBufferIndex,
const bool isWrite)
{
return (backBufferIndex << 16) | uint32_t(isWrite);
}
std::atomic<BackingState> mState{makeState(1u, false)}; uint32_t mReadIndex = 0u; uint32_t mWriteIndex = 2u;
std::array<T, 3> mBuffers{};
};
} }