Queue.h

Go to the documentation of this file.

 
#ifndef __ROGUE_QUEUE_H__
#define __ROGUE_QUEUE_H__
#include "rogue/Directives.h"
 
#include <stdint.h>
 
#include <atomic>
#include <condition_variable>
#include <mutex>
#include <queue>
 
namespace rogue {
template <typename T>
class Queue {
  private:
    std::queue<T> queue_;
    mutable std::mutex mtx_;
    std::condition_variable pushCond_;
    std::condition_variable popCond_;
    uint32_t max_   = 0;
    uint32_t thold_ = 0;
    std::atomic<bool> busy_{false};
    bool run_ = true;
 
  public:
    Queue() = default;
 
    void stop() {
        std::unique_lock<std::mutex> lock(mtx_);
        run_ = false;
        pushCond_.notify_all();
        popCond_.notify_all();
    }
 
    void setMax(uint32_t max) {
        max_ = max;
    }
 
    void setThold(uint32_t thold) {
        thold_ = thold;
    }
 
    void push(T const& data) {
        std::unique_lock<std::mutex> lock(mtx_);
 
        while (run_ && max_ > 0 && queue_.size() >= max_) pushCond_.wait(lock);
 
        if (run_) queue_.push(data);
        busy_ = (thold_ > 0 && queue_.size() >= thold_);
        popCond_.notify_all();
    }
 
    bool empty() {
        return queue_.empty();
    }
 
    uint32_t size() {
        std::unique_lock<std::mutex> lock(mtx_);
        return queue_.size();
    }
 
    bool busy() {
        return busy_;
    }
 
    void reset() {
        std::unique_lock<std::mutex> lock(mtx_);
        while (!queue_.empty()) queue_.pop();
        busy_ = false;
        pushCond_.notify_all();
    }
 
    T pop() {
        T ret;
        std::unique_lock<std::mutex> lock(mtx_);
        while (run_ && queue_.empty()) popCond_.wait(lock);
        if (run_) {
            ret = queue_.front();
            queue_.pop();
        }
        busy_ = (thold_ > 0 && queue_.size() >= thold_);
        pushCond_.notify_all();
        return (ret);
    }
};
}  // namespace rogue
 
#endif