WebcamCapture.h

#pragma once

#include <atomic>
#include <cstdint>
#include <functional>
#include <string>
#include <thread>
#include <vector>

extern "C" 
{
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
#include <libswscale/swscale.h>
}

class WebcamCapture 
{
public:

    using FrameCallback = std::function<void(std::vector<uint8_t> jpegFrame)>;

    WebcamCapture();
    ~WebcamCapture();

    WebcamCapture(const WebcamCapture&) = delete;
    WebcamCapture& operator=(const WebcamCapture&) = delete;

    bool init(const std::string& device, int width, int height, int fps, int quality, bool reencode);
    void start();
    void stop();
    void setFrameCallback(FrameCallback cb) { frameCallback_ = std::move(cb); }
    int width() const { return width_; }
    int height() const { return height_; }

private:
    void readLoop();
    void cleanup();
    bool encodeFrame(AVFrame* decoded);

    AVFormatContext* fmtCtx_ = nullptr;
    AVCodecContext*  decCtx_ = nullptr; 
    AVCodecContext*  encCtx_ = nullptr;
    SwsContext*      swsCtx_ = nullptr;
    int videoStreamIndex_ = -1;
    bool reencode_ = false;

    std::thread readThread_;
    std::atomic<bool> running_{false};

    FrameCallback frameCallback_;
    int width_ = 0;
    int height_ = 0;
};