wfree_message_queue.hpp

// Copyright 2019 Pokitec
// All rights reserved.
 
#pragma once
 
#include "TTauri/Foundation/required.hpp"
#include "TTauri/Foundation/atomic.hpp"
#include <array>
#include <atomic>
#include <memory>
#include <thread>
#include <optional>
#include <type_traits>
#include <functional>
 
namespace tt {
 
template<typename T, size_t Capacity>
class wfree_message_queue;
 
template<typename T, size_t Capacity, bool WriteOperation>
class wfree_message_queue_operation {
    wfree_message_queue<T,Capacity> *parent;
    size_t index;
 
public:
    wfree_message_queue_operation() noexcept : parent(nullptr), index(0) {}
    wfree_message_queue_operation(wfree_message_queue<T,Capacity> *parent, size_t index) noexcept : parent(parent), index(index) {}
 
    wfree_message_queue_operation(wfree_message_queue_operation const &other) = delete;
 
    wfree_message_queue_operation(wfree_message_queue_operation && other) :
        parent(other.parent), index(other.index)
    {
        tt_assume(this != &other);
        other.parent = nullptr;
    }
 
    ~wfree_message_queue_operation()
    {
        if (parent == nullptr) {
            return;
        }
 
        if constexpr (WriteOperation) {
            parent->write_finish(index);
        } else {
            parent->read_finish(index);
        }
    }
 
    wfree_message_queue_operation& operator=(wfree_message_queue_operation const &other) = delete;
 
    wfree_message_queue_operation& operator=(wfree_message_queue_operation && other)
    {
        tt_assume(this != &other);
        std::swap(index, other.index);
        std::swap(parent, other.parent);
    }
 
    T &operator*() noexcept
    {
        return (*parent)[index];
    }
 
    T *operator->() noexcept
    {
        return &(*parent)[index];
    }
};
 
template<typename T, size_t Capacity>
class wfree_message_queue {
    using index_type = size_t;
    using value_type = T;
    using scoped_write_operation = wfree_message_queue_operation<T,Capacity,true>;
    using scoped_read_operation = wfree_message_queue_operation<T,Capacity,false>;
 
    struct message_type {
        // The in_use atomic is first, to improve cache-line and prefetch.
        // There should not be much false sharing since the thread that uses the message is
        // also the one that updates the in_use atomic.
        std::atomic<bool> in_use = false;
        value_type value;
    };
    
    static constexpr index_type capacity = Capacity;
 
    static constexpr index_type slack = 16;
    static_assert(capacity > (slack * 2), "The capacity of the message queue should be much larger than its slack.");
 
    std::array<message_type,capacity> messages;
    alignas(cache_line_size) std::atomic<index_type> head = 0;
    alignas(cache_line_size) std::atomic<index_type> tail = 0;
 
public:
    wfree_message_queue() = default;
    wfree_message_queue(wfree_message_queue const &) = delete;
    wfree_message_queue(wfree_message_queue &&) = delete;
    wfree_message_queue &operator=(wfree_message_queue const &) = delete;
    wfree_message_queue &operator=(wfree_message_queue &&) = delete;
    ~wfree_message_queue() = default;
 
    index_type size() const noexcept {
        // head and tail are extremelly large integers, they will never wrap arround.
        return head.load(std::memory_order_relaxed) - tail.load(std::memory_order_relaxed);
    }
 
    bool empty() const noexcept {
        return head.load(std::memory_order_relaxed) <= tail.load(std::memory_order_relaxed);
    }
 
    bool full() const noexcept {
        return head.load(std::memory_order_relaxed) >= (tail.load(std::memory_order_relaxed) + (capacity - slack));
    }
 
    template<typename BlockCounterTag=void>
    scoped_write_operation write() noexcept {
        return {this, write_start<BlockCounterTag>()};
    }
 
    scoped_read_operation read() noexcept {
        return {this, read_start()};
    }
 
    value_type const &operator[](index_type index) const noexcept {
        return messages[index % capacity].value;
    }
 
    value_type &operator[](index_type index) noexcept {
        return messages[index % capacity].value;
    }
 
    template<typename CounterTag=void>
    index_type write_start() noexcept {
        ttlet index = head.fetch_add(1, std::memory_order_acquire);
        auto &message = messages[index % capacity];
 
        // We acquired the index before we knew if the queue was full.
        // So we have to wait until the message is empty, however when it is empty we are
        // the only one that holds the message, so we only need to mark it that we are done with
        // writing the message.
        wait_for_transition<CounterTag>(message.in_use, false, std::memory_order_acquire);
        return index;
    }
 
    void write_finish(index_type index) noexcept {
        auto &message = messages[index % capacity];
 
        // Mark that the message is finished with writing.
        message.in_use.store(true, std::memory_order_release);
    }
 
    template<typename CounterTag=void>
    index_type read_start() noexcept {
        ttlet index = tail.fetch_add(1, std::memory_order_acquire);
        auto &message = messages[index % capacity];
 
        // We acquired the index before we knew if the message was ready.
        wait_for_transition<CounterTag>(message.in_use, true, std::memory_order_acquire);
        return index;
    }
 
    void read_finish(index_type index) noexcept {
        auto &message = messages[index % capacity];
 
        // We acquired the index before we knew if the message was ready.
        message.in_use.store(false, std::memory_order_release);
 
        // The message itself does not need to be destructed.
        // This will happen automatically when wrapping around the ring buffer overwrites the message.
    }
};
 
}