整理文件录制

src/Common/MediaSink.h

@@ -40,6 +40,8 @@ namespace mediakit{
 
 class MediaSinkInterface : public FrameWriterInterface {
 public:
+    typedef std::shared_ptr<MediaSinkInterface> Ptr;
+
     MediaSinkInterface(){};
     virtual ~MediaSinkInterface(){};
 

src/Common/MediaSource.cpp

@@ -26,7 +26,7 @@
 
 
 #include "MediaSource.h"
-#include "MediaFile/MediaReader.h"
+#include "Record/MP4Reader.h"
 #include "Util/util.h"
 #include "Network/sockutil.h"
 #include "Network/TcpSession.h"
@@ -278,7 +278,7 @@ MediaSource::Ptr MediaSource::find(
     });
     if(!ret && bMake){
         //未查找媒体源，则创建一个
-        ret = MediaReader::onMakeMediaSource(schema, vhost,app,id);
+        ret = MP4Reader::onMakeMediaSource(schema, vhost,app,id);
     }
     return ret;
 }

src/Common/MultiMediaSourceMuxer.h

@@ -29,7 +29,7 @@
 
 #include "Rtsp/RtspMediaSourceMuxer.h"
 #include "Rtmp/RtmpMediaSourceMuxer.h"
-#include "MediaFile/MediaRecorder.h"
+#include "Record/Recorder.h"
 
 class MultiMediaSourceMuxer : public MediaSink , public std::enable_shared_from_this<MultiMediaSourceMuxer>{
 public:
@@ -42,15 +42,19 @@ public:
                           bool bEanbleRtsp = true,
                           bool bEanbleRtmp = true,
                           bool bEanbleHls = true,
-                          bool bEnableMp4 = false
-                          ){
+                          bool bEnableMp4 = false){
         if (bEanbleRtmp) {
             _rtmp = std::make_shared<RtmpMediaSourceMuxer>(vhost, strApp, strId, std::make_shared<TitleMeta>(dur_sec));
         }
         if (bEanbleRtsp) {
             _rtsp = std::make_shared<RtspMediaSourceMuxer>(vhost, strApp, strId, std::make_shared<TitleSdp>(dur_sec));
         }
-        _record = std::make_shared<MediaRecorder>(vhost,strApp,strId,bEanbleHls,bEnableMp4);
+        if(bEanbleHls){
+            _hls.reset(Recorder::createHlsRecorder(vhost, strApp, strId));
+        }
+        if(bEnableMp4){
+            _mp4.reset(Recorder::createMP4Recorder(vhost, strApp, strId));
+        }
     }
     virtual ~MultiMediaSourceMuxer(){}
 
@@ -64,7 +68,12 @@ public:
         if(_rtsp){
             _rtsp->resetTracks();
         }
-        _record->resetTracks();
+        if(_hls){
+            _hls->resetTracks();
+        }
+        if(_mp4){
+            _mp4->resetTracks();
+        }
     }
 
     /**
@@ -106,7 +115,12 @@ protected:
         if(_rtsp){
             _rtsp->addTrack(track);
         }
-        _record->addTrack(track);
+        if(_hls){
+            _hls->addTrack(track);
+        }
+        if(_mp4){
+            _mp4->addTrack(track);
+        }
     }
 
     /**
@@ -120,7 +134,12 @@ protected:
         if(_rtsp) {
             _rtsp->inputFrame(frame);
         }
-        _record->inputFrame(frame);
+        if(_hls){
+            _hls->inputFrame(frame);
+        }
+        if(_mp4){
+            _mp4->inputFrame(frame);
+        }
     }
 
     /**
@@ -139,7 +158,8 @@ protected:
 private:
     RtmpMediaSourceMuxer::Ptr _rtmp;
     RtspMediaSourceMuxer::Ptr _rtsp;
-    MediaRecorder::Ptr _record;
+    MediaSinkInterface::Ptr _hls;
+    MediaSinkInterface::Ptr _mp4;
 };
 
 

src/Common/Stamp.cpp

@@ -0,0 +1,92 @@
+/*
+ * 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 "Stamp.h"
+
+namespace mediakit {
+
+int64_t DeltaStamp::deltaStamp(int64_t stamp) {
+    if(!_last_stamp){
+        //第一次计算时间戳增量,时间戳增量为0
+        _last_stamp = stamp;
+        return 0;
+    }
+
+    int64_t ret = stamp - _last_stamp;
+    if(ret >= 0){
+        //时间戳增量为正，返回之
+        _last_stamp = stamp;
+        return ret;
+    }
+
+    //时间戳增量为负，说明时间戳回环了或回退了
+    _last_stamp = stamp;
+    return _playback ? ret : 0;
+}
+
+void DeltaStamp::setPlayBack(bool playback) {
+    _playback = playback;
+}
+
+void Stamp::revise(int64_t dts, int64_t pts, int64_t &dts_out, int64_t &pts_out,bool modifyStamp) {
+    if(!pts){
+        //没有播放时间戳,使其赋值为解码时间戳
+        pts = dts;
+    }
+
+    //pts和dts的差值
+    int pts_dts_diff = pts - dts;
+
+    if(_last_dts != dts){
+        //时间戳发生变更
+        _relativeStamp += deltaStamp(modifyStamp ? _ticker.elapsedTime() : dts);
+        _last_dts = dts;
+    }
+    dts_out = _relativeStamp;
+
+    //////////////以下是播放时间戳的计算//////////////////
+    if(pts_dts_diff > 200 || pts_dts_diff < -200){
+        //如果差值大于200毫秒，则认为由于回环导致时间戳错乱了
+        pts_dts_diff = 0;
+    }
+
+    pts_out = dts_out + pts_dts_diff;
+    if(pts_out < 0){
+        //时间戳不能小于0
+        pts_out = 0;
+    }
+}
+
+void Stamp::setRelativeStamp(int64_t relativeStamp) {
+    _relativeStamp = relativeStamp;
+}
+
+int64_t Stamp::getRelativeStamp() const {
+    return _relativeStamp;
+}
+
+
+}//namespace mediakit

src/Common/Stamp.h

@@ -0,0 +1,93 @@
+/*
+ * 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.
+ */
+
+#ifndef ZLMEDIAKIT_STAMP_H
+#define ZLMEDIAKIT_STAMP_H
+
+#include "Util/TimeTicker.h"
+#include <cstdint>
+using namespace toolkit;
+
+namespace mediakit {
+
+class DeltaStamp{
+public:
+    DeltaStamp() = default;
+    ~DeltaStamp() = default;
+
+    /**
+     * 计算时间戳增量
+     * @param stamp 绝对时间戳
+     * @return 时间戳增量
+     */
+    int64_t deltaStamp(int64_t stamp);
+
+    /**
+     * 设置是否为回放模式，回放模式运行时间戳回退
+     * @param playback 是否为回放模式
+     */
+    void setPlayBack(bool playback = true);
+private:
+    int64_t _last_stamp = 0;
+    bool _playback = false;
+};
+//该类解决时间戳回环、回退问题
+//计算相对时间戳或者产生平滑时间戳
+class Stamp : public DeltaStamp{
+public:
+    Stamp() = default;
+    ~Stamp() = default;
+
+    /**
+     * 修正时间戳
+     * @param dts 输入dts，如果为0则根据系统时间戳生成
+     * @param pts 输入pts，如果为0则等于dts
+     * @param dts_out 输出dts
+     * @param pts_out 输出pts
+     * @param modifyStamp 是否用系统时间戳覆盖
+     */
+    void revise(int64_t dts, int64_t pts, int64_t &dts_out, int64_t &pts_out,bool modifyStamp = false);
+
+    /**
+     * 再设置相对时间戳，用于seek用
+     * @param relativeStamp 相对时间戳
+     */
+    void setRelativeStamp(int64_t relativeStamp);
+
+    /**
+     * 获取当前相对时间戳
+     * @return
+     */
+    int64_t getRelativeStamp() const ;
+private:
+    int64_t _relativeStamp = 0;
+    int64_t _last_dts = -1;
+    SmoothTicker _ticker;
+};
+
+}//namespace mediakit
+
+#endif //ZLMEDIAKIT_STAMP_H