Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(673)

Side by Side Diff: webrtc/modules/audio_processing/agc/standalone_vad.cc

Issue 1212543002: Pull the Voice Activity Detector out from the AGC (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@master
Patch Set: Created 5 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 /*
2 * Copyright (c) 2012 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/modules/audio_processing/agc/standalone_vad.h"
12
13 #include <assert.h>
14
15 #include "webrtc/modules/interface/module_common_types.h"
16 #include "webrtc/modules/utility/interface/audio_frame_operations.h"
17 #include "webrtc/typedefs.h"
18
19 namespace webrtc {
20
21 static const int kDefaultStandaloneVadMode = 3;
22
23 StandaloneVad::StandaloneVad(VadInst* vad)
24 : vad_(vad),
25 buffer_(),
26 index_(0),
27 mode_(kDefaultStandaloneVadMode) {}
28
29 StandaloneVad::~StandaloneVad() {
30 WebRtcVad_Free(vad_);
31 }
32
33 StandaloneVad* StandaloneVad::Create() {
34 VadInst* vad = WebRtcVad_Create();
35 if (!vad)
36 return nullptr;
37
38 int err = WebRtcVad_Init(vad);
39 err |= WebRtcVad_set_mode(vad, kDefaultStandaloneVadMode);
40 if (err != 0) {
41 WebRtcVad_Free(vad);
42 return nullptr;
43 }
44 return new StandaloneVad(vad);
45 }
46
47 int StandaloneVad::AddAudio(const int16_t* data, int length) {
48 if (length != kLength10Ms)
49 return -1;
50
51 if (index_ + length > kLength10Ms * kMaxNum10msFrames)
52 // Reset the buffer if it's full.
53 // TODO(ajm): Instead, consider just processing every 10 ms frame. Then we
54 // can forgo the buffering.
55 index_ = 0;
56
57 memcpy(&buffer_[index_], data, sizeof(int16_t) * length);
58 index_ += length;
59 return 0;
60 }
61
62 int StandaloneVad::GetActivity(double* p, int length_p) {
63 if (index_ == 0)
64 return -1;
65
66 const int num_frames = index_ / kLength10Ms;
67 if (num_frames > length_p)
68 return -1;
69 assert(WebRtcVad_ValidRateAndFrameLength(kSampleRateHz, index_) == 0);
70
71 int activity = WebRtcVad_Process(vad_, kSampleRateHz, buffer_, index_);
72 if (activity < 0)
73 return -1;
74 else if (activity == 0)
75 p[0] = 0.01; // Arbitrary but small and non-zero.
76 else
77 p[0] = 0.5; // 0.5 is neutral values when combinned by other probabilities.
78 for (int n = 1; n < num_frames; n++)
79 p[n] = p[0];
80 // Reset the buffer to start from the beginning.
81 index_ = 0;
82 return activity;
83 }
84
85 int StandaloneVad::set_mode(int mode) {
86 if (mode < 0 || mode > 3)
87 return -1;
88 if (WebRtcVad_set_mode(vad_, mode) != 0)
89 return -1;
90
91 mode_ = mode;
92 return 0;
93 }
94
95 } // namespace webrtc
96
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698