87 lines
2.5 KiB
C++
87 lines
2.5 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.
|
|
*/
|
|
|
|
#include "Opus.h"
|
|
#include "Extension/Factory.h"
|
|
#include "Extension/CommonRtp.h"
|
|
#include "OpusRtmp.h"
|
|
#include "opus-head.h"
|
|
using namespace std;
|
|
using namespace toolkit;
|
|
|
|
namespace mediakit {
|
|
|
|
void OpusTrack::setExtraData(const uint8_t *data, size_t size) {
|
|
opus_head_t header;
|
|
if (opus_head_load(data, size, &header) > 0) {
|
|
// Successfully parsed Opus header
|
|
_sample_rate = header.input_sample_rate;
|
|
_channels = header.channels;
|
|
}
|
|
}
|
|
|
|
Buffer::Ptr OpusTrack::getExtraData() const {
|
|
struct opus_head_t opus {};
|
|
opus.version = 1;
|
|
opus.channels = getAudioChannel();
|
|
opus.input_sample_rate = getAudioSampleRate();
|
|
// opus.pre_skip = 120;
|
|
opus.channel_mapping_family = 0;
|
|
auto ret = BufferRaw::create(29);
|
|
ret->setSize(opus_head_save(&opus, (uint8_t *)ret->data(), ret->getCapacity()));
|
|
return ret;
|
|
}
|
|
|
|
namespace {
|
|
|
|
CodecId getCodec() {
|
|
return CodecOpus;
|
|
}
|
|
|
|
Track::Ptr getTrackByCodecId(int sample_rate, int channels, int sample_bit) {
|
|
return std::make_shared<OpusTrack>();
|
|
}
|
|
|
|
Track::Ptr getTrackBySdp(const SdpTrack::Ptr &track) {
|
|
return std::make_shared<OpusTrack>();
|
|
}
|
|
|
|
RtpCodec::Ptr getRtpEncoderByCodecId(uint8_t pt) {
|
|
return std::make_shared<CommonRtpEncoder>();
|
|
}
|
|
|
|
RtpCodec::Ptr getRtpDecoderByCodecId() {
|
|
return std::make_shared<CommonRtpDecoder>(CodecOpus);
|
|
}
|
|
|
|
RtmpCodec::Ptr getRtmpEncoderByTrack(const Track::Ptr &track) {
|
|
return std::make_shared<OpusRtmpEncoder>(track);
|
|
}
|
|
|
|
RtmpCodec::Ptr getRtmpDecoderByTrack(const Track::Ptr &track) {
|
|
return std::make_shared<OpusRtmpDecoder>(track);
|
|
}
|
|
|
|
Frame::Ptr getFrameFromPtr(const char *data, size_t bytes, uint64_t dts, uint64_t pts) {
|
|
return std::make_shared<FrameFromPtr>(CodecOpus, (char *)data, bytes, dts, pts);
|
|
}
|
|
|
|
} // namespace
|
|
|
|
CodecPlugin opus_plugin = { getCodec,
|
|
getTrackByCodecId,
|
|
getTrackBySdp,
|
|
getRtpEncoderByCodecId,
|
|
getRtpDecoderByCodecId,
|
|
getRtmpEncoderByTrack,
|
|
getRtmpDecoderByTrack,
|
|
getFrameFromPtr };
|
|
|
|
}//namespace mediakit
|