支持http-ts/websocket-ts直播

src/TS/TSMediaSource.h

@@ -0,0 +1,129 @@
+/*
+ * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/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_TSMEDIASOURCE_H
+#define ZLMEDIAKIT_TSMEDIASOURCE_H
+
+#include "Common/MediaSource.h"
+using namespace toolkit;
+#define TS_GOP_SIZE 512
+
+namespace mediakit {
+
+//TS直播数据包
+class TSPacket : public BufferRaw{
+public:
+    using Ptr = std::shared_ptr<TSPacket>;
+
+    template<typename ...ARGS>
+    TSPacket(ARGS && ...args) : BufferRaw(std::forward<ARGS>(args)...) {};
+    ~TSPacket() override = default;
+
+public:
+    uint32_t time_stamp = 0;
+};
+
+//TS直播合并写策略类
+class TSFlushPolicy : public FlushPolicy{
+public:
+    TSFlushPolicy() = default;
+    ~TSFlushPolicy() = default;
+
+    uint32_t getStamp(const TSPacket::Ptr &packet) {
+        return packet->time_stamp;
+    }
+};
+
+//TS直播源
+class TSMediaSource : public MediaSource, public RingDelegate<TSPacket::Ptr>, public PacketCache<TSPacket, TSFlushPolicy>{
+public:
+    using PoolType = ResourcePool<TSPacket>;
+    using Ptr = std::shared_ptr<TSMediaSource>;
+    using RingDataType = std::shared_ptr<List<TSPacket::Ptr> >;
+    using RingType = RingBuffer<RingDataType>;
+
+    TSMediaSource(const string &vhost,
+                  const string &app,
+                  const string &stream_id,
+                  int ring_size = TS_GOP_SIZE) : MediaSource(TS_SCHEMA, vhost, app, stream_id), _ring_size(ring_size) {}
+
+    ~TSMediaSource() override = default;
+
+    /**
+     * 获取媒体源的环形缓冲
+     */
+    const RingType::Ptr &getRing() const {
+        return _ring;
+    }
+
+    /**
+     * 获取播放器个数
+     */
+    int readerCount() override {
+        return _ring ? _ring->readerCount() : 0;
+    }
+
+    /**
+     * 输入TS包
+     * @param packet TS包
+     * @param key 是否为关键帧第一个包
+     */
+    void onWrite(const TSPacket::Ptr &packet, bool key) override {
+        if (!_ring) {
+            createRing();
+        }
+        if (key) {
+            _have_video = true;
+        }
+        PacketCache<TSPacket, TSFlushPolicy>::inputPacket(true, packet, key);
+    }
+
+    /**
+     * 情况GOP缓存
+     */
+    void clearCache() override {
+        PacketCache<TSPacket, TSFlushPolicy>::clearCache();
+        _ring->clearCache();
+    }
+
+private:
+    void createRing(){
+        weak_ptr<TSMediaSource> weak_self = dynamic_pointer_cast<TSMediaSource>(shared_from_this());
+        _ring = std::make_shared<RingType>(_ring_size, [weak_self](int size) {
+            auto strong_self = weak_self.lock();
+            if (!strong_self) {
+                return;
+            }
+            strong_self->onReaderChanged(size);
+        });
+        onReaderChanged(0);
+        //注册媒体源
+        regist();
+    }
+
+    /**
+     * 合并写回调
+     * @param packet_list 合并写缓存列队
+     * @param key_pos 是否包含关键帧
+     */
+    void onFlush(std::shared_ptr<List<TSPacket::Ptr> > &packet_list, bool key_pos) override {
+        //如果不存在视频，那么就没有存在GOP缓存的意义，所以确保一直清空GOP缓存
+        _ring->write(packet_list, _have_video ? key_pos : true);
+    }
+
+private:
+    bool _have_video = false;
+    int _ring_size;
+    RingType::Ptr _ring;
+};
+
+
+}//namespace mediakit
+#endif //ZLMEDIAKIT_TSMEDIASOURCE_H

src/TS/TSMediaSourceMuxer.h

@@ -0,0 +1,84 @@
+/*
+ * Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
+ *
+ * This file is part of ZLMediaKit(https://github.com/xiongziliang/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_TSMEDIASOURCEMUXER_H
+#define ZLMEDIAKIT_TSMEDIASOURCEMUXER_H
+
+#include "TSMediaSource.h"
+#include "Record/TsMuxer.h"
+
+namespace mediakit {
+
+class TSMediaSourceMuxer : public TsMuxer, public MediaSourceEventInterceptor,
+                           public std::enable_shared_from_this<TSMediaSourceMuxer> {
+public:
+    using Ptr = std::shared_ptr<TSMediaSourceMuxer>;
+
+    TSMediaSourceMuxer(const string &vhost,
+                       const string &app,
+                       const string &stream_id) {
+        _media_src = std::make_shared<TSMediaSource>(vhost, app, stream_id);
+        _pool.setSize(256);
+    }
+
+    ~TSMediaSourceMuxer() override = default;
+
+    void setListener(const std::weak_ptr<MediaSourceEvent> &listener){
+        _listener = listener;
+        _media_src->setListener(shared_from_this());
+    }
+
+    int readerCount() const{
+        return _media_src->readerCount();
+    }
+
+    void onReaderChanged(MediaSource &sender, int size) override {
+        _enabled = size;
+        if (!size) {
+            _clear_cache = true;
+        }
+        MediaSourceEventInterceptor::onReaderChanged(sender, size);
+    }
+
+    void inputFrame(const Frame::Ptr &frame) override {
+        if (_clear_cache) {
+            _clear_cache = false;
+            _media_src->clearCache();
+        }
+        if (_enabled) {
+            TsMuxer::inputFrame(frame);
+        }
+    }
+
+    bool isEnabled() {
+        //缓存尚未清空时，还允许触发inputFrame函数，以便及时清空缓存
+        return _clear_cache ? true : _enabled;
+    }
+
+protected:
+    void onTs(const void *data, int len,uint32_t timestamp,bool is_idr_fast_packet) override{
+        if(!data || !len){
+            return;
+        }
+        TSPacket::Ptr packet = _pool.obtain();
+        packet->assign((char *) data, len);
+        packet->time_stamp = timestamp;
+        _media_src->onWrite(packet, is_idr_fast_packet);
+    }
+
+private:
+    bool _enabled = true;
+    bool _clear_cache = false;
+    TSMediaSource::PoolType _pool;
+    TSMediaSource::Ptr _media_src;
+};
+
+}//namespace mediakit
+#endif //ZLMEDIAKIT_TSMEDIASOURCEMUXER_H