From 662c7b1e7139486373ecf74479659ff555c59cd2 Mon Sep 17 00:00:00 2001 From: gingerBill Date: Sat, 7 Aug 2021 14:39:26 +0100 Subject: Minor clean-up --- src/queue.cpp | 24 +++++++++++++----------- 1 file changed, 13 insertions(+), 11 deletions(-) (limited to 'src/queue.cpp') diff --git a/src/queue.cpp b/src/queue.cpp index a9bbbeee7..d69a2845c 100644 --- a/src/queue.cpp +++ b/src/queue.cpp @@ -1,27 +1,29 @@ #define MPMC_CACHE_LINE_SIZE 64 +typedef std::atomic MPMCQueueAtomicIdx; + // Multiple Producer Multiple Consumer Queue template struct MPMCQueue { - static size_t const PAD0_OFFSET = (sizeof(T *) + sizeof(std::atomic *) + sizeof(gbAllocator) + sizeof(BlockingMutex) + sizeof(i32) + sizeof(i32)); + static size_t const PAD0_OFFSET = (sizeof(T *) + sizeof(MPMCQueueAtomicIdx *) + sizeof(gbAllocator) + sizeof(BlockingMutex) + sizeof(i32) + sizeof(i32)); - T * nodes; - std::atomic *indices; - gbAllocator allocator; - BlockingMutex mutex; - std::atomic count; - i32 mask; + T * nodes; + MPMCQueueAtomicIdx *indices; + gbAllocator allocator; + BlockingMutex mutex; + MPMCQueueAtomicIdx count; + i32 mask; // capacity-1, because capacity must be a power of 2 char pad0[(MPMC_CACHE_LINE_SIZE*2 - PAD0_OFFSET) % MPMC_CACHE_LINE_SIZE]; - std::atomic head_idx; + MPMCQueueAtomicIdx head_idx; char pad1[MPMC_CACHE_LINE_SIZE - sizeof(i32)]; - std::atomic tail_idx; + MPMCQueueAtomicIdx tail_idx; }; -void mpmc_internal_init_indices(std::atomic *indices, i32 offset, i32 size) { +void mpmc_internal_init_indices(MPMCQueueAtomicIdx *indices, i32 offset, i32 size) { GB_ASSERT(offset % 8 == 0); GB_ASSERT(size % 8 == 0); @@ -54,7 +56,7 @@ void mpmc_init(MPMCQueue *q, gbAllocator a, isize size_i) { q->mask = size-1; q->allocator = a; q->nodes = gb_alloc_array(a, T, size); - q->indices = cast(std::atomic *)gb_alloc_array(a, i32, size); + q->indices = gb_alloc_array(a, MPMCQueueAtomicIdx, size); mpmc_internal_init_indices(q->indices, 0, q->mask+1); } -- cgit v1.2.3