blob: c0b3d52f3f994a79cf9aeb38edaee1e3cf712d5e [file] [log] [blame]
niklase@google.com470e71d2011-07-07 08:21:25 +00001/*
andrew@webrtc.org648af742012-02-08 01:57:29 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
niklase@google.com470e71d2011-07-07 08:21:25 +00003 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#ifndef MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_
12#define MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_
niklase@google.com470e71d2011-07-07 08:21:25 +000013
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -070014// MSVC++ requires this to be set before any other includes to get M_PI.
Patrik Höglund3ff90f12017-12-12 14:41:53 +010015#ifndef _USE_MATH_DEFINES
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -070016#define _USE_MATH_DEFINES
Patrik Höglund3ff90f12017-12-12 14:41:53 +010017#endif
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -070018
19#include <math.h>
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000020#include <stddef.h> // size_t
Yves Gerey665174f2018-06-19 15:03:05 +020021#include <stdio.h> // FILE
peah8cee56f2017-08-24 22:36:53 -070022#include <string.h>
Jonas Olssona4d87372019-07-05 19:08:33 +020023
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +000024#include <vector>
ajm@google.com22e65152011-07-18 18:03:01 +000025
Ali Tofigh1fa87c42022-07-25 22:07:08 +020026#include "absl/strings/string_view.h"
Danil Chapovalovdb9f7ab2018-06-19 10:50:11 +020027#include "absl/types/optional.h"
Sam Zackrissonab866a22020-05-07 13:07:49 +020028#include "api/array_view.h"
Gustaf Ullbergbffa3002018-02-14 15:12:00 +010029#include "api/audio/echo_canceller3_config.h"
Gustaf Ullbergfd4ce502018-02-15 10:09:09 +010030#include "api/audio/echo_control.h"
Mirko Bonadeid9708072019-01-25 20:26:48 +010031#include "api/scoped_refptr.h"
Ivo Creusen56d46092017-11-24 17:29:59 +010032#include "modules/audio_processing/include/audio_processing_statistics.h"
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020033#include "rtc_base/arraysize.h"
Steve Anton10542f22019-01-11 09:11:00 -080034#include "rtc_base/ref_count.h"
Per Åhgren09e9a832020-05-11 11:03:47 +020035#include "rtc_base/system/file_wrapper.h"
Mirko Bonadei3d255302018-10-11 10:50:45 +020036#include "rtc_base/system/rtc_export.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000037
Per Åhgren09e9a832020-05-11 11:03:47 +020038namespace rtc {
39class TaskQueue;
40} // namespace rtc
41
niklase@google.com470e71d2011-07-07 08:21:25 +000042namespace webrtc {
43
aleloi868f32f2017-05-23 07:20:05 -070044class AecDump;
Sam Zackrisson0beac582017-09-25 12:04:02 +020045class AudioBuffer;
Michael Graczykdfa36052015-03-25 16:37:27 -070046
Michael Graczyk86c6d332015-07-23 11:41:39 -070047class StreamConfig;
48class ProcessingConfig;
49
Ivo Creusen09fa4b02018-01-11 16:08:54 +010050class EchoDetector;
Valeriia Nemychnikovaf06eb572018-08-29 10:37:09 +020051class CustomAudioAnalyzer;
Alex Loiko5825aa62017-12-18 16:02:40 +010052class CustomProcessing;
niklase@google.com470e71d2011-07-07 08:21:25 +000053
54// The Audio Processing Module (APM) provides a collection of voice processing
55// components designed for real-time communications software.
56//
57// APM operates on two audio streams on a frame-by-frame basis. Frames of the
58// primary stream, on which all processing is applied, are passed to
Artem Titov0b489302021-07-28 20:50:03 +020059// `ProcessStream()`. Frames of the reverse direction stream are passed to
60// `ProcessReverseStream()`. On the client-side, this will typically be the
aluebsb0319552016-03-17 20:39:53 -070061// near-end (capture) and far-end (render) streams, respectively. APM should be
62// placed in the signal chain as close to the audio hardware abstraction layer
63// (HAL) as possible.
niklase@google.com470e71d2011-07-07 08:21:25 +000064//
65// On the server-side, the reverse stream will normally not be used, with
66// processing occurring on each incoming stream.
67//
68// Component interfaces follow a similar pattern and are accessed through
69// corresponding getters in APM. All components are disabled at create-time,
70// with default settings that are recommended for most situations. New settings
71// can be applied without enabling a component. Enabling a component triggers
72// memory allocation and initialization to allow it to start processing the
73// streams.
74//
75// Thread safety is provided with the following assumptions to reduce locking
76// overhead:
77// 1. The stream getters and setters are called from the same thread as
78// ProcessStream(). More precisely, stream functions are never called
79// concurrently with ProcessStream().
80// 2. Parameter getters are never called concurrently with the corresponding
81// setter.
82//
Sam Zackrisson3bd444f2022-08-03 14:37:00 +020083// APM accepts only linear PCM audio data in chunks of ~10 ms (see
84// AudioProcessing::GetFrameSize() for details). The int16 interfaces use
85// interleaved data, while the float interfaces use deinterleaved data.
niklase@google.com470e71d2011-07-07 08:21:25 +000086//
87// Usage example, omitting error checking:
Ivo Creusen62337e52018-01-09 14:17:33 +010088// AudioProcessing* apm = AudioProcessingBuilder().Create();
niklase@google.com470e71d2011-07-07 08:21:25 +000089//
peah88ac8532016-09-12 16:47:25 -070090// AudioProcessing::Config config;
Sam Zackrissoncdf0e6d2018-09-17 11:05:17 +020091// config.echo_canceller.enabled = true;
92// config.echo_canceller.mobile_mode = false;
Sam Zackrisson41478c72019-10-15 10:10:26 +020093//
94// config.gain_controller1.enabled = true;
95// config.gain_controller1.mode =
96// AudioProcessing::Config::GainController1::kAdaptiveAnalog;
97// config.gain_controller1.analog_level_minimum = 0;
98// config.gain_controller1.analog_level_maximum = 255;
99//
Sam Zackrissonab1aee02018-03-05 15:59:06 +0100100// config.gain_controller2.enabled = true;
Sam Zackrisson41478c72019-10-15 10:10:26 +0200101//
102// config.high_pass_filter.enabled = true;
103//
peah88ac8532016-09-12 16:47:25 -0700104// apm->ApplyConfig(config)
105//
niklase@google.com470e71d2011-07-07 08:21:25 +0000106// apm->noise_reduction()->set_level(kHighSuppression);
107// apm->noise_reduction()->Enable(true);
108//
niklase@google.com470e71d2011-07-07 08:21:25 +0000109// // Start a voice call...
110//
111// // ... Render frame arrives bound for the audio HAL ...
aluebsb0319552016-03-17 20:39:53 -0700112// apm->ProcessReverseStream(render_frame);
niklase@google.com470e71d2011-07-07 08:21:25 +0000113//
114// // ... Capture frame arrives from the audio HAL ...
115// // Call required set_stream_ functions.
116// apm->set_stream_delay_ms(delay_ms);
Sam Zackrisson41478c72019-10-15 10:10:26 +0200117// apm->set_stream_analog_level(analog_level);
niklase@google.com470e71d2011-07-07 08:21:25 +0000118//
119// apm->ProcessStream(capture_frame);
120//
121// // Call required stream_ functions.
Sam Zackrisson41478c72019-10-15 10:10:26 +0200122// analog_level = apm->recommended_stream_analog_level();
niklase@google.com470e71d2011-07-07 08:21:25 +0000123// has_voice = apm->stream_has_voice();
124//
Hua, Chunboe61a40e2021-01-08 16:34:49 +0800125// // Repeat render and capture processing for the duration of the call...
niklase@google.com470e71d2011-07-07 08:21:25 +0000126// // Start a new call...
127// apm->Initialize();
128//
129// // Close the application...
andrew@webrtc.orgf3930e92013-09-18 22:37:32 +0000130// delete apm;
niklase@google.com470e71d2011-07-07 08:21:25 +0000131//
Mirko Bonadei35214fc2019-09-23 14:54:28 +0200132class RTC_EXPORT AudioProcessing : public rtc::RefCountInterface {
niklase@google.com470e71d2011-07-07 08:21:25 +0000133 public:
peah88ac8532016-09-12 16:47:25 -0700134 // The struct below constitutes the new parameter scheme for the audio
135 // processing. It is being introduced gradually and until it is fully
136 // introduced, it is prone to change.
137 // TODO(peah): Remove this comment once the new config scheme is fully rolled
138 // out.
139 //
140 // The parameters and behavior of the audio processing module are controlled
141 // by changing the default values in the AudioProcessing::Config struct.
142 // The config is applied by passing the struct to the ApplyConfig method.
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100143 //
144 // This config is intended to be used during setup, and to enable/disable
145 // top-level processing effects. Use during processing may cause undesired
146 // submodule resets, affecting the audio quality. Use the RuntimeSetting
147 // construct for runtime configuration.
Mirko Bonadeid4002a72019-11-12 20:11:48 +0100148 struct RTC_EXPORT Config {
Per Åhgrenfcbe4072019-09-15 00:27:58 +0200149 // Sets the properties of the audio processing pipeline.
Mirko Bonadeid4002a72019-11-12 20:11:48 +0100150 struct RTC_EXPORT Pipeline {
Per Åhgrenfcbe4072019-09-15 00:27:58 +0200151 // Maximum allowed processing rate used internally. May only be set to
Per Åhgren68c225d2021-01-21 23:03:32 +0100152 // 32000 or 48000 and any differing values will be treated as 48000.
153 int maximum_internal_processing_rate = 48000;
Per Åhgrene14cb992019-11-27 09:34:22 +0100154 // Allow multi-channel processing of render audio.
155 bool multi_channel_render = false;
156 // Allow multi-channel processing of capture audio when AEC3 is active
157 // or a custom AEC is injected..
158 bool multi_channel_capture = false;
Per Åhgrenfcbe4072019-09-15 00:27:58 +0200159 } pipeline;
160
Sam Zackrisson23513132019-01-11 15:10:32 +0100161 // Enabled the pre-amplifier. It amplifies the capture signal
162 // before any other processing is done.
Per Åhgrendb5d7282021-03-15 16:31:04 +0000163 // TODO(webrtc:5298): Deprecate and use the pre-gain functionality in
164 // capture_level_adjustment instead.
Sam Zackrisson23513132019-01-11 15:10:32 +0100165 struct PreAmplifier {
166 bool enabled = false;
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200167 float fixed_gain_factor = 1.0f;
Sam Zackrisson23513132019-01-11 15:10:32 +0100168 } pre_amplifier;
169
Per Åhgrendb5d7282021-03-15 16:31:04 +0000170 // Functionality for general level adjustment in the capture pipeline. This
171 // should not be used together with the legacy PreAmplifier functionality.
172 struct CaptureLevelAdjustment {
173 bool operator==(const CaptureLevelAdjustment& rhs) const;
174 bool operator!=(const CaptureLevelAdjustment& rhs) const {
175 return !(*this == rhs);
176 }
177 bool enabled = false;
178 // The `pre_gain_factor` scales the signal before any processing is done.
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200179 float pre_gain_factor = 1.0f;
Per Åhgrendb5d7282021-03-15 16:31:04 +0000180 // The `post_gain_factor` scales the signal after all processing is done.
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200181 float post_gain_factor = 1.0f;
Per Åhgrendb5d7282021-03-15 16:31:04 +0000182 struct AnalogMicGainEmulation {
183 bool operator==(const AnalogMicGainEmulation& rhs) const;
184 bool operator!=(const AnalogMicGainEmulation& rhs) const {
185 return !(*this == rhs);
186 }
187 bool enabled = false;
188 // Initial analog gain level to use for the emulated analog gain. Must
189 // be in the range [0...255].
190 int initial_level = 255;
191 } analog_mic_gain_emulation;
192 } capture_level_adjustment;
193
Sam Zackrisson23513132019-01-11 15:10:32 +0100194 struct HighPassFilter {
195 bool enabled = false;
Per Åhgrenc0424252019-12-10 13:04:15 +0100196 bool apply_in_full_band = true;
Sam Zackrisson23513132019-01-11 15:10:32 +0100197 } high_pass_filter;
198
Sam Zackrisson8b5d2cc2018-07-27 13:27:23 +0200199 struct EchoCanceller {
200 bool enabled = false;
201 bool mobile_mode = false;
Per Åhgrenc20a19c2019-11-13 11:12:29 +0100202 bool export_linear_aec_output = false;
Per Åhgrenb8106462019-12-04 08:34:12 +0100203 // Enforce the highpass filter to be on (has no effect for the mobile
204 // mode).
Per Åhgrenbcce4532019-12-03 13:52:40 +0100205 bool enforce_high_pass_filtering = true;
Sam Zackrisson8b5d2cc2018-07-27 13:27:23 +0200206 } echo_canceller;
207
Sam Zackrisson23513132019-01-11 15:10:32 +0100208 // Enables background noise suppression.
209 struct NoiseSuppression {
peah8271d042016-11-22 07:24:52 -0800210 bool enabled = false;
Sam Zackrisson23513132019-01-11 15:10:32 +0100211 enum Level { kLow, kModerate, kHigh, kVeryHigh };
212 Level level = kModerate;
Per Åhgren2e8e1c62019-12-20 00:42:22 +0100213 bool analyze_linear_aec_output_when_available = false;
Sam Zackrisson23513132019-01-11 15:10:32 +0100214 } noise_suppression;
peahe0eae3c2016-12-14 01:16:23 -0800215
Per Åhgrenc0734712020-01-02 15:15:36 +0100216 // Enables transient suppression.
217 struct TransientSuppression {
218 bool enabled = false;
219 } transient_suppression;
220
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100221 // Enables automatic gain control (AGC) functionality.
222 // The automatic gain control (AGC) component brings the signal to an
223 // appropriate range. This is done by applying a digital gain directly and,
224 // in the analog mode, prescribing an analog gain to be applied at the audio
225 // HAL.
226 // Recommended to be enabled on the client-side.
Alessio Bazzicadfc11d52021-05-07 11:58:11 +0200227 struct RTC_EXPORT GainController1 {
Alessio Bazzica3438a932020-10-14 12:47:50 +0200228 bool operator==(const GainController1& rhs) const;
229 bool operator!=(const GainController1& rhs) const {
230 return !(*this == rhs);
231 }
232
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100233 bool enabled = false;
234 enum Mode {
235 // Adaptive mode intended for use if an analog volume control is
236 // available on the capture device. It will require the user to provide
237 // coupling between the OS mixer controls and AGC through the
238 // stream_analog_level() functions.
239 // It consists of an analog gain prescription for the audio device and a
240 // digital compression stage.
241 kAdaptiveAnalog,
242 // Adaptive mode intended for situations in which an analog volume
243 // control is unavailable. It operates in a similar fashion to the
244 // adaptive analog mode, but with scaling instead applied in the digital
245 // domain. As with the analog mode, it additionally uses a digital
246 // compression stage.
247 kAdaptiveDigital,
248 // Fixed mode which enables only the digital compression stage also used
249 // by the two adaptive modes.
250 // It is distinguished from the adaptive modes by considering only a
251 // short time-window of the input signal. It applies a fixed gain
252 // through most of the input level range, and compresses (gradually
253 // reduces gain with increasing level) the input signal at higher
254 // levels. This mode is preferred on embedded devices where the capture
255 // signal level is predictable, so that a known gain can be applied.
256 kFixedDigital
257 };
258 Mode mode = kAdaptiveAnalog;
259 // Sets the target peak level (or envelope) of the AGC in dBFs (decibels
260 // from digital full-scale). The convention is to use positive values. For
261 // instance, passing in a value of 3 corresponds to -3 dBFs, or a target
262 // level 3 dB below full-scale. Limited to [0, 31].
263 int target_level_dbfs = 3;
264 // Sets the maximum gain the digital compression stage may apply, in dB. A
265 // higher number corresponds to greater compression, while a value of 0
266 // will leave the signal uncompressed. Limited to [0, 90].
267 // For updates after APM setup, use a RuntimeSetting instead.
268 int compression_gain_db = 9;
269 // When enabled, the compression stage will hard limit the signal to the
270 // target level. Otherwise, the signal will be compressed but not limited
271 // above the target level.
272 bool enable_limiter = true;
Per Åhgren0695df12020-01-13 14:43:13 +0100273
274 // Enables the analog gain controller functionality.
275 struct AnalogGainController {
276 bool enabled = true;
Alessio Bazzica7afd6982022-10-13 17:15:36 +0200277 // TODO(bugs.webrtc.org/7494): Deprecated. Stop using and remove.
278 int startup_min_volume = 0;
Per Åhgren0695df12020-01-13 14:43:13 +0100279 // Lowest analog microphone level that will be applied in response to
280 // clipping.
Alessio Bazzica488f6692022-10-13 13:06:05 +0200281 int clipped_level_min = 70;
Alessio Bazzica866caeb2022-07-19 12:18:38 +0200282 // If true, an adaptive digital gain is applied.
Per Åhgren0695df12020-01-13 14:43:13 +0100283 bool enable_digital_adaptive = true;
Hanna Silenb8dc7fa2021-05-20 17:37:56 +0200284 // Amount the microphone level is lowered with every clipping event.
285 // Limited to (0, 255].
286 int clipped_level_step = 15;
287 // Proportion of clipped samples required to declare a clipping event.
288 // Limited to (0.f, 1.f).
289 float clipped_ratio_threshold = 0.1f;
290 // Time in frames to wait after a clipping event before checking again.
291 // Limited to values higher than 0.
292 int clipped_wait_frames = 300;
Hanna Silena43953a2021-06-02 17:13:24 +0200293
294 // Enables clipping prediction functionality.
295 struct ClippingPredictor {
296 bool enabled = false;
297 enum Mode {
Alessio Bazzicab237a872021-06-11 12:37:54 +0200298 // Clipping event prediction mode with fixed step estimation.
Hanna Silena43953a2021-06-02 17:13:24 +0200299 kClippingEventPrediction,
Alessio Bazzicab237a872021-06-11 12:37:54 +0200300 // Clipped peak estimation mode with adaptive step estimation.
Hanna Silena43953a2021-06-02 17:13:24 +0200301 kAdaptiveStepClippingPeakPrediction,
Alessio Bazzicab237a872021-06-11 12:37:54 +0200302 // Clipped peak estimation mode with fixed step estimation.
Hanna Silena43953a2021-06-02 17:13:24 +0200303 kFixedStepClippingPeakPrediction,
304 };
305 Mode mode = kClippingEventPrediction;
Alessio Bazzicab237a872021-06-11 12:37:54 +0200306 // Number of frames in the sliding analysis window.
Hanna Silena43953a2021-06-02 17:13:24 +0200307 int window_length = 5;
Alessio Bazzicab237a872021-06-11 12:37:54 +0200308 // Number of frames in the sliding reference window.
Hanna Silena43953a2021-06-02 17:13:24 +0200309 int reference_window_length = 5;
Alessio Bazzicab237a872021-06-11 12:37:54 +0200310 // Reference window delay (unit: number of frames).
Hanna Silena43953a2021-06-02 17:13:24 +0200311 int reference_window_delay = 5;
Alessio Bazzicab237a872021-06-11 12:37:54 +0200312 // Clipping prediction threshold (dBFS).
Hanna Silena43953a2021-06-02 17:13:24 +0200313 float clipping_threshold = -1.0f;
314 // Crest factor drop threshold (dB).
315 float crest_factor_margin = 3.0f;
Alessio Bazzica42dacda2021-06-17 17:18:46 +0200316 // If true, the recommended clipped level step is used to modify the
317 // analog gain. Otherwise, the predictor runs without affecting the
318 // analog gain.
319 bool use_predicted_step = true;
Hanna Silena43953a2021-06-02 17:13:24 +0200320 } clipping_predictor;
Per Åhgren0695df12020-01-13 14:43:13 +0100321 } analog_gain_controller;
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100322 } gain_controller1;
323
Alex Loikoe5831742018-08-24 11:28:36 +0200324 // Enables the next generation AGC functionality. This feature replaces the
325 // standard methods of gain control in the previous AGC. Enabling this
326 // submodule enables an adaptive digital AGC followed by a limiter. By
Artem Titov0b489302021-07-28 20:50:03 +0200327 // setting `fixed_gain_db`, the limiter can be turned into a compressor that
Alex Loikoe5831742018-08-24 11:28:36 +0200328 // first applies a fixed gain. The adaptive digital AGC can be turned off by
329 // setting |adaptive_digital_mode=false|.
Alessio Bazzicadfc11d52021-05-07 11:58:11 +0200330 struct RTC_EXPORT GainController2 {
Alessio Bazzica3438a932020-10-14 12:47:50 +0200331 bool operator==(const GainController2& rhs) const;
332 bool operator!=(const GainController2& rhs) const {
333 return !(*this == rhs);
334 }
335
alessiob3ec96df2017-05-22 06:57:06 -0700336 bool enabled = false;
Alessio Bazzica253f8362020-11-27 16:02:38 +0100337 struct FixedDigital {
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200338 float gain_db = 0.0f;
Alessio Bazzica1e2542f2018-11-13 14:44:15 +0100339 } fixed_digital;
Alessio Bazzicadfc11d52021-05-07 11:58:11 +0200340 struct RTC_EXPORT AdaptiveDigital {
Alessio Bazzicaa2efd152021-04-29 16:17:49 +0200341 bool operator==(const AdaptiveDigital& rhs) const;
342 bool operator!=(const AdaptiveDigital& rhs) const {
343 return !(*this == rhs);
344 }
345
Alessio Bazzica8da7b352018-11-21 10:50:58 +0100346 bool enabled = false;
Alessio Bazzicaa850e6c2021-10-04 13:35:55 +0200347 // When true, the adaptive digital controller runs but the signal is not
348 // modified.
Alessio Bazzicad66a6052021-04-29 16:13:25 +0200349 bool dry_run = false;
Alessio Bazzicaa850e6c2021-10-04 13:35:55 +0200350 float headroom_db = 6.0f;
351 // TODO(bugs.webrtc.org/7494): Consider removing and inferring from
352 // `max_output_noise_level_dbfs`.
353 float max_gain_db = 30.0f;
354 float initial_gain_db = 8.0f;
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200355 int vad_reset_period_ms = 1500;
Alessio Bazzica980c4602021-04-14 19:09:17 +0200356 int adjacent_speech_frames_threshold = 12;
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200357 float max_gain_change_db_per_second = 3.0f;
Alessio Bazzica980c4602021-04-14 19:09:17 +0200358 float max_output_noise_level_dbfs = -50.0f;
Alessio Bazzica1e2542f2018-11-13 14:44:15 +0100359 } adaptive_digital;
Hanna Silen9f06ef12022-11-01 17:17:54 +0100360
361 // Enables input volume control in AGC2.
362 struct InputVolumeController {
363 bool operator==(const InputVolumeController& rhs) const;
364 bool operator!=(const InputVolumeController& rhs) const {
365 return !(*this == rhs);
366 }
367 bool enabled = false;
368 } input_volume_controller;
alessiob3ec96df2017-05-22 06:57:06 -0700369 } gain_controller2;
peah8cee56f2017-08-24 22:36:53 -0700370
Artem Titov59bbd652019-08-02 11:31:37 +0200371 std::string ToString() const;
peah88ac8532016-09-12 16:47:25 -0700372 };
373
Alessio Bazzicac054e782018-04-16 12:10:09 +0200374 // Specifies the properties of a setting to be passed to AudioProcessing at
375 // runtime.
376 class RuntimeSetting {
377 public:
Alex Loiko73ec0192018-05-15 10:52:28 +0200378 enum class Type {
379 kNotSpecified,
380 kCapturePreGain,
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100381 kCaptureCompressionGain,
Per Åhgren6ee75fd2019-04-26 11:33:37 +0200382 kCaptureFixedPostGain,
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200383 kPlayoutVolumeChange,
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100384 kCustomRenderProcessingRuntimeSetting,
Per Åhgren552d3e32020-08-12 08:46:47 +0200385 kPlayoutAudioDeviceChange,
Per Åhgrendb5d7282021-03-15 16:31:04 +0000386 kCapturePostGain,
Per Åhgren552d3e32020-08-12 08:46:47 +0200387 kCaptureOutputUsed
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100388 };
389
390 // Play-out audio device properties.
391 struct PlayoutAudioDeviceInfo {
392 int id; // Identifies the audio device.
393 int max_volume; // Maximum play-out volume.
Alex Loiko73ec0192018-05-15 10:52:28 +0200394 };
Alessio Bazzicac054e782018-04-16 12:10:09 +0200395
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200396 RuntimeSetting() : type_(Type::kNotSpecified), value_(0.0f) {}
Alessio Bazzicac054e782018-04-16 12:10:09 +0200397 ~RuntimeSetting() = default;
398
399 static RuntimeSetting CreateCapturePreGain(float gain) {
Alessio Bazzicac054e782018-04-16 12:10:09 +0200400 return {Type::kCapturePreGain, gain};
401 }
402
Per Åhgrendb5d7282021-03-15 16:31:04 +0000403 static RuntimeSetting CreateCapturePostGain(float gain) {
404 return {Type::kCapturePostGain, gain};
405 }
406
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100407 // Corresponds to Config::GainController1::compression_gain_db, but for
408 // runtime configuration.
409 static RuntimeSetting CreateCompressionGainDb(int gain_db) {
410 RTC_DCHECK_GE(gain_db, 0);
411 RTC_DCHECK_LE(gain_db, 90);
412 return {Type::kCaptureCompressionGain, static_cast<float>(gain_db)};
413 }
414
Per Åhgren6ee75fd2019-04-26 11:33:37 +0200415 // Corresponds to Config::GainController2::fixed_digital::gain_db, but for
416 // runtime configuration.
417 static RuntimeSetting CreateCaptureFixedPostGain(float gain_db) {
Alessio Bazzica841d74e2021-03-31 15:04:03 +0200418 RTC_DCHECK_GE(gain_db, 0.0f);
419 RTC_DCHECK_LE(gain_db, 90.0f);
Per Åhgren6ee75fd2019-04-26 11:33:37 +0200420 return {Type::kCaptureFixedPostGain, gain_db};
421 }
422
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100423 // Creates a runtime setting to notify play-out (aka render) audio device
424 // changes.
425 static RuntimeSetting CreatePlayoutAudioDeviceChange(
426 PlayoutAudioDeviceInfo audio_device) {
427 return {Type::kPlayoutAudioDeviceChange, audio_device};
428 }
429
430 // Creates a runtime setting to notify play-out (aka render) volume changes.
Artem Titov0b489302021-07-28 20:50:03 +0200431 // `volume` is the unnormalized volume, the maximum of which
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200432 static RuntimeSetting CreatePlayoutVolumeChange(int volume) {
433 return {Type::kPlayoutVolumeChange, volume};
434 }
435
Alex Loiko73ec0192018-05-15 10:52:28 +0200436 static RuntimeSetting CreateCustomRenderSetting(float payload) {
437 return {Type::kCustomRenderProcessingRuntimeSetting, payload};
438 }
439
Per Åhgren652ada52021-03-03 10:52:44 +0000440 static RuntimeSetting CreateCaptureOutputUsedSetting(
441 bool capture_output_used) {
442 return {Type::kCaptureOutputUsed, capture_output_used};
Per Åhgren552d3e32020-08-12 08:46:47 +0200443 }
444
Alessio Bazzicac054e782018-04-16 12:10:09 +0200445 Type type() const { return type_; }
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100446 // Getters do not return a value but instead modify the argument to protect
447 // from implicit casting.
Alessio Bazzicac054e782018-04-16 12:10:09 +0200448 void GetFloat(float* value) const {
449 RTC_DCHECK(value);
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200450 *value = value_.float_value;
451 }
452 void GetInt(int* value) const {
453 RTC_DCHECK(value);
454 *value = value_.int_value;
Alessio Bazzicac054e782018-04-16 12:10:09 +0200455 }
Per Åhgren552d3e32020-08-12 08:46:47 +0200456 void GetBool(bool* value) const {
457 RTC_DCHECK(value);
458 *value = value_.bool_value;
459 }
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100460 void GetPlayoutAudioDeviceInfo(PlayoutAudioDeviceInfo* value) const {
461 RTC_DCHECK(value);
462 *value = value_.playout_audio_device_info;
463 }
Alessio Bazzicac054e782018-04-16 12:10:09 +0200464
465 private:
466 RuntimeSetting(Type id, float value) : type_(id), value_(value) {}
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200467 RuntimeSetting(Type id, int value) : type_(id), value_(value) {}
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100468 RuntimeSetting(Type id, PlayoutAudioDeviceInfo value)
469 : type_(id), value_(value) {}
Alessio Bazzicac054e782018-04-16 12:10:09 +0200470 Type type_;
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200471 union U {
472 U() {}
473 U(int value) : int_value(value) {}
474 U(float value) : float_value(value) {}
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100475 U(PlayoutAudioDeviceInfo value) : playout_audio_device_info(value) {}
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200476 float float_value;
477 int int_value;
Per Åhgren552d3e32020-08-12 08:46:47 +0200478 bool bool_value;
Alessio Bazzica7c19a702019-11-07 13:22:00 +0100479 PlayoutAudioDeviceInfo playout_audio_device_info;
Fredrik Hernqvistca362852019-05-10 15:50:02 +0200480 } value_;
Alessio Bazzicac054e782018-04-16 12:10:09 +0200481 };
482
peaha9cc40b2017-06-29 08:32:09 -0700483 ~AudioProcessing() override {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000484
niklase@google.com470e71d2011-07-07 08:21:25 +0000485 // Initializes internal states, while retaining all user settings. This
486 // should be called before beginning to process a new audio stream. However,
487 // it is not necessary to call before processing the first stream after
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000488 // creation.
489 //
490 // It is also not necessary to call if the audio parameters (sample
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000491 // rate and number of channels) have changed. Passing updated parameters
Artem Titov0b489302021-07-28 20:50:03 +0200492 // directly to `ProcessStream()` and `ProcessReverseStream()` is permissible.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000493 // If the parameters are known at init-time though, they may be provided.
Per Åhgren0ade9832020-09-01 23:57:20 +0200494 // TODO(webrtc:5298): Change to return void.
niklase@google.com470e71d2011-07-07 08:21:25 +0000495 virtual int Initialize() = 0;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000496
497 // The int16 interfaces require:
Artem Titov0b489302021-07-28 20:50:03 +0200498 // - only `NativeRate`s be used
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000499 // - that the input, output and reverse rates must match
Artem Titovcfea2182021-08-10 01:22:31 +0200500 // - that `processing_config.output_stream()` matches
501 // `processing_config.input_stream()`.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000502 //
Michael Graczyk86c6d332015-07-23 11:41:39 -0700503 // The float interfaces accept arbitrary rates and support differing input and
504 // output layouts, but the output must have either one channel or the same
505 // number of channels as the input.
506 virtual int Initialize(const ProcessingConfig& processing_config) = 0;
507
peah88ac8532016-09-12 16:47:25 -0700508 // TODO(peah): This method is a temporary solution used to take control
509 // over the parameters in the audio processing module and is likely to change.
510 virtual void ApplyConfig(const Config& config) = 0;
511
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000512 // TODO(ajm): Only intended for internal use. Make private and friend the
513 // necessary classes?
514 virtual int proc_sample_rate_hz() const = 0;
515 virtual int proc_split_sample_rate_hz() const = 0;
Peter Kasting69558702016-01-12 16:26:35 -0800516 virtual size_t num_input_channels() const = 0;
517 virtual size_t num_proc_channels() const = 0;
518 virtual size_t num_output_channels() const = 0;
519 virtual size_t num_reverse_channels() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000520
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000521 // Set to true when the output of AudioProcessing will be muted or in some
522 // other way not used. Ideally, the captured audio would still be processed,
523 // but some components may change behavior based on this information.
Per Åhgren0a144a72021-02-09 08:47:51 +0100524 // Default false. This method takes a lock. To achieve this in a lock-less
525 // manner the PostRuntimeSetting can instead be used.
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000526 virtual void set_output_will_be_muted(bool muted) = 0;
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000527
Per Åhgren0a144a72021-02-09 08:47:51 +0100528 // Enqueues a runtime setting.
Alessio Bazzicac054e782018-04-16 12:10:09 +0200529 virtual void SetRuntimeSetting(RuntimeSetting setting) = 0;
530
Per Åhgren0a144a72021-02-09 08:47:51 +0100531 // Enqueues a runtime setting. Returns a bool indicating whether the
532 // enqueueing was successfull.
Per Åhgren8eea1172021-02-09 23:15:07 +0100533 virtual bool PostRuntimeSetting(RuntimeSetting setting) = 0;
Per Åhgren0a144a72021-02-09 08:47:51 +0100534
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200535 // Accepts and produces a ~10 ms frame of interleaved 16 bit integer audio as
Artem Titov0b489302021-07-28 20:50:03 +0200536 // specified in `input_config` and `output_config`. `src` and `dest` may use
Per Åhgren645f24c2020-03-16 12:06:02 +0100537 // the same memory, if desired.
538 virtual int ProcessStream(const int16_t* const src,
539 const StreamConfig& input_config,
540 const StreamConfig& output_config,
Per Åhgrendc5522b2020-03-19 14:55:58 +0100541 int16_t* const dest) = 0;
Per Åhgren645f24c2020-03-16 12:06:02 +0100542
Michael Graczyk86c6d332015-07-23 11:41:39 -0700543 // Accepts deinterleaved float audio with the range [-1, 1]. Each element of
Artem Titov0b489302021-07-28 20:50:03 +0200544 // `src` points to a channel buffer, arranged according to `input_stream`. At
545 // output, the channels will be arranged according to `output_stream` in
546 // `dest`.
Michael Graczyk86c6d332015-07-23 11:41:39 -0700547 //
Artem Titov0b489302021-07-28 20:50:03 +0200548 // The output must have one channel or as many channels as the input. `src`
549 // and `dest` may use the same memory, if desired.
Michael Graczyk86c6d332015-07-23 11:41:39 -0700550 virtual int ProcessStream(const float* const* src,
551 const StreamConfig& input_config,
552 const StreamConfig& output_config,
553 float* const* dest) = 0;
554
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200555 // Accepts and produces a ~10 ms frame of interleaved 16 bit integer audio for
Artem Titov0b489302021-07-28 20:50:03 +0200556 // the reverse direction audio stream as specified in `input_config` and
557 // `output_config`. `src` and `dest` may use the same memory, if desired.
Per Åhgren645f24c2020-03-16 12:06:02 +0100558 virtual int ProcessReverseStream(const int16_t* const src,
559 const StreamConfig& input_config,
560 const StreamConfig& output_config,
561 int16_t* const dest) = 0;
562
Michael Graczyk86c6d332015-07-23 11:41:39 -0700563 // Accepts deinterleaved float audio with the range [-1, 1]. Each element of
Artem Titov0b489302021-07-28 20:50:03 +0200564 // `data` points to a channel buffer, arranged according to `reverse_config`.
ekmeyerson60d9b332015-08-14 10:35:55 -0700565 virtual int ProcessReverseStream(const float* const* src,
peahde65ddc2016-09-16 15:02:15 -0700566 const StreamConfig& input_config,
567 const StreamConfig& output_config,
ekmeyerson60d9b332015-08-14 10:35:55 -0700568 float* const* dest) = 0;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700569
Gustaf Ullbergcb307262019-10-29 09:30:44 +0100570 // Accepts deinterleaved float audio with the range [-1, 1]. Each element
Artem Titov0b489302021-07-28 20:50:03 +0200571 // of `data` points to a channel buffer, arranged according to
572 // `reverse_config`.
Gustaf Ullbergcb307262019-10-29 09:30:44 +0100573 virtual int AnalyzeReverseStream(const float* const* data,
574 const StreamConfig& reverse_config) = 0;
575
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200576 // Returns the most recently produced ~10 ms of the linear AEC output at a
577 // rate of 16 kHz. If there is more than one capture channel, a mono
578 // representation of the input is returned. Returns true/false to indicate
579 // whether an output returned.
Per Åhgrenc20a19c2019-11-13 11:12:29 +0100580 virtual bool GetLinearAecOutput(
581 rtc::ArrayView<std::array<float, 160>> linear_output) const = 0;
582
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100583 // This must be called prior to ProcessStream() if and only if adaptive analog
584 // gain control is enabled, to pass the current analog level from the audio
Hanna Silencd597042021-11-02 11:02:48 +0100585 // HAL. Must be within the range [0, 255].
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100586 virtual void set_stream_analog_level(int level) = 0;
587
Alessio Bazzicafcf1af32022-09-07 17:14:26 +0200588 // When an analog mode is set, this should be called after
589 // `set_stream_analog_level()` and `ProcessStream()` to obtain the recommended
590 // new analog level for the audio HAL. It is the user's responsibility to
591 // apply this level.
Sam Zackrissonf0d1c032019-03-27 13:28:08 +0100592 virtual int recommended_stream_analog_level() const = 0;
593
niklase@google.com470e71d2011-07-07 08:21:25 +0000594 // This must be called if and only if echo processing is enabled.
595 //
Artem Titov0b489302021-07-28 20:50:03 +0200596 // Sets the `delay` in ms between ProcessReverseStream() receiving a far-end
niklase@google.com470e71d2011-07-07 08:21:25 +0000597 // frame and ProcessStream() receiving a near-end frame containing the
598 // corresponding echo. On the client-side this can be expressed as
599 // delay = (t_render - t_analyze) + (t_process - t_capture)
600 // where,
aluebsb0319552016-03-17 20:39:53 -0700601 // - t_analyze is the time a frame is passed to ProcessReverseStream() and
niklase@google.com470e71d2011-07-07 08:21:25 +0000602 // t_render is the time the first sample of the same frame is rendered by
603 // the audio hardware.
604 // - t_capture is the time the first sample of a frame is captured by the
alessiob13fc1802017-04-19 05:35:51 -0700605 // audio hardware and t_process is the time the same frame is passed to
niklase@google.com470e71d2011-07-07 08:21:25 +0000606 // ProcessStream().
607 virtual int set_stream_delay_ms(int delay) = 0;
608 virtual int stream_delay_ms() const = 0;
609
andrew@webrtc.org75dd2882014-02-11 20:52:30 +0000610 // Call to signal that a key press occurred (true) or did not occur (false)
611 // with this chunk of audio.
612 virtual void set_stream_key_pressed(bool key_pressed) = 0;
andrew@webrtc.org75dd2882014-02-11 20:52:30 +0000613
Per Åhgren09e9a832020-05-11 11:03:47 +0200614 // Creates and attaches an webrtc::AecDump for recording debugging
615 // information.
Artem Titov0b489302021-07-28 20:50:03 +0200616 // The `worker_queue` may not be null and must outlive the created
Per Åhgren09e9a832020-05-11 11:03:47 +0200617 // AecDump instance. |max_log_size_bytes == -1| means the log size
Artem Titov0b489302021-07-28 20:50:03 +0200618 // will be unlimited. `handle` may not be null. The AecDump takes
619 // responsibility for `handle` and closes it in the destructor. A
Per Åhgren09e9a832020-05-11 11:03:47 +0200620 // return value of true indicates that the file has been
621 // sucessfully opened, while a value of false indicates that
622 // opening the file failed.
Ali Tofigh1fa87c42022-07-25 22:07:08 +0200623 virtual bool CreateAndAttachAecDump(absl::string_view file_name,
624 int64_t max_log_size_bytes,
Ali Tofigh980ad0c2022-08-09 09:21:17 +0200625 rtc::TaskQueue* worker_queue) = 0;
Per Åhgren09e9a832020-05-11 11:03:47 +0200626 virtual bool CreateAndAttachAecDump(FILE* handle,
627 int64_t max_log_size_bytes,
628 rtc::TaskQueue* worker_queue) = 0;
629
630 // TODO(webrtc:5298) Deprecated variant.
aleloi868f32f2017-05-23 07:20:05 -0700631 // Attaches provided webrtc::AecDump for recording debugging
632 // information. Log file and maximum file size logic is supposed to
633 // be handled by implementing instance of AecDump. Calling this
634 // method when another AecDump is attached resets the active AecDump
635 // with a new one. This causes the d-tor of the earlier AecDump to
636 // be called. The d-tor call may block until all pending logging
637 // tasks are completed.
Alex Loikobe767e02017-06-08 09:45:03 +0200638 virtual void AttachAecDump(std::unique_ptr<AecDump> aec_dump) = 0;
aleloi868f32f2017-05-23 07:20:05 -0700639
640 // If no AecDump is attached, this has no effect. If an AecDump is
641 // attached, it's destructor is called. The d-tor may block until
642 // all pending logging tasks are completed.
Alex Loikobe767e02017-06-08 09:45:03 +0200643 virtual void DetachAecDump() = 0;
aleloi868f32f2017-05-23 07:20:05 -0700644
Per Åhgrencf4c8722019-12-30 14:32:14 +0100645 // Get audio processing statistics.
646 virtual AudioProcessingStats GetStatistics() = 0;
Artem Titov0b489302021-07-28 20:50:03 +0200647 // TODO(webrtc:5298) Deprecated variant. The `has_remote_tracks` argument
Per Åhgrencf4c8722019-12-30 14:32:14 +0100648 // should be set if there are active remote tracks (this would usually be true
649 // during a call). If there are no remote tracks some of the stats will not be
650 // set by AudioProcessing, because they only make sense if there is at least
651 // one remote track.
652 virtual AudioProcessingStats GetStatistics(bool has_remote_tracks) = 0;
Ivo Creusenae026092017-11-20 13:07:16 +0100653
henrik.lundinadf06352017-04-05 05:48:24 -0700654 // Returns the last applied configuration.
henrik.lundin77492862017-04-06 23:28:09 -0700655 virtual AudioProcessing::Config GetConfig() const = 0;
henrik.lundinadf06352017-04-05 05:48:24 -0700656
andrew@webrtc.org648af742012-02-08 01:57:29 +0000657 enum Error {
658 // Fatal errors.
niklase@google.com470e71d2011-07-07 08:21:25 +0000659 kNoError = 0,
660 kUnspecifiedError = -1,
661 kCreationFailedError = -2,
662 kUnsupportedComponentError = -3,
663 kUnsupportedFunctionError = -4,
664 kNullPointerError = -5,
665 kBadParameterError = -6,
666 kBadSampleRateError = -7,
667 kBadDataLengthError = -8,
668 kBadNumberChannelsError = -9,
669 kFileError = -10,
670 kStreamParameterNotSetError = -11,
andrew@webrtc.org648af742012-02-08 01:57:29 +0000671 kNotEnabledError = -12,
niklase@google.com470e71d2011-07-07 08:21:25 +0000672
andrew@webrtc.org648af742012-02-08 01:57:29 +0000673 // Warnings are non-fatal.
niklase@google.com470e71d2011-07-07 08:21:25 +0000674 // This results when a set_stream_ parameter is out of range. Processing
675 // will continue, but the parameter may have been truncated.
andrew@webrtc.org648af742012-02-08 01:57:29 +0000676 kBadStreamParameterWarning = -13
niklase@google.com470e71d2011-07-07 08:21:25 +0000677 };
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000678
Per Åhgren2507f8c2020-03-19 12:33:29 +0100679 // Native rates supported by the integer interfaces.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000680 enum NativeRate {
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000681 kSampleRate8kHz = 8000,
682 kSampleRate16kHz = 16000,
aluebs@webrtc.org087da132014-11-17 23:01:23 +0000683 kSampleRate32kHz = 32000,
684 kSampleRate48kHz = 48000
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000685 };
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000686
kwibergd59d3bb2016-09-13 07:49:33 -0700687 // TODO(kwiberg): We currently need to support a compiler (Visual C++) that
688 // complains if we don't explicitly state the size of the array here. Remove
689 // the size when that's no longer the case.
690 static constexpr int kNativeSampleRatesHz[4] = {
691 kSampleRate8kHz, kSampleRate16kHz, kSampleRate32kHz, kSampleRate48kHz};
692 static constexpr size_t kNumNativeSampleRates =
693 arraysize(kNativeSampleRatesHz);
694 static constexpr int kMaxNativeSampleRateHz =
695 kNativeSampleRatesHz[kNumNativeSampleRates - 1];
Alejandro Luebscdfe20b2015-09-23 12:49:12 -0700696
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200697 // APM processes audio in chunks of about 10 ms. See GetFrameSize() for
698 // details.
Per Åhgren12dc2742020-12-08 09:40:35 +0100699 static constexpr int kChunkSizeMs = 10;
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200700
701 // Returns floor(sample_rate_hz/100): the number of samples per channel used
702 // as input and output to the audio processing module in calls to
703 // ProcessStream, ProcessReverseStream, AnalyzeReverseStream, and
704 // GetLinearAecOutput.
705 //
706 // This is exactly 10 ms for sample rates divisible by 100. For example:
707 // - 48000 Hz (480 samples per channel),
708 // - 44100 Hz (441 samples per channel),
709 // - 16000 Hz (160 samples per channel).
710 //
711 // Sample rates not divisible by 100 are received/produced in frames of
712 // approximately 10 ms. For example:
713 // - 22050 Hz (220 samples per channel, or ~9.98 ms per frame),
714 // - 11025 Hz (110 samples per channel, or ~9.98 ms per frame).
715 // These nondivisible sample rates yield lower audio quality compared to
716 // multiples of 100. Internal resampling to 10 ms frames causes a simulated
717 // clock drift effect which impacts the performance of (for example) echo
718 // cancellation.
719 static int GetFrameSize(int sample_rate_hz) { return sample_rate_hz / 100; }
niklase@google.com470e71d2011-07-07 08:21:25 +0000720};
721
Mirko Bonadei3d255302018-10-11 10:50:45 +0200722class RTC_EXPORT AudioProcessingBuilder {
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100723 public:
724 AudioProcessingBuilder();
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200725 AudioProcessingBuilder(const AudioProcessingBuilder&) = delete;
726 AudioProcessingBuilder& operator=(const AudioProcessingBuilder&) = delete;
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100727 ~AudioProcessingBuilder();
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200728
729 // Sets the APM configuration.
730 AudioProcessingBuilder& SetConfig(const AudioProcessing::Config& config) {
731 config_ = config;
732 return *this;
733 }
734
735 // Sets the echo controller factory to inject when APM is created.
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100736 AudioProcessingBuilder& SetEchoControlFactory(
Per Åhgrencc73ed32020-04-26 23:56:17 +0200737 std::unique_ptr<EchoControlFactory> echo_control_factory) {
738 echo_control_factory_ = std::move(echo_control_factory);
739 return *this;
740 }
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200741
742 // Sets the capture post-processing sub-module to inject when APM is created.
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100743 AudioProcessingBuilder& SetCapturePostProcessing(
Per Åhgrencc73ed32020-04-26 23:56:17 +0200744 std::unique_ptr<CustomProcessing> capture_post_processing) {
745 capture_post_processing_ = std::move(capture_post_processing);
746 return *this;
747 }
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200748
749 // Sets the render pre-processing sub-module to inject when APM is created.
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100750 AudioProcessingBuilder& SetRenderPreProcessing(
Per Åhgrencc73ed32020-04-26 23:56:17 +0200751 std::unique_ptr<CustomProcessing> render_pre_processing) {
752 render_pre_processing_ = std::move(render_pre_processing);
753 return *this;
754 }
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200755
756 // Sets the echo detector to inject when APM is created.
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100757 AudioProcessingBuilder& SetEchoDetector(
Per Åhgrencc73ed32020-04-26 23:56:17 +0200758 rtc::scoped_refptr<EchoDetector> echo_detector) {
759 echo_detector_ = std::move(echo_detector);
760 return *this;
761 }
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200762
763 // Sets the capture analyzer sub-module to inject when APM is created.
Valeriia Nemychnikovaf06eb572018-08-29 10:37:09 +0200764 AudioProcessingBuilder& SetCaptureAnalyzer(
Per Åhgrencc73ed32020-04-26 23:56:17 +0200765 std::unique_ptr<CustomAudioAnalyzer> capture_analyzer) {
766 capture_analyzer_ = std::move(capture_analyzer);
767 return *this;
768 }
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200769
770 // Creates an APM instance with the specified config or the default one if
771 // unspecified. Injects the specified components transferring the ownership
772 // to the newly created APM instance - i.e., except for the config, the
773 // builder is reset to its initial state.
Niels Möller4f776ac2021-07-02 11:30:54 +0200774 rtc::scoped_refptr<AudioProcessing> Create();
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100775
776 private:
Alessio Bazzica20a9ac62021-10-14 10:55:08 +0200777 AudioProcessing::Config config_;
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100778 std::unique_ptr<EchoControlFactory> echo_control_factory_;
779 std::unique_ptr<CustomProcessing> capture_post_processing_;
780 std::unique_ptr<CustomProcessing> render_pre_processing_;
Ivo Creusend1f970d2018-06-14 11:02:03 +0200781 rtc::scoped_refptr<EchoDetector> echo_detector_;
Valeriia Nemychnikovaf06eb572018-08-29 10:37:09 +0200782 std::unique_ptr<CustomAudioAnalyzer> capture_analyzer_;
Ivo Creusen5ec7e122017-12-22 11:35:59 +0100783};
784
Michael Graczyk86c6d332015-07-23 11:41:39 -0700785class StreamConfig {
786 public:
787 // sample_rate_hz: The sampling rate of the stream.
Henrik Lundin64253a92022-02-04 09:02:48 +0000788 // num_channels: The number of audio channels in the stream.
Alessio Bazzicac7d0e422022-02-04 17:06:55 +0100789 StreamConfig(int sample_rate_hz = 0, size_t num_channels = 0)
Michael Graczyk86c6d332015-07-23 11:41:39 -0700790 : sample_rate_hz_(sample_rate_hz),
791 num_channels_(num_channels),
Michael Graczyk86c6d332015-07-23 11:41:39 -0700792 num_frames_(calculate_frames(sample_rate_hz)) {}
793
794 void set_sample_rate_hz(int value) {
795 sample_rate_hz_ = value;
796 num_frames_ = calculate_frames(value);
797 }
Peter Kasting69558702016-01-12 16:26:35 -0800798 void set_num_channels(size_t value) { num_channels_ = value; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700799
800 int sample_rate_hz() const { return sample_rate_hz_; }
801
Henrik Lundin64253a92022-02-04 09:02:48 +0000802 // The number of channels in the stream.
Peter Kasting69558702016-01-12 16:26:35 -0800803 size_t num_channels() const { return num_channels_; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700804
Peter Kastingdce40cf2015-08-24 14:52:23 -0700805 size_t num_frames() const { return num_frames_; }
806 size_t num_samples() const { return num_channels_ * num_frames_; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700807
808 bool operator==(const StreamConfig& other) const {
809 return sample_rate_hz_ == other.sample_rate_hz_ &&
Henrik Lundin64253a92022-02-04 09:02:48 +0000810 num_channels_ == other.num_channels_;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700811 }
812
813 bool operator!=(const StreamConfig& other) const { return !(*this == other); }
814
815 private:
Peter Kastingdce40cf2015-08-24 14:52:23 -0700816 static size_t calculate_frames(int sample_rate_hz) {
Sam Zackrisson3bd444f2022-08-03 14:37:00 +0200817 return static_cast<size_t>(AudioProcessing::GetFrameSize(sample_rate_hz));
Michael Graczyk86c6d332015-07-23 11:41:39 -0700818 }
819
820 int sample_rate_hz_;
Peter Kasting69558702016-01-12 16:26:35 -0800821 size_t num_channels_;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700822 size_t num_frames_;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700823};
824
825class ProcessingConfig {
826 public:
827 enum StreamName {
828 kInputStream,
829 kOutputStream,
ekmeyerson60d9b332015-08-14 10:35:55 -0700830 kReverseInputStream,
831 kReverseOutputStream,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700832 kNumStreamNames,
833 };
834
835 const StreamConfig& input_stream() const {
836 return streams[StreamName::kInputStream];
837 }
838 const StreamConfig& output_stream() const {
839 return streams[StreamName::kOutputStream];
840 }
ekmeyerson60d9b332015-08-14 10:35:55 -0700841 const StreamConfig& reverse_input_stream() const {
842 return streams[StreamName::kReverseInputStream];
843 }
844 const StreamConfig& reverse_output_stream() const {
845 return streams[StreamName::kReverseOutputStream];
Michael Graczyk86c6d332015-07-23 11:41:39 -0700846 }
847
848 StreamConfig& input_stream() { return streams[StreamName::kInputStream]; }
849 StreamConfig& output_stream() { return streams[StreamName::kOutputStream]; }
ekmeyerson60d9b332015-08-14 10:35:55 -0700850 StreamConfig& reverse_input_stream() {
851 return streams[StreamName::kReverseInputStream];
852 }
853 StreamConfig& reverse_output_stream() {
854 return streams[StreamName::kReverseOutputStream];
855 }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700856
857 bool operator==(const ProcessingConfig& other) const {
858 for (int i = 0; i < StreamName::kNumStreamNames; ++i) {
859 if (this->streams[i] != other.streams[i]) {
860 return false;
861 }
862 }
863 return true;
864 }
865
866 bool operator!=(const ProcessingConfig& other) const {
867 return !(*this == other);
868 }
869
870 StreamConfig streams[StreamName::kNumStreamNames];
871};
872
Valeriia Nemychnikovaf06eb572018-08-29 10:37:09 +0200873// Experimental interface for a custom analysis submodule.
874class CustomAudioAnalyzer {
875 public:
876 // (Re-) Initializes the submodule.
877 virtual void Initialize(int sample_rate_hz, int num_channels) = 0;
878 // Analyzes the given capture or render signal.
879 virtual void Analyze(const AudioBuffer* audio) = 0;
880 // Returns a string representation of the module state.
881 virtual std::string ToString() const = 0;
882
883 virtual ~CustomAudioAnalyzer() {}
884};
885
Alex Loiko5825aa62017-12-18 16:02:40 +0100886// Interface for a custom processing submodule.
887class CustomProcessing {
Sam Zackrisson0beac582017-09-25 12:04:02 +0200888 public:
889 // (Re-)Initializes the submodule.
890 virtual void Initialize(int sample_rate_hz, int num_channels) = 0;
891 // Processes the given capture or render signal.
892 virtual void Process(AudioBuffer* audio) = 0;
893 // Returns a string representation of the module state.
894 virtual std::string ToString() const = 0;
Alex Loiko73ec0192018-05-15 10:52:28 +0200895 // Handles RuntimeSettings. TODO(webrtc:9262): make pure virtual
896 // after updating dependencies.
897 virtual void SetRuntimeSetting(AudioProcessing::RuntimeSetting setting);
Sam Zackrisson0beac582017-09-25 12:04:02 +0200898
Alex Loiko5825aa62017-12-18 16:02:40 +0100899 virtual ~CustomProcessing() {}
Sam Zackrisson0beac582017-09-25 12:04:02 +0200900};
901
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100902// Interface for an echo detector submodule.
Ivo Creusend1f970d2018-06-14 11:02:03 +0200903class EchoDetector : public rtc::RefCountInterface {
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100904 public:
905 // (Re-)Initializes the submodule.
Ivo Creusen647ef092018-03-14 17:13:48 +0100906 virtual void Initialize(int capture_sample_rate_hz,
907 int num_capture_channels,
908 int render_sample_rate_hz,
909 int num_render_channels) = 0;
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100910
Sam Zackrisson03cb7e52021-12-06 15:40:04 +0100911 // Analysis (not changing) of the first channel of the render signal.
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100912 virtual void AnalyzeRenderAudio(rtc::ArrayView<const float> render_audio) = 0;
913
914 // Analysis (not changing) of the capture signal.
915 virtual void AnalyzeCaptureAudio(
916 rtc::ArrayView<const float> capture_audio) = 0;
917
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100918 struct Metrics {
Ivo Creusenbb826c92020-04-29 14:34:48 +0200919 absl::optional<double> echo_likelihood;
920 absl::optional<double> echo_likelihood_recent_max;
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100921 };
922
923 // Collect current metrics from the echo detector.
924 virtual Metrics GetMetrics() const = 0;
Ivo Creusen09fa4b02018-01-11 16:08:54 +0100925};
926
niklase@google.com470e71d2011-07-07 08:21:25 +0000927} // namespace webrtc
928
Mirko Bonadei92ea95e2017-09-15 06:47:31 +0200929#endif // MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_