command_queue.hpp

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