MPEG: 整合复用ts/ps生成代码

src/Record/MPEG.h

@@ -0,0 +1,114 @@
+/*
+ * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
+ *
+ * Use of this source code is governed by MIT license that can be found in the
+ * LICENSE file in the root of the source tree. All contributing project authors
+ * may be found in the AUTHORS file in the root of the source tree.
+ */
+
+#ifndef ZLMEDIAKIT_MPEG_H
+#define ZLMEDIAKIT_MPEG_H
+
+#if defined(ENABLE_HLS) || defined(ENABLE_RTPPROXY)
+
+#include <cstdio>
+#include <cstdint>
+#include <unordered_map>
+#include "mpeg-ps.h"
+#include "Extension/Frame.h"
+#include "Extension/Track.h"
+#include "Util/File.h"
+#include "Common/MediaSink.h"
+#include "Common/Stamp.h"
+
+namespace mediakit {
+
+typedef struct mpeg_muxer_t mpeg_muxer_t;
+typedef struct ps_muxer_func_t mpeg_muxer_func_t;
+
+mpeg_muxer_t *mpeg_muxer_create(int is_ps, const mpeg_muxer_func_t *func, void *param);
+int mpeg_muxer_destroy(mpeg_muxer_t *muxer);
+int mpeg_muxer_add_stream(mpeg_muxer_t *muxer, int codecid, const void *extradata, size_t extradata_size);
+int mpeg_muxer_input(mpeg_muxer_t *muxer, int stream, int flags, int64_t pts, int64_t dts, const void *data, size_t bytes);
+int mpeg_muxer_reset(mpeg_muxer_t *muxer);
+int mpeg_muxer_add_program(mpeg_muxer_t *muxer, uint16_t pn, const void *info, int bytes);
+int mpeg_muxer_remove_program(mpeg_muxer_t *muxer, uint16_t pn);
+int peg_muxer_add_program_stream(mpeg_muxer_t *muxer, uint16_t pn, int codecid, const void *extra_data, size_t extra_data_size);
+
+///////////////////////////////////////////////////////////////
+
+//该类用于产生MPEG-TS/MPEG-PS
+class MpegMuxer : public MediaSinkInterface {
+public:
+    MpegMuxer(bool is_ps);
+    ~MpegMuxer() override;
+
+    /**
+     * 添加音视频轨道
+     */
+    bool addTrack(const Track::Ptr &track) override;
+
+    /**
+     * 重置音视频轨道
+     */
+    void resetTracks() override;
+
+    /**
+     * 输入帧数据
+     */
+    bool inputFrame(const Frame::Ptr &frame) override;
+
+protected:
+    /**
+     * 输出ts/ps数据回调
+     * @param buffer ts/ps数据包
+     * @param timestamp 时间戳，单位毫秒
+     * @param key_pos 是否为关键帧的第一个ts/ps包，用于确保ts切片第一帧为关键帧
+     */
+    virtual void onWrite(std::shared_ptr<Buffer> buffer, uint32_t timestamp, bool key_pos) = 0;
+
+private:
+    void createContext();
+    void releaseContext();
+    void flushCache();
+    void onWrite_l(const void *packet, size_t bytes);
+
+private:
+    bool _is_ps = false;
+    bool _have_video = false;
+    bool _key_pos = false;
+    uint32_t _timestamp = 0;
+    mpeg_muxer_t *_context = nullptr;
+    BufferRaw::Ptr _buffer;
+    unordered_map<int, int/*track_id*/> _codec_to_trackid;
+    FrameMerger _frame_merger{FrameMerger::h264_prefix};
+    std::shared_ptr<BufferLikeString> _cache;
+};
+
+}//mediakit
+
+#else
+
+#include "Common/MediaSink.h"
+
+namespace mediakit {
+
+class MpegMuxer : public MediaSinkInterface {
+public:
+    MpegMuxer(bool is_ps) {};
+    ~MpegMuxer() override = default;
+    bool addTrack(const Track::Ptr &track) override { return false; }
+    void resetTracks() override {}
+    bool inputFrame(const Frame::Ptr &frame) override { return false; }
+
+protected:
+    virtual void onWrite(std::shared_ptr<Buffer> buffer, uint32_t timestamp, bool key_pos) = 0;
+};
+
+}//namespace mediakit
+
+#endif
+
+#endif //ZLMEDIAKIT_MPEG_H