mirror of
https://github.com/mollyim/webrtc.git
synced 2025-05-13 22:00:47 +01:00
7120742701
Now that we have moved WebRTC from src/webrtc to src/, common_types.h and typedefs.h are triggering a cpplint error. The cpplint complaint is: Include the directory when naming .h files [build/include] [4] This CL disables the error but we have to remove these two headers from the root directory. NOPRESUBMIT=true Bug: webrtc:5876 Change-Id: I08e1b69aadcc4b28ab83bf25e3819d135d41d333 Reviewed-on: https://webrtc-review.googlesource.com/1577 Commit-Queue: Mirko Bonadei <mbonadei@webrtc.org> Reviewed-by: Henrik Kjellander <kjellander@google.com> Reviewed-by: Karl Wiberg <kwiberg@webrtc.org> Cr-Commit-Position: refs/heads/master@{#19859}
263 lines
10 KiB
C
263 lines
10 KiB
C
/*
|
|
* 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.
|
|
*/
|
|
|
|
#ifndef MODULES_AUDIO_PROCESSING_NS_MAIN_SOURCE_NSX_CORE_H_
|
|
#define MODULES_AUDIO_PROCESSING_NS_MAIN_SOURCE_NSX_CORE_H_
|
|
|
|
#ifdef NS_FILEDEBUG
|
|
#include <stdio.h>
|
|
#endif
|
|
|
|
#include "common_audio/signal_processing/include/signal_processing_library.h"
|
|
#include "modules/audio_processing/ns/nsx_defines.h"
|
|
#include "typedefs.h" // NOLINT(build/include)
|
|
|
|
typedef struct NoiseSuppressionFixedC_ {
|
|
uint32_t fs;
|
|
|
|
const int16_t* window;
|
|
int16_t analysisBuffer[ANAL_BLOCKL_MAX];
|
|
int16_t synthesisBuffer[ANAL_BLOCKL_MAX];
|
|
uint16_t noiseSupFilter[HALF_ANAL_BLOCKL];
|
|
uint16_t overdrive; /* Q8 */
|
|
uint16_t denoiseBound; /* Q14 */
|
|
const int16_t* factor2Table;
|
|
int16_t noiseEstLogQuantile[SIMULT* HALF_ANAL_BLOCKL];
|
|
int16_t noiseEstDensity[SIMULT* HALF_ANAL_BLOCKL];
|
|
int16_t noiseEstCounter[SIMULT];
|
|
int16_t noiseEstQuantile[HALF_ANAL_BLOCKL];
|
|
|
|
size_t anaLen;
|
|
size_t anaLen2;
|
|
size_t magnLen;
|
|
int aggrMode;
|
|
int stages;
|
|
int initFlag;
|
|
int gainMap;
|
|
|
|
int32_t maxLrt;
|
|
int32_t minLrt;
|
|
// Log LRT factor with time-smoothing in Q8.
|
|
int32_t logLrtTimeAvgW32[HALF_ANAL_BLOCKL];
|
|
int32_t featureLogLrt;
|
|
int32_t thresholdLogLrt;
|
|
int16_t weightLogLrt;
|
|
|
|
uint32_t featureSpecDiff;
|
|
uint32_t thresholdSpecDiff;
|
|
int16_t weightSpecDiff;
|
|
|
|
uint32_t featureSpecFlat;
|
|
uint32_t thresholdSpecFlat;
|
|
int16_t weightSpecFlat;
|
|
|
|
// Conservative estimate of noise spectrum.
|
|
int32_t avgMagnPause[HALF_ANAL_BLOCKL];
|
|
uint32_t magnEnergy;
|
|
uint32_t sumMagn;
|
|
uint32_t curAvgMagnEnergy;
|
|
uint32_t timeAvgMagnEnergy;
|
|
uint32_t timeAvgMagnEnergyTmp;
|
|
|
|
uint32_t whiteNoiseLevel; // Initial noise estimate.
|
|
// Initial magnitude spectrum estimate.
|
|
uint32_t initMagnEst[HALF_ANAL_BLOCKL];
|
|
// Pink noise parameters:
|
|
int32_t pinkNoiseNumerator; // Numerator.
|
|
int32_t pinkNoiseExp; // Power of freq.
|
|
int minNorm; // Smallest normalization factor.
|
|
int zeroInputSignal; // Zero input signal flag.
|
|
|
|
// Noise spectrum from previous frame.
|
|
uint32_t prevNoiseU32[HALF_ANAL_BLOCKL];
|
|
// Magnitude spectrum from previous frame.
|
|
uint16_t prevMagnU16[HALF_ANAL_BLOCKL];
|
|
// Prior speech/noise probability in Q14.
|
|
int16_t priorNonSpeechProb;
|
|
|
|
int blockIndex; // Frame index counter.
|
|
// Parameter for updating or estimating thresholds/weights for prior model.
|
|
int modelUpdate;
|
|
int cntThresUpdate;
|
|
|
|
// Histograms for parameter estimation.
|
|
int16_t histLrt[HIST_PAR_EST];
|
|
int16_t histSpecFlat[HIST_PAR_EST];
|
|
int16_t histSpecDiff[HIST_PAR_EST];
|
|
|
|
// Quantities for high band estimate.
|
|
int16_t dataBufHBFX[NUM_HIGH_BANDS_MAX][ANAL_BLOCKL_MAX];
|
|
|
|
int qNoise;
|
|
int prevQNoise;
|
|
int prevQMagn;
|
|
size_t blockLen10ms;
|
|
|
|
int16_t real[ANAL_BLOCKL_MAX];
|
|
int16_t imag[ANAL_BLOCKL_MAX];
|
|
int32_t energyIn;
|
|
int scaleEnergyIn;
|
|
int normData;
|
|
|
|
struct RealFFT* real_fft;
|
|
} NoiseSuppressionFixedC;
|
|
|
|
#ifdef __cplusplus
|
|
extern "C"
|
|
{
|
|
#endif
|
|
|
|
/****************************************************************************
|
|
* WebRtcNsx_InitCore(...)
|
|
*
|
|
* This function initializes a noise suppression instance
|
|
*
|
|
* Input:
|
|
* - inst : Instance that should be initialized
|
|
* - fs : Sampling frequency
|
|
*
|
|
* Output:
|
|
* - inst : Initialized instance
|
|
*
|
|
* Return value : 0 - Ok
|
|
* -1 - Error
|
|
*/
|
|
int32_t WebRtcNsx_InitCore(NoiseSuppressionFixedC* inst, uint32_t fs);
|
|
|
|
/****************************************************************************
|
|
* WebRtcNsx_set_policy_core(...)
|
|
*
|
|
* This changes the aggressiveness of the noise suppression method.
|
|
*
|
|
* Input:
|
|
* - inst : Instance that should be initialized
|
|
* - mode : 0: Mild (6 dB), 1: Medium (10 dB), 2: Aggressive (15 dB)
|
|
*
|
|
* Output:
|
|
* - inst : Initialized instance
|
|
*
|
|
* Return value : 0 - Ok
|
|
* -1 - Error
|
|
*/
|
|
int WebRtcNsx_set_policy_core(NoiseSuppressionFixedC* inst, int mode);
|
|
|
|
/****************************************************************************
|
|
* WebRtcNsx_ProcessCore
|
|
*
|
|
* Do noise suppression.
|
|
*
|
|
* Input:
|
|
* - inst : Instance that should be initialized
|
|
* - inFrame : Input speech frame for each band
|
|
* - num_bands : Number of bands
|
|
*
|
|
* Output:
|
|
* - inst : Updated instance
|
|
* - outFrame : Output speech frame for each band
|
|
*/
|
|
void WebRtcNsx_ProcessCore(NoiseSuppressionFixedC* inst,
|
|
const short* const* inFrame,
|
|
int num_bands,
|
|
short* const* outFrame);
|
|
|
|
/****************************************************************************
|
|
* Some function pointers, for internal functions shared by ARM NEON and
|
|
* generic C code.
|
|
*/
|
|
// Noise Estimation.
|
|
typedef void (*NoiseEstimation)(NoiseSuppressionFixedC* inst,
|
|
uint16_t* magn,
|
|
uint32_t* noise,
|
|
int16_t* q_noise);
|
|
extern NoiseEstimation WebRtcNsx_NoiseEstimation;
|
|
|
|
// Filter the data in the frequency domain, and create spectrum.
|
|
typedef void (*PrepareSpectrum)(NoiseSuppressionFixedC* inst,
|
|
int16_t* freq_buff);
|
|
extern PrepareSpectrum WebRtcNsx_PrepareSpectrum;
|
|
|
|
// For the noise supression process, synthesis, read out fully processed
|
|
// segment, and update synthesis buffer.
|
|
typedef void (*SynthesisUpdate)(NoiseSuppressionFixedC* inst,
|
|
int16_t* out_frame,
|
|
int16_t gain_factor);
|
|
extern SynthesisUpdate WebRtcNsx_SynthesisUpdate;
|
|
|
|
// Update analysis buffer for lower band, and window data before FFT.
|
|
typedef void (*AnalysisUpdate)(NoiseSuppressionFixedC* inst,
|
|
int16_t* out,
|
|
int16_t* new_speech);
|
|
extern AnalysisUpdate WebRtcNsx_AnalysisUpdate;
|
|
|
|
// Denormalize the real-valued signal |in|, the output from inverse FFT.
|
|
typedef void (*Denormalize)(NoiseSuppressionFixedC* inst,
|
|
int16_t* in,
|
|
int factor);
|
|
extern Denormalize WebRtcNsx_Denormalize;
|
|
|
|
// Normalize the real-valued signal |in|, the input to forward FFT.
|
|
typedef void (*NormalizeRealBuffer)(NoiseSuppressionFixedC* inst,
|
|
const int16_t* in,
|
|
int16_t* out);
|
|
extern NormalizeRealBuffer WebRtcNsx_NormalizeRealBuffer;
|
|
|
|
// Compute speech/noise probability.
|
|
// Intended to be private.
|
|
void WebRtcNsx_SpeechNoiseProb(NoiseSuppressionFixedC* inst,
|
|
uint16_t* nonSpeechProbFinal,
|
|
uint32_t* priorLocSnr,
|
|
uint32_t* postLocSnr);
|
|
|
|
#if defined(WEBRTC_HAS_NEON)
|
|
// For the above function pointers, functions for generic platforms are declared
|
|
// and defined as static in file nsx_core.c, while those for ARM Neon platforms
|
|
// are declared below and defined in file nsx_core_neon.c.
|
|
void WebRtcNsx_NoiseEstimationNeon(NoiseSuppressionFixedC* inst,
|
|
uint16_t* magn,
|
|
uint32_t* noise,
|
|
int16_t* q_noise);
|
|
void WebRtcNsx_SynthesisUpdateNeon(NoiseSuppressionFixedC* inst,
|
|
int16_t* out_frame,
|
|
int16_t gain_factor);
|
|
void WebRtcNsx_AnalysisUpdateNeon(NoiseSuppressionFixedC* inst,
|
|
int16_t* out,
|
|
int16_t* new_speech);
|
|
void WebRtcNsx_PrepareSpectrumNeon(NoiseSuppressionFixedC* inst,
|
|
int16_t* freq_buff);
|
|
#endif
|
|
|
|
#if defined(MIPS32_LE)
|
|
// For the above function pointers, functions for generic platforms are declared
|
|
// and defined as static in file nsx_core.c, while those for MIPS platforms
|
|
// are declared below and defined in file nsx_core_mips.c.
|
|
void WebRtcNsx_SynthesisUpdate_mips(NoiseSuppressionFixedC* inst,
|
|
int16_t* out_frame,
|
|
int16_t gain_factor);
|
|
void WebRtcNsx_AnalysisUpdate_mips(NoiseSuppressionFixedC* inst,
|
|
int16_t* out,
|
|
int16_t* new_speech);
|
|
void WebRtcNsx_PrepareSpectrum_mips(NoiseSuppressionFixedC* inst,
|
|
int16_t* freq_buff);
|
|
void WebRtcNsx_NormalizeRealBuffer_mips(NoiseSuppressionFixedC* inst,
|
|
const int16_t* in,
|
|
int16_t* out);
|
|
#if defined(MIPS_DSP_R1_LE)
|
|
void WebRtcNsx_Denormalize_mips(NoiseSuppressionFixedC* inst,
|
|
int16_t* in,
|
|
int factor);
|
|
#endif
|
|
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // MODULES_AUDIO_PROCESSING_NS_MAIN_SOURCE_NSX_CORE_H_
|