amass/ZLMediaKit
1
0
Fork 0
mirror of https://github.com/ZLMediaKit/ZLMediaKit.git synced 2024-11-22 10:40:05 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

add RtspMediaSource::Clone

Browse Source
This commit is contained in:
Johnny 2023-05-11 20:47:40 +08:00 committed by 夏楚
parent 51e9313275
commit f4ee607feb
6 changed files with 21 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/Rtsp/RtspMediaSource.h
View File

@ -76,6 +76,10 @@ public:
return _sdp; return _sdp;
} }
virtual RtspMediaSource::Ptr Clone(const std::string& stream) {
return nullptr;
}
/** /**
* 获取相应轨道的ssrc * 获取相应轨道的ssrc
*/ */

6
src/Rtsp/RtspMediaSourceImp.cpp
View File

@ -126,6 +126,12 @@ void RtspMediaSourceImp::setProtocolOption(const ProtocolOption &option)
} }
} }
RtspMediaSource::Ptr RtspMediaSourceImp::Clone(const std::string &stream) {
auto src_imp = std::make_shared<RtspMediaSourceImp>(getVhost(), getApp(), stream);
src_imp->setSdp(getSdp());
src_imp->setProtocolOption(getProtocolOption());
return src_imp;
}
} }

1
src/Rtsp/RtspMediaSourceImp.h
View File

@ -107,6 +107,7 @@ public:
} }
} }
RtspMediaSource::Ptr Clone(const std::string& stream) override;
private: private:
bool _all_track_ready = false; bool _all_track_ready = false;
ProtocolOption _option; ProtocolOption _option;

9
webrtc/WebRtcPusher.cpp
View File

@ -10,13 +10,14 @@
#include "WebRtcPusher.h" #include "WebRtcPusher.h"
#include "Common/config.h" #include "Common/config.h"
#include "Rtsp/RtspMediaSourceImp.h"
using namespace std; using namespace std;
namespace mediakit { namespace mediakit {
WebRtcPusher::Ptr WebRtcPusher::create(const EventPoller::Ptr &poller, WebRtcPusher::Ptr WebRtcPusher::create(const EventPoller::Ptr &poller,
const RtspMediaSourceImp::Ptr &src, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const std::shared_ptr<void> &ownership,
const MediaInfo &info, const MediaInfo &info,
const ProtocolOption &option, const ProtocolOption &option,
@ -30,7 +31,7 @@ WebRtcPusher::Ptr WebRtcPusher::create(const EventPoller::Ptr &poller,
} }
WebRtcPusher::WebRtcPusher(const EventPoller::Ptr &poller, WebRtcPusher::WebRtcPusher(const EventPoller::Ptr &poller,
const RtspMediaSourceImp::Ptr &src, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const std::shared_ptr<void> &ownership,
const MediaInfo &info, const MediaInfo &info,
const ProtocolOption &option, const ProtocolOption &option,
@ -98,10 +99,8 @@ void WebRtcPusher::onRecvRtp(MediaTrack &track, const string &rid, RtpPacket::Pt
auto &src = _push_src_sim[rid]; auto &src = _push_src_sim[rid];
if (!src) { if (!src) {
auto stream_id = rid.empty() ? _push_src->getId() : _push_src->getId() + "_" + rid; auto stream_id = rid.empty() ? _push_src->getId() : _push_src->getId() + "_" + rid;
auto src_imp = std::make_shared<RtspMediaSourceImp>(_push_src->getVhost(), _push_src->getApp(), stream_id); auto src_imp = _push_src->Clone(stream_id);
_push_src_sim_ownership[rid] = src_imp->getOwnership(); _push_src_sim_ownership[rid] = src_imp->getOwnership();
src_imp->setSdp(_push_src->getSdp());
src_imp->setProtocolOption(_push_src->getProtocolOption());
src_imp->setListener(static_pointer_cast<WebRtcPusher>(shared_from_this())); src_imp->setListener(static_pointer_cast<WebRtcPusher>(shared_from_this()));
src = src_imp; src = src_imp;
} }

8
webrtc/WebRtcPusher.h
View File

@ -12,7 +12,7 @@
#define ZLMEDIAKIT_WEBRTCPUSHER_H #define ZLMEDIAKIT_WEBRTCPUSHER_H
#include "WebRtcTransport.h" #include "WebRtcTransport.h"
#include "Rtsp/RtspMediaSourceImp.h" #include "Rtsp/RtspMediaSource.h"
namespace mediakit { namespace mediakit {
@ -20,7 +20,7 @@ class WebRtcPusher : public WebRtcTransportImp, public MediaSourceEvent {
public: public:
using Ptr = std::shared_ptr<WebRtcPusher>; using Ptr = std::shared_ptr<WebRtcPusher>;
~WebRtcPusher() override = default; ~WebRtcPusher() override = default;
static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSourceImp::Ptr &src, static Ptr create(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp = false); const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp = false);
protected: protected:
@ -51,7 +51,7 @@ protected:
float getLossRate(MediaSource &sender,TrackType type) override; float getLossRate(MediaSource &sender,TrackType type) override;
private: private:
WebRtcPusher(const EventPoller::Ptr &poller, const RtspMediaSourceImp::Ptr &src, WebRtcPusher(const EventPoller::Ptr &poller, const RtspMediaSource::Ptr &src,
const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp); const std::shared_ptr<void> &ownership, const MediaInfo &info, const ProtocolOption &option, bool preferred_tcp);
private: private:
@ -61,7 +61,7 @@ private:
//媒体相关元数据 //媒体相关元数据
MediaInfo _media_info; MediaInfo _media_info;
//推流的rtsp源 //推流的rtsp源
RtspMediaSourceImp::Ptr _push_src; RtspMediaSource::Ptr _push_src;
//推流所有权 //推流所有权
std::shared_ptr<void> _push_src_ownership; std::shared_ptr<void> _push_src_ownership;
//推流的rtsp源,支持simulcast //推流的rtsp源,支持simulcast

2
webrtc/WebRtcTransport.cpp
View File

@ -23,6 +23,8 @@
#include "WebRtcPlayer.h" #include "WebRtcPlayer.h"
#include "WebRtcPusher.h" #include "WebRtcPusher.h"
#include "Rtsp/RtspMediaSourceImp.h"
#define RTP_SSRC_OFFSET 1 #define RTP_SSRC_OFFSET 1
#define RTX_SSRC_OFFSET 2 #define RTX_SSRC_OFFSET 2
#define RTP_CNAME "zlmediakit-rtp" #define RTP_CNAME "zlmediakit-rtp"