Index: webrtc/media/engine/apm_helpers.cc |
diff --git a/webrtc/media/engine/apm_helpers.cc b/webrtc/media/engine/apm_helpers.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a806523d62bc97d3653da416a0e80baf30f91163 |
--- /dev/null |
+++ b/webrtc/media/engine/apm_helpers.cc |
@@ -0,0 +1,173 @@ |
+/* |
+ * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include "webrtc/media/engine/apm_helpers.h" |
+ |
+#include "webrtc/base/logging.h" |
+#include "webrtc/modules/audio_device/include/audio_device.h" |
+#include "webrtc/modules/audio_processing/include/audio_processing.h" |
+#include "webrtc/voice_engine/transmit_mixer.h" |
+ |
+namespace webrtc { |
+namespace apm_helpers { |
+ |
+AgcConfig GetAgcConfig(AudioProcessing* apm) { |
+ RTC_DCHECK(apm); |
+ AgcConfig result; |
+ result.targetLeveldBOv = apm->gain_control()->target_level_dbfs(); |
+ result.digitalCompressionGaindB = apm->gain_control()->compression_gain_db(); |
+ result.limiterEnable = apm->gain_control()->is_limiter_enabled(); |
+ return result; |
+} |
+ |
+void SetAgcConfig(AudioProcessing* apm, |
+ const AgcConfig& config) { |
+ RTC_DCHECK(apm); |
+ GainControl* gc = apm->gain_control(); |
+ if (gc->set_target_level_dbfs(config.targetLeveldBOv) != 0) { |
+ LOG(LS_ERROR) << "Failed to set target level: " << config.targetLeveldBOv; |
+ } |
+ if (gc->set_compression_gain_db(config.digitalCompressionGaindB) != 0) { |
+ LOG(LS_ERROR) << "Failed to set compression gain: " |
+ << config.digitalCompressionGaindB; |
+ } |
+ if (gc->enable_limiter(config.limiterEnable) != 0) { |
+ LOG(LS_ERROR) << "Failed to set limiter on/off: " << config.limiterEnable; |
+ } |
+} |
+ |
+void SetAgcStatus(AudioProcessing* apm, |
+ AudioDeviceModule* adm, |
+ bool enable, |
+ AgcModes mode) { |
+ RTC_DCHECK(apm); |
+ RTC_DCHECK(adm); |
+#if defined(WEBRTC_IOS) || defined(WEBRTC_ANDROID) |
+ RTC_DCHECK_EQ(kAgcFixedDigital, mode); |
+ GainControl::Mode agc_mode = GainControl::kFixedDigital; |
+#else |
+ RTC_DCHECK_EQ(kAgcAdaptiveAnalog, mode); |
+ GainControl::Mode agc_mode = GainControl::kAdaptiveAnalog; |
+#endif |
+ GainControl* gc = apm->gain_control(); |
+ if (gc->set_mode(agc_mode) != 0) { |
+ LOG(LS_ERROR) << "Failed to set AGC mode: " << agc_mode; |
+ return; |
+ } |
+ if (gc->Enable(enable) != 0) { |
+ LOG(LS_ERROR) << "Failed to enable/disable AGC: " << enable; |
+ return; |
+ } |
+ // Set AGC state in the ADM when adaptive AGC mode has been selected. |
+ if (adm->SetAGC(enable && agc_mode == GainControl::kAdaptiveAnalog) != 0) { |
+ LOG(LS_ERROR) << "Failed to set AGC mode in ADM: " << enable; |
+ return; |
+ } |
+ LOG(LS_INFO) << "AGC set to " << enable << " with mode " << mode; |
+} |
+ |
+void SetEcStatus(AudioProcessing* apm, |
+ bool enable, |
+ EcModes mode) { |
+ RTC_DCHECK(apm); |
+ RTC_DCHECK(mode == kEcConference || mode == kEcAecm) << "mode: " << mode; |
+ EchoCancellation* ec = apm->echo_cancellation(); |
+ EchoControlMobile* ecm = apm->echo_control_mobile(); |
+ if (mode == kEcConference) { |
+ // Disable the AECM before enabling the AEC. |
+ if (enable && ecm->is_enabled() && ecm->Enable(false) != 0) { |
+ LOG(LS_ERROR) << "Failed to disable AECM."; |
+ return; |
+ } |
+ if (ec->Enable(enable) != 0) { |
+ LOG(LS_ERROR) << "Failed to enable/disable AEC: " << enable; |
+ return; |
+ } |
+ if (ec->set_suppression_level(EchoCancellation::kHighSuppression) |
+ != 0) { |
+ LOG(LS_ERROR) << "Failed to set high AEC aggressiveness."; |
+ return; |
+ } |
+ } else { |
+ // Disable the AEC before enabling the AECM. |
+ if (enable && ec->is_enabled() && ec->Enable(false) != 0) { |
+ LOG(LS_ERROR) << "Failed to disable AEC."; |
+ return; |
+ } |
+ if (ecm->Enable(enable) != 0) { |
+ LOG(LS_ERROR) << "Failed to enable/disable AECM: " << enable; |
+ return; |
+ } |
+ } |
+ LOG(LS_INFO) << "Echo control set to " << enable << " with mode " << mode; |
+} |
+ |
+void SetEcMetricsStatus(AudioProcessing* apm, bool enable) { |
+ RTC_DCHECK(apm); |
+ if ((apm->echo_cancellation()->enable_metrics(enable) != 0) || |
+ (apm->echo_cancellation()->enable_delay_logging(enable) != 0)) { |
+ LOG(LS_ERROR) << "Failed to enable/disable EC metrics: " << enable; |
+ return; |
+ } |
+ LOG(LS_INFO) << "EC metrics set to " << enable; |
+} |
+ |
+void SetAecmMode(AudioProcessing* apm, bool enable) { |
+ RTC_DCHECK(apm); |
+ EchoControlMobile* ecm = apm->echo_control_mobile(); |
+ RTC_DCHECK_EQ(EchoControlMobile::kSpeakerphone, ecm->routing_mode()); |
+ if (ecm->enable_comfort_noise(enable) != 0) { |
+ LOG(LS_ERROR) << "Failed to enable/disable CNG: " << enable; |
+ return; |
+ } |
+ LOG(LS_INFO) << "CNG set to " << enable; |
+} |
+ |
+void SetNsStatus(AudioProcessing* apm, bool enable) { |
+ RTC_DCHECK(apm); |
+ NoiseSuppression* ns = apm->noise_suppression(); |
+ if (ns->set_level(NoiseSuppression::kHigh) != 0) { |
+ LOG(LS_ERROR) << "Failed to set high NS level."; |
+ return; |
+ } |
+ if (ns->Enable(enable) != 0) { |
+ LOG(LS_ERROR) << "Failed to enable/disable NS: " << enable; |
+ return; |
+ } |
+ LOG(LS_INFO) << "NS set to " << enable; |
+} |
+ |
+void SetTypingDetectionStatus(AudioProcessing* apm, bool enable) { |
+ RTC_DCHECK(apm); |
+#if WEBRTC_VOICE_ENGINE_TYPING_DETECTION |
+ // Typing detection takes place in TransmitMixer::PrepareDemux() and |
+ // TransmitMixer::TypingDetection(). The typing detection algorithm takes as |
+ // input two booleans: |
+ // 1. A signal whether a key was pressed during the audio frame. |
+ // 2. Whether VAD is active or not. |
+ // TransmitMixer will not even call the detector if APM has set kVadUnknown in |
+ // the audio frame after near end processing, so enabling/disabling VAD is |
+ // sufficient for turning typing detection on/off. |
+ // TODO(solenberg): Rather than relying on a side effect, consider forcing the |
+ // feature on/off in TransmitMixer. |
+ VoiceDetection* vd = apm->voice_detection(); |
+ if (vd->Enable(enable)) { |
+ LOG(LS_ERROR) << "Failed to enable/disable VAD: " << enable; |
+ return; |
+ } |
+ if (vd->set_likelihood(VoiceDetection::kVeryLowLikelihood)) { |
+ LOG(LS_ERROR) << "Failed to set low VAD likelihood."; |
+ return; |
+ } |
+ LOG(LS_INFO) << "VAD set to " << enable << " for typing detection."; |
+#endif |
+} |
+} // namespace apm_helpers |
+} // namespace webrtc |