blob: 0c93a984b241817636f980a3d2e1375dd4b40e90 [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
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000011#ifndef WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_
12#define WEBRTC_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.
15#define _USE_MATH_DEFINES
16
17#include <math.h>
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000018#include <stddef.h> // size_t
henrikg@webrtc.org863b5362013-12-06 16:05:17 +000019#include <stdio.h> // FILE
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +000020#include <vector>
ajm@google.com22e65152011-07-18 18:03:01 +000021
Alejandro Luebscdfe20b2015-09-23 12:49:12 -070022#include "webrtc/base/arraysize.h"
xians@webrtc.orge46bc772014-10-10 08:36:56 +000023#include "webrtc/base/platform_file.h"
andrew@webrtc.org61e596f2013-07-25 18:28:29 +000024#include "webrtc/common.h"
aluebs@webrtc.org1d883942015-03-05 20:38:21 +000025#include "webrtc/modules/audio_processing/beamformer/array_util.h"
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000026#include "webrtc/typedefs.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000027
28namespace webrtc {
29
peah50e21bd2016-03-05 08:39:21 -080030struct AecCore;
31
niklase@google.com470e71d2011-07-07 08:21:25 +000032class AudioFrame;
Michael Graczykdfa36052015-03-25 16:37:27 -070033
34template<typename T>
35class Beamformer;
36
Michael Graczyk86c6d332015-07-23 11:41:39 -070037class StreamConfig;
38class ProcessingConfig;
39
niklase@google.com470e71d2011-07-07 08:21:25 +000040class EchoCancellation;
41class EchoControlMobile;
42class GainControl;
43class HighPassFilter;
44class LevelEstimator;
45class NoiseSuppression;
46class VoiceDetection;
47
Henrik Lundin441f6342015-06-09 16:03:13 +020048// Use to enable the extended filter mode in the AEC, along with robustness
49// measures around the reported system delays. It comes with a significant
50// increase in AEC complexity, but is much more robust to unreliable reported
51// delays.
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +000052//
53// Detailed changes to the algorithm:
54// - The filter length is changed from 48 to 128 ms. This comes with tuning of
55// several parameters: i) filter adaptation stepsize and error threshold;
56// ii) non-linear processing smoothing and overdrive.
57// - Option to ignore the reported delays on platforms which we deem
58// sufficiently unreliable. See WEBRTC_UNTRUSTED_DELAY in echo_cancellation.c.
59// - Faster startup times by removing the excessive "startup phase" processing
60// of reported delays.
61// - Much more conservative adjustments to the far-end read pointer. We smooth
62// the delay difference more heavily, and back off from the difference more.
63// Adjustments force a readaptation of the filter, so they should be avoided
64// except when really necessary.
Henrik Lundin441f6342015-06-09 16:03:13 +020065struct ExtendedFilter {
66 ExtendedFilter() : enabled(false) {}
67 explicit ExtendedFilter(bool enabled) : enabled(enabled) {}
aluebs688e3082016-01-14 04:32:46 -080068 static const ConfigOptionID identifier = ConfigOptionID::kExtendedFilter;
Henrik Lundin441f6342015-06-09 16:03:13 +020069 bool enabled;
70};
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +000071
peaha332e2d2016-02-17 01:11:16 -080072// Enables the next generation AEC functionality. This feature replaces the
73// standard methods for echo removal in the AEC. This configuration only applies
74// to EchoCancellation and not EchoControlMobile. It can be set in the
75// constructor or using AudioProcessing::SetExtraOptions().
peah6ebc4d32016-03-07 16:59:39 -080076struct EchoCanceller3 {
77 EchoCanceller3() : enabled(false) {}
78 explicit EchoCanceller3(bool enabled) : enabled(enabled) {}
79 static const ConfigOptionID identifier = ConfigOptionID::kEchoCanceller3;
peaha332e2d2016-02-17 01:11:16 -080080 bool enabled;
81};
82
peah0332c2d2016-04-15 11:23:33 -070083// Enables the refined linear filter adaptation in the echo canceller.
84// This configuration only applies to EchoCancellation and not
85// EchoControlMobile. It can be set in the constructor
86// or using AudioProcessing::SetExtraOptions().
87struct RefinedAdaptiveFilter {
88 RefinedAdaptiveFilter() : enabled(false) {}
89 explicit RefinedAdaptiveFilter(bool enabled) : enabled(enabled) {}
90 static const ConfigOptionID identifier =
91 ConfigOptionID::kAecRefinedAdaptiveFilter;
92 bool enabled;
93};
94
henrik.lundin366e9522015-07-03 00:50:05 -070095// Enables delay-agnostic echo cancellation. This feature relies on internally
96// estimated delays between the process and reverse streams, thus not relying
97// on reported system delays. This configuration only applies to
98// EchoCancellation and not EchoControlMobile. It can be set in the constructor
99// or using AudioProcessing::SetExtraOptions().
henrik.lundin0f133b92015-07-02 00:17:55 -0700100struct DelayAgnostic {
101 DelayAgnostic() : enabled(false) {}
102 explicit DelayAgnostic(bool enabled) : enabled(enabled) {}
aluebs688e3082016-01-14 04:32:46 -0800103 static const ConfigOptionID identifier = ConfigOptionID::kDelayAgnostic;
henrik.lundin0f133b92015-07-02 00:17:55 -0700104 bool enabled;
105};
bjornv@webrtc.org3f830722014-06-11 04:48:11 +0000106
Bjorn Volckeradc46c42015-04-15 11:42:40 +0200107// Use to enable experimental gain control (AGC). At startup the experimental
108// AGC moves the microphone volume up to |startup_min_volume| if the current
109// microphone volume is set too low. The value is clamped to its operating range
110// [12, 255]. Here, 255 maps to 100%.
111//
112// Must be provided through AudioProcessing::Create(Confg&).
Bjorn Volckerfb494512015-04-22 06:39:58 +0200113#if defined(WEBRTC_CHROMIUM_BUILD)
Bjorn Volckeradc46c42015-04-15 11:42:40 +0200114static const int kAgcStartupMinVolume = 85;
Bjorn Volckerfb494512015-04-22 06:39:58 +0200115#else
116static const int kAgcStartupMinVolume = 0;
117#endif // defined(WEBRTC_CHROMIUM_BUILD)
andrew@webrtc.orgc7c7a532014-01-29 04:57:25 +0000118struct ExperimentalAgc {
Bjorn Volckeradc46c42015-04-15 11:42:40 +0200119 ExperimentalAgc() : enabled(true), startup_min_volume(kAgcStartupMinVolume) {}
Michael Graczyk86c6d332015-07-23 11:41:39 -0700120 explicit ExperimentalAgc(bool enabled)
Bjorn Volckeradc46c42015-04-15 11:42:40 +0200121 : enabled(enabled), startup_min_volume(kAgcStartupMinVolume) {}
122 ExperimentalAgc(bool enabled, int startup_min_volume)
123 : enabled(enabled), startup_min_volume(startup_min_volume) {}
aluebs688e3082016-01-14 04:32:46 -0800124 static const ConfigOptionID identifier = ConfigOptionID::kExperimentalAgc;
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +0000125 bool enabled;
Bjorn Volckeradc46c42015-04-15 11:42:40 +0200126 int startup_min_volume;
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +0000127};
128
aluebs@webrtc.org9825afc2014-06-30 17:39:53 +0000129// Use to enable experimental noise suppression. It can be set in the
130// constructor or using AudioProcessing::SetExtraOptions().
131struct ExperimentalNs {
132 ExperimentalNs() : enabled(false) {}
133 explicit ExperimentalNs(bool enabled) : enabled(enabled) {}
aluebs688e3082016-01-14 04:32:46 -0800134 static const ConfigOptionID identifier = ConfigOptionID::kExperimentalNs;
aluebs@webrtc.org9825afc2014-06-30 17:39:53 +0000135 bool enabled;
136};
137
aluebs@webrtc.orgae643ce2014-12-19 19:57:34 +0000138// Use to enable beamforming. Must be provided through the constructor. It will
139// have no impact if used with AudioProcessing::SetExtraOptions().
140struct Beamforming {
eblima894ad942015-07-03 08:34:33 -0700141 Beamforming()
142 : enabled(false),
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -0700143 array_geometry(),
144 target_direction(
145 SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) {}
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +0000146 Beamforming(bool enabled, const std::vector<Point>& array_geometry)
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -0700147 : Beamforming(enabled,
148 array_geometry,
149 SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) {
150 }
151 Beamforming(bool enabled,
152 const std::vector<Point>& array_geometry,
153 SphericalPointf target_direction)
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +0000154 : enabled(enabled),
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -0700155 array_geometry(array_geometry),
156 target_direction(target_direction) {}
aluebs688e3082016-01-14 04:32:46 -0800157 static const ConfigOptionID identifier = ConfigOptionID::kBeamforming;
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +0000158 const bool enabled;
159 const std::vector<Point> array_geometry;
Alejandro Luebscb3f9bd2015-10-29 18:21:34 -0700160 const SphericalPointf target_direction;
aluebs@webrtc.orgae643ce2014-12-19 19:57:34 +0000161};
162
ekmeyerson60d9b332015-08-14 10:35:55 -0700163// Use to enable intelligibility enhancer in audio processing. Must be provided
164// though the constructor. It will have no impact if used with
165// AudioProcessing::SetExtraOptions().
166//
167// Note: If enabled and the reverse stream has more than one output channel,
168// the reverse stream will become an upmixed mono signal.
169struct Intelligibility {
170 Intelligibility() : enabled(false) {}
171 explicit Intelligibility(bool enabled) : enabled(enabled) {}
aluebs688e3082016-01-14 04:32:46 -0800172 static const ConfigOptionID identifier = ConfigOptionID::kIntelligibility;
ekmeyerson60d9b332015-08-14 10:35:55 -0700173 bool enabled;
174};
175
niklase@google.com470e71d2011-07-07 08:21:25 +0000176// The Audio Processing Module (APM) provides a collection of voice processing
177// components designed for real-time communications software.
178//
179// APM operates on two audio streams on a frame-by-frame basis. Frames of the
180// primary stream, on which all processing is applied, are passed to
aluebsb0319552016-03-17 20:39:53 -0700181// |ProcessStream()|. Frames of the reverse direction stream are passed to
182// |ProcessReverseStream()|. On the client-side, this will typically be the
183// near-end (capture) and far-end (render) streams, respectively. APM should be
184// placed in the signal chain as close to the audio hardware abstraction layer
185// (HAL) as possible.
niklase@google.com470e71d2011-07-07 08:21:25 +0000186//
187// On the server-side, the reverse stream will normally not be used, with
188// processing occurring on each incoming stream.
189//
190// Component interfaces follow a similar pattern and are accessed through
191// corresponding getters in APM. All components are disabled at create-time,
192// with default settings that are recommended for most situations. New settings
193// can be applied without enabling a component. Enabling a component triggers
194// memory allocation and initialization to allow it to start processing the
195// streams.
196//
197// Thread safety is provided with the following assumptions to reduce locking
198// overhead:
199// 1. The stream getters and setters are called from the same thread as
200// ProcessStream(). More precisely, stream functions are never called
201// concurrently with ProcessStream().
202// 2. Parameter getters are never called concurrently with the corresponding
203// setter.
204//
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000205// APM accepts only linear PCM audio data in chunks of 10 ms. The int16
206// interfaces use interleaved data, while the float interfaces use deinterleaved
207// data.
niklase@google.com470e71d2011-07-07 08:21:25 +0000208//
209// Usage example, omitting error checking:
210// AudioProcessing* apm = AudioProcessing::Create(0);
niklase@google.com470e71d2011-07-07 08:21:25 +0000211//
212// apm->high_pass_filter()->Enable(true);
213//
214// apm->echo_cancellation()->enable_drift_compensation(false);
215// apm->echo_cancellation()->Enable(true);
216//
217// apm->noise_reduction()->set_level(kHighSuppression);
218// apm->noise_reduction()->Enable(true);
219//
220// apm->gain_control()->set_analog_level_limits(0, 255);
221// apm->gain_control()->set_mode(kAdaptiveAnalog);
222// apm->gain_control()->Enable(true);
223//
224// apm->voice_detection()->Enable(true);
225//
226// // Start a voice call...
227//
228// // ... Render frame arrives bound for the audio HAL ...
aluebsb0319552016-03-17 20:39:53 -0700229// apm->ProcessReverseStream(render_frame);
niklase@google.com470e71d2011-07-07 08:21:25 +0000230//
231// // ... Capture frame arrives from the audio HAL ...
232// // Call required set_stream_ functions.
233// apm->set_stream_delay_ms(delay_ms);
234// apm->gain_control()->set_stream_analog_level(analog_level);
235//
236// apm->ProcessStream(capture_frame);
237//
238// // Call required stream_ functions.
239// analog_level = apm->gain_control()->stream_analog_level();
240// has_voice = apm->stream_has_voice();
241//
242// // Repeate render and capture processing for the duration of the call...
243// // Start a new call...
244// apm->Initialize();
245//
246// // Close the application...
andrew@webrtc.orgf3930e92013-09-18 22:37:32 +0000247// delete apm;
niklase@google.com470e71d2011-07-07 08:21:25 +0000248//
andrew@webrtc.orgf92aaff2014-02-15 04:22:49 +0000249class AudioProcessing {
niklase@google.com470e71d2011-07-07 08:21:25 +0000250 public:
Michael Graczyk86c6d332015-07-23 11:41:39 -0700251 // TODO(mgraczyk): Remove once all methods that use ChannelLayout are gone.
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000252 enum ChannelLayout {
253 kMono,
254 // Left, right.
255 kStereo,
256 // Mono, keyboard mic.
257 kMonoAndKeyboard,
258 // Left, right, keyboard mic.
259 kStereoAndKeyboard
260 };
261
andrew@webrtc.org54744912014-02-05 06:30:29 +0000262 // Creates an APM instance. Use one instance for every primary audio stream
263 // requiring processing. On the client-side, this would typically be one
264 // instance for the near-end stream, and additional instances for each far-end
265 // stream which requires processing. On the server-side, this would typically
266 // be one instance for every incoming stream.
andrew@webrtc.orge84978f2014-01-25 02:09:06 +0000267 static AudioProcessing* Create();
andrew@webrtc.org54744912014-02-05 06:30:29 +0000268 // Allows passing in an optional configuration at create-time.
andrew@webrtc.orge84978f2014-01-25 02:09:06 +0000269 static AudioProcessing* Create(const Config& config);
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000270 // Only for testing.
mgraczyk@chromium.org0f663de2015-03-13 00:13:32 +0000271 static AudioProcessing* Create(const Config& config,
Michael Graczykdfa36052015-03-25 16:37:27 -0700272 Beamformer<float>* beamformer);
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000273 virtual ~AudioProcessing() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000274
niklase@google.com470e71d2011-07-07 08:21:25 +0000275 // Initializes internal states, while retaining all user settings. This
276 // should be called before beginning to process a new audio stream. However,
277 // it is not necessary to call before processing the first stream after
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000278 // creation.
279 //
280 // It is also not necessary to call if the audio parameters (sample
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000281 // rate and number of channels) have changed. Passing updated parameters
aluebsb0319552016-03-17 20:39:53 -0700282 // directly to |ProcessStream()| and |ProcessReverseStream()| is permissible.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000283 // If the parameters are known at init-time though, they may be provided.
niklase@google.com470e71d2011-07-07 08:21:25 +0000284 virtual int Initialize() = 0;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000285
286 // The int16 interfaces require:
287 // - only |NativeRate|s be used
288 // - that the input, output and reverse rates must match
Michael Graczyk86c6d332015-07-23 11:41:39 -0700289 // - that |processing_config.output_stream()| matches
290 // |processing_config.input_stream()|.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000291 //
Michael Graczyk86c6d332015-07-23 11:41:39 -0700292 // The float interfaces accept arbitrary rates and support differing input and
293 // output layouts, but the output must have either one channel or the same
294 // number of channels as the input.
295 virtual int Initialize(const ProcessingConfig& processing_config) = 0;
296
297 // Initialize with unpacked parameters. See Initialize() above for details.
298 //
299 // TODO(mgraczyk): Remove once clients are updated to use the new interface.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000300 virtual int Initialize(int input_sample_rate_hz,
301 int output_sample_rate_hz,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000302 int reverse_sample_rate_hz,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000303 ChannelLayout input_layout,
304 ChannelLayout output_layout,
305 ChannelLayout reverse_layout) = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000306
andrew@webrtc.org61e596f2013-07-25 18:28:29 +0000307 // Pass down additional options which don't have explicit setters. This
308 // ensures the options are applied immediately.
309 virtual void SetExtraOptions(const Config& config) = 0;
310
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000311 // TODO(ajm): Only intended for internal use. Make private and friend the
312 // necessary classes?
313 virtual int proc_sample_rate_hz() const = 0;
314 virtual int proc_split_sample_rate_hz() const = 0;
Peter Kasting69558702016-01-12 16:26:35 -0800315 virtual size_t num_input_channels() const = 0;
316 virtual size_t num_proc_channels() const = 0;
317 virtual size_t num_output_channels() const = 0;
318 virtual size_t num_reverse_channels() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000319
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000320 // Set to true when the output of AudioProcessing will be muted or in some
321 // other way not used. Ideally, the captured audio would still be processed,
322 // but some components may change behavior based on this information.
323 // Default false.
324 virtual void set_output_will_be_muted(bool muted) = 0;
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000325
niklase@google.com470e71d2011-07-07 08:21:25 +0000326 // Processes a 10 ms |frame| of the primary audio stream. On the client-side,
327 // this is the near-end (or captured) audio.
328 //
329 // If needed for enabled functionality, any function with the set_stream_ tag
330 // must be called prior to processing the current frame. Any getter function
331 // with the stream_ tag which is needed should be called after processing.
332 //
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000333 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_|
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000334 // members of |frame| must be valid. If changed from the previous call to this
335 // method, it will trigger an initialization.
niklase@google.com470e71d2011-07-07 08:21:25 +0000336 virtual int ProcessStream(AudioFrame* frame) = 0;
337
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000338 // Accepts deinterleaved float audio with the range [-1, 1]. Each element
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000339 // of |src| points to a channel buffer, arranged according to
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000340 // |input_layout|. At output, the channels will be arranged according to
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000341 // |output_layout| at |output_sample_rate_hz| in |dest|.
342 //
Michael Graczyk86c6d332015-07-23 11:41:39 -0700343 // The output layout must have one channel or as many channels as the input.
344 // |src| and |dest| may use the same memory, if desired.
345 //
346 // TODO(mgraczyk): Remove once clients are updated to use the new interface.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000347 virtual int ProcessStream(const float* const* src,
Peter Kastingdce40cf2015-08-24 14:52:23 -0700348 size_t samples_per_channel,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000349 int input_sample_rate_hz,
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000350 ChannelLayout input_layout,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000351 int output_sample_rate_hz,
352 ChannelLayout output_layout,
353 float* const* dest) = 0;
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000354
Michael Graczyk86c6d332015-07-23 11:41:39 -0700355 // Accepts deinterleaved float audio with the range [-1, 1]. Each element of
356 // |src| points to a channel buffer, arranged according to |input_stream|. At
357 // output, the channels will be arranged according to |output_stream| in
358 // |dest|.
359 //
360 // The output must have one channel or as many channels as the input. |src|
361 // and |dest| may use the same memory, if desired.
362 virtual int ProcessStream(const float* const* src,
363 const StreamConfig& input_config,
364 const StreamConfig& output_config,
365 float* const* dest) = 0;
366
aluebsb0319552016-03-17 20:39:53 -0700367 // Processes a 10 ms |frame| of the reverse direction audio stream. The frame
368 // may be modified. On the client-side, this is the far-end (or to be
niklase@google.com470e71d2011-07-07 08:21:25 +0000369 // rendered) audio.
370 //
aluebsb0319552016-03-17 20:39:53 -0700371 // It is necessary to provide this if echo processing is enabled, as the
niklase@google.com470e71d2011-07-07 08:21:25 +0000372 // reverse stream forms the echo reference signal. It is recommended, but not
373 // necessary, to provide if gain control is enabled. On the server-side this
374 // typically will not be used. If you're not sure what to pass in here,
375 // chances are you don't need to use it.
376 //
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000377 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_|
aluebsda116c42016-03-17 16:43:29 -0700378 // members of |frame| must be valid.
ekmeyerson60d9b332015-08-14 10:35:55 -0700379 virtual int ProcessReverseStream(AudioFrame* frame) = 0;
380
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000381 // Accepts deinterleaved float audio with the range [-1, 1]. Each element
382 // of |data| points to a channel buffer, arranged according to |layout|.
Michael Graczyk86c6d332015-07-23 11:41:39 -0700383 // TODO(mgraczyk): Remove once clients are updated to use the new interface.
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000384 virtual int AnalyzeReverseStream(const float* const* data,
Peter Kastingdce40cf2015-08-24 14:52:23 -0700385 size_t samples_per_channel,
ekmeyerson60d9b332015-08-14 10:35:55 -0700386 int rev_sample_rate_hz,
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000387 ChannelLayout layout) = 0;
388
Michael Graczyk86c6d332015-07-23 11:41:39 -0700389 // Accepts deinterleaved float audio with the range [-1, 1]. Each element of
390 // |data| points to a channel buffer, arranged according to |reverse_config|.
ekmeyerson60d9b332015-08-14 10:35:55 -0700391 virtual int ProcessReverseStream(const float* const* src,
392 const StreamConfig& reverse_input_config,
393 const StreamConfig& reverse_output_config,
394 float* const* dest) = 0;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700395
niklase@google.com470e71d2011-07-07 08:21:25 +0000396 // This must be called if and only if echo processing is enabled.
397 //
aluebsb0319552016-03-17 20:39:53 -0700398 // Sets the |delay| in ms between ProcessReverseStream() receiving a far-end
niklase@google.com470e71d2011-07-07 08:21:25 +0000399 // frame and ProcessStream() receiving a near-end frame containing the
400 // corresponding echo. On the client-side this can be expressed as
401 // delay = (t_render - t_analyze) + (t_process - t_capture)
402 // where,
aluebsb0319552016-03-17 20:39:53 -0700403 // - t_analyze is the time a frame is passed to ProcessReverseStream() and
niklase@google.com470e71d2011-07-07 08:21:25 +0000404 // t_render is the time the first sample of the same frame is rendered by
405 // the audio hardware.
406 // - t_capture is the time the first sample of a frame is captured by the
407 // audio hardware and t_pull is the time the same frame is passed to
408 // ProcessStream().
409 virtual int set_stream_delay_ms(int delay) = 0;
410 virtual int stream_delay_ms() const = 0;
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000411 virtual bool was_stream_delay_set() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000412
andrew@webrtc.org75dd2882014-02-11 20:52:30 +0000413 // Call to signal that a key press occurred (true) or did not occur (false)
414 // with this chunk of audio.
415 virtual void set_stream_key_pressed(bool key_pressed) = 0;
andrew@webrtc.org75dd2882014-02-11 20:52:30 +0000416
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000417 // Sets a delay |offset| in ms to add to the values passed in through
418 // set_stream_delay_ms(). May be positive or negative.
419 //
420 // Note that this could cause an otherwise valid value passed to
421 // set_stream_delay_ms() to return an error.
422 virtual void set_delay_offset_ms(int offset) = 0;
423 virtual int delay_offset_ms() const = 0;
424
niklase@google.com470e71d2011-07-07 08:21:25 +0000425 // Starts recording debugging information to a file specified by |filename|,
426 // a NULL-terminated string. If there is an ongoing recording, the old file
427 // will be closed, and recording will continue in the newly specified file.
ivocd66b44d2016-01-15 03:06:36 -0800428 // An already existing file will be overwritten without warning. A maximum
429 // file size (in bytes) for the log can be specified. The logging is stopped
430 // once the limit has been reached. If max_log_size_bytes is set to a value
431 // <= 0, no limit will be used.
andrew@webrtc.org5ae19de2011-12-13 22:59:33 +0000432 static const size_t kMaxFilenameSize = 1024;
ivocd66b44d2016-01-15 03:06:36 -0800433 virtual int StartDebugRecording(const char filename[kMaxFilenameSize],
434 int64_t max_log_size_bytes) = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000435
henrikg@webrtc.org863b5362013-12-06 16:05:17 +0000436 // Same as above but uses an existing file handle. Takes ownership
437 // of |handle| and closes it at StopDebugRecording().
ivocd66b44d2016-01-15 03:06:36 -0800438 virtual int StartDebugRecording(FILE* handle, int64_t max_log_size_bytes) = 0;
439
440 // TODO(ivoc): Remove this function after Chrome stops using it.
441 int StartDebugRecording(FILE* handle) {
442 return StartDebugRecording(handle, -1);
443 }
henrikg@webrtc.org863b5362013-12-06 16:05:17 +0000444
xians@webrtc.orge46bc772014-10-10 08:36:56 +0000445 // Same as above but uses an existing PlatformFile handle. Takes ownership
446 // of |handle| and closes it at StopDebugRecording().
447 // TODO(xians): Make this interface pure virtual.
448 virtual int StartDebugRecordingForPlatformFile(rtc::PlatformFile handle) {
449 return -1;
450 }
451
niklase@google.com470e71d2011-07-07 08:21:25 +0000452 // Stops recording debugging information, and closes the file. Recording
453 // cannot be resumed in the same file (without overwriting it).
454 virtual int StopDebugRecording() = 0;
455
Bjorn Volcker4e7aa432015-07-07 11:50:05 +0200456 // Use to send UMA histograms at end of a call. Note that all histogram
457 // specific member variables are reset.
458 virtual void UpdateHistogramsOnCallEnd() = 0;
459
niklase@google.com470e71d2011-07-07 08:21:25 +0000460 // These provide access to the component interfaces and should never return
461 // NULL. The pointers will be valid for the lifetime of the APM instance.
462 // The memory for these objects is entirely managed internally.
463 virtual EchoCancellation* echo_cancellation() const = 0;
464 virtual EchoControlMobile* echo_control_mobile() const = 0;
465 virtual GainControl* gain_control() const = 0;
466 virtual HighPassFilter* high_pass_filter() const = 0;
467 virtual LevelEstimator* level_estimator() const = 0;
468 virtual NoiseSuppression* noise_suppression() const = 0;
469 virtual VoiceDetection* voice_detection() const = 0;
470
471 struct Statistic {
472 int instant; // Instantaneous value.
473 int average; // Long-term average.
474 int maximum; // Long-term maximum.
475 int minimum; // Long-term minimum.
476 };
477
andrew@webrtc.org648af742012-02-08 01:57:29 +0000478 enum Error {
479 // Fatal errors.
niklase@google.com470e71d2011-07-07 08:21:25 +0000480 kNoError = 0,
481 kUnspecifiedError = -1,
482 kCreationFailedError = -2,
483 kUnsupportedComponentError = -3,
484 kUnsupportedFunctionError = -4,
485 kNullPointerError = -5,
486 kBadParameterError = -6,
487 kBadSampleRateError = -7,
488 kBadDataLengthError = -8,
489 kBadNumberChannelsError = -9,
490 kFileError = -10,
491 kStreamParameterNotSetError = -11,
andrew@webrtc.org648af742012-02-08 01:57:29 +0000492 kNotEnabledError = -12,
niklase@google.com470e71d2011-07-07 08:21:25 +0000493
andrew@webrtc.org648af742012-02-08 01:57:29 +0000494 // Warnings are non-fatal.
niklase@google.com470e71d2011-07-07 08:21:25 +0000495 // This results when a set_stream_ parameter is out of range. Processing
496 // will continue, but the parameter may have been truncated.
andrew@webrtc.org648af742012-02-08 01:57:29 +0000497 kBadStreamParameterWarning = -13
niklase@google.com470e71d2011-07-07 08:21:25 +0000498 };
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000499
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000500 enum NativeRate {
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000501 kSampleRate8kHz = 8000,
502 kSampleRate16kHz = 16000,
aluebs@webrtc.org087da132014-11-17 23:01:23 +0000503 kSampleRate32kHz = 32000,
504 kSampleRate48kHz = 48000
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000505 };
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000506
Alejandro Luebscdfe20b2015-09-23 12:49:12 -0700507 static const int kNativeSampleRatesHz[];
508 static const size_t kNumNativeSampleRates;
509 static const int kMaxNativeSampleRateHz;
Alejandro Luebscdfe20b2015-09-23 12:49:12 -0700510
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000511 static const int kChunkSizeMs = 10;
niklase@google.com470e71d2011-07-07 08:21:25 +0000512};
513
Michael Graczyk86c6d332015-07-23 11:41:39 -0700514class StreamConfig {
515 public:
516 // sample_rate_hz: The sampling rate of the stream.
517 //
518 // num_channels: The number of audio channels in the stream, excluding the
519 // keyboard channel if it is present. When passing a
520 // StreamConfig with an array of arrays T*[N],
521 //
522 // N == {num_channels + 1 if has_keyboard
523 // {num_channels if !has_keyboard
524 //
525 // has_keyboard: True if the stream has a keyboard channel. When has_keyboard
526 // is true, the last channel in any corresponding list of
527 // channels is the keyboard channel.
528 StreamConfig(int sample_rate_hz = 0,
Peter Kasting69558702016-01-12 16:26:35 -0800529 size_t num_channels = 0,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700530 bool has_keyboard = false)
531 : sample_rate_hz_(sample_rate_hz),
532 num_channels_(num_channels),
533 has_keyboard_(has_keyboard),
534 num_frames_(calculate_frames(sample_rate_hz)) {}
535
536 void set_sample_rate_hz(int value) {
537 sample_rate_hz_ = value;
538 num_frames_ = calculate_frames(value);
539 }
Peter Kasting69558702016-01-12 16:26:35 -0800540 void set_num_channels(size_t value) { num_channels_ = value; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700541 void set_has_keyboard(bool value) { has_keyboard_ = value; }
542
543 int sample_rate_hz() const { return sample_rate_hz_; }
544
545 // The number of channels in the stream, not including the keyboard channel if
546 // present.
Peter Kasting69558702016-01-12 16:26:35 -0800547 size_t num_channels() const { return num_channels_; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700548
549 bool has_keyboard() const { return has_keyboard_; }
Peter Kastingdce40cf2015-08-24 14:52:23 -0700550 size_t num_frames() const { return num_frames_; }
551 size_t num_samples() const { return num_channels_ * num_frames_; }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700552
553 bool operator==(const StreamConfig& other) const {
554 return sample_rate_hz_ == other.sample_rate_hz_ &&
555 num_channels_ == other.num_channels_ &&
556 has_keyboard_ == other.has_keyboard_;
557 }
558
559 bool operator!=(const StreamConfig& other) const { return !(*this == other); }
560
561 private:
Peter Kastingdce40cf2015-08-24 14:52:23 -0700562 static size_t calculate_frames(int sample_rate_hz) {
563 return static_cast<size_t>(
564 AudioProcessing::kChunkSizeMs * sample_rate_hz / 1000);
Michael Graczyk86c6d332015-07-23 11:41:39 -0700565 }
566
567 int sample_rate_hz_;
Peter Kasting69558702016-01-12 16:26:35 -0800568 size_t num_channels_;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700569 bool has_keyboard_;
Peter Kastingdce40cf2015-08-24 14:52:23 -0700570 size_t num_frames_;
Michael Graczyk86c6d332015-07-23 11:41:39 -0700571};
572
573class ProcessingConfig {
574 public:
575 enum StreamName {
576 kInputStream,
577 kOutputStream,
ekmeyerson60d9b332015-08-14 10:35:55 -0700578 kReverseInputStream,
579 kReverseOutputStream,
Michael Graczyk86c6d332015-07-23 11:41:39 -0700580 kNumStreamNames,
581 };
582
583 const StreamConfig& input_stream() const {
584 return streams[StreamName::kInputStream];
585 }
586 const StreamConfig& output_stream() const {
587 return streams[StreamName::kOutputStream];
588 }
ekmeyerson60d9b332015-08-14 10:35:55 -0700589 const StreamConfig& reverse_input_stream() const {
590 return streams[StreamName::kReverseInputStream];
591 }
592 const StreamConfig& reverse_output_stream() const {
593 return streams[StreamName::kReverseOutputStream];
Michael Graczyk86c6d332015-07-23 11:41:39 -0700594 }
595
596 StreamConfig& input_stream() { return streams[StreamName::kInputStream]; }
597 StreamConfig& output_stream() { return streams[StreamName::kOutputStream]; }
ekmeyerson60d9b332015-08-14 10:35:55 -0700598 StreamConfig& reverse_input_stream() {
599 return streams[StreamName::kReverseInputStream];
600 }
601 StreamConfig& reverse_output_stream() {
602 return streams[StreamName::kReverseOutputStream];
603 }
Michael Graczyk86c6d332015-07-23 11:41:39 -0700604
605 bool operator==(const ProcessingConfig& other) const {
606 for (int i = 0; i < StreamName::kNumStreamNames; ++i) {
607 if (this->streams[i] != other.streams[i]) {
608 return false;
609 }
610 }
611 return true;
612 }
613
614 bool operator!=(const ProcessingConfig& other) const {
615 return !(*this == other);
616 }
617
618 StreamConfig streams[StreamName::kNumStreamNames];
619};
620
niklase@google.com470e71d2011-07-07 08:21:25 +0000621// The acoustic echo cancellation (AEC) component provides better performance
622// than AECM but also requires more processing power and is dependent on delay
623// stability and reporting accuracy. As such it is well-suited and recommended
624// for PC and IP phone applications.
625//
626// Not recommended to be enabled on the server-side.
627class EchoCancellation {
628 public:
629 // EchoCancellation and EchoControlMobile may not be enabled simultaneously.
630 // Enabling one will disable the other.
631 virtual int Enable(bool enable) = 0;
632 virtual bool is_enabled() const = 0;
633
634 // Differences in clock speed on the primary and reverse streams can impact
635 // the AEC performance. On the client-side, this could be seen when different
636 // render and capture devices are used, particularly with webcams.
637 //
638 // This enables a compensation mechanism, and requires that
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000639 // set_stream_drift_samples() be called.
niklase@google.com470e71d2011-07-07 08:21:25 +0000640 virtual int enable_drift_compensation(bool enable) = 0;
641 virtual bool is_drift_compensation_enabled() const = 0;
642
niklase@google.com470e71d2011-07-07 08:21:25 +0000643 // Sets the difference between the number of samples rendered and captured by
644 // the audio devices since the last call to |ProcessStream()|. Must be called
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000645 // if drift compensation is enabled, prior to |ProcessStream()|.
646 virtual void set_stream_drift_samples(int drift) = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000647 virtual int stream_drift_samples() const = 0;
648
649 enum SuppressionLevel {
650 kLowSuppression,
651 kModerateSuppression,
652 kHighSuppression
653 };
654
655 // Sets the aggressiveness of the suppressor. A higher level trades off
656 // double-talk performance for increased echo suppression.
657 virtual int set_suppression_level(SuppressionLevel level) = 0;
658 virtual SuppressionLevel suppression_level() const = 0;
659
660 // Returns false if the current frame almost certainly contains no echo
661 // and true if it _might_ contain echo.
662 virtual bool stream_has_echo() const = 0;
663
664 // Enables the computation of various echo metrics. These are obtained
665 // through |GetMetrics()|.
666 virtual int enable_metrics(bool enable) = 0;
667 virtual bool are_metrics_enabled() const = 0;
668
669 // Each statistic is reported in dB.
670 // P_far: Far-end (render) signal power.
671 // P_echo: Near-end (capture) echo signal power.
672 // P_out: Signal power at the output of the AEC.
673 // P_a: Internal signal power at the point before the AEC's non-linear
674 // processor.
675 struct Metrics {
676 // RERL = ERL + ERLE
677 AudioProcessing::Statistic residual_echo_return_loss;
678
679 // ERL = 10log_10(P_far / P_echo)
680 AudioProcessing::Statistic echo_return_loss;
681
682 // ERLE = 10log_10(P_echo / P_out)
683 AudioProcessing::Statistic echo_return_loss_enhancement;
684
685 // (Pre non-linear processing suppression) A_NLP = 10log_10(P_echo / P_a)
686 AudioProcessing::Statistic a_nlp;
minyue50453372016-04-07 06:36:43 -0700687
688 // Fraction of time that the AEC linear filter is divergent, in a 0.5-second
689 // non-overlapped aggregation window.
690 float divergent_filter_fraction;
niklase@google.com470e71d2011-07-07 08:21:25 +0000691 };
692
693 // TODO(ajm): discuss the metrics update period.
694 virtual int GetMetrics(Metrics* metrics) = 0;
695
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000696 // Enables computation and logging of delay values. Statistics are obtained
697 // through |GetDelayMetrics()|.
698 virtual int enable_delay_logging(bool enable) = 0;
699 virtual bool is_delay_logging_enabled() const = 0;
700
701 // The delay metrics consists of the delay |median| and the delay standard
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000702 // deviation |std|. It also consists of the fraction of delay estimates
703 // |fraction_poor_delays| that can make the echo cancellation perform poorly.
704 // The values are aggregated until the first call to |GetDelayMetrics()| and
705 // afterwards aggregated and updated every second.
706 // Note that if there are several clients pulling metrics from
707 // |GetDelayMetrics()| during a session the first call from any of them will
708 // change to one second aggregation window for all.
709 // TODO(bjornv): Deprecated, remove.
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000710 virtual int GetDelayMetrics(int* median, int* std) = 0;
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000711 virtual int GetDelayMetrics(int* median, int* std,
712 float* fraction_poor_delays) = 0;
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000713
bjornv@webrtc.org91d11b32013-03-05 16:53:09 +0000714 // Returns a pointer to the low level AEC component. In case of multiple
715 // channels, the pointer to the first one is returned. A NULL pointer is
716 // returned when the AEC component is disabled or has not been initialized
717 // successfully.
718 virtual struct AecCore* aec_core() const = 0;
719
niklase@google.com470e71d2011-07-07 08:21:25 +0000720 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000721 virtual ~EchoCancellation() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000722};
723
724// The acoustic echo control for mobile (AECM) component is a low complexity
725// robust option intended for use on mobile devices.
726//
727// Not recommended to be enabled on the server-side.
728class EchoControlMobile {
729 public:
730 // EchoCancellation and EchoControlMobile may not be enabled simultaneously.
731 // Enabling one will disable the other.
732 virtual int Enable(bool enable) = 0;
733 virtual bool is_enabled() const = 0;
734
735 // Recommended settings for particular audio routes. In general, the louder
736 // the echo is expected to be, the higher this value should be set. The
737 // preferred setting may vary from device to device.
738 enum RoutingMode {
739 kQuietEarpieceOrHeadset,
740 kEarpiece,
741 kLoudEarpiece,
742 kSpeakerphone,
743 kLoudSpeakerphone
744 };
745
746 // Sets echo control appropriate for the audio routing |mode| on the device.
747 // It can and should be updated during a call if the audio routing changes.
748 virtual int set_routing_mode(RoutingMode mode) = 0;
749 virtual RoutingMode routing_mode() const = 0;
750
751 // Comfort noise replaces suppressed background noise to maintain a
752 // consistent signal level.
753 virtual int enable_comfort_noise(bool enable) = 0;
754 virtual bool is_comfort_noise_enabled() const = 0;
755
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000756 // A typical use case is to initialize the component with an echo path from a
ajm@google.com22e65152011-07-18 18:03:01 +0000757 // previous call. The echo path is retrieved using |GetEchoPath()|, typically
758 // at the end of a call. The data can then be stored for later use as an
759 // initializer before the next call, using |SetEchoPath()|.
760 //
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000761 // Controlling the echo path this way requires the data |size_bytes| to match
762 // the internal echo path size. This size can be acquired using
763 // |echo_path_size_bytes()|. |SetEchoPath()| causes an entire reset, worth
ajm@google.com22e65152011-07-18 18:03:01 +0000764 // noting if it is to be called during an ongoing call.
765 //
766 // It is possible that version incompatibilities may result in a stored echo
767 // path of the incorrect size. In this case, the stored path should be
768 // discarded.
769 virtual int SetEchoPath(const void* echo_path, size_t size_bytes) = 0;
770 virtual int GetEchoPath(void* echo_path, size_t size_bytes) const = 0;
771
772 // The returned path size is guaranteed not to change for the lifetime of
773 // the application.
774 static size_t echo_path_size_bytes();
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000775
niklase@google.com470e71d2011-07-07 08:21:25 +0000776 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000777 virtual ~EchoControlMobile() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000778};
779
780// The automatic gain control (AGC) component brings the signal to an
781// appropriate range. This is done by applying a digital gain directly and, in
782// the analog mode, prescribing an analog gain to be applied at the audio HAL.
783//
784// Recommended to be enabled on the client-side.
785class GainControl {
786 public:
787 virtual int Enable(bool enable) = 0;
788 virtual bool is_enabled() const = 0;
789
790 // When an analog mode is set, this must be called prior to |ProcessStream()|
791 // to pass the current analog level from the audio HAL. Must be within the
792 // range provided to |set_analog_level_limits()|.
793 virtual int set_stream_analog_level(int level) = 0;
794
795 // When an analog mode is set, this should be called after |ProcessStream()|
796 // to obtain the recommended new analog level for the audio HAL. It is the
797 // users responsibility to apply this level.
798 virtual int stream_analog_level() = 0;
799
800 enum Mode {
801 // Adaptive mode intended for use if an analog volume control is available
802 // on the capture device. It will require the user to provide coupling
803 // between the OS mixer controls and AGC through the |stream_analog_level()|
804 // functions.
805 //
806 // It consists of an analog gain prescription for the audio device and a
807 // digital compression stage.
808 kAdaptiveAnalog,
809
810 // Adaptive mode intended for situations in which an analog volume control
811 // is unavailable. It operates in a similar fashion to the adaptive analog
812 // mode, but with scaling instead applied in the digital domain. As with
813 // the analog mode, it additionally uses a digital compression stage.
814 kAdaptiveDigital,
815
816 // Fixed mode which enables only the digital compression stage also used by
817 // the two adaptive modes.
818 //
819 // It is distinguished from the adaptive modes by considering only a
820 // short time-window of the input signal. It applies a fixed gain through
821 // most of the input level range, and compresses (gradually reduces gain
822 // with increasing level) the input signal at higher levels. This mode is
823 // preferred on embedded devices where the capture signal level is
824 // predictable, so that a known gain can be applied.
825 kFixedDigital
826 };
827
828 virtual int set_mode(Mode mode) = 0;
829 virtual Mode mode() const = 0;
830
831 // Sets the target peak |level| (or envelope) of the AGC in dBFs (decibels
832 // from digital full-scale). The convention is to use positive values. For
833 // instance, passing in a value of 3 corresponds to -3 dBFs, or a target
834 // level 3 dB below full-scale. Limited to [0, 31].
835 //
836 // TODO(ajm): use a negative value here instead, if/when VoE will similarly
837 // update its interface.
838 virtual int set_target_level_dbfs(int level) = 0;
839 virtual int target_level_dbfs() const = 0;
840
841 // Sets the maximum |gain| the digital compression stage may apply, in dB. A
842 // higher number corresponds to greater compression, while a value of 0 will
843 // leave the signal uncompressed. Limited to [0, 90].
844 virtual int set_compression_gain_db(int gain) = 0;
845 virtual int compression_gain_db() const = 0;
846
847 // When enabled, the compression stage will hard limit the signal to the
848 // target level. Otherwise, the signal will be compressed but not limited
849 // above the target level.
850 virtual int enable_limiter(bool enable) = 0;
851 virtual bool is_limiter_enabled() const = 0;
852
853 // Sets the |minimum| and |maximum| analog levels of the audio capture device.
854 // Must be set if and only if an analog mode is used. Limited to [0, 65535].
855 virtual int set_analog_level_limits(int minimum,
856 int maximum) = 0;
857 virtual int analog_level_minimum() const = 0;
858 virtual int analog_level_maximum() const = 0;
859
860 // Returns true if the AGC has detected a saturation event (period where the
861 // signal reaches digital full-scale) in the current frame and the analog
862 // level cannot be reduced.
863 //
864 // This could be used as an indicator to reduce or disable analog mic gain at
865 // the audio HAL.
866 virtual bool stream_is_saturated() const = 0;
867
868 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000869 virtual ~GainControl() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000870};
871
872// A filtering component which removes DC offset and low-frequency noise.
873// Recommended to be enabled on the client-side.
874class HighPassFilter {
875 public:
876 virtual int Enable(bool enable) = 0;
877 virtual bool is_enabled() const = 0;
878
879 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000880 virtual ~HighPassFilter() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000881};
882
883// An estimation component used to retrieve level metrics.
884class LevelEstimator {
885 public:
886 virtual int Enable(bool enable) = 0;
887 virtual bool is_enabled() const = 0;
888
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000889 // Returns the root mean square (RMS) level in dBFs (decibels from digital
890 // full-scale), or alternately dBov. It is computed over all primary stream
891 // frames since the last call to RMS(). The returned value is positive but
892 // should be interpreted as negative. It is constrained to [0, 127].
893 //
andrew@webrtc.org382c0c22014-05-05 18:22:21 +0000894 // The computation follows: https://tools.ietf.org/html/rfc6465
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000895 // with the intent that it can provide the RTP audio level indication.
896 //
897 // Frames passed to ProcessStream() with an |_energy| of zero are considered
898 // to have been muted. The RMS of the frame will be interpreted as -127.
899 virtual int RMS() = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000900
901 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000902 virtual ~LevelEstimator() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000903};
904
905// The noise suppression (NS) component attempts to remove noise while
906// retaining speech. Recommended to be enabled on the client-side.
907//
908// Recommended to be enabled on the client-side.
909class NoiseSuppression {
910 public:
911 virtual int Enable(bool enable) = 0;
912 virtual bool is_enabled() const = 0;
913
914 // Determines the aggressiveness of the suppression. Increasing the level
915 // will reduce the noise level at the expense of a higher speech distortion.
916 enum Level {
917 kLow,
918 kModerate,
919 kHigh,
920 kVeryHigh
921 };
922
923 virtual int set_level(Level level) = 0;
924 virtual Level level() const = 0;
925
bjornv@webrtc.org08329f42012-07-12 21:00:43 +0000926 // Returns the internally computed prior speech probability of current frame
927 // averaged over output channels. This is not supported in fixed point, for
928 // which |kUnsupportedFunctionError| is returned.
929 virtual float speech_probability() const = 0;
930
Alejandro Luebsfa639f02016-02-09 11:24:32 -0800931 // Returns the noise estimate per frequency bin averaged over all channels.
932 virtual std::vector<float> NoiseEstimate() = 0;
933
niklase@google.com470e71d2011-07-07 08:21:25 +0000934 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000935 virtual ~NoiseSuppression() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000936};
937
938// The voice activity detection (VAD) component analyzes the stream to
939// determine if voice is present. A facility is also provided to pass in an
940// external VAD decision.
andrew@webrtc.orged083d42011-09-19 15:28:51 +0000941//
942// In addition to |stream_has_voice()| the VAD decision is provided through the
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000943// |AudioFrame| passed to |ProcessStream()|. The |vad_activity_| member will be
andrew@webrtc.orged083d42011-09-19 15:28:51 +0000944// modified to reflect the current decision.
niklase@google.com470e71d2011-07-07 08:21:25 +0000945class VoiceDetection {
946 public:
947 virtual int Enable(bool enable) = 0;
948 virtual bool is_enabled() const = 0;
949
950 // Returns true if voice is detected in the current frame. Should be called
951 // after |ProcessStream()|.
952 virtual bool stream_has_voice() const = 0;
953
954 // Some of the APM functionality requires a VAD decision. In the case that
955 // a decision is externally available for the current frame, it can be passed
956 // in here, before |ProcessStream()| is called.
957 //
958 // VoiceDetection does _not_ need to be enabled to use this. If it happens to
959 // be enabled, detection will be skipped for any frame in which an external
960 // VAD decision is provided.
961 virtual int set_stream_has_voice(bool has_voice) = 0;
962
963 // Specifies the likelihood that a frame will be declared to contain voice.
964 // A higher value makes it more likely that speech will not be clipped, at
965 // the expense of more noise being detected as voice.
966 enum Likelihood {
967 kVeryLowLikelihood,
968 kLowLikelihood,
969 kModerateLikelihood,
970 kHighLikelihood
971 };
972
973 virtual int set_likelihood(Likelihood likelihood) = 0;
974 virtual Likelihood likelihood() const = 0;
975
976 // Sets the |size| of the frames in ms on which the VAD will operate. Larger
977 // frames will improve detection accuracy, but reduce the frequency of
978 // updates.
979 //
980 // This does not impact the size of frames passed to |ProcessStream()|.
981 virtual int set_frame_size_ms(int size) = 0;
982 virtual int frame_size_ms() const = 0;
983
984 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000985 virtual ~VoiceDetection() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000986};
987} // namespace webrtc
988
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000989#endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_