68 lines
2.3 KiB
C++
68 lines
2.3 KiB
C++
#include "OpusCodec.h"
|
||
#include "BoostLog.h"
|
||
|
||
namespace Opus {
|
||
|
||
bool Encoder::open(int32_t samplerate, int32_t channels, uint32_t sampleSize) {
|
||
int error = 0;
|
||
|
||
uint32_t maxFrameSize = samplerate / 1000 * 60 * sampleSize * channels; // opus最大60ms一帧编码
|
||
m_buffer.resize(maxFrameSize);
|
||
|
||
m_encoder = opus_encoder_create(samplerate, channels, OPUS_APPLICATION_AUDIO, &error);
|
||
if (error != OPUS_OK || m_encoder == nullptr) {
|
||
LOG(error) << "opus_encoder_create() failed, status: " << error;
|
||
return false;
|
||
}
|
||
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_VBR(0)); // 0:CBR, 1:VBR
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_VBR_CONSTRAINT(true));
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_BITRATE(96000));
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_COMPLEXITY(8)); // 8 0~10
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_SIGNAL(OPUS_SIGNAL_VOICE));
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_LSB_DEPTH(16)); // 每个采样16个bit，2个byte
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_DTX(0));
|
||
opus_encoder_ctl(m_encoder, OPUS_SET_INBAND_FEC(0));
|
||
|
||
return true;
|
||
}
|
||
|
||
void Encoder::close() {
|
||
if (m_encoder != nullptr) {
|
||
opus_encoder_destroy(m_encoder);
|
||
m_encoder = nullptr;
|
||
}
|
||
}
|
||
|
||
Encoder::~Encoder() {
|
||
close();
|
||
}
|
||
|
||
const Opus::Packet Encoder::encode(const int16_t *pcm, int32_t framesize) {
|
||
Packet ret;
|
||
if (m_encoder == nullptr) return ret;
|
||
ret.byteSize = opus_encode(m_encoder, pcm, framesize, m_buffer.data(), m_buffer.size());
|
||
ret.data = m_buffer.data();
|
||
LOG(info) << "framesize: " << framesize << ", record size: " << ret.byteSize;
|
||
return ret;
|
||
}
|
||
|
||
bool Decoder::open(int32_t samplerate, int32_t channels) {
|
||
|
||
int error = 0;
|
||
m_decoder = opus_decoder_create(samplerate, channels, &error);
|
||
if (error != OPUS_OK || m_decoder == NULL) {
|
||
LOG(error) << "opus_decoder_create() failed, status: " << error;
|
||
return false;
|
||
}
|
||
opus_decoder_ctl(m_decoder, OPUS_SET_LSB_DEPTH(16));
|
||
return true;
|
||
}
|
||
|
||
int32_t Decoder::decode(const uint8_t *data, int32_t dataSize, int16_t *pcm, int32_t pcmBuffer) {
|
||
int32_t decodedSize = opus_decode(m_decoder, data, dataSize, pcm, pcmBuffer, 0);
|
||
LOG(info) << "dataSize: " << dataSize << ", decodes frame size: " << decodedSize;
|
||
return decodedSize;
|
||
}
|
||
|
||
} // namespace Opus
|