GrabBag/Module/FFMediaStream/Inc/IVrFFMediaPusher.h

#pragma once

#include <cstdint>
#include <cstddef>
#include <string>
#include <memory>

// FFMediaStream 现为静态库（CONFIG += staticlib），不使用 DLL 导入/导出修饰，
// 否则 MSVC 会以 __imp_ 方式查找符号导致 LNK2019。
#define VR_FFMEDIA_API

/**
 * @brief 推流帧像素格式
 */
enum class EFFPushPixelFormat
{
    NV12 = 0,   // 4:2:0 半平面 (Y + UV)
    YUV420P,    // 4:2:0 平面
    BGR24,      // OpenCV 默认
    RGB24,
    BGRA32,
    RGBA32,
};

/**
 * @brief 编码类型
 */
enum class EFFEncodeType
{
    H264 = 0,
    H265,
    MJPEG,
};

/**
 * @brief 码率控制模式
 */
enum class EFFRcMode
{
    CBR = 0,
    VBR,
};

/**
 * @brief 推流参数
 */
struct VrFFPushConfig
{
    // 帧参数
    unsigned int    width{1920};
    unsigned int    height{1080};
    EFFPushPixelFormat pixelFormat{EFFPushPixelFormat::NV12};

    // 编码参数
    EFFEncodeType   encodeType{EFFEncodeType::H264};
    unsigned int    fps{30};
    unsigned int    gop{60};
    unsigned int    bitrateKbps{4096};   // 码率 (kbps)
    EFFRcMode       rcMode{EFFRcMode::CBR};

    // RTSP 服务参数
    std::string     rtspHost{"127.0.0.1"};      // mediamtx 服务端 IP（主动推流目标）
    std::string     rtspPath{"/live/stream"};   // URL 路径
    unsigned short  rtspPort{8554};
    std::string     authRealm;
    std::string     authUser;
    std::string     authPass;
};

/**
 * @brief IVrFFMediaPusher
 *
 * 基于 MPP 硬件编码 + FFmpeg libavformat 的 RTSP 推流（内存输入 -> swscale 格式转换 -> MPP 编码 -> RTSP）。
 *
 * 典型用法：
 *   IVrFFMediaPusher* pusher = nullptr;
 *   IVrFFMediaPusher::CreateObject(&pusher);
 *   pusher->Init(cfg);
 *   pusher->Start();
 *   pusher->PushFrame(data, size);
 *   ...
 *   pusher->Stop();
 *   delete pusher;
 */
class VR_FFMEDIA_API IVrFFMediaPusher
{
public:
    virtual ~IVrFFMediaPusher() = default;

    // 工厂
    static bool CreateObject(IVrFFMediaPusher** ppPusher);

    /**
     * @brief 初始化推流链路
     * @return 0 成功
     */
    virtual int Init(const VrFFPushConfig& config) = 0;

    /**
     * @brief 启动推流（建立 RTSP 服务，等待客户端拉流）
     */
    virtual int Start() = 0;

    /**
     * @brief 喂入一帧待编码的图像
     * @param data 帧数据指针
     * @param size 帧数据大小（字节）
     * @param ptsUs PTS，单位微秒；传 0 由内部生成
     * @return 0 成功
     */
    virtual int PushFrame(const void* data, size_t size, int64_t ptsUs = 0) = 0;

    /**
     * @brief 停止推流
     */
    virtual int Stop() = 0;

    /**
     * @brief 反初始化
     */
    virtual int UnInit() = 0;

    /**
     * @brief 获取当前推流 URL（含 IP/端口/路径）
     */
    virtual std::string GetRtspUrl(const std::string& localIp) const = 0;
};