CommandQueue.hpp

#pragma once
 
#include <condition_variable>
#include <functional>
#include <mutex>
#include <queue>
#include <chrono>
 
namespace PixelBullet
{
    class CommandQueue
    {
    public:
        void Push(const std::function<void()>& command)
        {
            {
                std::lock_guard<std::mutex> lock(m_Mutex);
                m_Queue.push(command);
            }
            m_Cond.notify_one();
        }
 
        bool Pop(std::function<void()>& command, std::chrono::milliseconds waitDuration)
        {
            std::unique_lock<std::mutex> lock(m_Mutex);
            if (m_Cond.wait_for(lock, waitDuration, [this]() { return !m_Queue.empty(); }))
            {
                command = m_Queue.front();
                m_Queue.pop();
                return true;
            }
            return false;
        }
 
        void NotifyAll()
        {
            m_Cond.notify_all();
        }
 
    private:
        std::mutex m_Mutex;
        std::condition_variable m_Cond;
        std::queue<std::function<void()>> m_Queue;
    };
} // namespace PixelBullet