| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 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/intelligibility/intelligibility_enhanc
er.h" | 11 #include "webrtc/modules/audio_processing/intelligibility/intelligibility_enhanc
er.h" |
| 12 | 12 |
| 13 #include <math.h> | 13 #include <math.h> |
| 14 #include <stdlib.h> | 14 #include <stdlib.h> |
| 15 #include <algorithm> | 15 #include <algorithm> |
| 16 #include <limits> | 16 #include <limits> |
| 17 #include <numeric> | 17 #include <numeric> |
| 18 | 18 |
| 19 #include "webrtc/base/checks.h" | 19 #include "webrtc/base/checks.h" |
| 20 #include "webrtc/base/logging.h" | 20 #include "webrtc/base/logging.h" |
| 21 #include "webrtc/base/safe_minmax.h" |
| 21 #include "webrtc/common_audio/include/audio_util.h" | 22 #include "webrtc/common_audio/include/audio_util.h" |
| 22 #include "webrtc/common_audio/window_generator.h" | 23 #include "webrtc/common_audio/window_generator.h" |
| 23 | 24 |
| 24 namespace webrtc { | 25 namespace webrtc { |
| 25 | 26 |
| 26 namespace { | 27 namespace { |
| 27 | 28 |
| 28 const size_t kErbResolution = 2; | 29 const size_t kErbResolution = 2; |
| 29 const int kWindowSizeMs = 16; | 30 const int kWindowSizeMs = 16; |
| 30 const int kChunkSizeMs = 10; // Size provided by APM. | 31 const int kChunkSizeMs = 10; // Size provided by APM. |
| (...skipping 249 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 280 float last_center_freq = center_freqs_[bank_size_ - 1]; | 281 float last_center_freq = center_freqs_[bank_size_ - 1]; |
| 281 for (size_t i = 0; i < bank_size_; ++i) { | 282 for (size_t i = 0; i < bank_size_; ++i) { |
| 282 center_freqs_[i] *= 0.5f * sample_rate_hz_ / last_center_freq; | 283 center_freqs_[i] *= 0.5f * sample_rate_hz_ / last_center_freq; |
| 283 } | 284 } |
| 284 | 285 |
| 285 for (size_t i = 0; i < bank_size_; ++i) { | 286 for (size_t i = 0; i < bank_size_; ++i) { |
| 286 filter_bank[i].resize(num_freqs); | 287 filter_bank[i].resize(num_freqs); |
| 287 } | 288 } |
| 288 | 289 |
| 289 for (size_t i = 1; i <= bank_size_; ++i) { | 290 for (size_t i = 1; i <= bank_size_; ++i) { |
| 290 static const size_t kOne = 1; // Avoids repeated static_cast<>s below. | 291 size_t lll = static_cast<size_t>( |
| 291 size_t lll = | 292 round(center_freqs_[rtc::SafeMax<size_t>(1, i - lf) - 1] * num_freqs / |
| 292 static_cast<size_t>(round(center_freqs_[std::max(kOne, i - lf) - 1] * | 293 (0.5f * sample_rate_hz_))); |
| 293 num_freqs / (0.5f * sample_rate_hz_))); | 294 size_t ll = static_cast<size_t>( |
| 294 size_t ll = static_cast<size_t>(round(center_freqs_[std::max(kOne, i) - 1] * | 295 round(center_freqs_[rtc::SafeMax<size_t>(1, i) - 1] * num_freqs / |
| 295 num_freqs / (0.5f * sample_rate_hz_))); | 296 (0.5f * sample_rate_hz_))); |
| 296 lll = std::min(num_freqs, std::max(lll, kOne)) - 1; | 297 lll = rtc::SafeClamp<size_t>(lll, 1, num_freqs) - 1; |
| 297 ll = std::min(num_freqs, std::max(ll, kOne)) - 1; | 298 ll = rtc::SafeClamp<size_t>(ll, 1, num_freqs) - 1; |
| 298 | 299 |
| 299 size_t rrr = static_cast<size_t>( | 300 size_t rrr = static_cast<size_t>( |
| 300 round(center_freqs_[std::min(bank_size_, i + rf) - 1] * num_freqs / | 301 round(center_freqs_[rtc::SafeMin<size_t>(bank_size_, i + rf) - 1] * |
| 301 (0.5f * sample_rate_hz_))); | 302 num_freqs / (0.5f * sample_rate_hz_))); |
| 302 size_t rr = static_cast<size_t>( | 303 size_t rr = static_cast<size_t>( |
| 303 round(center_freqs_[std::min(bank_size_, i + 1) - 1] * num_freqs / | 304 round(center_freqs_[rtc::SafeMin<size_t>(bank_size_, i + 1) - 1] * |
| 304 (0.5f * sample_rate_hz_))); | 305 num_freqs / (0.5f * sample_rate_hz_))); |
| 305 rrr = std::min(num_freqs, std::max(rrr, kOne)) - 1; | 306 rrr = rtc::SafeClamp<size_t>(rrr, 1, num_freqs) - 1; |
| 306 rr = std::min(num_freqs, std::max(rr, kOne)) - 1; | 307 rr = rtc::SafeClamp<size_t>(rr, 1, num_freqs) - 1; |
| 307 | 308 |
| 308 float step = ll == lll ? 0.f : 1.f / (ll - lll); | 309 float step = ll == lll ? 0.f : 1.f / (ll - lll); |
| 309 float element = 0.f; | 310 float element = 0.f; |
| 310 for (size_t j = lll; j <= ll; ++j) { | 311 for (size_t j = lll; j <= ll; ++j) { |
| 311 filter_bank[i - 1][j] = element; | 312 filter_bank[i - 1][j] = element; |
| 312 element += step; | 313 element += step; |
| 313 } | 314 } |
| 314 step = rr == rrr ? 0.f : 1.f / (rrr - rr); | 315 step = rr == rrr ? 0.f : 1.f / (rrr - rr); |
| 315 element = 1.f; | 316 element = 1.f; |
| 316 for (size_t j = rr; j <= rrr; ++j) { | 317 for (size_t j = rr; j <= rrr; ++j) { |
| (...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 381 | 382 |
| 382 void IntelligibilityEnhancer::DelayHighBands(AudioBuffer* audio) { | 383 void IntelligibilityEnhancer::DelayHighBands(AudioBuffer* audio) { |
| 383 RTC_DCHECK_EQ(audio->num_bands(), high_bands_buffers_.size() + 1); | 384 RTC_DCHECK_EQ(audio->num_bands(), high_bands_buffers_.size() + 1); |
| 384 for (size_t i = 0u; i < high_bands_buffers_.size(); ++i) { | 385 for (size_t i = 0u; i < high_bands_buffers_.size(); ++i) { |
| 385 Band band = static_cast<Band>(i + 1); | 386 Band band = static_cast<Band>(i + 1); |
| 386 high_bands_buffers_[i]->Delay(audio->split_channels_f(band), chunk_length_); | 387 high_bands_buffers_[i]->Delay(audio->split_channels_f(band), chunk_length_); |
| 387 } | 388 } |
| 388 } | 389 } |
| 389 | 390 |
| 390 } // namespace webrtc | 391 } // namespace webrtc |
| OLD | NEW |