1 /*
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include "webrtc/common_audio/vad/include/vad.h"
12
13 #include <memory>
14
15 #include "webrtc/base/checks.h"
16
17 namespace webrtc {
18
19 namespace {
20
21 class VadImpl final : public Vad {
22 public:
VadImpl(Aggressiveness aggressiveness)23 explicit VadImpl(Aggressiveness aggressiveness)
24 : handle_(nullptr), aggressiveness_(aggressiveness) {
25 Reset();
26 }
27
~VadImpl()28 ~VadImpl() override { WebRtcVad_Free(handle_); }
29
VoiceActivity(const int16_t * audio,size_t num_samples,int sample_rate_hz)30 Activity VoiceActivity(const int16_t* audio,
31 size_t num_samples,
32 int sample_rate_hz) override {
33 int ret = WebRtcVad_Process(handle_, sample_rate_hz, audio, num_samples);
34 switch (ret) {
35 case 0:
36 return kPassive;
37 case 1:
38 return kActive;
39 default:
40 RTC_NOTREACHED() << "WebRtcVad_Process returned an error.";
41 return kError;
42 }
43 }
44
Reset()45 void Reset() override {
46 if (handle_)
47 WebRtcVad_Free(handle_);
48 handle_ = WebRtcVad_Create();
49 RTC_CHECK(handle_);
50 RTC_CHECK_EQ(WebRtcVad_Init(handle_), 0);
51 RTC_CHECK_EQ(WebRtcVad_set_mode(handle_, aggressiveness_), 0);
52 }
53
54 private:
55 VadInst* handle_;
56 Aggressiveness aggressiveness_;
57 };
58
59 } // namespace
60
CreateVad(Vad::Aggressiveness aggressiveness)61 std::unique_ptr<Vad> CreateVad(Vad::Aggressiveness aggressiveness) {
62 return std::unique_ptr<Vad>(new VadImpl(aggressiveness));
63 }
64
65 } // namespace webrtc
66