修复hls由于缓存导致花屏的问题

src/Extension/Frame.cpp

@@ -0,0 +1,42 @@
+/*
+ * MIT License
+ *
+ * Copyright (c) 2016-2019 xiongziliang <771730766@qq.com>
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/ZLMediaKit).
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to deal
+ * in the Software without restriction, including without limitation the rights
+ * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+ * copies of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included in all
+ * copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
+ */
+
+#include "Frame.h"
+
+using namespace std;
+using namespace toolkit;
+
+namespace mediakit{
+
+Frame::Ptr Frame::getCacheAbleFrame(const Frame::Ptr &frame){
+    if(frame->cacheAble()){
+        return frame;
+    }
+    return std::make_shared<FrameCacheAble>(frame);
+}
+
+}//namespace mediakit
+

src/Extension/Frame.h

@@ -121,6 +121,12 @@ public:
      * 是否可以缓存
      */
     virtual bool cacheAble() const { return true; }
+
+    /**
+     * 返回可缓存的frame
+     * @return
+     */
+    static Ptr getCacheAbleFrame(const Ptr &frame);
 };
 
 /**
@@ -326,6 +332,57 @@ protected:
     uint32_t _prefixSize;
 };
 
+class FrameCacheAble : public FrameNoCacheAble {
+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();
+    }
+
+    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;
+    }
+private:
+    Frame::Ptr _frame;
+    BufferRaw::Ptr _buffer;
+    TrackType _trackType;
+    CodecId _codec;
+    bool _key;
+};
+
 
 }//namespace mediakit
 

src/MediaFile/TsMuxer.cpp

@@ -78,7 +78,7 @@ void TsMuxer::inputFrame(const Frame::Ptr &frame) {
                 mpeg_ts_write(_context, it->second, back->keyFrame() ? 0x0001 : 0, back->pts() * 90LL, back->dts() * 90LL, merged_frame->data(),  merged_frame->size());
                 _frameCached.clear();
             }
-            _frameCached.emplace_back(frame);
+            _frameCached.emplace_back(Frame::getCacheAbleFrame(frame));
         }
             break;
         default: {