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

Side by Side Diff: webrtc/modules/audio_processing/noise_suppression_impl.cc

Issue 1424663003: Lock scheme #8: Introduced the new locking scheme (Closed) Base URL: https://chromium.googlesource.com/external/webrtc.git@add_threadcheckers_CL
Patch Set: Fixed a bad merge error for the beamformer settings and updated with the latest merge from master Created 5 years, 1 month 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
1 /* 1 /*
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
3 * 3 *
4 * Use of this source code is governed by a BSD-style license 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 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 6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may 7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree. 8 * be found in the AUTHORS file in the root of the source tree.
9 */ 9 */
10 10
11 #include "webrtc/modules/audio_processing/noise_suppression_impl.h" 11 #include "webrtc/modules/audio_processing/noise_suppression_impl.h"
12 12
13 #include <assert.h> 13 #include <assert.h>
14 14
15 #include "webrtc/modules/audio_processing/audio_buffer.h" 15 #include "webrtc/modules/audio_processing/audio_buffer.h"
16 #if defined(WEBRTC_NS_FLOAT) 16 #if defined(WEBRTC_NS_FLOAT)
17 #include "webrtc/modules/audio_processing/ns/noise_suppression.h" 17 #include "webrtc/modules/audio_processing/ns/noise_suppression.h"
18 #elif defined(WEBRTC_NS_FIXED) 18 #elif defined(WEBRTC_NS_FIXED)
19 #include "webrtc/modules/audio_processing/ns/noise_suppression_x.h" 19 #include "webrtc/modules/audio_processing/ns/noise_suppression_x.h"
20 #endif 20 #endif
21 #include "webrtc/system_wrappers/include/critical_section_wrapper.h"
22 21
23 22
24 namespace webrtc { 23 namespace webrtc {
25 24
26 #if defined(WEBRTC_NS_FLOAT) 25 #if defined(WEBRTC_NS_FLOAT)
27 typedef NsHandle Handle; 26 typedef NsHandle Handle;
28 #elif defined(WEBRTC_NS_FIXED) 27 #elif defined(WEBRTC_NS_FIXED)
29 typedef NsxHandle Handle; 28 typedef NsxHandle Handle;
30 #endif 29 #endif
31 30
32 namespace { 31 namespace {
33 int MapSetting(NoiseSuppression::Level level) { 32 int MapSetting(NoiseSuppression::Level level) {
34 switch (level) { 33 switch (level) {
35 case NoiseSuppression::kLow: 34 case NoiseSuppression::kLow:
36 return 0; 35 return 0;
37 case NoiseSuppression::kModerate: 36 case NoiseSuppression::kModerate:
38 return 1; 37 return 1;
39 case NoiseSuppression::kHigh: 38 case NoiseSuppression::kHigh:
40 return 2; 39 return 2;
41 case NoiseSuppression::kVeryHigh: 40 case NoiseSuppression::kVeryHigh:
42 return 3; 41 return 3;
43 } 42 }
44 assert(false); 43 assert(false);
45 return -1; 44 return -1;
46 } 45 }
47 } // namespace 46 } // namespace
48 47
49 NoiseSuppressionImpl::NoiseSuppressionImpl(const AudioProcessing* apm, 48 NoiseSuppressionImpl::NoiseSuppressionImpl(const AudioProcessing* apm,
50 CriticalSectionWrapper* crit) 49 rtc::CriticalSection* crit)
51 : ProcessingComponent(), 50 : ProcessingComponent(), apm_(apm), crit_(crit), level_(kModerate) {}
52 apm_(apm),
53 crit_(crit),
54 level_(kModerate) {}
55 51
56 NoiseSuppressionImpl::~NoiseSuppressionImpl() {} 52 NoiseSuppressionImpl::~NoiseSuppressionImpl() {}
57 53
58 int NoiseSuppressionImpl::AnalyzeCaptureAudio(AudioBuffer* audio) { 54 int NoiseSuppressionImpl::AnalyzeCaptureAudio(AudioBuffer* audio) {
59 #if defined(WEBRTC_NS_FLOAT) 55 #if defined(WEBRTC_NS_FLOAT)
60 if (!is_component_enabled()) { 56 if (!is_component_enabled()) {
61 return apm_->kNoError; 57 return apm_->kNoError;
62 } 58 }
63 assert(audio->num_frames_per_band() <= 160); 59 assert(audio->num_frames_per_band() <= 160);
64 assert(audio->num_channels() == num_handles()); 60 assert(audio->num_channels() == num_handles());
(...skipping 25 matching lines...) Expand all
90 WebRtcNsx_Process(my_handle, 86 WebRtcNsx_Process(my_handle,
91 audio->split_bands_const(i), 87 audio->split_bands_const(i),
92 audio->num_bands(), 88 audio->num_bands(),
93 audio->split_bands(i)); 89 audio->split_bands(i));
94 #endif 90 #endif
95 } 91 }
96 return apm_->kNoError; 92 return apm_->kNoError;
97 } 93 }
98 94
99 int NoiseSuppressionImpl::Enable(bool enable) { 95 int NoiseSuppressionImpl::Enable(bool enable) {
100 CriticalSectionScoped crit_scoped(crit_); 96 rtc::CritScope cs(crit_);
101 return EnableComponent(enable); 97 return EnableComponent(enable);
102 } 98 }
103 99
104 bool NoiseSuppressionImpl::is_enabled() const { 100 bool NoiseSuppressionImpl::is_enabled() const {
101 rtc::CritScope cs(crit_);
105 return is_component_enabled(); 102 return is_component_enabled();
106 } 103 }
107 104
108 int NoiseSuppressionImpl::set_level(Level level) { 105 int NoiseSuppressionImpl::set_level(Level level) {
109 CriticalSectionScoped crit_scoped(crit_); 106 rtc::CritScope cs(crit_);
110 if (MapSetting(level) == -1) { 107 if (MapSetting(level) == -1) {
111 return apm_->kBadParameterError; 108 return apm_->kBadParameterError;
112 } 109 }
113 110
114 level_ = level; 111 level_ = level;
115 return Configure(); 112 return Configure();
116 } 113 }
117 114
118 NoiseSuppression::Level NoiseSuppressionImpl::level() const { 115 NoiseSuppression::Level NoiseSuppressionImpl::level() const {
116 rtc::CritScope cs(crit_);
119 return level_; 117 return level_;
120 } 118 }
121 119
122 float NoiseSuppressionImpl::speech_probability() const { 120 float NoiseSuppressionImpl::speech_probability() const {
121 rtc::CritScope cs(crit_);
123 #if defined(WEBRTC_NS_FLOAT) 122 #if defined(WEBRTC_NS_FLOAT)
124 float probability_average = 0.0f; 123 float probability_average = 0.0f;
125 for (int i = 0; i < num_handles(); i++) { 124 for (int i = 0; i < num_handles(); i++) {
126 Handle* my_handle = static_cast<Handle*>(handle(i)); 125 Handle* my_handle = static_cast<Handle*>(handle(i));
127 probability_average += WebRtcNs_prior_speech_probability(my_handle); 126 probability_average += WebRtcNs_prior_speech_probability(my_handle);
128 } 127 }
129 return probability_average / num_handles(); 128 return probability_average / num_handles();
130 #elif defined(WEBRTC_NS_FIXED) 129 #elif defined(WEBRTC_NS_FIXED)
131 // Currently not available for the fixed point implementation. 130 // Currently not available for the fixed point implementation.
132 return apm_->kUnsupportedFunctionError; 131 return apm_->kUnsupportedFunctionError;
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after
172 int NoiseSuppressionImpl::num_handles_required() const { 171 int NoiseSuppressionImpl::num_handles_required() const {
173 return apm_->num_output_channels(); 172 return apm_->num_output_channels();
174 } 173 }
175 174
176 int NoiseSuppressionImpl::GetHandleError(void* handle) const { 175 int NoiseSuppressionImpl::GetHandleError(void* handle) const {
177 // The NS has no get_error() function. 176 // The NS has no get_error() function.
178 assert(handle != NULL); 177 assert(handle != NULL);
179 return apm_->kUnspecifiedError; 178 return apm_->kUnspecifiedError;
180 } 179 }
181 } // namespace webrtc 180 } // namespace webrtc
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698