OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2017 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 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ | 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ |
12 #define WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ | 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ |
13 | 13 |
14 #include <array> | 14 #include <array> |
15 | 15 |
16 #include "webrtc/modules/audio_processing/aec3/aec3_common.h" | 16 #include "webrtc/modules/audio_processing/aec3/aec3_common.h" |
17 #include "webrtc/rtc_base/constructormagic.h" | 17 #include "webrtc/rtc_base/constructormagic.h" |
18 | 18 |
19 namespace webrtc { | 19 namespace webrtc { |
20 | 20 |
21 // Estimates the echo return loss enhancement based on the signal spectra. | 21 // Estimates the echo return loss enhancement based on the signal spectra. |
22 class ErleEstimator { | 22 class ErleEstimator { |
23 public: | 23 public: |
24 ErleEstimator(); | 24 ErleEstimator(float min_erle, float max_erle_lf, float max_erle_hf); |
25 ~ErleEstimator(); | 25 ~ErleEstimator(); |
26 | 26 |
27 // Updates the ERLE estimate. | 27 // Updates the ERLE estimate. |
28 void Update(const std::array<float, kFftLengthBy2Plus1>& render_spectrum, | 28 void Update(const std::array<float, kFftLengthBy2Plus1>& render_spectrum, |
29 const std::array<float, kFftLengthBy2Plus1>& capture_spectrum, | 29 const std::array<float, kFftLengthBy2Plus1>& capture_spectrum, |
30 const std::array<float, kFftLengthBy2Plus1>& subtractor_spectrum); | 30 const std::array<float, kFftLengthBy2Plus1>& subtractor_spectrum); |
31 | 31 |
32 // Returns the most recent ERLE estimate. | 32 // Returns the most recent ERLE estimate. |
33 const std::array<float, kFftLengthBy2Plus1>& Erle() const { return erle_; } | 33 const std::array<float, kFftLengthBy2Plus1>& Erle() const { return erle_; } |
34 | 34 |
35 private: | 35 private: |
36 std::array<float, kFftLengthBy2Plus1> erle_; | 36 std::array<float, kFftLengthBy2Plus1> erle_; |
37 std::array<int, kFftLengthBy2Minus1> hold_counters_; | 37 std::array<int, kFftLengthBy2Minus1> hold_counters_; |
| 38 const float min_erle_; |
| 39 const float max_erle_lf_; |
| 40 const float max_erle_hf_; |
38 | 41 |
39 RTC_DISALLOW_COPY_AND_ASSIGN(ErleEstimator); | 42 RTC_DISALLOW_COPY_AND_ASSIGN(ErleEstimator); |
40 }; | 43 }; |
41 | 44 |
42 } // namespace webrtc | 45 } // namespace webrtc |
43 | 46 |
44 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ | 47 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_ERLE_ESTIMATOR_H_ |
OLD | NEW |