Turn webrtc::Vad into a pure virtual interface
Review URL: https://codereview.webrtc.org/1317243005 Cr-Commit-Position: refs/heads/master@{#9899}
This commit is contained in:
@ -12,12 +12,12 @@
|
||||
#define WEBRTC_COMMON_AUDIO_VAD_INCLUDE_VAD_H_
|
||||
|
||||
#include "webrtc/base/checks.h"
|
||||
#include "webrtc/base/scoped_ptr.h"
|
||||
#include "webrtc/common_audio/vad/include/webrtc_vad.h"
|
||||
#include "webrtc/typedefs.h"
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
// This is a C++ wrapper class for WebRtcVad.
|
||||
class Vad {
|
||||
public:
|
||||
enum Aggressiveness {
|
||||
@ -29,21 +29,22 @@ class Vad {
|
||||
|
||||
enum Activity { kPassive = 0, kActive = 1, kError = -1 };
|
||||
|
||||
explicit Vad(enum Aggressiveness mode);
|
||||
|
||||
virtual ~Vad();
|
||||
virtual ~Vad() = default;
|
||||
|
||||
// Calculates a VAD decision for the given audio frame. Valid sample rates
|
||||
// are 8000, 16000, and 32000 Hz; the number of samples must be such that the
|
||||
// frame is 10, 20, or 30 ms long.
|
||||
virtual Activity VoiceActivity(const int16_t* audio,
|
||||
size_t num_samples,
|
||||
int sample_rate_hz);
|
||||
int sample_rate_hz) = 0;
|
||||
|
||||
// Reset VAD state.
|
||||
virtual void Reset();
|
||||
|
||||
private:
|
||||
VadInst* handle_;
|
||||
Aggressiveness aggressiveness_;
|
||||
// Resets VAD state.
|
||||
virtual void Reset() = 0;
|
||||
};
|
||||
|
||||
// Returns a Vad instance that's implemented on top of WebRtcVad.
|
||||
rtc::scoped_ptr<Vad> CreateVad(Vad::Aggressiveness aggressiveness);
|
||||
|
||||
} // namespace webrtc
|
||||
|
||||
#endif // WEBRTC_COMMON_AUDIO_VAD_INCLUDE_VAD_H_
|
||||
|
@ -19,7 +19,6 @@ namespace webrtc {
|
||||
|
||||
class MockVad : public Vad {
|
||||
public:
|
||||
explicit MockVad(enum Aggressiveness mode) : Vad(mode) {}
|
||||
virtual ~MockVad() { Die(); }
|
||||
MOCK_METHOD0(Die, void());
|
||||
|
||||
|
@ -14,36 +14,50 @@
|
||||
|
||||
namespace webrtc {
|
||||
|
||||
Vad::Vad(enum Aggressiveness mode) : handle_(nullptr), aggressiveness_(mode) {
|
||||
Reset();
|
||||
}
|
||||
namespace {
|
||||
|
||||
Vad::~Vad() {
|
||||
WebRtcVad_Free(handle_);
|
||||
}
|
||||
|
||||
enum Vad::Activity Vad::VoiceActivity(const int16_t* audio,
|
||||
size_t num_samples,
|
||||
int sample_rate_hz) {
|
||||
int ret = WebRtcVad_Process(handle_, sample_rate_hz, audio, num_samples);
|
||||
switch (ret) {
|
||||
case 0:
|
||||
return kPassive;
|
||||
case 1:
|
||||
return kActive;
|
||||
default:
|
||||
DCHECK(false) << "WebRtcVad_Process returned an error.";
|
||||
return kError;
|
||||
class VadImpl final : public Vad {
|
||||
public:
|
||||
explicit VadImpl(Aggressiveness aggressiveness)
|
||||
: handle_(nullptr), aggressiveness_(aggressiveness) {
|
||||
Reset();
|
||||
}
|
||||
}
|
||||
|
||||
void Vad::Reset() {
|
||||
if (handle_)
|
||||
WebRtcVad_Free(handle_);
|
||||
handle_ = WebRtcVad_Create();
|
||||
CHECK(handle_);
|
||||
CHECK_EQ(WebRtcVad_Init(handle_), 0);
|
||||
CHECK_EQ(WebRtcVad_set_mode(handle_, aggressiveness_), 0);
|
||||
~VadImpl() override { WebRtcVad_Free(handle_); }
|
||||
|
||||
Activity VoiceActivity(const int16_t* audio,
|
||||
size_t num_samples,
|
||||
int sample_rate_hz) override {
|
||||
int ret = WebRtcVad_Process(handle_, sample_rate_hz, audio, num_samples);
|
||||
switch (ret) {
|
||||
case 0:
|
||||
return kPassive;
|
||||
case 1:
|
||||
return kActive;
|
||||
default:
|
||||
DCHECK(false) << "WebRtcVad_Process returned an error.";
|
||||
return kError;
|
||||
}
|
||||
}
|
||||
|
||||
void Reset() override {
|
||||
if (handle_)
|
||||
WebRtcVad_Free(handle_);
|
||||
handle_ = WebRtcVad_Create();
|
||||
CHECK(handle_);
|
||||
CHECK_EQ(WebRtcVad_Init(handle_), 0);
|
||||
CHECK_EQ(WebRtcVad_set_mode(handle_, aggressiveness_), 0);
|
||||
}
|
||||
|
||||
private:
|
||||
VadInst* handle_;
|
||||
Aggressiveness aggressiveness_;
|
||||
};
|
||||
|
||||
} // namespace
|
||||
|
||||
rtc::scoped_ptr<Vad> CreateVad(Vad::Aggressiveness aggressiveness) {
|
||||
return rtc::scoped_ptr<Vad>(new VadImpl(aggressiveness));
|
||||
}
|
||||
|
||||
} // namespace webrtc
|
||||
|
@ -54,7 +54,8 @@ AudioEncoderCng::AudioEncoderCng(const Config& config)
|
||||
num_cng_coefficients_(config.num_cng_coefficients),
|
||||
sid_frame_interval_ms_(config.sid_frame_interval_ms),
|
||||
last_frame_active_(true),
|
||||
vad_(config.vad ? config.vad : new Vad(config.vad_mode)) {
|
||||
vad_(config.vad ? rtc_make_scoped_ptr(config.vad)
|
||||
: CreateVad(config.vad_mode)) {
|
||||
CHECK(config.IsOk()) << "Invalid configuration.";
|
||||
cng_inst_ = CreateCngInst(SampleRateHz(), sid_frame_interval_ms_,
|
||||
num_cng_coefficients_);
|
||||
|
@ -34,7 +34,7 @@ static const int kCngPayloadType = 18;
|
||||
class AudioEncoderCngTest : public ::testing::Test {
|
||||
protected:
|
||||
AudioEncoderCngTest()
|
||||
: mock_vad_(new MockVad(Vad::kVadNormal)),
|
||||
: mock_vad_(new MockVad),
|
||||
timestamp_(4711),
|
||||
num_audio_samples_10ms_(0),
|
||||
sample_rate_hz_(8000) {
|
||||
|
Reference in New Issue
Block a user