整理MediaSink

src/Rtsp/RtspMuxer.cpp

@@ -38,29 +38,34 @@ RtspMuxer::RtspMuxer(const TitleSdp::Ptr &title){
     _rtpRing = std::make_shared<RtpRingInterface::RingType>();
 }
 
-void RtspMuxer::onTrackReady(const Track::Ptr &track) {
+void RtspMuxer::addTrack(const Track::Ptr &track) {
     //根据track生成sdp
     Sdp::Ptr sdp = track->getSdp();
     if (!sdp) {
         return;
     }
-    auto encoder = Factory::getRtpEncoderBySdp(sdp);
+
+    auto &encoder = _encoder[track->getTrackType()];
+    encoder = Factory::getRtpEncoderBySdp(sdp);
     if (!encoder) {
         return;
     }
+
+    //设置rtp输出环形缓存
+    encoder->setRtpRing(_rtpRing);
+
     //添加其sdp
     _sdp.append(sdp->getSdp());
-    //设置Track的代理，这样输入frame至Track时，最终数据将输出到RtpEncoder中
-    track->addDelegate(encoder);
-    //rtp编码器共用同一个环形缓存
-    encoder->setRtpRing(_rtpRing);
+}
+
+void RtspMuxer::inputFrame(const Frame::Ptr &frame) {
+    auto &encoder = _encoder[frame->getTrackType()];
+    if(encoder){
+        encoder->inputFrame(frame);
+    }
 }
 
 string RtspMuxer::getSdp() {
-    if(!isAllTrackReady()){
-        //尚未就绪
-        return "";
-    }
     return _sdp;
 }
 
@@ -68,5 +73,12 @@ RtpRingInterface::RingType::Ptr RtspMuxer::getRtpRing() const {
     return _rtpRing;
 }
 
+void RtspMuxer::resetTracks() {
+    _sdp.clear();
+    for(auto &encoder : _encoder){
+        encoder = nullptr;
+    }
+}
+
 
 } /* namespace mediakit */