mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-30 07:26:53 +08:00
582 lines
14 KiB
C++
582 lines
14 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT license that can be found in the
|
||
* LICENSE file in the root of the source tree. All contributing project authors
|
||
* may be found in the AUTHORS file in the root of the source tree.
|
||
*/
|
||
|
||
#ifndef ZLMEDIAKIT_FRAME_H
|
||
#define ZLMEDIAKIT_FRAME_H
|
||
|
||
#include <mutex>
|
||
#include <functional>
|
||
#include "Util/RingBuffer.h"
|
||
#include "Network/Socket.h"
|
||
|
||
namespace mediakit{
|
||
|
||
typedef enum {
|
||
TrackInvalid = -1,
|
||
TrackVideo = 0,
|
||
TrackAudio,
|
||
TrackTitle,
|
||
TrackApplication,
|
||
TrackMax
|
||
} TrackType;
|
||
|
||
#define CODEC_MAP(XX) \
|
||
XX(CodecH264, TrackVideo, 0, "H264", PSI_STREAM_H264) \
|
||
XX(CodecH265, TrackVideo, 1, "H265", PSI_STREAM_H265) \
|
||
XX(CodecAAC, TrackAudio, 2, "mpeg4-generic", PSI_STREAM_AAC) \
|
||
XX(CodecG711A, TrackAudio, 3, "PCMA", PSI_STREAM_AUDIO_G711A) \
|
||
XX(CodecG711U, TrackAudio, 4, "PCMU", PSI_STREAM_AUDIO_G711U) \
|
||
XX(CodecOpus, TrackAudio, 5, "opus", PSI_STREAM_AUDIO_OPUS) \
|
||
XX(CodecL16, TrackAudio, 6, "L16", PSI_STREAM_RESERVED) \
|
||
XX(CodecVP8, TrackVideo, 7, "VP8", PSI_STREAM_VP8) \
|
||
XX(CodecVP9, TrackVideo, 8, "VP9", PSI_STREAM_VP9) \
|
||
XX(CodecAV1, TrackVideo, 9, "AV1X", PSI_STREAM_AV1)
|
||
|
||
typedef enum {
|
||
CodecInvalid = -1,
|
||
#define XX(name, type, value, str, mpeg_id) name = value,
|
||
CODEC_MAP(XX)
|
||
#undef XX
|
||
CodecMax
|
||
} CodecId;
|
||
|
||
/**
|
||
* 字符串转媒体类型转
|
||
*/
|
||
TrackType getTrackType(const std::string &str);
|
||
|
||
/**
|
||
* 媒体类型转字符串
|
||
*/
|
||
const char* getTrackString(TrackType type);
|
||
|
||
/**
|
||
* 根据SDP中描述获取codec_id
|
||
* @param str
|
||
* @return
|
||
*/
|
||
CodecId getCodecId(const std::string &str);
|
||
|
||
/**
|
||
* 获取编码器名称
|
||
*/
|
||
const char *getCodecName(CodecId codecId);
|
||
|
||
/**
|
||
* 获取音视频类型
|
||
*/
|
||
TrackType getTrackType(CodecId codecId);
|
||
|
||
/**
|
||
* 编码信息的抽象接口
|
||
*/
|
||
class CodecInfo {
|
||
public:
|
||
typedef std::shared_ptr<CodecInfo> Ptr;
|
||
|
||
CodecInfo(){}
|
||
virtual ~CodecInfo(){}
|
||
|
||
/**
|
||
* 获取编解码器类型
|
||
*/
|
||
virtual CodecId getCodecId() const = 0;
|
||
|
||
/**
|
||
* 获取编码器名称
|
||
*/
|
||
const char *getCodecName() const;
|
||
|
||
/**
|
||
* 获取音视频类型
|
||
*/
|
||
TrackType getTrackType() const;
|
||
};
|
||
|
||
/**
|
||
* 帧类型的抽象接口
|
||
*/
|
||
class Frame : public toolkit::Buffer, public CodecInfo {
|
||
public:
|
||
typedef std::shared_ptr<Frame> Ptr;
|
||
virtual ~Frame(){}
|
||
|
||
/**
|
||
* 返回解码时间戳，单位毫秒
|
||
*/
|
||
virtual uint32_t dts() const = 0;
|
||
|
||
/**
|
||
* 返回显示时间戳，单位毫秒
|
||
*/
|
||
virtual uint32_t pts() const {
|
||
return dts();
|
||
}
|
||
|
||
/**
|
||
* 前缀长度，譬如264前缀为0x00 00 00 01,那么前缀长度就是4
|
||
* aac前缀则为7个字节
|
||
*/
|
||
virtual size_t prefixSize() const = 0;
|
||
|
||
/**
|
||
* 返回是否为关键帧
|
||
*/
|
||
virtual bool keyFrame() const = 0;
|
||
|
||
/**
|
||
* 是否为配置帧，譬如sps pps vps
|
||
*/
|
||
virtual bool configFrame() const = 0;
|
||
|
||
/**
|
||
* 是否可以缓存
|
||
*/
|
||
virtual bool cacheAble() const { return true; }
|
||
|
||
/**
|
||
* 该帧是否可以丢弃
|
||
* SEI/AUD帧可以丢弃
|
||
* 默认都不能丢帧
|
||
*/
|
||
virtual bool dropAble() const { return false; }
|
||
|
||
/**
|
||
* 是否为可解码帧
|
||
* sps pps等帧不能解码
|
||
*/
|
||
virtual bool decodeAble() const {
|
||
if (getTrackType() != TrackVideo) {
|
||
//非视频帧都可以解码
|
||
return true;
|
||
}
|
||
//默认非sps pps帧都可以解码
|
||
return !configFrame();
|
||
}
|
||
|
||
/**
|
||
* 返回可缓存的frame
|
||
*/
|
||
static Ptr getCacheAbleFrame(const Ptr &frame);
|
||
|
||
private:
|
||
//对象个数统计
|
||
toolkit::ObjectStatistic<Frame> _statistic;
|
||
};
|
||
|
||
class FrameImp : public Frame {
|
||
public:
|
||
using Ptr = std::shared_ptr<FrameImp>;
|
||
|
||
template<typename C=FrameImp>
|
||
static std::shared_ptr<C> create() {
|
||
#if 0
|
||
static ResourcePool<C> packet_pool;
|
||
static onceToken token([]() {
|
||
packet_pool.setSize(1024);
|
||
});
|
||
auto ret = packet_pool.obtain2();
|
||
ret->_buffer.clear();
|
||
ret->_prefix_size = 0;
|
||
ret->_dts = 0;
|
||
ret->_pts = 0;
|
||
return ret;
|
||
#else
|
||
return std::shared_ptr<C>(new C());
|
||
#endif
|
||
}
|
||
|
||
char *data() const override{
|
||
return (char *)_buffer.data();
|
||
}
|
||
|
||
size_t size() const override {
|
||
return _buffer.size();
|
||
}
|
||
|
||
uint32_t dts() const override {
|
||
return _dts;
|
||
}
|
||
|
||
uint32_t pts() const override{
|
||
return _pts ? _pts : _dts;
|
||
}
|
||
|
||
size_t prefixSize() const override{
|
||
return _prefix_size;
|
||
}
|
||
|
||
CodecId getCodecId() const override{
|
||
return _codec_id;
|
||
}
|
||
|
||
bool keyFrame() const override {
|
||
return false;
|
||
}
|
||
|
||
bool configFrame() const override{
|
||
return false;
|
||
}
|
||
|
||
public:
|
||
CodecId _codec_id = CodecInvalid;
|
||
uint32_t _dts = 0;
|
||
uint32_t _pts = 0;
|
||
size_t _prefix_size = 0;
|
||
toolkit::BufferLikeString _buffer;
|
||
|
||
private:
|
||
//对象个数统计
|
||
toolkit::ObjectStatistic<FrameImp> _statistic;
|
||
|
||
protected:
|
||
friend class toolkit::ResourcePool_l<FrameImp>;
|
||
FrameImp() = default;
|
||
};
|
||
|
||
/**
|
||
* 一个Frame类中可以有多个帧，他们通过 0x 00 00 01 分隔
|
||
* ZLMediaKit会先把这种复合帧split成单个帧然后再处理
|
||
* 一个复合帧可以通过无内存拷贝的方式切割成多个子Frame
|
||
* 提供该类的目的是切割复合帧时防止内存拷贝，提高性能
|
||
*/
|
||
template<typename Parent>
|
||
class FrameInternal : public Parent{
|
||
public:
|
||
typedef std::shared_ptr<FrameInternal> Ptr;
|
||
FrameInternal(const Frame::Ptr &parent_frame, char *ptr, size_t size, size_t prefix_size)
|
||
: Parent(ptr, size, parent_frame->dts(), parent_frame->pts(), prefix_size) {
|
||
_parent_frame = parent_frame;
|
||
}
|
||
bool cacheAble() const override {
|
||
return _parent_frame->cacheAble();
|
||
}
|
||
private:
|
||
Frame::Ptr _parent_frame;
|
||
};
|
||
|
||
/**
|
||
* 写帧接口的抽象接口类
|
||
*/
|
||
class FrameWriterInterface {
|
||
public:
|
||
typedef std::shared_ptr<FrameWriterInterface> Ptr;
|
||
FrameWriterInterface(){}
|
||
virtual ~FrameWriterInterface(){}
|
||
|
||
/**
|
||
* 写入帧数据
|
||
*/
|
||
virtual bool inputFrame(const Frame::Ptr &frame) = 0;
|
||
};
|
||
|
||
/**
|
||
* 写帧接口转function，辅助类
|
||
*/
|
||
class FrameWriterInterfaceHelper : public FrameWriterInterface {
|
||
public:
|
||
typedef std::shared_ptr<FrameWriterInterfaceHelper> Ptr;
|
||
typedef std::function<bool(const Frame::Ptr &frame)> onWriteFrame;
|
||
|
||
/**
|
||
* inputFrame后触发onWriteFrame回调
|
||
*/
|
||
FrameWriterInterfaceHelper(const onWriteFrame& cb){
|
||
_writeCallback = cb;
|
||
}
|
||
|
||
virtual ~FrameWriterInterfaceHelper(){}
|
||
|
||
/**
|
||
* 写入帧数据
|
||
*/
|
||
bool inputFrame(const Frame::Ptr &frame) override {
|
||
return _writeCallback(frame);
|
||
}
|
||
|
||
private:
|
||
onWriteFrame _writeCallback;
|
||
};
|
||
|
||
/**
|
||
* 支持代理转发的帧环形缓存
|
||
*/
|
||
class FrameDispatcher : public FrameWriterInterface {
|
||
public:
|
||
typedef std::shared_ptr<FrameDispatcher> Ptr;
|
||
|
||
FrameDispatcher(){}
|
||
virtual ~FrameDispatcher(){}
|
||
|
||
/**
|
||
* 添加代理
|
||
*/
|
||
void addDelegate(const FrameWriterInterface::Ptr &delegate){
|
||
//_delegates_write可能多线程同时操作
|
||
std::lock_guard<std::mutex> lck(_mtx);
|
||
_delegates_write.emplace(delegate.get(),delegate);
|
||
_need_update = true;
|
||
}
|
||
|
||
/**
|
||
* 删除代理
|
||
*/
|
||
void delDelegate(FrameWriterInterface *ptr){
|
||
//_delegates_write可能多线程同时操作
|
||
std::lock_guard<std::mutex> lck(_mtx);
|
||
_delegates_write.erase(ptr);
|
||
_need_update = true;
|
||
}
|
||
|
||
/**
|
||
* 写入帧并派发
|
||
*/
|
||
bool inputFrame(const Frame::Ptr &frame) override{
|
||
if(_need_update){
|
||
//发现代理列表发生变化了，这里同步一次
|
||
std::lock_guard<std::mutex> lck(_mtx);
|
||
_delegates_read = _delegates_write;
|
||
_need_update = false;
|
||
}
|
||
|
||
//_delegates_read能确保是单线程操作的
|
||
bool ret = false;
|
||
for (auto &pr : _delegates_read) {
|
||
if (pr.second->inputFrame(frame)) {
|
||
ret = true;
|
||
}
|
||
}
|
||
return ret;
|
||
}
|
||
|
||
/**
|
||
* 返回代理个数
|
||
*/
|
||
size_t size() const {
|
||
return _delegates_write.size();
|
||
}
|
||
private:
|
||
std::mutex _mtx;
|
||
std::map<void *,FrameWriterInterface::Ptr> _delegates_read;
|
||
std::map<void *,FrameWriterInterface::Ptr> _delegates_write;
|
||
bool _need_update = false;
|
||
};
|
||
|
||
/**
|
||
* 通过Frame接口包装指针，方便使用者把自己的数据快速接入ZLMediaKit
|
||
*/
|
||
class FrameFromPtr : public Frame{
|
||
public:
|
||
typedef std::shared_ptr<FrameFromPtr> Ptr;
|
||
|
||
FrameFromPtr(CodecId codec_id, char *ptr, size_t size, uint32_t dts, uint32_t pts = 0, size_t prefix_size = 0)
|
||
: FrameFromPtr(ptr, size, dts, pts, prefix_size) {
|
||
_codec_id = codec_id;
|
||
}
|
||
|
||
FrameFromPtr(char *ptr, size_t size, uint32_t dts, uint32_t pts = 0, size_t prefix_size = 0){
|
||
_ptr = ptr;
|
||
_size = size;
|
||
_dts = dts;
|
||
_pts = pts;
|
||
_prefix_size = prefix_size;
|
||
}
|
||
|
||
char *data() const override{
|
||
return _ptr;
|
||
}
|
||
|
||
size_t size() const override {
|
||
return _size;
|
||
}
|
||
|
||
uint32_t dts() const override {
|
||
return _dts;
|
||
}
|
||
|
||
uint32_t pts() const override{
|
||
return _pts ? _pts : dts();
|
||
}
|
||
|
||
size_t prefixSize() const override{
|
||
return _prefix_size;
|
||
}
|
||
|
||
bool cacheAble() const override {
|
||
return false;
|
||
}
|
||
|
||
CodecId getCodecId() const override {
|
||
if (_codec_id == CodecInvalid) {
|
||
throw std::invalid_argument("FrameFromPtr对象未设置codec类型");
|
||
}
|
||
return _codec_id;
|
||
}
|
||
|
||
void setCodecId(CodecId codec_id) {
|
||
_codec_id = codec_id;
|
||
}
|
||
|
||
bool keyFrame() const override {
|
||
return false;
|
||
}
|
||
|
||
bool configFrame() const override{
|
||
return false;
|
||
}
|
||
|
||
protected:
|
||
FrameFromPtr() {}
|
||
|
||
protected:
|
||
char *_ptr;
|
||
uint32_t _dts;
|
||
uint32_t _pts = 0;
|
||
size_t _size;
|
||
size_t _prefix_size;
|
||
CodecId _codec_id = CodecInvalid;
|
||
};
|
||
|
||
/**
|
||
* 该对象的功能是把一个不可缓存的帧转换成可缓存的帧
|
||
*/
|
||
class FrameCacheAble : public FrameFromPtr {
|
||
public:
|
||
typedef std::shared_ptr<FrameCacheAble> Ptr;
|
||
|
||
FrameCacheAble(const Frame::Ptr &frame, bool force_key_frame = false) {
|
||
if (frame->cacheAble()) {
|
||
_frame = frame;
|
||
_ptr = frame->data();
|
||
} else {
|
||
_buffer = FrameImp::create();
|
||
_buffer->_buffer.assign(frame->data(), frame->size());
|
||
_ptr = _buffer->data();
|
||
}
|
||
_size = frame->size();
|
||
_dts = frame->dts();
|
||
_pts = frame->pts();
|
||
_prefix_size = frame->prefixSize();
|
||
_codec_id = frame->getCodecId();
|
||
_key = force_key_frame ? true : frame->keyFrame();
|
||
_config = frame->configFrame();
|
||
_drop_able = frame->dropAble();
|
||
_decode_able = frame->decodeAble();
|
||
}
|
||
|
||
~FrameCacheAble() override = default;
|
||
|
||
/**
|
||
* 可以被缓存
|
||
*/
|
||
bool cacheAble() const override {
|
||
return true;
|
||
}
|
||
|
||
bool keyFrame() const override{
|
||
return _key;
|
||
}
|
||
|
||
bool configFrame() const override{
|
||
return _config;
|
||
}
|
||
|
||
bool dropAble() const override {
|
||
return _drop_able;
|
||
}
|
||
|
||
bool decodeAble() const override {
|
||
return _decode_able;
|
||
}
|
||
|
||
private:
|
||
bool _key;
|
||
bool _config;
|
||
bool _drop_able;
|
||
bool _decode_able;
|
||
Frame::Ptr _frame;
|
||
FrameImp::Ptr _buffer;
|
||
};
|
||
|
||
/**
|
||
* 该对象可以把Buffer对象转换成可缓存的Frame对象
|
||
*/
|
||
template <typename Parent>
|
||
class FrameWrapper : public Parent{
|
||
public:
|
||
~FrameWrapper() = default;
|
||
|
||
/**
|
||
* 构造frame
|
||
* @param buf 数据缓存
|
||
* @param dts 解码时间戳
|
||
* @param pts 显示时间戳
|
||
* @param prefix 帧前缀长度
|
||
* @param offset buffer有效数据偏移量
|
||
*/
|
||
FrameWrapper(const toolkit::Buffer::Ptr &buf, uint32_t dts, uint32_t pts, size_t prefix, size_t offset) : Parent(buf->data() + offset, buf->size() - offset, dts, pts, prefix){
|
||
_buf = buf;
|
||
}
|
||
|
||
/**
|
||
* 构造frame
|
||
* @param buf 数据缓存
|
||
* @param dts 解码时间戳
|
||
* @param pts 显示时间戳
|
||
* @param prefix 帧前缀长度
|
||
* @param offset buffer有效数据偏移量
|
||
* @param codec 帧类型
|
||
*/
|
||
FrameWrapper(const toolkit::Buffer::Ptr &buf, uint32_t dts, uint32_t pts, size_t prefix, size_t offset, CodecId codec) : Parent(codec, buf->data() + offset, buf->size() - offset, dts, pts, prefix){
|
||
_buf = buf;
|
||
}
|
||
|
||
/**
|
||
* 该帧可缓存
|
||
*/
|
||
bool cacheAble() const override {
|
||
return true;
|
||
}
|
||
|
||
private:
|
||
toolkit::Buffer::Ptr _buf;
|
||
};
|
||
|
||
/**
|
||
* 合并一些时间戳相同的frame
|
||
*/
|
||
class FrameMerger {
|
||
public:
|
||
using onOutput = std::function<void(uint32_t dts, uint32_t pts, const toolkit::Buffer::Ptr &buffer, bool have_key_frame)>;
|
||
using Ptr = std::shared_ptr<FrameMerger>;
|
||
enum {
|
||
none = 0,
|
||
h264_prefix,
|
||
mp4_nal_size,
|
||
};
|
||
|
||
FrameMerger(int type);
|
||
~FrameMerger() = default;
|
||
|
||
void clear();
|
||
bool inputFrame(const Frame::Ptr &frame, const onOutput &cb, toolkit::BufferLikeString *buffer = nullptr);
|
||
|
||
private:
|
||
bool willFlush(const Frame::Ptr &frame) const;
|
||
void doMerge(toolkit::BufferLikeString &buffer, const Frame::Ptr &frame) const;
|
||
|
||
private:
|
||
int _type;
|
||
bool _have_decode_able_frame = false;
|
||
toolkit::List<Frame::Ptr> _frame_cache;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif //ZLMEDIAKIT_FRAME_H
|