Mirror/ZLMediaKit
1
0
Fork 0
mirror of https://github.com/ZLMediaKit/ZLMediaKit.git synced 2026-07-04 18:17:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

统一使用MultiMediaSourceMuxer转协议

Browse Source
This commit is contained in:
xiongziliang
2019-07-22 18:37:32 +08:00
parent 46f5a853ac
commit cf4482983b
2 changed files with 44 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/Rtmp/RtmpToRtspMediaSource.h
View File

@@ -38,8 +38,8 @@
#include "Rtmp.h"
#include "RtmpMediaSource.h"
#include "RtmpDemuxer.h"
#include "MediaFile/MediaRecorder.h"
#include "Rtsp/RtspMediaSourceMuxer.h"
#include "Common/MultiMediaSourceMuxer.h"
using namespace std;
using namespace toolkit;
@@ -54,8 +54,9 @@ public:
const string &id,
bool bEnableHls = true,
bool bEnableMp4 = false,
int ringSize = 0):RtmpMediaSource(vhost, app, id,ringSize){
_recorder = std::make_shared<MediaRecorder>(vhost, app, id, bEnableHls, bEnableMp4);
int ringSize = 0) : RtmpMediaSource(vhost, app, id,ringSize){
_bEnableHls = bEnableHls;
_bEnableMp4 = bEnableMp4;
_rtmpDemuxer = std::make_shared<RtmpDemuxer>();
}
virtual ~RtmpToRtspMediaSource(){}
@@ -67,36 +68,39 @@ public:
void onWrite(const RtmpPacket::Ptr &pkt,bool key_pos) override {
_rtmpDemuxer->inputRtmp(pkt);
if(!_rtspMuxer && _rtmpDemuxer->isInited(2000)){
_rtspMuxer = std::make_shared<RtspMediaSourceMuxer>(getVhost(),
getApp(),
getId(),
std::make_shared<TitleSdp>(_rtmpDemuxer->getDuration()));
if(!_muxer && _rtmpDemuxer->isInited(2000)){
_muxer = std::make_shared<MultiMediaSourceMuxer>(getVhost(),
getApp(),
getId(),
_rtmpDemuxer->getDuration(),
true,//转rtsp
false,//不重复生成rtmp
_bEnableHls,
_bEnableMp4);
for (auto &track : _rtmpDemuxer->getTracks(false)){
_rtspMuxer->addTrack(track);
_recorder->addTrack(track);
track->addDelegate(_rtspMuxer);
track->addDelegate(_recorder);
_muxer->addTrack(track);
track->addDelegate(_muxer);
}
_rtspMuxer->setListener(_listener);
_muxer->setListener(_listener);
}
RtmpMediaSource::onWrite(pkt,key_pos);
}
void setListener(const std::weak_ptr<MediaSourceEvent> &listener) override {
RtmpMediaSource::setListener(listener);
if(_rtspMuxer){
_rtspMuxer->setListener(listener);
if(_muxer){
_muxer->setListener(listener);
}
}
int readerCount() override {
return RtmpMediaSource::readerCount() + (_rtspMuxer ? _rtspMuxer->readerCount() : 0);
return RtmpMediaSource::readerCount() + (_muxer ? _muxer->readerCount() : 0);
}
private:
RtmpDemuxer::Ptr _rtmpDemuxer;
RtspMediaSourceMuxer::Ptr _rtspMuxer;
MediaRecorder::Ptr _recorder;
MultiMediaSourceMuxer::Ptr _muxer;
bool _bEnableHls;
bool _bEnableMp4;
};
} /* namespace mediakit */