mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-12-02 00:12:33 +08:00
137 lines
4.0 KiB
C++
137 lines
4.0 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 DEVICE_DEVICE_H_
|
||
#define DEVICE_DEVICE_H_
|
||
|
||
#include <memory>
|
||
#include <string>
|
||
#include <functional>
|
||
#include "Util/TimeTicker.h"
|
||
#include "Common/MultiMediaSourceMuxer.h"
|
||
|
||
namespace mediakit {
|
||
|
||
class H264Encoder;
|
||
class AACEncoder;
|
||
|
||
class VideoInfo {
|
||
public:
|
||
CodecId codecId = CodecH264;
|
||
int iWidth;
|
||
int iHeight;
|
||
float iFrameRate;
|
||
int iBitRate = 2 * 1024 * 1024;
|
||
};
|
||
|
||
class AudioInfo {
|
||
public:
|
||
CodecId codecId = CodecAAC;
|
||
int iChannel;
|
||
int iSampleBit;
|
||
int iSampleRate;
|
||
};
|
||
|
||
/**
|
||
* MultiMediaSourceMuxer类的包装，方便初学者使用
|
||
*/
|
||
class DevChannel : public MultiMediaSourceMuxer{
|
||
public:
|
||
using Ptr = std::shared_ptr<DevChannel>;
|
||
|
||
//fDuration<=0为直播，否则为点播
|
||
DevChannel(const MediaTuple& tuple, float duration = 0, const ProtocolOption &option = ProtocolOption())
|
||
: MultiMediaSourceMuxer(tuple, duration, option) {}
|
||
|
||
/**
|
||
* 初始化视频Track
|
||
* 相当于MultiMediaSourceMuxer::addTrack(VideoTrack::Ptr );
|
||
* @param info 视频相关信息
|
||
*/
|
||
bool initVideo(const VideoInfo &info);
|
||
|
||
/**
|
||
* 初始化音频Track
|
||
* 相当于MultiMediaSourceMuxer::addTrack(AudioTrack::Ptr );
|
||
* @param info 音频相关信息
|
||
*/
|
||
bool initAudio(const AudioInfo &info);
|
||
|
||
/**
|
||
* 输入264帧
|
||
* @param data 264单帧数据指针
|
||
* @param len 数据指针长度
|
||
* @param dts 解码时间戳，单位毫秒；等于0时内部会自动生成时间戳
|
||
* @param pts 播放时间戳，单位毫秒；等于0时内部会赋值为dts
|
||
*/
|
||
bool inputH264(const char *data, int len, uint64_t dts, uint64_t pts = 0);
|
||
|
||
/**
|
||
* 输入265帧
|
||
* @param data 265单帧数据指针
|
||
* @param len 数据指针长度
|
||
* @param dts 解码时间戳，单位毫秒；等于0时内部会自动生成时间戳
|
||
* @param pts 播放时间戳，单位毫秒；等于0时内部会赋值为dts
|
||
*/
|
||
bool inputH265(const char *data, int len, uint64_t dts, uint64_t pts = 0);
|
||
|
||
/**
|
||
* 输入aac帧
|
||
* @param data_without_adts 不带adts头的aac帧
|
||
* @param len 帧数据长度
|
||
* @param dts 时间戳，单位毫秒
|
||
* @param adts_header adts头
|
||
*/
|
||
bool inputAAC(const char *data_without_adts, int len, uint64_t dts, const char *adts_header);
|
||
|
||
/**
|
||
* 输入OPUS/G711音频帧
|
||
* @param data 音频帧
|
||
* @param len 帧数据长度
|
||
* @param dts 时间戳，单位毫秒
|
||
*/
|
||
bool inputAudio(const char *data, int len, uint64_t dts);
|
||
|
||
/**
|
||
* 输入yuv420p视频帧，内部会完成编码并调用inputH264方法
|
||
* @param yuv yuv420p数据指针
|
||
* @param linesize yuv420p数据linesize
|
||
* @param cts 采集时间戳，单位毫秒
|
||
*/
|
||
bool inputYUV(char *yuv[3], int linesize[3], uint64_t cts);
|
||
|
||
/**
|
||
* 输入pcm数据，内部会完成编码并调用inputAAC方法
|
||
* @param data pcm数据指针，int16整形
|
||
* @param len pcm数据长度
|
||
* @param cts 采集时间戳，单位毫秒
|
||
*/
|
||
bool inputPCM(char *data, int len, uint64_t cts);
|
||
|
||
//// 重载基类方法，确保线程安全 ////
|
||
bool inputFrame(const Frame::Ptr &frame) override;
|
||
bool addTrack(const Track::Ptr & track) override;
|
||
void addTrackCompleted() override;
|
||
|
||
private:
|
||
MediaOriginType getOriginType(MediaSource &sender) const override;
|
||
|
||
private:
|
||
std::shared_ptr<H264Encoder> _pH264Enc;
|
||
std::shared_ptr<AACEncoder> _pAacEnc;
|
||
std::shared_ptr<VideoInfo> _video;
|
||
std::shared_ptr<AudioInfo> _audio;
|
||
toolkit::SmoothTicker _aTicker[2];
|
||
};
|
||
|
||
} /* namespace mediakit */
|
||
|
||
#endif /* DEVICE_DEVICE_H_ */
|