134 lines
2.8 KiB
C
134 lines
2.8 KiB
C
|
#pragma once
|
|||
|
|
|
|||
|
|
#include <cstdint>
|
|||
|
|
#include <cstddef>
|
|||
|
|
#include <string>
|
|||
|
|
#include <memory>
|
|||
|
|
|
|||
|
|
// DLL 导出宏（FFMEDIASTREAM_LIBRARY 在 .pro 中已定义）
|
|||
|
|
#ifdef _WIN32
|
|||
|
|
# ifdef FFMEDIASTREAM_LIBRARY
|
|||
|
|
# define VR_FFMEDIA_API __declspec(dllexport)
|
|||
|
|
# else
|
|||
|
|
# define VR_FFMEDIA_API __declspec(dllimport)
|
|||
|
|
# endif
|
|||
|
|
#else
|
|||
|
|
# define VR_FFMEDIA_API
|
|||
|
|
#endif
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @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 rtspPath{"/live/stream"}; // URL 路径
|
|||
|
|
unsigned short rtspPort{8554};
|
|||
|
|
std::string authRealm;
|
|||
|
|
std::string authUser;
|
|||
|
|
std::string authPass;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @brief IVrFFMediaPusher
|
|||
|
|
*
|
|||
|
|
* 基于 ffmedia_release 的 RTSP 推流（内存输入 -> RGA 格式转换 -> 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;
|
|||
|
|
};
|