OLD | NEW |
---|---|
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 |
(...skipping 159 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
170 // TODO(peah): Returning error code as a float! Remove this. | 170 // TODO(peah): Returning error code as a float! Remove this. |
171 // Currently not available for the fixed point implementation. | 171 // Currently not available for the fixed point implementation. |
172 return AudioProcessing::kUnsupportedFunctionError; | 172 return AudioProcessing::kUnsupportedFunctionError; |
173 #endif | 173 #endif |
174 } | 174 } |
175 | 175 |
176 std::vector<float> NoiseSuppressionImpl::NoiseEstimate() { | 176 std::vector<float> NoiseSuppressionImpl::NoiseEstimate() { |
177 rtc::CritScope cs(crit_); | 177 rtc::CritScope cs(crit_); |
178 std::vector<float> noise_estimate; | 178 std::vector<float> noise_estimate; |
179 #if defined(WEBRTC_NS_FLOAT) | 179 #if defined(WEBRTC_NS_FLOAT) |
180 const float kNormalizationFactor = 1.f / (1 << 15); | |
181 noise_estimate.assign(WebRtcNs_num_freq(), 0.f); | 180 noise_estimate.assign(WebRtcNs_num_freq(), 0.f); |
182 for (auto& suppressor : suppressors_) { | 181 for (auto& suppressor : suppressors_) { |
183 const float* noise = WebRtcNs_noise_estimate(suppressor->state()); | 182 const float* noise = WebRtcNs_noise_estimate(suppressor->state()); |
184 for (size_t i = 0; i < noise_estimate.size(); ++i) { | 183 for (size_t i = 0; i < noise_estimate.size(); ++i) { |
185 noise_estimate[i] += | 184 noise_estimate[i] += noise[i] / suppressors_.size(); |
186 kNormalizationFactor * noise[i] / suppressors_.size(); | |
187 } | 185 } |
188 } | 186 } |
189 #elif defined(WEBRTC_NS_FIXED) | 187 #elif defined(WEBRTC_NS_FIXED) |
190 const float kNormalizationFactor = 1.f / (1 << 23); | |
191 noise_estimate.assign(WebRtcNsx_num_freq(), 0.f); | 188 noise_estimate.assign(WebRtcNsx_num_freq(), 0.f); |
192 for (auto& suppressor : suppressors_) { | 189 for (auto& suppressor : suppressors_) { |
193 const uint32_t* noise = WebRtcNsx_noise_estimate(suppressor->state()); | 190 const uint32_t* noise = WebRtcNsx_noise_estimate(suppressor->state()); |
turaj
2016/03/31 15:20:48
Just a suggestion, up to you if you want to change
aluebs-webrtc
2016/03/31 17:48:13
In general I think it can be confusing when passin
| |
191 const float kNormalizationFactor = | |
192 1.f / (1 << WebRtcNsx_noise_estimate_domain(suppressor->state())); | |
194 for (size_t i = 0; i < noise_estimate.size(); ++i) { | 193 for (size_t i = 0; i < noise_estimate.size(); ++i) { |
195 noise_estimate[i] += kNormalizationFactor * | 194 noise_estimate[i] += kNormalizationFactor * |
196 static_cast<float>(noise[i]) / suppressors_.size(); | 195 static_cast<float>(noise[i]) / suppressors_.size(); |
197 } | 196 } |
198 } | 197 } |
199 #endif | 198 #endif |
200 return noise_estimate; | 199 return noise_estimate; |
201 } | 200 } |
202 | 201 |
203 size_t NoiseSuppressionImpl::num_noise_bins() { | 202 size_t NoiseSuppressionImpl::num_noise_bins() { |
204 #if defined(WEBRTC_NS_FLOAT) | 203 #if defined(WEBRTC_NS_FLOAT) |
205 return WebRtcNs_num_freq(); | 204 return WebRtcNs_num_freq(); |
206 #elif defined(WEBRTC_NS_FIXED) | 205 #elif defined(WEBRTC_NS_FIXED) |
207 return WebRtcNsx_num_freq(); | 206 return WebRtcNsx_num_freq(); |
208 #endif | 207 #endif |
209 } | 208 } |
210 | 209 |
211 } // namespace webrtc | 210 } // namespace webrtc |
OLD | NEW |