| Index: webrtc/modules/audio_coding/codecs/cng/webrtc_cng.cc
|
| diff --git a/webrtc/modules/audio_coding/codecs/cng/webrtc_cng.cc b/webrtc/modules/audio_coding/codecs/cng/webrtc_cng.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..b4da260dba2eabc4118cebdda43fdf13233a06f9
|
| --- /dev/null
|
| +++ b/webrtc/modules/audio_coding/codecs/cng/webrtc_cng.cc
|
| @@ -0,0 +1,442 @@
|
| +/*
|
| + * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#include "webrtc/modules/audio_coding/codecs/cng/webrtc_cng.h"
|
| +
|
| +#include <algorithm>
|
| +
|
| +#include "webrtc/common_audio/signal_processing/include/signal_processing_library.h"
|
| +
|
| +namespace webrtc {
|
| +
|
| +namespace {
|
| +
|
| +const size_t kCngMaxOutsizeOrder = 640;
|
| +
|
| +// TODO(ossu): Rename the left-over WebRtcCng according to style guide.
|
| +void WebRtcCng_K2a16(int16_t* k, int useOrder, int16_t* a);
|
| +
|
| +const int32_t WebRtcCng_kDbov[94] = {
|
| + 1081109975, 858756178, 682134279, 541838517, 430397633, 341876992,
|
| + 271562548, 215709799, 171344384, 136103682, 108110997, 85875618,
|
| + 68213428, 54183852, 43039763, 34187699, 27156255, 21570980,
|
| + 17134438, 13610368, 10811100, 8587562, 6821343, 5418385,
|
| + 4303976, 3418770, 2715625, 2157098, 1713444, 1361037,
|
| + 1081110, 858756, 682134, 541839, 430398, 341877,
|
| + 271563, 215710, 171344, 136104, 108111, 85876,
|
| + 68213, 54184, 43040, 34188, 27156, 21571,
|
| + 17134, 13610, 10811, 8588, 6821, 5418,
|
| + 4304, 3419, 2716, 2157, 1713, 1361,
|
| + 1081, 859, 682, 542, 430, 342,
|
| + 272, 216, 171, 136, 108, 86,
|
| + 68, 54, 43, 34, 27, 22,
|
| + 17, 14, 11, 9, 7, 5,
|
| + 4, 3, 3, 2, 2, 1,
|
| + 1, 1, 1, 1
|
| +};
|
| +
|
| +const int16_t WebRtcCng_kCorrWindow[WEBRTC_CNG_MAX_LPC_ORDER] = {
|
| + 32702, 32636, 32570, 32505, 32439, 32374,
|
| + 32309, 32244, 32179, 32114, 32049, 31985
|
| +};
|
| +
|
| +} // namespace
|
| +
|
| +ComfortNoiseDecoder::ComfortNoiseDecoder() {
|
| + /* Needed to get the right function pointers in SPLIB. */
|
| + WebRtcSpl_Init();
|
| + Reset();
|
| +}
|
| +
|
| +void ComfortNoiseDecoder::Reset() {
|
| + dec_seed_ = 7777; /* For debugging only. */
|
| + dec_target_energy_ = 0;
|
| + dec_used_energy_ = 0;
|
| + for (auto& c : dec_target_reflCoefs_)
|
| + c = 0;
|
| + for (auto& c : dec_used_reflCoefs_)
|
| + c = 0;
|
| + for (auto& c : dec_filtstate_)
|
| + c = 0;
|
| + for (auto& c : dec_filtstateLow_)
|
| + c = 0;
|
| + dec_order_ = 5;
|
| + dec_target_scale_factor_ = 0;
|
| + dec_used_scale_factor_ = 0;
|
| +}
|
| +
|
| +void ComfortNoiseDecoder::UpdateSid(rtc::ArrayView<const uint8_t> sid) {
|
| + int16_t refCs[WEBRTC_CNG_MAX_LPC_ORDER];
|
| + int32_t targetEnergy;
|
| + size_t length = sid.size();
|
| + /* Throw away reflection coefficients of higher order than we can handle. */
|
| + if (length > (WEBRTC_CNG_MAX_LPC_ORDER + 1))
|
| + length = WEBRTC_CNG_MAX_LPC_ORDER + 1;
|
| +
|
| + dec_order_ = static_cast<uint16_t>(length - 1);
|
| +
|
| + uint8_t sid0 = std::min<uint8_t>(sid[0], 93);
|
| + targetEnergy = WebRtcCng_kDbov[sid0];
|
| + /* Take down target energy to 75%. */
|
| + targetEnergy = targetEnergy >> 1;
|
| + targetEnergy += targetEnergy >> 2;
|
| +
|
| + dec_target_energy_ = targetEnergy;
|
| +
|
| + /* Reconstruct coeffs with tweak for WebRtc implementation of RFC3389. */
|
| + if (dec_order_ == WEBRTC_CNG_MAX_LPC_ORDER) {
|
| + for (size_t i = 0; i < (dec_order_); i++) {
|
| + refCs[i] = sid[i + 1] << 8; /* Q7 to Q15*/
|
| + dec_target_reflCoefs_[i] = refCs[i];
|
| + }
|
| + } else {
|
| + for (size_t i = 0; i < (dec_order_); i++) {
|
| + refCs[i] = (sid[i + 1] - 127) << 8; /* Q7 to Q15. */
|
| + dec_target_reflCoefs_[i] = refCs[i];
|
| + }
|
| + }
|
| +
|
| + for (size_t i = (dec_order_); i < WEBRTC_CNG_MAX_LPC_ORDER; i++) {
|
| + refCs[i] = 0;
|
| + dec_target_reflCoefs_[i] = refCs[i];
|
| + }
|
| +}
|
| +
|
| +bool ComfortNoiseDecoder::Generate(rtc::ArrayView<int16_t> out_data,
|
| + bool new_period) {
|
| + int16_t excitation[kCngMaxOutsizeOrder];
|
| + int16_t low[kCngMaxOutsizeOrder];
|
| + int16_t lpPoly[WEBRTC_CNG_MAX_LPC_ORDER + 1];
|
| + int16_t ReflBetaStd = 26214; /* 0.8 in q15. */
|
| + int16_t ReflBetaCompStd = 6553; /* 0.2 in q15. */
|
| + int16_t ReflBetaNewP = 19661; /* 0.6 in q15. */
|
| + int16_t ReflBetaCompNewP = 13107; /* 0.4 in q15. */
|
| + int16_t Beta, BetaC, tmp1, tmp2, tmp3;
|
| + int32_t targetEnergy;
|
| + int16_t En;
|
| + int16_t temp16;
|
| + const size_t num_samples = out_data.size();
|
| +
|
| + if (num_samples > kCngMaxOutsizeOrder) {
|
| + return false;
|
| + }
|
| +
|
| + if (new_period) {
|
| + dec_used_scale_factor_ = dec_target_scale_factor_;
|
| + Beta = ReflBetaNewP;
|
| + BetaC = ReflBetaCompNewP;
|
| + } else {
|
| + Beta = ReflBetaStd;
|
| + BetaC = ReflBetaCompStd;
|
| + }
|
| +
|
| + /* Here we use a 0.5 weighting, should possibly be modified to 0.6. */
|
| + tmp1 = dec_used_scale_factor_ << 2; /* Q13->Q15 */
|
| + tmp2 = dec_target_scale_factor_ << 2; /* Q13->Q15 */
|
| + tmp3 = (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(tmp1, Beta, 15);
|
| + tmp3 += (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(tmp2, BetaC, 15);
|
| + dec_used_scale_factor_ = tmp3 >> 2; /* Q15->Q13 */
|
| +
|
| + dec_used_energy_ = dec_used_energy_ >> 1;
|
| + dec_used_energy_ += dec_target_energy_ >> 1;
|
| +
|
| + /* Do the same for the reflection coeffs. */
|
| + for (size_t i = 0; i < WEBRTC_CNG_MAX_LPC_ORDER; i++) {
|
| + dec_used_reflCoefs_[i] = (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(
|
| + dec_used_reflCoefs_[i], Beta, 15);
|
| + dec_used_reflCoefs_[i] += (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(
|
| + dec_target_reflCoefs_[i], BetaC, 15);
|
| + }
|
| +
|
| + /* Compute the polynomial coefficients. */
|
| + WebRtcCng_K2a16(dec_used_reflCoefs_, WEBRTC_CNG_MAX_LPC_ORDER, lpPoly);
|
| +
|
| +
|
| + targetEnergy = dec_used_energy_;
|
| +
|
| + /* Calculate scaling factor based on filter energy. */
|
| + En = 8192; /* 1.0 in Q13. */
|
| + for (size_t i = 0; i < (WEBRTC_CNG_MAX_LPC_ORDER); i++) {
|
| + /* Floating point value for reference.
|
| + E *= 1.0 - (dec_used_reflCoefs_[i] / 32768.0) *
|
| + (dec_used_reflCoefs_[i] / 32768.0);
|
| + */
|
| +
|
| + /* Same in fixed point. */
|
| + /* K(i).^2 in Q15. */
|
| + temp16 = (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(
|
| + dec_used_reflCoefs_[i], dec_used_reflCoefs_[i], 15);
|
| + /* 1 - K(i).^2 in Q15. */
|
| + temp16 = 0x7fff - temp16;
|
| + En = (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(En, temp16, 15);
|
| + }
|
| +
|
| + /* float scaling= sqrt(E * dec_target_energy_ / (1 << 24)); */
|
| +
|
| + /* Calculate sqrt(En * target_energy / excitation energy) */
|
| + targetEnergy = WebRtcSpl_Sqrt(dec_used_energy_);
|
| +
|
| + En = (int16_t) WebRtcSpl_Sqrt(En) << 6;
|
| + En = (En * 3) >> 1; /* 1.5 estimates sqrt(2). */
|
| + dec_used_scale_factor_ = (int16_t)((En * targetEnergy) >> 12);
|
| +
|
| + /* Generate excitation. */
|
| + /* Excitation energy per sample is 2.^24 - Q13 N(0,1). */
|
| + for (size_t i = 0; i < num_samples; i++) {
|
| + excitation[i] = WebRtcSpl_RandN(&dec_seed_) >> 1;
|
| + }
|
| +
|
| + /* Scale to correct energy. */
|
| + WebRtcSpl_ScaleVector(excitation, excitation, dec_used_scale_factor_,
|
| + num_samples, 13);
|
| +
|
| + /* |lpPoly| - Coefficients in Q12.
|
| + * |excitation| - Speech samples.
|
| + * |nst->dec_filtstate| - State preservation.
|
| + * |out_data| - Filtered speech samples. */
|
| + WebRtcSpl_FilterAR(lpPoly, WEBRTC_CNG_MAX_LPC_ORDER + 1, excitation,
|
| + num_samples, dec_filtstate_, WEBRTC_CNG_MAX_LPC_ORDER,
|
| + dec_filtstateLow_, WEBRTC_CNG_MAX_LPC_ORDER,
|
| + out_data.data(), low, num_samples);
|
| +
|
| + return true;
|
| +}
|
| +
|
| +ComfortNoiseEncoder::ComfortNoiseEncoder(int fs, int interval, int quality)
|
| + : enc_nrOfCoefs_(quality),
|
| + enc_sampfreq_(fs),
|
| + enc_interval_(interval),
|
| + enc_msSinceSid_(0),
|
| + enc_Energy_(0),
|
| + enc_reflCoefs_{0},
|
| + enc_corrVector_{0},
|
| + enc_seed_(7777) /* For debugging only. */ {
|
| + RTC_CHECK(quality <= WEBRTC_CNG_MAX_LPC_ORDER && quality > 0);
|
| + /* Needed to get the right function pointers in SPLIB. */
|
| + WebRtcSpl_Init();
|
| +}
|
| +
|
| +void ComfortNoiseEncoder::Reset(int fs, int interval, int quality) {
|
| + RTC_CHECK(quality <= WEBRTC_CNG_MAX_LPC_ORDER && quality > 0);
|
| + enc_nrOfCoefs_ = quality;
|
| + enc_sampfreq_ = fs;
|
| + enc_interval_ = interval;
|
| + enc_msSinceSid_ = 0;
|
| + enc_Energy_ = 0;
|
| + for (auto& c : enc_reflCoefs_)
|
| + c = 0;
|
| + for (auto& c : enc_corrVector_)
|
| + c = 0;
|
| + enc_seed_ = 7777; /* For debugging only. */
|
| +}
|
| +
|
| +size_t ComfortNoiseEncoder::Encode(rtc::ArrayView<const int16_t> speech,
|
| + bool force_sid,
|
| + rtc::Buffer* output) {
|
| + int16_t arCoefs[WEBRTC_CNG_MAX_LPC_ORDER + 1];
|
| + int32_t corrVector[WEBRTC_CNG_MAX_LPC_ORDER + 1];
|
| + int16_t refCs[WEBRTC_CNG_MAX_LPC_ORDER + 1];
|
| + int16_t hanningW[kCngMaxOutsizeOrder];
|
| + int16_t ReflBeta = 19661; /* 0.6 in q15. */
|
| + int16_t ReflBetaComp = 13107; /* 0.4 in q15. */
|
| + int32_t outEnergy;
|
| + int outShifts;
|
| + size_t i;
|
| + int stab;
|
| + int acorrScale;
|
| + size_t index;
|
| + size_t ind, factor;
|
| + int32_t* bptr;
|
| + int32_t blo, bhi;
|
| + int16_t negate;
|
| + const int16_t* aptr;
|
| + int16_t speechBuf[kCngMaxOutsizeOrder];
|
| +
|
| + const size_t num_samples = speech.size();
|
| + RTC_CHECK_LE(num_samples, static_cast<size_t>(kCngMaxOutsizeOrder));
|
| +
|
| + for (i = 0; i < num_samples; i++) {
|
| + speechBuf[i] = speech[i];
|
| + }
|
| +
|
| + factor = num_samples;
|
| +
|
| + /* Calculate energy and a coefficients. */
|
| + outEnergy = WebRtcSpl_Energy(speechBuf, num_samples, &outShifts);
|
| + while (outShifts > 0) {
|
| + /* We can only do 5 shifts without destroying accuracy in
|
| + * division factor. */
|
| + if (outShifts > 5) {
|
| + outEnergy <<= (outShifts - 5);
|
| + outShifts = 5;
|
| + } else {
|
| + factor /= 2;
|
| + outShifts--;
|
| + }
|
| + }
|
| + outEnergy = WebRtcSpl_DivW32W16(outEnergy, (int16_t)factor);
|
| +
|
| + if (outEnergy > 1) {
|
| + /* Create Hanning Window. */
|
| + WebRtcSpl_GetHanningWindow(hanningW, num_samples / 2);
|
| + for (i = 0; i < (num_samples / 2); i++)
|
| + hanningW[num_samples - i - 1] = hanningW[i];
|
| +
|
| + WebRtcSpl_ElementwiseVectorMult(speechBuf, hanningW, speechBuf, num_samples,
|
| + 14);
|
| +
|
| + WebRtcSpl_AutoCorrelation(speechBuf, num_samples, enc_nrOfCoefs_,
|
| + corrVector, &acorrScale);
|
| +
|
| + if (*corrVector == 0)
|
| + *corrVector = WEBRTC_SPL_WORD16_MAX;
|
| +
|
| + /* Adds the bandwidth expansion. */
|
| + aptr = WebRtcCng_kCorrWindow;
|
| + bptr = corrVector;
|
| +
|
| + /* (zzz) lpc16_1 = 17+1+820+2+2 = 842 (ordo2=700). */
|
| + for (ind = 0; ind < enc_nrOfCoefs_; ind++) {
|
| + /* The below code multiplies the 16 b corrWindow values (Q15) with
|
| + * the 32 b corrvector (Q0) and shifts the result down 15 steps. */
|
| + negate = *bptr < 0;
|
| + if (negate)
|
| + *bptr = -*bptr;
|
| +
|
| + blo = (int32_t) * aptr * (*bptr & 0xffff);
|
| + bhi = ((blo >> 16) & 0xffff)
|
| + + ((int32_t)(*aptr++) * ((*bptr >> 16) & 0xffff));
|
| + blo = (blo & 0xffff) | ((bhi & 0xffff) << 16);
|
| +
|
| + *bptr = (((bhi >> 16) & 0x7fff) << 17) | ((uint32_t) blo >> 15);
|
| + if (negate)
|
| + *bptr = -*bptr;
|
| + bptr++;
|
| + }
|
| + /* End of bandwidth expansion. */
|
| +
|
| + stab = WebRtcSpl_LevinsonDurbin(corrVector, arCoefs, refCs,
|
| + enc_nrOfCoefs_);
|
| +
|
| + if (!stab) {
|
| + /* Disregard from this frame */
|
| + return 0;
|
| + }
|
| +
|
| + } else {
|
| + for (i = 0; i < enc_nrOfCoefs_; i++)
|
| + refCs[i] = 0;
|
| + }
|
| +
|
| + if (force_sid) {
|
| + /* Read instantaneous values instead of averaged. */
|
| + for (i = 0; i < enc_nrOfCoefs_; i++)
|
| + enc_reflCoefs_[i] = refCs[i];
|
| + enc_Energy_ = outEnergy;
|
| + } else {
|
| + /* Average history with new values. */
|
| + for (i = 0; i < enc_nrOfCoefs_; i++) {
|
| + enc_reflCoefs_[i] = (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(
|
| + enc_reflCoefs_[i], ReflBeta, 15);
|
| + enc_reflCoefs_[i] +=
|
| + (int16_t) WEBRTC_SPL_MUL_16_16_RSFT(refCs[i], ReflBetaComp, 15);
|
| + }
|
| + enc_Energy_ =
|
| + (outEnergy >> 2) + (enc_Energy_ >> 1) + (enc_Energy_ >> 2);
|
| + }
|
| +
|
| + if (enc_Energy_ < 1) {
|
| + enc_Energy_ = 1;
|
| + }
|
| +
|
| + if ((enc_msSinceSid_ > (enc_interval_ - 1)) || force_sid) {
|
| + /* Search for best dbov value. */
|
| + index = 0;
|
| + for (i = 1; i < 93; i++) {
|
| + /* Always round downwards. */
|
| + if ((enc_Energy_ - WebRtcCng_kDbov[i]) > 0) {
|
| + index = i;
|
| + break;
|
| + }
|
| + }
|
| + if ((i == 93) && (index == 0))
|
| + index = 94;
|
| +
|
| + const size_t output_coefs = enc_nrOfCoefs_ + 1;
|
| + output->AppendData(output_coefs, [&] (rtc::ArrayView<uint8_t> output) {
|
| + output[0] = (uint8_t)index;
|
| +
|
| + /* Quantize coefficients with tweak for WebRtc implementation of
|
| + * RFC3389. */
|
| + if (enc_nrOfCoefs_ == WEBRTC_CNG_MAX_LPC_ORDER) {
|
| + for (i = 0; i < enc_nrOfCoefs_; i++) {
|
| + /* Q15 to Q7 with rounding. */
|
| + output[i + 1] = ((enc_reflCoefs_[i] + 128) >> 8);
|
| + }
|
| + } else {
|
| + for (i = 0; i < enc_nrOfCoefs_; i++) {
|
| + /* Q15 to Q7 with rounding. */
|
| + output[i + 1] = (127 + ((enc_reflCoefs_[i] + 128) >> 8));
|
| + }
|
| + }
|
| +
|
| + return output_coefs;
|
| + });
|
| +
|
| + enc_msSinceSid_ =
|
| + static_cast<int16_t>((1000 * num_samples) / enc_sampfreq_);
|
| + return output_coefs;
|
| + } else {
|
| + enc_msSinceSid_ +=
|
| + static_cast<int16_t>((1000 * num_samples) / enc_sampfreq_);
|
| + return 0;
|
| + }
|
| +}
|
| +
|
| +namespace {
|
| +/* Values in |k| are Q15, and |a| Q12. */
|
| +void WebRtcCng_K2a16(int16_t* k, int useOrder, int16_t* a) {
|
| + int16_t any[WEBRTC_SPL_MAX_LPC_ORDER + 1];
|
| + int16_t* aptr;
|
| + int16_t* aptr2;
|
| + int16_t* anyptr;
|
| + const int16_t* kptr;
|
| + int m, i;
|
| +
|
| + kptr = k;
|
| + *a = 4096; /* i.e., (Word16_MAX >> 3) + 1 */
|
| + *any = *a;
|
| + a[1] = (*k + 4) >> 3;
|
| + for (m = 1; m < useOrder; m++) {
|
| + kptr++;
|
| + aptr = a;
|
| + aptr++;
|
| + aptr2 = &a[m];
|
| + anyptr = any;
|
| + anyptr++;
|
| +
|
| + any[m + 1] = (*kptr + 4) >> 3;
|
| + for (i = 0; i < m; i++) {
|
| + *anyptr++ =
|
| + (*aptr++) +
|
| + (int16_t)((((int32_t)(*aptr2--) * (int32_t)*kptr) + 16384) >> 15);
|
| + }
|
| +
|
| + aptr = a;
|
| + anyptr = any;
|
| + for (i = 0; i < (m + 2); i++) {
|
| + *aptr++ = *anyptr++;
|
| + }
|
| + }
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| +} // namespace webrtc
|
|
|