Merge branch 'FrameDispatcher_addDelegate_refactor' of github.com:mtdxc/ZLMediaKit

src/Common/MediaSink.cpp

@@ -37,7 +37,7 @@ bool MediaSink::addTrack(const Track::Ptr &track_in) {
     };
     _ticker.resetTime();
 
-    track->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([this](const Frame::Ptr &frame) {
+    track->addDelegate([this](const Frame::Ptr &frame) {
         if (_all_track_ready) {
             return onTrackFrame(frame);
         }
@@ -52,7 +52,7 @@ bool MediaSink::addTrack(const Track::Ptr &track_in) {
         //还有Track未就绪，先缓存之
         frame_unread.emplace_back(Frame::getCacheAbleFrame(frame));
         return true;
-    }));
+    });
     return true;
 }
 
@@ -247,13 +247,13 @@ bool MediaSink::addMuteAudioTrack() {
     }
     auto audio = std::make_shared<AACTrack>(makeAacConfig(MUTE_ADTS_DATA, ADTS_HEADER_LEN));
     _track_map[audio->getTrackType()] = std::make_pair(audio, true);
-    audio->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([this](const Frame::Ptr &frame) {
+    audio->addDelegate([this](const Frame::Ptr &frame) {
         return onTrackFrame(frame);
-    }));
+    });
     _mute_audio_maker = std::make_shared<MuteAudioMaker>();
-    _mute_audio_maker->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([audio](const Frame::Ptr &frame) {
+    _mute_audio_maker->addDelegate([audio](const Frame::Ptr &frame) {
         return audio->inputFrame(frame);
-    }));
+    });
     onTrackReady(audio);
     TraceL << "mute aac track added";
     return true;

src/Extension/Frame.cpp

@@ -210,5 +210,39 @@ void FrameMerger::flush() {
     }
     clear();
 }
+/**
+ * 写帧接口转function，辅助类
+ */
+class FrameWriterInterfaceHelper : public FrameWriterInterface {
+public:
+    typedef std::shared_ptr<FrameWriterInterfaceHelper> Ptr;
+    typedef std::function<bool(const Frame::Ptr &frame)> onWriteFrame;
+
+    /**
+     * inputFrame后触发onWriteFrame回调
+     */
+    FrameWriterInterfaceHelper(const onWriteFrame& cb){
+        _writeCallback = cb;
+    }
+
+    virtual ~FrameWriterInterfaceHelper(){}
+
+    /**
+     * 写入帧数据
+     */
+    bool inputFrame(const Frame::Ptr &frame) override {
+        return _writeCallback(frame);
+    }
+
+private:
+    onWriteFrame _writeCallback;
+};
+
+FrameWriterInterface* FrameDispatcher::addDelegate(const std::function<bool(const Frame::Ptr &frame)> &cb) {
+    auto delegate = std::make_shared<FrameWriterInterfaceHelper>(cb);
+    std::lock_guard<std::mutex> lck(_mtx);
+    _delegates.emplace(delegate.get(), delegate);
+    return delegate.get();
+}
 
 }//namespace mediakit

src/Extension/Frame.h

@@ -278,29 +278,6 @@ public:
     virtual void flush() {};
 };
 
-/**
- * 写帧接口转function，辅助类
- */
-class FrameWriterInterfaceHelper : public FrameWriterInterface {
-public:
-    typedef std::shared_ptr<FrameWriterInterfaceHelper> Ptr;
-    typedef std::function<bool(const Frame::Ptr &frame)> onWriteFrame;
-
-    /**
-     * inputFrame后触发onWriteFrame回调
-     */
-    FrameWriterInterfaceHelper(const onWriteFrame &cb) { _writeCallback = cb; }
-    virtual ~FrameWriterInterfaceHelper() = default;
-
-    /**
-     * 写入帧数据
-     */
-    bool inputFrame(const Frame::Ptr &frame) override { return _writeCallback(frame); }
-
-private:
-    onWriteFrame _writeCallback;
-};
-
 /**
  * 支持代理转发的帧环形缓存
  */
@@ -318,6 +295,7 @@ public:
         _delegates.emplace(delegate.get(), delegate);
     }
 
+    FrameWriterInterface* addDelegate(const std::function<bool(const Frame::Ptr &frame)> &cb);
     /**
      * 删除代理
      */

src/Player/PlayerBase.cpp

@@ -112,9 +112,9 @@ bool Demuxer::addTrack(const Track::Ptr &track) {
     }
 
     if (_sink->addTrack(track)) {
-        track->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([this](const Frame::Ptr &frame) {
+        track->addDelegate([this](const Frame::Ptr &frame) {
             return _sink->inputFrame(frame);
-        }));
+        });
         return true;
     }
     return false;

src/Rtp/GB28181Process.cpp

@@ -136,10 +136,10 @@ bool GB28181Process::inputRtp(bool, const char *data, size_t data_len) {
             }
         }
         // 设置frame回调
-        _rtp_decoder[pt]->addDelegate(std::make_shared<FrameWriterInterfaceHelper>([this](const Frame::Ptr &frame) {
+        _rtp_decoder[pt]->addDelegate([this](const Frame::Ptr &frame) {
             onRtpDecode(frame);
             return true;
-        }));
+        });
     }
 
     return ref->inputRtp(TrackVideo, (unsigned char *)data, data_len);