132 lines
3.2 KiB
C++
132 lines
3.2 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 SRC_RTMP_RTMPTORTSPMEDIASOURCE_H_
|
|
#define SRC_RTMP_RTMPTORTSPMEDIASOURCE_H_
|
|
|
|
#include <mutex>
|
|
#include <string>
|
|
#include <memory>
|
|
#include <functional>
|
|
#include <unordered_map>
|
|
#include "amf.h"
|
|
#include "Rtmp.h"
|
|
#include "RtmpDemuxer.h"
|
|
#include "RtmpMediaSource.h"
|
|
#include "Common/MultiMediaSourceMuxer.h"
|
|
|
|
namespace mediakit {
|
|
|
|
class RtmpMediaSourceImp final : public RtmpMediaSource, private TrackListener, public MultiMediaSourceMuxer::Listener {
|
|
public:
|
|
using Ptr = std::shared_ptr<RtmpMediaSourceImp>;
|
|
|
|
/**
|
|
* 构造函数
|
|
* @param vhost 虚拟主机
|
|
* @param app 应用名
|
|
* @param id 流id
|
|
* @param ringSize 环形缓存大小
|
|
* Constructor
|
|
* @param vhost Virtual host
|
|
* @param app Application name
|
|
* @param id Stream id
|
|
* @param ringSize Ring buffer size
|
|
|
|
* [AUTO-TRANSLATED:7679d212]
|
|
*/
|
|
RtmpMediaSourceImp(const MediaTuple& tuple, int ringSize = RTMP_GOP_SIZE);
|
|
|
|
/**
|
|
* 设置metadata
|
|
* Set metadata
|
|
|
|
* [AUTO-TRANSLATED:e32234cf]
|
|
*/
|
|
void setMetaData(const AMFValue &metadata) override;
|
|
|
|
/**
|
|
* 输入rtmp并解析
|
|
* Input rtmp and parse
|
|
|
|
* [AUTO-TRANSLATED:de255b79]
|
|
*/
|
|
void onWrite(RtmpPacket::Ptr pkt, bool = true) override;
|
|
|
|
/**
|
|
* 获取观看总人数，包括(hls/rtsp/rtmp)
|
|
* Get total number of viewers, including (hls/rtsp/rtmp)
|
|
|
|
* [AUTO-TRANSLATED:19a26d5a]
|
|
*/
|
|
int totalReaderCount() override;
|
|
|
|
/**
|
|
* 设置协议转换
|
|
* Set protocol conversion
|
|
|
|
* [AUTO-TRANSLATED:f7206bf3]
|
|
*/
|
|
void setProtocolOption(const ProtocolOption &option);
|
|
|
|
const ProtocolOption &getProtocolOption() const {
|
|
return _option;
|
|
}
|
|
|
|
/**
|
|
* _demuxer触发的添加Track事件
|
|
* _demuxer triggered add Track event
|
|
|
|
* [AUTO-TRANSLATED:80dbcf16]
|
|
*/
|
|
bool addTrack(const Track::Ptr &track) override;
|
|
|
|
/**
|
|
* _demuxer触发的Track添加完毕事件
|
|
* _demuxer triggered Track add complete event
|
|
|
|
* [AUTO-TRANSLATED:939cb312]
|
|
*/
|
|
void addTrackCompleted() override;
|
|
|
|
void resetTracks() override;
|
|
|
|
/**
|
|
* _muxer触发的所有Track就绪的事件
|
|
* _muxer triggered all Track ready event
|
|
|
|
* [AUTO-TRANSLATED:1d34b7e0]
|
|
*/
|
|
void onAllTrackReady() override;
|
|
|
|
/**
|
|
* 设置事件监听器
|
|
* @param listener 监听器
|
|
* Set event listener
|
|
* @param listener Listener
|
|
|
|
|
|
* [AUTO-TRANSLATED:d829419b]
|
|
*/
|
|
void setListener(const std::weak_ptr<MediaSourceEvent> &listener) override;
|
|
|
|
private:
|
|
bool _all_track_ready = false;
|
|
bool _recreate_metadata = false;
|
|
ProtocolOption _option;
|
|
AMFValue _metadata;
|
|
RtmpDemuxer::Ptr _demuxer;
|
|
MultiMediaSourceMuxer::Ptr _muxer;
|
|
|
|
};
|
|
} /* namespace mediakit */
|
|
|
|
#endif /* SRC_RTMP_RTMPTORTSPMEDIASOURCE_H_ */
|