| 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 |
| (...skipping 13 matching lines...) Expand all Loading... |
| 24 #include "webrtc/modules/audio_processing/aec3/erle_estimator.h" | 24 #include "webrtc/modules/audio_processing/aec3/erle_estimator.h" |
| 25 #include "webrtc/modules/audio_processing/aec3/render_buffer.h" | 25 #include "webrtc/modules/audio_processing/aec3/render_buffer.h" |
| 26 | 26 |
| 27 namespace webrtc { | 27 namespace webrtc { |
| 28 | 28 |
| 29 class ApmDataDumper; | 29 class ApmDataDumper; |
| 30 | 30 |
| 31 // Handles the state and the conditions for the echo removal functionality. | 31 // Handles the state and the conditions for the echo removal functionality. |
| 32 class AecState { | 32 class AecState { |
| 33 public: | 33 public: |
| 34 AecState(); | 34 explicit AecState(float echo_decay); |
| 35 ~AecState(); | 35 ~AecState(); |
| 36 | 36 |
| 37 // Returns whether the linear filter estimate is usable. | 37 // Returns whether the linear filter estimate is usable. |
| 38 bool UsableLinearEstimate() const { return usable_linear_estimate_; } | 38 bool UsableLinearEstimate() const { return usable_linear_estimate_; } |
| 39 | 39 |
| 40 // Returns whether there has been echo leakage detected. | 40 // Returns whether there has been echo leakage detected. |
| 41 bool EchoLeakageDetected() const { return echo_leakage_detected_; } | 41 bool EchoLeakageDetected() const { return echo_leakage_detected_; } |
| 42 | 42 |
| 43 // Returns whether the render signal is currently active. | 43 // Returns whether the render signal is currently active. |
| 44 // TODO(peah): Deprecate this in an upcoming CL. | 44 // TODO(peah): Deprecate this in an upcoming CL. |
| (...skipping 27 matching lines...) Expand all Loading... |
| 72 } | 72 } |
| 73 | 73 |
| 74 // Returns whether a probable headset setup has been detected. | 74 // Returns whether a probable headset setup has been detected. |
| 75 bool HeadsetDetected() const { return headset_detected_; } | 75 bool HeadsetDetected() const { return headset_detected_; } |
| 76 | 76 |
| 77 // Takes appropriate action at an echo path change. | 77 // Takes appropriate action at an echo path change. |
| 78 void HandleEchoPathChange(const EchoPathVariability& echo_path_variability); | 78 void HandleEchoPathChange(const EchoPathVariability& echo_path_variability); |
| 79 | 79 |
| 80 // Returns the decay factor for the echo reverberation. | 80 // Returns the decay factor for the echo reverberation. |
| 81 // TODO(peah): Make this adaptive. | 81 // TODO(peah): Make this adaptive. |
| 82 float ReverbDecayFactor() const { return 0.f; } | 82 float ReverbDecayFactor() const { return echo_decay_factor_; } |
| 83 | 83 |
| 84 // Returns whether the echo suppression gain should be forced to zero. | 84 // Returns whether the echo suppression gain should be forced to zero. |
| 85 bool ForcedZeroGain() const { return force_zero_gain_; } | 85 bool ForcedZeroGain() const { return force_zero_gain_; } |
| 86 | 86 |
| 87 // Updates the aec state. | 87 // Updates the aec state. |
| 88 void Update(const std::vector<std::array<float, kFftLengthBy2Plus1>>& | 88 void Update(const std::vector<std::array<float, kFftLengthBy2Plus1>>& |
| 89 adaptive_filter_frequency_response, | 89 adaptive_filter_frequency_response, |
| 90 const rtc::Optional<size_t>& external_delay_samples, | 90 const rtc::Optional<size_t>& external_delay_samples, |
| 91 const RenderBuffer& render_buffer, | 91 const RenderBuffer& render_buffer, |
| 92 const std::array<float, kFftLengthBy2Plus1>& E2_main, | 92 const std::array<float, kFftLengthBy2Plus1>& E2_main, |
| (...skipping 13 matching lines...) Expand all Loading... |
| 106 bool capture_signal_saturation_ = false; | 106 bool capture_signal_saturation_ = false; |
| 107 bool echo_saturation_ = false; | 107 bool echo_saturation_ = false; |
| 108 bool headset_detected_ = false; | 108 bool headset_detected_ = false; |
| 109 float previous_max_sample_ = 0.f; | 109 float previous_max_sample_ = 0.f; |
| 110 bool force_zero_gain_ = false; | 110 bool force_zero_gain_ = false; |
| 111 bool render_received_ = false; | 111 bool render_received_ = false; |
| 112 size_t force_zero_gain_counter_ = 0; | 112 size_t force_zero_gain_counter_ = 0; |
| 113 rtc::Optional<size_t> filter_delay_; | 113 rtc::Optional<size_t> filter_delay_; |
| 114 rtc::Optional<size_t> external_delay_; | 114 rtc::Optional<size_t> external_delay_; |
| 115 size_t blocks_since_last_saturation_ = 1000; | 115 size_t blocks_since_last_saturation_ = 1000; |
| 116 | 116 const float echo_decay_factor_; |
| 117 RTC_DISALLOW_COPY_AND_ASSIGN(AecState); | 117 RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(AecState); |
| 118 }; | 118 }; |
| 119 | 119 |
| 120 } // namespace webrtc | 120 } // namespace webrtc |
| 121 | 121 |
| 122 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_AEC_STATE_H_ | 122 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_AEC3_AEC_STATE_H_ |
| OLD | NEW |