mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-30 07:26:53 +08:00
486 lines
11 KiB
C++
486 lines
11 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"
|
||
|
||
using namespace std;
|
||
using namespace toolkit;
|
||
|
||
namespace mediakit{
|
||
|
||
typedef enum {
|
||
TrackInvalid = -1,
|
||
TrackVideo = 0,
|
||
TrackAudio,
|
||
TrackTitle,
|
||
TrackApplication,
|
||
TrackMax
|
||
} TrackType;
|
||
|
||
#define CODEC_MAP(XX) \
|
||
XX(CodecH264, TrackVideo, 0, "H264") \
|
||
XX(CodecH265, TrackVideo, 1, "H265") \
|
||
XX(CodecAAC, TrackAudio, 2, "mpeg4-generic") \
|
||
XX(CodecG711A, TrackAudio, 3, "PCMA") \
|
||
XX(CodecG711U, TrackAudio, 4, "PCMU") \
|
||
XX(CodecOpus, TrackAudio, 5, "opus") \
|
||
XX(CodecL16, TrackAudio, 6, "L16") \
|
||
XX(CodecVP8, TrackVideo, 7, "VP8") \
|
||
XX(CodecVP9, TrackVideo, 8, "VP9") \
|
||
XX(CodecAV1, TrackVideo, 9, "AV1X")
|
||
|
||
typedef enum {
|
||
CodecInvalid = -1,
|
||
#define XX(name, type, value, str) name = value,
|
||
CODEC_MAP(XX)
|
||
#undef XX
|
||
CodecMax
|
||
} CodecId;
|
||
|
||
/**
|
||
* 字符串转媒体类型转
|
||
*/
|
||
TrackType getTrackType(const string &str);
|
||
|
||
/**
|
||
* 媒体类型转字符串
|
||
*/
|
||
const char* getTrackString(TrackType type);
|
||
|
||
/**
|
||
* 根据SDP中描述获取codec_id
|
||
* @param str
|
||
* @return
|
||
*/
|
||
CodecId getCodecId(const 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();
|
||
|
||
/**
|
||
* 获取音视频类型
|
||
*/
|
||
TrackType getTrackType();
|
||
};
|
||
|
||
/**
|
||
* 帧类型的抽象接口
|
||
*/
|
||
class Frame : public 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; }
|
||
|
||
/**
|
||
* 返回可缓存的frame
|
||
*/
|
||
static Ptr getCacheAbleFrame(const Ptr &frame);
|
||
|
||
private:
|
||
//对象个数统计
|
||
ObjectStatistic<Frame> _statistic;
|
||
};
|
||
|
||
class FrameImp : public Frame {
|
||
public:
|
||
using Ptr = std::shared_ptr<FrameImp>;
|
||
|
||
template<typename C=FrameImp>
|
||
static std::shared_ptr<C> create();
|
||
|
||
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;
|
||
BufferLikeString _buffer;
|
||
|
||
private:
|
||
//对象个数统计
|
||
ObjectStatistic<FrameImp> _statistic;
|
||
|
||
protected:
|
||
friend class ResourcePool_l<FrameImp>;
|
||
FrameImp() = default;
|
||
|
||
template<typename C>
|
||
static std::shared_ptr<C> create_l();
|
||
};
|
||
|
||
/**
|
||
* 一个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 void inputFrame(const Frame::Ptr &frame) = 0;
|
||
};
|
||
|
||
/**
|
||
* 写帧接口转function，辅助类
|
||
*/
|
||
class FrameWriterInterfaceHelper : public FrameWriterInterface {
|
||
public:
|
||
typedef std::shared_ptr<FrameWriterInterfaceHelper> Ptr;
|
||
typedef std::function<void(const Frame::Ptr &frame)> onWriteFrame;
|
||
|
||
/**
|
||
* inputFrame后触发onWriteFrame回调
|
||
*/
|
||
FrameWriterInterfaceHelper(const onWriteFrame& cb){
|
||
_writeCallback = cb;
|
||
}
|
||
|
||
virtual ~FrameWriterInterfaceHelper(){}
|
||
|
||
/**
|
||
* 写入帧数据
|
||
*/
|
||
void inputFrame(const Frame::Ptr &frame) override {
|
||
_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可能多线程同时操作
|
||
lock_guard<mutex> lck(_mtx);
|
||
_delegates_write.emplace(delegate.get(),delegate);
|
||
_need_update = true;
|
||
}
|
||
|
||
/**
|
||
* 删除代理
|
||
*/
|
||
void delDelegate(FrameWriterInterface *ptr){
|
||
//_delegates_write可能多线程同时操作
|
||
lock_guard<mutex> lck(_mtx);
|
||
_delegates_write.erase(ptr);
|
||
_need_update = true;
|
||
}
|
||
|
||
/**
|
||
* 写入帧并派发
|
||
*/
|
||
void inputFrame(const Frame::Ptr &frame) override{
|
||
if(_need_update){
|
||
//发现代理列表发生变化了，这里同步一次
|
||
lock_guard<mutex> lck(_mtx);
|
||
_delegates_read = _delegates_write;
|
||
_need_update = false;
|
||
}
|
||
|
||
//_delegates_read能确保是单线程操作的
|
||
for(auto &pr : _delegates_read){
|
||
pr.second->inputFrame(frame);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 返回代理个数
|
||
*/
|
||
size_t size() const {
|
||
return _delegates_write.size();
|
||
}
|
||
private:
|
||
mutex _mtx;
|
||
map<void *,FrameWriterInterface::Ptr> _delegates_read;
|
||
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;
|
||
};
|
||
|
||
/**
|
||
* 该对象可以把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 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 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:
|
||
Buffer::Ptr _buf;
|
||
};
|
||
|
||
/**
|
||
* 合并一些时间戳相同的frame
|
||
*/
|
||
class FrameMerger {
|
||
public:
|
||
using onOutput = function<void(uint32_t dts, uint32_t pts, const Buffer::Ptr &buffer, bool have_idr)>;
|
||
enum {
|
||
none = 0,
|
||
h264_prefix,
|
||
mp4_nal_size,
|
||
};
|
||
|
||
FrameMerger(int type);
|
||
~FrameMerger() = default;
|
||
|
||
void clear();
|
||
void inputFrame(const Frame::Ptr &frame, const onOutput &cb);
|
||
|
||
private:
|
||
bool willFlush(const Frame::Ptr &frame) const;
|
||
void doMerge(BufferLikeString &buffer, const Frame::Ptr &frame) const;
|
||
bool shouldDrop(const Frame::Ptr &frame) const;
|
||
|
||
private:
|
||
int _type;
|
||
List<Frame::Ptr> _frameCached;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif //ZLMEDIAKIT_FRAME_H
|