ZLMediaKit/src/Rtsp/RtspMediaSourceMuxer.h

87 lines
2.7 KiB
C++
Raw Normal View History

2018-10-25 10:00:17 +08:00
/*
2020-04-04 20:30:09 +08:00
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
2018-10-25 10:00:17 +08:00
*
* This file is part of ZLMediaKit(https://github.com/xia-chu/ZLMediaKit).
2018-10-25 10:00:17 +08:00
*
2020-04-04 20:30:09 +08:00
* 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.
2018-10-25 10:00:17 +08:00
*/
2018-10-24 17:17:55 +08:00
2018-10-25 16:46:00 +08:00
#ifndef ZLMEDIAKIT_RTSPMEDIASOURCEMUXER_H
#define ZLMEDIAKIT_RTSPMEDIASOURCEMUXER_H
2018-10-24 17:17:55 +08:00
2019-06-28 17:30:13 +08:00
#include "RtspMuxer.h"
2018-10-24 17:17:55 +08:00
#include "Rtsp/RtspMediaSource.h"
namespace mediakit {
2022-10-16 19:49:56 +08:00
class RtspMediaSourceMuxer final : public RtspMuxer, public MediaSourceEventInterceptor,
public std::enable_shared_from_this<RtspMediaSourceMuxer> {
2018-10-24 17:17:55 +08:00
public:
2022-12-02 14:43:06 +08:00
using Ptr = std::shared_ptr<RtspMediaSourceMuxer>;
2018-10-25 16:46:00 +08:00
RtspMediaSourceMuxer(const std::string &vhost,
const std::string &strApp,
const std::string &strId,
const ProtocolOption &option,
const TitleSdp::Ptr &title = nullptr) : RtspMuxer(title) {
_option = option;
2020-09-12 19:09:56 +08:00
_media_src = std::make_shared<RtspMediaSource>(vhost,strApp,strId);
getRtpRing()->setDelegate(_media_src);
2018-10-25 16:46:00 +08:00
}
2020-09-12 19:09:56 +08:00
2022-10-16 19:49:56 +08:00
~RtspMediaSourceMuxer() override { RtspMuxer::flush(); }
2018-10-25 16:46:00 +08:00
void setListener(const std::weak_ptr<MediaSourceEvent> &listener){
setDelegate(listener);
2020-09-12 19:20:18 +08:00
_media_src->setListener(shared_from_this());
2018-10-25 16:46:00 +08:00
}
2019-12-03 12:32:57 +08:00
2018-10-26 14:12:16 +08:00
int readerCount() const{
2020-09-12 19:09:56 +08:00
return _media_src->readerCount();
2018-10-26 14:12:16 +08:00
}
2019-12-03 12:32:57 +08:00
2018-10-26 14:19:13 +08:00
void setTimeStamp(uint32_t stamp){
2020-09-12 19:09:56 +08:00
_media_src->setTimeStamp(stamp);
2018-10-26 14:12:16 +08:00
}
2019-12-03 12:32:57 +08:00
void onAllTrackReady(){
2020-09-12 19:09:56 +08:00
_media_src->setSdp(getSdp());
}
void onReaderChanged(MediaSource &sender, int size) override {
_enabled = _option.rtsp_demand ? size : true;
if (!size && _option.rtsp_demand) {
2020-09-12 19:09:56 +08:00
_clear_cache = true;
}
MediaSourceEventInterceptor::onReaderChanged(sender, size);
}
bool inputFrame(const Frame::Ptr &frame) override {
if (_clear_cache && _option.rtsp_demand) {
2020-09-12 19:09:56 +08:00
_clear_cache = false;
_media_src->clearCache();
}
if (_enabled || !_option.rtsp_demand) {
return RtspMuxer::inputFrame(frame);
2020-09-12 19:09:56 +08:00
}
return false;
2020-09-12 19:09:56 +08:00
}
bool isEnabled() {
//缓存尚未清空时还允许触发inputFrame函数以便及时清空缓存
return _option.rtsp_demand ? (_clear_cache ? true : _enabled) : true;
2018-10-25 16:46:00 +08:00
}
2019-12-03 16:10:02 +08:00
2018-10-25 16:46:00 +08:00
private:
2020-09-12 19:09:56 +08:00
bool _enabled = true;
bool _clear_cache = false;
ProtocolOption _option;
2020-09-12 19:09:56 +08:00
RtspMediaSource::Ptr _media_src;
2018-10-24 17:17:55 +08:00
};
2018-10-25 16:46:00 +08:00
}//namespace mediakit
#endif //ZLMEDIAKIT_RTSPMEDIASOURCEMUXER_H