添加MediaSink类，简化重复逻辑代码

src/RtspMuxer/RtspMediaSourceMuxer.h

@@ -55,7 +55,7 @@ public:
         _mediaSouce->setTimeStamp(stamp);
     }
 private:
-    void onInited() override {
+    void onAllTrackReady() override {
         _mediaSouce->onGetSDP(getSdp());
     }
 private:

src/RtspMuxer/RtspMuxer.cpp

@@ -29,79 +29,47 @@
 
 namespace mediakit {
 
-void RtspMuxer::addTrack(const Track::Ptr &track_in, uint32_t ssrc, int mtu) {
-    //克隆Track，只拷贝其数据，不拷贝其数据转发关系
-    auto track = track_in->clone();
-    auto codec_id = track->getCodecId();
-    _track_map[codec_id] = track;
-    if(mtu == 0){
-        mtu = (track->getTrackType() == TrackVideo ? 1400 : 600);
+RtspMuxer::RtspMuxer(const TitleSdp::Ptr &title){
+    if(!title){
+        _sdp = std::make_shared<TitleSdp>()->getSdp();
+    } else{
+        _sdp = title->getSdp();
     }
-    auto lam = [this,ssrc,mtu,track](){
-        //异步生成rtp编码器
-        //根据track生产sdp
-        Sdp::Ptr sdp = Factory::getSdpByTrack(track);
-        if (!sdp) {
-            return;
-        }
+    _rtpRing = std::make_shared<RtpRingInterface::RingType>();
+}
 
-        // 根据sdp生成rtp编码器
-        auto encoder = sdp->createRtpEncoder(ssrc ? ssrc : ((uint64_t) sdp.get()) & 0xFFFFFFFF, mtu);
-        if (!encoder) {
-            return;
-        }
-        //添加其sdp
-        _sdp.append(sdp->getSdp());
-        //设置Track的代理，这样输入frame至Track时，最终数据将输出到RtpEncoder中
-        track->addDelegate(encoder);
-        //rtp编码器共用同一个环形缓存
-        encoder->setRtpRing(_rtpRing);
-    };
-    if(track->ready()){
-        lam();
-    }else{
-        _trackReadyCallback[codec_id] = lam;
+void RtspMuxer::onTrackReady(const Track::Ptr &track) {
+    //根据track生产sdp
+    Sdp::Ptr sdp = Factory::getSdpByTrack(track);
+    if (!sdp) {
+        return;
     }
+    uint32_t ssrc = ((uint64_t) sdp.get()) & 0xFFFFFFFF;
+    auto mtu = (track->getTrackType() == TrackVideo ? 1400 : 600);
+    // 根据sdp生成rtp编码器ssrc
+    auto encoder = sdp->createRtpEncoder(ssrc, mtu);
+    if (!encoder) {
+        return;
+    }
+    //添加其sdp
+    _sdp.append(sdp->getSdp());
+    //设置Track的代理，这样输入frame至Track时，最终数据将输出到RtpEncoder中
+    track->addDelegate(encoder);
+    //rtp编码器共用同一个环形缓存
+    encoder->setRtpRing(_rtpRing);
 }
 
 string RtspMuxer::getSdp() {
-    if(!_trackReadyCallback.empty()){
+    if(!isAllTrackReady()){
         //尚未就绪
         return "";
     }
     return _sdp;
 }
 
-
-void RtspMuxer::inputFrame(const Frame::Ptr &frame) {
-    auto codec_id = frame->getCodecId();
-    auto it = _track_map.find(codec_id);
-    if (it == _track_map.end()) {
-        return;
-    }
-    it->second->inputFrame(frame);
-    if(!_inited && !_trackReadyCallback.empty() && it->second->ready()){
-        //Track由未就绪状态装换成就绪状态，我们就生成sdp以及rtp编码器
-        auto it_callback = _trackReadyCallback.find(codec_id);
-        if(it_callback != _trackReadyCallback.end()){
-            it_callback->second();
-            _trackReadyCallback.erase(it_callback);
-        }
-    }
-
-    if(!_inited && _trackReadyCallback.empty()){
-        _inited = true;
-        onInited();
-    }
-}
-
-bool RtspMuxer::inputRtp(const RtpPacket::Ptr &rtp, bool key_pos) {
-   _rtpRing->write(rtp,key_pos);
-    return key_pos;
-}
-
 RtpRingInterface::RingType::Ptr RtspMuxer::getRtpRing() const {
     return _rtpRing;
 }
 
+
 } /* namespace mediakit */

src/RtspMuxer/RtspMuxer.h

@@ -29,35 +29,22 @@
 
 #include "RtspSdp.h"
 #include "Player/Frame.h"
+#include "Common/MediaSink.h"
 
 namespace mediakit{
 /**
 * rtsp生成器
 */
-class RtspMuxer : public  FrameRingWriterInterface{
+class RtspMuxer : public MediaSink{
 public:
     typedef std::shared_ptr<RtspMuxer> Ptr;
 
     /**
      * 构造函数
      */
-    RtspMuxer(const TitleSdp::Ptr &title = nullptr){
-        if(!title){
-            _sdp = std::make_shared<TitleSdp>()->getSdp();
-        } else{
-            _sdp = title->getSdp();
-        }
-        _rtpRing = std::make_shared<RtpRingInterface::RingType>();
-    }
-    virtual ~RtspMuxer(){}
+    RtspMuxer(const TitleSdp::Ptr &title = nullptr);
 
-    /**
-     * 添加音视频媒体
-     * @param track 媒体描述
-     * @param ssrc 媒体rtp ssrc
-     * @param mtu 媒体rtp mtu
-     */
-    void addTrack(const Track::Ptr & track,uint32_t ssrc = 0,int mtu = 0) ;
+    virtual ~RtspMuxer(){}
 
     /**
      * 获取完整的SDP字符串
@@ -65,33 +52,21 @@ public:
      */
     string getSdp() ;
 
-    /**
-     * 写入帧数据然后打包rtp
-     * @param frame 帧数据
-     */
-    void inputFrame(const Frame::Ptr &frame) override ;
-
-    /**
-     * 也可以在外部打包好rtp然后再写入
-     * @param rtp rtp包
-     * @param key_pos 是否为关键帧的第一个rtp包
-     */
-    bool inputRtp(const RtpPacket::Ptr &rtp, bool key_pos = true);
-
     /**
      * 获取rtp环形缓存
      * @return
      */
     RtpRingInterface::RingType::Ptr getRtpRing() const;
-
 protected:
-    virtual void onInited(){};
+    /**
+    * 某track已经准备好，其ready()状态返回true，
+    * 此时代表可以获取其例如sps pps等相关信息了
+    * @param track
+    */
+    void onTrackReady(const Track::Ptr & track) override ;
 private:
-    map<int,Track::Ptr> _track_map;
-    map<int,function<void()> > _trackReadyCallback;
     RtpRingInterface::RingType::Ptr _rtpRing;
     string _sdp;
-    bool _inited = false;
 };