circularBuffer.h

// Copyright 2009-2022 NTESS. Under the terms
// of Contract DE-NA0003525 with NTESS, the U.S.
// Government retains certain rights in this software.
//
// Copyright (c) 2009-2022, NTESS
// All rights reserved.
//
// This file is part of the SST software package. For license
// information, see the LICENSE file in the top level directory of the
// distribution.

#ifndef SST_CORE_INTERPROCESS_CIRCULARBUFFER_H
#define SST_CORE_INTERPROCESS_CIRCULARBUFFER_H

#include "sstmutex.h"

namespace SST {
namespace Core {
namespace Interprocess {

template <typename T>
class CircularBuffer
{

public:
    CircularBuffer(size_t mSize = 0)
    {
        buffSize   = mSize;
        readIndex  = 0;
        writeIndex = 0;
    }

    bool setBufferSize(const size_t bufferSize)
    {
        if ( buffSize != 0 ) {
            fprintf(stderr, "Already specified size for buffer\n");
            return false;
        }

        buffSize = bufferSize;
        __sync_synchronize();
        return true;
    }

    T read()
    {
        int loop_counter = 0;

        while ( true ) {
            bufferMutex.lock();

            if ( readIndex != writeIndex ) {
                const T result = buffer[readIndex];
                readIndex      = (readIndex + 1) % buffSize;

                bufferMutex.unlock();
                return result;
            }

            bufferMutex.unlock();
            bufferMutex.processorPause(loop_counter++);
        }
    }

    bool readNB(T* result)
    {
        if ( bufferMutex.try_lock() ) {
            if ( readIndex != writeIndex ) {
                *result   = buffer[readIndex];
                readIndex = (readIndex + 1) % buffSize;

                bufferMutex.unlock();
                return true;
            }

            bufferMutex.unlock();
        }

        return false;
    }

    void write(const T& v)
    {
        int loop_counter = 0;

        while ( true ) {
            bufferMutex.lock();

            if ( ((writeIndex + 1) % buffSize) != readIndex ) {
                buffer[writeIndex] = v;
                writeIndex         = (writeIndex + 1) % buffSize;

                __sync_synchronize();
                bufferMutex.unlock();
                return;
            }

            bufferMutex.unlock();
            bufferMutex.processorPause(loop_counter++);
        }
    }

    ~CircularBuffer() {}

    void clearBuffer()
    {
        bufferMutex.lock();
        readIndex = writeIndex;
        __sync_synchronize();
        bufferMutex.unlock();
    }

private:
    SSTMutex bufferMutex;
    size_t   buffSize;
    size_t   readIndex;
    size_t   writeIndex;
    T        buffer[0];
};

} // namespace Interprocess
} // namespace Core
} // namespace SST

#endif // SST_CORE_INTERPROCESS_CIRCULARBUFFER_H