ZLMediaKit/src/Rtmp/RtmpCodec.h
2022-12-02 14:43:06 +08:00

67 lines
1.5 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_RTMPCODEC_H
#define ZLMEDIAKIT_RTMPCODEC_H
#include "Rtmp/Rtmp.h"
#include "Extension/Frame.h"
#include "Util/RingBuffer.h"
namespace mediakit{
class RtmpRing {
public:
using Ptr = std::shared_ptr<RtmpRing>;
using RingType = toolkit::RingBuffer<RtmpPacket::Ptr>;
RtmpRing() = default;
virtual ~RtmpRing() = default;
/**
* 获取rtmp环形缓存
*/
virtual RingType::Ptr getRtmpRing() const {
return _ring;
}
/**
* 设置rtmp环形缓存
*/
virtual void setRtmpRing(const RingType::Ptr &ring) {
_ring = ring;
}
/**
* 输入rtmp包
* @param rtmp rtmp包
*/
virtual void inputRtmp(const RtmpPacket::Ptr &rtmp) {
if (_ring) {
_ring->write(rtmp, rtmp->isVideoKeyFrame());
}
}
protected:
RingType::Ptr _ring;
};
class RtmpCodec : public RtmpRing, public FrameDispatcher, public CodecInfo {
public:
using Ptr = std::shared_ptr<RtmpCodec>;
RtmpCodec() = default;
~RtmpCodec() override = default;
virtual void makeConfigPacket() {};
};
}//namespace mediakit
#endif //ZLMEDIAKIT_RTMPCODEC_H