90 lines
2.8 KiB
C++
90 lines
2.8 KiB
C++
/*
|
||
* Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
|
||
*
|
||
* Use of this source code is governed by MIT-like 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/MPEG.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class TSMediaSourceMuxer final : public MpegMuxer, public MediaSourceEventInterceptor,
|
||
public std::enable_shared_from_this<TSMediaSourceMuxer> {
|
||
public:
|
||
using Ptr = std::shared_ptr<TSMediaSourceMuxer>;
|
||
|
||
TSMediaSourceMuxer(const MediaTuple& tuple, const ProtocolOption &option) : MpegMuxer(false) {
|
||
_option = option;
|
||
_media_src = std::make_shared<TSMediaSource>(tuple);
|
||
}
|
||
|
||
~TSMediaSourceMuxer() override {
|
||
try {
|
||
MpegMuxer::flush();
|
||
} catch (std::exception &ex) {
|
||
WarnL << ex.what();
|
||
}
|
||
};
|
||
|
||
void setListener(const std::weak_ptr<MediaSourceEvent> &listener){
|
||
setDelegate(listener);
|
||
_media_src->setListener(shared_from_this());
|
||
}
|
||
|
||
int readerCount() const{
|
||
return _media_src->readerCount();
|
||
}
|
||
|
||
void onReaderChanged(MediaSource &sender, int size) override {
|
||
_enabled = _option.ts_demand ? size : true;
|
||
if (!size && _option.ts_demand) {
|
||
_clear_cache = true;
|
||
}
|
||
MediaSourceEventInterceptor::onReaderChanged(sender, size);
|
||
}
|
||
|
||
bool inputFrame(const Frame::Ptr &frame) override {
|
||
if (_clear_cache && _option.ts_demand) {
|
||
_clear_cache = false;
|
||
_media_src->clearCache();
|
||
}
|
||
if (_enabled || !_option.ts_demand) {
|
||
return MpegMuxer::inputFrame(frame);
|
||
}
|
||
return false;
|
||
}
|
||
|
||
bool isEnabled() {
|
||
// 缓存尚未清空时，还允许触发inputFrame函数，以便及时清空缓存 [AUTO-TRANSLATED:7cfd4d49]
|
||
// Allow the inputFrame function to be triggered even when the cache is not yet cleared, so that the cache can be cleared in time.
|
||
return _option.ts_demand ? (_clear_cache ? true : _enabled) : true;
|
||
}
|
||
|
||
protected:
|
||
void onWrite(std::shared_ptr<toolkit::Buffer> buffer, uint64_t timestamp, bool key_pos) override {
|
||
if (!buffer) {
|
||
return;
|
||
}
|
||
auto packet = std::make_shared<TSPacket>(std::move(buffer));
|
||
packet->time_stamp = timestamp;
|
||
_media_src->onWrite(std::move(packet), key_pos);
|
||
}
|
||
|
||
private:
|
||
bool _enabled = true;
|
||
bool _clear_cache = false;
|
||
ProtocolOption _option;
|
||
TSMediaSource::Ptr _media_src;
|
||
};
|
||
|
||
}//namespace mediakit
|
||
#endif //ZLMEDIAKIT_TSMEDIASOURCEMUXER_H
|