src/AtomicQueue.cpp - master - shmblg4/spectrum-visualizer

src/AtomicQueue.cpp
86 строк3 KB
Matvey Gotfrid
initial commit
04 фев 2025, 07:58
04 фев 2025, 07:58f5566ef
#include "AtomicQueue.hpp"

template< typename T >
AtomicQueue<T>::AtomicQueue(unsigned int length){
    m_length = length;
    buffer = new T[m_length];
    reset(); //возврат в начальное состояние
}

template<typename T>
AtomicQueue<T>::~AtomicQueue(){
    reset();
    delete[] buffer;
}

template< typename T >
void AtomicQueue<T>::reset() {
    read_ind.store(0);  // Выполняем сброс очереди
    write_ind.store(0); // Выполняем сброс очереди
    size.store(0);      // Выполняем сброс очереди
}

template< typename T >
bool AtomicQueue<T>::Push(const T val)             // Запись в конец очереди
{
    unsigned int ind = write_ind.load();            //читаем текущий индекс последнего элемента (он указывает на следующий пустой элемент)
    unsigned int next_ind = (ind + 1) % m_length;   //увеличиваем на единицу по модулю

    // если новый индекс не совпадает с индексом первого элемента очереди
    if(next_ind != read_ind.load())
    {
        buffer[ind] = val;          //пишем значение переданное по ссылке в конец
        size++;                     //увеличиваем размер очереди
        write_ind.store(next_ind);  //и только после записи модифицируем индекс
        return true;
    }
    else    //иначе буфер заполнен, свободного места нет
    {
        return false;
    }
}

template< typename T >
bool AtomicQueue<T>::Pull(T &val)       //чтение из начала очереди
{
    unsigned int ind = read_ind.load(); //читаем текущий индекс первого элемента очереди

    if(ind != write_ind.load())         //если индекс начала не совпадает с индексом последнего элемента
    {
        val = buffer[ind];                          // читаем элемент из начала и передаем его по ссылке
        size--;                                     //уменьшаем размер очереди
        unsigned int next_ind = (ind + 1) % m_length; //смещаем указатель на начало на один элемент
        read_ind.store(next_ind);                   //модифицируем индекс перврго элемента
        return true;
    }
    else     //иначе буфер пуст
    {
        return false;
    }
}

template< typename T >
unsigned int AtomicQueue<T>::Size()
{
    return size.load();
}

template< typename T >
unsigned int AtomicQueue<T>::getMaxLength(){
    return m_length;
}

/**
 * @brief Добавляет cnt нулей 
 * 
 * @param cnt Количество нулей в AtomicQueue 
 */
template< typename T >
void AtomicQueue<T>::push_zeros(size_t cnt) {
    T zero_value{}; // Создаём значение, инициализированное по умолчанию (например, 0 для чисел)
    for (size_t i = 0; i < cnt; ++i) {
        if (!Push(zero_value)) {
            break;
        }
    }
}