EncodingQueue.h

#pragma once

#include <cstddef>
#include <condition_variable>
#include <deque>
#include <functional>
#include <mutex>
#include <thread>
#include <vector>

#include "LogBuffer.h"
#include "ScreenCapture.h"

namespace vr_image_client
{

class EncodingQueue
{
public:
    EncodingQueue(std::size_t max_size, std::size_t worker_count);
    ~EncodingQueue();

    void Start();
    void Stop();

    void Enqueue(RawFrame frame, LogBuffer &logs);
    void SetEncoder(std::function<void(RawFrame &&)> encoder);

private:
    void WorkerLoop();

    std::mutex mutex_;
    std::condition_variable cond_;
    std::deque<RawFrame> queue_;
    std::function<void(RawFrame &&)> encoder_;
    std::vector<std::thread> workers_;
    bool stop_requested_{false};
    std::size_t max_size_{0};
    std::size_t worker_count_{1};
};

} // namespace vr_image_client