mirror of
https://github.com/ZLMediaKit/ZLMediaKit.git
synced 2024-11-25 04:08:57 +08:00
209 lines
6.9 KiB
C++
209 lines
6.9 KiB
C++
/*
|
||
* Copyright (c) 2016 The ZLMediaKit project authors. All Rights Reserved.
|
||
*
|
||
* This file is part of ZLMediaKit(https://github.com/xia-chu/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.
|
||
*/
|
||
|
||
#include "Device.h"
|
||
#include "Util/logger.h"
|
||
#include "Util/base64.h"
|
||
#include "Extension/AAC.h"
|
||
#include "Extension/Opus.h"
|
||
#include "Extension/G711.h"
|
||
#include "Extension/H264.h"
|
||
#include "Extension/H265.h"
|
||
#ifdef ENABLE_FAAC
|
||
#include "Codec/AACEncoder.h"
|
||
#endif //ENABLE_FAAC
|
||
|
||
#ifdef ENABLE_X264
|
||
#include "Codec/H264Encoder.h"
|
||
#endif //ENABLE_X264
|
||
using namespace toolkit;
|
||
using namespace std;
|
||
|
||
namespace mediakit {
|
||
|
||
bool DevChannel::inputYUV(char *yuv[3], int linesize[3], uint64_t cts) {
|
||
#ifdef ENABLE_X264
|
||
//TimeTicker1(50);
|
||
if (!_pH264Enc) {
|
||
_pH264Enc.reset(new H264Encoder());
|
||
if (!_pH264Enc->init(_video->iWidth, _video->iHeight, _video->iFrameRate, _video->iBitRate)) {
|
||
_pH264Enc.reset();
|
||
WarnL << "H264Encoder init failed!";
|
||
}
|
||
}
|
||
if (_pH264Enc) {
|
||
H264Encoder::H264Frame *out_frames;
|
||
int frames = _pH264Enc->inputData(yuv, linesize, cts, &out_frames);
|
||
bool ret = false;
|
||
for (int i = 0; i < frames; i++) {
|
||
ret = inputH264((char *) out_frames[i].pucData, out_frames[i].iLength, cts) ? true : ret;
|
||
}
|
||
return ret;
|
||
}
|
||
return false;
|
||
#else
|
||
WarnL << "h264编码未启用,该方法无效,编译时请打开ENABLE_X264选项";
|
||
return false;
|
||
#endif //ENABLE_X264
|
||
}
|
||
|
||
bool DevChannel::inputPCM(char* pcData, int iDataLen, uint64_t uiStamp) {
|
||
#ifdef ENABLE_FAAC
|
||
if (!_pAacEnc) {
|
||
_pAacEnc.reset(new AACEncoder());
|
||
if (!_pAacEnc->init(_audio->iSampleRate, _audio->iChannel, _audio->iSampleBit)) {
|
||
_pAacEnc.reset();
|
||
WarnL << "AACEncoder init failed!";
|
||
}
|
||
}
|
||
if (_pAacEnc) {
|
||
unsigned char *pucOut;
|
||
int iRet = _pAacEnc->inputData(pcData, iDataLen, &pucOut);
|
||
if (iRet > 7) {
|
||
return inputAAC((char *) pucOut + 7, iRet - 7, uiStamp, (char *)pucOut);
|
||
}
|
||
}
|
||
return false;
|
||
#else
|
||
WarnL << "aac编码未启用,该方法无效,编译时请打开ENABLE_FAAC选项";
|
||
return false;
|
||
#endif //ENABLE_FAAC
|
||
}
|
||
|
||
bool DevChannel::inputH264(const char *data, int len, uint64_t dts, uint64_t pts) {
|
||
if (dts == 0) {
|
||
dts = _aTicker[0].elapsedTime();
|
||
}
|
||
if (pts == 0) {
|
||
pts = dts;
|
||
}
|
||
|
||
//由于rtmp/hls/mp4需要缓存时间戳相同的帧，
|
||
//所以使用FrameNoCacheAble类型的帧反而会在转换成FrameCacheAble时多次内存拷贝
|
||
//在此处只拷贝一次，性能开销更低
|
||
auto frame = FrameImp::create<H264Frame>();
|
||
frame->_dts = dts;
|
||
frame->_pts = pts;
|
||
frame->_buffer.assign(data, len);
|
||
frame->_prefix_size = prefixSize(data,len);
|
||
return inputFrame(frame);
|
||
}
|
||
|
||
bool DevChannel::inputH265(const char *data, int len, uint64_t dts, uint64_t pts) {
|
||
if (dts == 0) {
|
||
dts = _aTicker[0].elapsedTime();
|
||
}
|
||
if (pts == 0) {
|
||
pts = dts;
|
||
}
|
||
|
||
//由于rtmp/hls/mp4需要缓存时间戳相同的帧，
|
||
//所以使用FrameNoCacheAble类型的帧反而会在转换成FrameCacheAble时多次内存拷贝
|
||
//在此处只拷贝一次，性能开销更低
|
||
auto frame = FrameImp::create<H265Frame>();
|
||
frame->_dts = dts;
|
||
frame->_pts = pts;
|
||
frame->_buffer.assign(data, len);
|
||
frame->_prefix_size = prefixSize(data,len);
|
||
return inputFrame(frame);
|
||
}
|
||
|
||
class FrameAutoDelete : public FrameFromPtr{
|
||
public:
|
||
template <typename ... ARGS>
|
||
FrameAutoDelete(ARGS && ...args) : FrameFromPtr(std::forward<ARGS>(args)...){}
|
||
|
||
~FrameAutoDelete() override {
|
||
delete [] _ptr;
|
||
};
|
||
|
||
bool cacheAble() const override {
|
||
return true;
|
||
}
|
||
};
|
||
|
||
bool DevChannel::inputAAC(const char *data_without_adts, int len, uint64_t dts, const char *adts_header){
|
||
if (dts == 0) {
|
||
dts = _aTicker[1].elapsedTime();
|
||
}
|
||
|
||
if (!adts_header) {
|
||
//没有adts头
|
||
return inputFrame(std::make_shared<FrameFromPtr>(_audio->codecId, (char *) data_without_adts, len, dts, 0, 0));
|
||
}
|
||
|
||
if (adts_header + ADTS_HEADER_LEN == data_without_adts) {
|
||
//adts头和帧在一起
|
||
return inputFrame(std::make_shared<FrameFromPtr>(_audio->codecId, (char *) data_without_adts - ADTS_HEADER_LEN, len + ADTS_HEADER_LEN, dts, 0, ADTS_HEADER_LEN));
|
||
}
|
||
|
||
//adts头和帧不在一起
|
||
char *data_with_adts = new char[len + ADTS_HEADER_LEN];
|
||
memcpy(data_with_adts, adts_header, ADTS_HEADER_LEN);
|
||
memcpy(data_with_adts + ADTS_HEADER_LEN, data_without_adts, len);
|
||
return inputFrame(std::make_shared<FrameAutoDelete>(_audio->codecId, data_with_adts, len + ADTS_HEADER_LEN, dts, 0, ADTS_HEADER_LEN));
|
||
|
||
}
|
||
|
||
bool DevChannel::inputAudio(const char *data, int len, uint64_t dts){
|
||
if (dts == 0) {
|
||
dts = _aTicker[1].elapsedTime();
|
||
}
|
||
return inputFrame(std::make_shared<FrameFromPtr>(_audio->codecId, (char *) data, len, dts, 0));
|
||
}
|
||
|
||
bool DevChannel::initVideo(const VideoInfo &info) {
|
||
_video = std::make_shared<VideoInfo>(info);
|
||
switch (info.codecId){
|
||
case CodecH265 : return addTrack(std::make_shared<H265Track>());
|
||
case CodecH264 : return addTrack(std::make_shared<H264Track>());
|
||
default: WarnL << "不支持该类型的视频编码类型:" << info.codecId; return false;
|
||
}
|
||
}
|
||
|
||
bool DevChannel::initAudio(const AudioInfo &info) {
|
||
_audio = std::make_shared<AudioInfo>(info);
|
||
switch (info.codecId) {
|
||
case CodecAAC : return addTrack(std::make_shared<AACTrack>());
|
||
case CodecG711A :
|
||
case CodecG711U : return addTrack(std::make_shared<G711Track>(info.codecId, info.iSampleRate, info.iChannel, info.iSampleBit));
|
||
case CodecOpus : return addTrack(std::make_shared<OpusTrack>());
|
||
default: WarnL << "不支持该类型的音频编码类型:" << info.codecId; return false;
|
||
}
|
||
}
|
||
|
||
bool DevChannel::inputFrame(const Frame::Ptr &frame) {
|
||
auto cached_frame = Frame::getCacheAbleFrame(frame);
|
||
weak_ptr<MultiMediaSourceMuxer> weak_self = shared_from_this();
|
||
getOwnerPoller(MediaSource::NullMediaSource())->async([weak_self, cached_frame]() {
|
||
if (auto strong_self = weak_self.lock()) {
|
||
strong_self->MultiMediaSourceMuxer::inputFrame(cached_frame);
|
||
}
|
||
});
|
||
return true;
|
||
}
|
||
|
||
bool DevChannel::addTrack(const Track::Ptr &track) {
|
||
bool ret;
|
||
getOwnerPoller(MediaSource::NullMediaSource())->sync([&]() { ret = MultiMediaSourceMuxer::addTrack(track); });
|
||
return ret;
|
||
}
|
||
|
||
void DevChannel::addTrackCompleted() {
|
||
getOwnerPoller(MediaSource::NullMediaSource())->sync([&]() { MultiMediaSourceMuxer::addTrackCompleted(); });
|
||
}
|
||
|
||
MediaOriginType DevChannel::getOriginType(MediaSource &sender) const {
|
||
return MediaOriginType::device_chn;
|
||
}
|
||
|
||
} /* namespace mediakit */
|
||
|