整理和精简代码

src/Extension/Frame.h

@@ -28,6 +28,7 @@ typedef enum {
     CodecAAC,
     CodecG711A,
     CodecG711U,
+    CodecOpus,
     CodecMax = 0x7FFF
 } CodecId;
 
@@ -49,11 +50,6 @@ public:
     CodecInfo(){}
     virtual ~CodecInfo(){}
 
-    /**
-     * 获取音视频类型
-     */
-    virtual TrackType getTrackType() const  = 0;
-
     /**
      * 获取编解码器类型
      */
@@ -61,9 +57,13 @@ public:
 
     /**
      * 获取编码器名称
-     * @return 编码器名称
      */
     const char *getCodecName();
+
+    /**
+     * 获取音视频类型
+     */
+    TrackType getTrackType();
 };
 
 /**
@@ -76,15 +76,11 @@ public:
 
     /**
      * 返回解码时间戳，单位毫秒
-     * @return
      */
     virtual uint32_t dts() const = 0;
 
-
-
     /**
      * 返回显示时间戳，单位毫秒
-     * @return
      */
     virtual uint32_t pts() const {
         return dts();
@@ -98,13 +94,11 @@ public:
 
     /**
      * 返回是否为关键帧
-     * @return
      */
     virtual bool keyFrame() const = 0;
 
     /**
      * 是否为配置帧，譬如sps pps vps
-     * @return
      */
     virtual bool configFrame() const = 0;
 
@@ -115,14 +109,77 @@ public:
 
     /**
      * 返回可缓存的frame
-     * @return
      */
     static Ptr getCacheAbleFrame(const Ptr &frame);
 };
 
+class FrameImp : public Frame {
+public:
+    typedef std::shared_ptr<FrameImp> Ptr;
+
+    char *data() const override{
+        return (char *)_buffer.data();
+    }
+
+    uint32_t size() const override {
+        return _buffer.size();
+    }
+
+    uint32_t dts() const override {
+        return _dts;
+    }
+
+    uint32_t pts() const override{
+        return _pts ? _pts : _dts;
+    }
+
+    uint32_t prefixSize() const override{
+        return _prefix_size;
+    }
+
+    CodecId getCodecId() const override{
+        return _codecid;
+    }
+
+    bool keyFrame() const override {
+        return false;
+    }
+
+    bool configFrame() const override{
+        return false;
+    }
+
+public:
+    CodecId _codecid = CodecInvalid;
+    string _buffer;
+    uint32_t _dts = 0;
+    uint32_t _pts = 0;
+    uint32_t _prefix_size = 0;
+};
+
+/**
+ * 一个Frame类中可以有多个帧，他们通过 0x 00 00 01 分隔
+ * ZLMediaKit会先把这种复合帧split成单个帧然后再处理
+ * 一个复合帧可以通过无内存拷贝的方式切割成多个子Frame
+ * 提供该类的目的是切割复合帧时防止内存拷贝，提高性能
+ */
+template<typename Parent>
+class FrameInternal : public Parent{
+public:
+    typedef std::shared_ptr<FrameInternal> Ptr;
+    FrameInternal(const Frame::Ptr &parent_frame, char *ptr, uint32_t size, int prefix_size)
+            : Parent(ptr, size, parent_frame->dts(), parent_frame->pts(), prefix_size) {
+        _parent_frame = parent_frame;
+    }
+    bool cacheAble() const override {
+        return _parent_frame->cacheAble();
+    }
+private:
+    Frame::Ptr _parent_frame;
+};
+
 /**
  * 循环池辅助类
- * @tparam T
  */
 template <typename T>
 class ResourcePoolHelper{
@@ -140,18 +197,17 @@ private:
 };
 
 /**
- * 写帧接口的抽闲接口
+ * 写帧接口的抽象接口类
  */
 class FrameWriterInterface {
 public:
     typedef std::shared_ptr<FrameWriterInterface> Ptr;
-
     FrameWriterInterface(){}
     virtual ~FrameWriterInterface(){}
+
     /**
-    * 写入帧数据
-    * @param frame 帧
-    */
+     * 写入帧数据
+     */
     virtual void inputFrame(const Frame::Ptr &frame) = 0;
 };
 
@@ -165,16 +221,16 @@ public:
 
     /**
      * inputFrame后触发onWriteFrame回调
-     * @param cb
      */
     FrameWriterInterfaceHelper(const onWriteFrame& cb){
         _writeCallback = cb;
     }
+
     virtual ~FrameWriterInterfaceHelper(){}
+
     /**
-    * 写入帧数据
-    * @param frame 帧
-    */
+     * 写入帧数据
+     */
     void inputFrame(const Frame::Ptr &frame) override {
         _writeCallback(frame);
     }
@@ -182,7 +238,6 @@ private:
     onWriteFrame _writeCallback;
 };
 
-
 /**
  * 支持代理转发的帧环形缓存
  */
@@ -193,6 +248,9 @@ public:
     FrameDispatcher(){}
     virtual ~FrameDispatcher(){}
 
+    /**
+     * 添加代理
+     */
     void addDelegate(const FrameWriterInterface::Ptr &delegate){
         //_delegates_write可能多线程同时操作
         lock_guard<mutex> lck(_mtx);
@@ -200,6 +258,9 @@ public:
         _need_update = true;
     }
 
+    /**
+     * 删除代理
+     */
     void delDelegate(void *ptr){
         //_delegates_write可能多线程同时操作
         lock_guard<mutex> lck(_mtx);
@@ -208,8 +269,7 @@ public:
     }
 
     /**
-     * 写入帧数据
-     * @param frame 帧
+     * 写入帧并派发
      */
     void inputFrame(const Frame::Ptr &frame) override{
         if(_need_update){
@@ -223,7 +283,13 @@ public:
         for(auto &pr : _delegates_read){
             pr.second->inputFrame(frame);
         }
+    }
 
+    /**
+     * 返回代理个数
+     */
+    int size() const {
+        return _delegates_write.size();
     }
 private:
     mutex _mtx;
@@ -250,105 +316,23 @@ public:
     }
 
     uint32_t pts() const override{
-        if(_pts){
-            return _pts;
-        }
-        return dts();
+        return _pts ? _pts : dts();
     }
 
     uint32_t prefixSize() const override{
-        return _prefixSize;
+        return _prefix_size;
+    }
+
+    bool cacheAble() const override {
+        return false;
     }
 protected:
     char *_ptr;
     uint32_t _size;
     uint32_t _dts;
     uint32_t _pts = 0;
-    uint32_t _prefixSize;
+    uint32_t _prefix_size;
 };
 
-/**
- * 不可缓存的帧，在DevChannel类中有用到。
- * 该帧类型用于防止内存拷贝，直接使用指针传递数据
- * 在大多数情况下，ZLMediaKit是同步对帧数据进行使用和处理的
- * 所以提供此类型的帧很有必要，但是有时又无法避免缓存帧做后续处理
- * 所以可以通过Frame::getCacheAbleFrame方法拷贝一个可缓存的帧
- */
-class FrameNoCacheAble : public FrameFromPtr{
-public:
-    typedef std::shared_ptr<FrameNoCacheAble> Ptr;
-
-    /**
-     * 该帧不可缓存
-     * @return
-     */
-    bool cacheAble() const override {
-        return false;
-    }
-};
-
-/**
- * 该对象的功能是把一个不可缓存的帧转换成可缓存的帧
- * @see FrameNoCacheAble
- */
-class FrameCacheAble : public FrameFromPtr {
-public:
-    typedef std::shared_ptr<FrameCacheAble> Ptr;
-
-    FrameCacheAble(const Frame::Ptr &frame){
-        if(frame->cacheAble()){
-            _frame = frame;
-            _ptr = frame->data();
-        }else{
-            _buffer = std::make_shared<BufferRaw>();
-            _buffer->assign(frame->data(),frame->size());
-            _ptr = _buffer->data();
-        }
-        _size = frame->size();
-        _dts = frame->dts();
-        _pts = frame->pts();
-        _prefixSize = frame->prefixSize();
-        _trackType = frame->getTrackType();
-        _codec = frame->getCodecId();
-        _key = frame->keyFrame();
-        _config = frame->configFrame();
-    }
-
-    virtual ~FrameCacheAble() = default;
-
-    /**
-     * 可以被缓存
-     * @return
-     */
-    bool cacheAble() const override {
-        return true;
-    }
-
-    TrackType getTrackType() const override{
-        return _trackType;
-    }
-
-    CodecId getCodecId() const override{
-        return _codec;
-    }
-
-    bool keyFrame() const override{
-        return _key;
-    }
-
-    bool configFrame() const override{
-        return _config;
-    }
-private:
-    Frame::Ptr _frame;
-    BufferRaw::Ptr _buffer;
-    TrackType _trackType;
-    CodecId _codec;
-    bool _key;
-    bool _config;
-};
-
-
 }//namespace mediakit
-
-#endif //ZLMEDIAKIT_FRAME_H
+#endif //ZLMEDIAKIT_FRAME_H