mirror of
https://github.com/danog/libtgvoip.git
synced 2025-01-09 06:08:17 +01:00
5caaaafa42
I'm now using the entire audio processing module from WebRTC as opposed to individual DSP algorithms pulled from there before. Seems to work better this way.
70 lines
2.3 KiB
C++
70 lines
2.3 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_AGC_STANDALONE_VAD_H_
|
|
#define MODULES_AUDIO_PROCESSING_AGC_STANDALONE_VAD_H_
|
|
|
|
#include <stddef.h>
|
|
#include <stdint.h>
|
|
|
|
#include "common_audio/vad/include/webrtc_vad.h"
|
|
#include "modules/audio_processing/vad/common.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class StandaloneVad {
|
|
public:
|
|
static StandaloneVad* Create();
|
|
~StandaloneVad();
|
|
|
|
// Outputs
|
|
// p: a buffer where probabilities are written to.
|
|
// length_p: number of elements of |p|.
|
|
//
|
|
// return value:
|
|
// -1: if no audio is stored or VAD returns error.
|
|
// 0: in success.
|
|
// In case of error the content of |activity| is unchanged.
|
|
//
|
|
// Note that due to a high false-positive (VAD decision is active while the
|
|
// processed audio is just background noise) rate, stand-alone VAD is used as
|
|
// a one-sided indicator. The activity probability is 0.5 if the frame is
|
|
// classified as active, and the probability is 0.01 if the audio is
|
|
// classified as passive. In this way, when probabilities are combined, the
|
|
// effect of the stand-alone VAD is neutral if the input is classified as
|
|
// active.
|
|
int GetActivity(double* p, size_t length_p);
|
|
|
|
// Expecting 10 ms of 16 kHz audio to be pushed in.
|
|
int AddAudio(const int16_t* data, size_t length);
|
|
|
|
// Set aggressiveness of VAD, 0 is the least aggressive and 3 is the most
|
|
// aggressive mode. Returns -1 if the input is less than 0 or larger than 3,
|
|
// otherwise 0 is returned.
|
|
int set_mode(int mode);
|
|
// Get the agressiveness of the current VAD.
|
|
int mode() const { return mode_; }
|
|
|
|
private:
|
|
explicit StandaloneVad(VadInst* vad);
|
|
|
|
static const size_t kMaxNum10msFrames = 3;
|
|
|
|
// TODO(turajs): Is there a way to use scoped-pointer here?
|
|
VadInst* vad_;
|
|
int16_t buffer_[kMaxNum10msFrames * kLength10Ms];
|
|
size_t index_;
|
|
int mode_;
|
|
};
|
|
|
|
} // namespace webrtc
|
|
|
|
#endif // MODULES_AUDIO_PROCESSING_AGC_STANDALONE_VAD_H_
|