blob: 72553ffa6e7dbaa1cd04d027943213e4891723bb [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
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000014#include <stddef.h> // size_t
henrikg@webrtc.org863b5362013-12-06 16:05:17 +000015#include <stdio.h> // FILE
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +000016#include <vector>
ajm@google.com22e65152011-07-18 18:03:01 +000017
xians@webrtc.orge46bc772014-10-10 08:36:56 +000018#include "webrtc/base/platform_file.h"
andrew@webrtc.org61e596f2013-07-25 18:28:29 +000019#include "webrtc/common.h"
aluebs@webrtc.org1d883942015-03-05 20:38:21 +000020#include "webrtc/modules/audio_processing/beamformer/array_util.h"
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +000021#include "webrtc/typedefs.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000022
bjornv@webrtc.org91d11b32013-03-05 16:53:09 +000023struct AecCore;
24
niklase@google.com470e71d2011-07-07 08:21:25 +000025namespace webrtc {
26
27class AudioFrame;
Michael Graczykdfa36052015-03-25 16:37:27 -070028
29template<typename T>
30class Beamformer;
31
niklase@google.com470e71d2011-07-07 08:21:25 +000032class EchoCancellation;
33class EchoControlMobile;
34class GainControl;
35class HighPassFilter;
36class LevelEstimator;
37class NoiseSuppression;
38class VoiceDetection;
39
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +000040// Use to enable the delay correction feature. This now engages an extended
41// filter mode in the AEC, along with robustness measures around the reported
42// system delays. It comes with a significant increase in AEC complexity, but is
43// much more robust to unreliable reported delays.
44//
45// Detailed changes to the algorithm:
46// - The filter length is changed from 48 to 128 ms. This comes with tuning of
47// several parameters: i) filter adaptation stepsize and error threshold;
48// ii) non-linear processing smoothing and overdrive.
49// - Option to ignore the reported delays on platforms which we deem
50// sufficiently unreliable. See WEBRTC_UNTRUSTED_DELAY in echo_cancellation.c.
51// - Faster startup times by removing the excessive "startup phase" processing
52// of reported delays.
53// - Much more conservative adjustments to the far-end read pointer. We smooth
54// the delay difference more heavily, and back off from the difference more.
55// Adjustments force a readaptation of the filter, so they should be avoided
56// except when really necessary.
57struct DelayCorrection {
58 DelayCorrection() : enabled(false) {}
andrew@webrtc.orgc7c7a532014-01-29 04:57:25 +000059 explicit DelayCorrection(bool enabled) : enabled(enabled) {}
60 bool enabled;
61};
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +000062
bjornv@webrtc.org3f830722014-06-11 04:48:11 +000063// Use to disable the reported system delays. By disabling the reported system
64// delays the echo cancellation algorithm assumes the process and reverse
65// streams to be aligned. This configuration only applies to EchoCancellation
66// and not EchoControlMobile and is set with AudioProcessing::SetExtraOptions().
67// Note that by disabling reported system delays the EchoCancellation may
68// regress in performance.
69struct ReportedDelay {
70 ReportedDelay() : enabled(true) {}
71 explicit ReportedDelay(bool enabled) : enabled(enabled) {}
72 bool enabled;
73};
74
andrew@webrtc.orgc7c7a532014-01-29 04:57:25 +000075// Must be provided through AudioProcessing::Create(Confg&). It will have no
76// impact if used with AudioProcessing::SetExtraOptions().
77struct ExperimentalAgc {
78 ExperimentalAgc() : enabled(true) {}
79 explicit ExperimentalAgc(bool enabled) : enabled(enabled) {}
andrew@webrtc.org6b1e2192013-09-25 23:46:20 +000080 bool enabled;
81};
82
aluebs@webrtc.org9825afc2014-06-30 17:39:53 +000083// Use to enable experimental noise suppression. It can be set in the
84// constructor or using AudioProcessing::SetExtraOptions().
85struct ExperimentalNs {
86 ExperimentalNs() : enabled(false) {}
87 explicit ExperimentalNs(bool enabled) : enabled(enabled) {}
88 bool enabled;
89};
90
aluebs@webrtc.orgae643ce2014-12-19 19:57:34 +000091// Use to enable beamforming. Must be provided through the constructor. It will
92// have no impact if used with AudioProcessing::SetExtraOptions().
93struct Beamforming {
94 Beamforming() : enabled(false) {}
aluebs@webrtc.orgfb7a0392015-01-05 21:58:58 +000095 Beamforming(bool enabled, const std::vector<Point>& array_geometry)
96 : enabled(enabled),
97 array_geometry(array_geometry) {}
98 const bool enabled;
99 const std::vector<Point> array_geometry;
aluebs@webrtc.orgae643ce2014-12-19 19:57:34 +0000100};
101
aluebs@webrtc.orgc9ce07e2015-03-02 20:07:31 +0000102// Use to enable 48kHz support in audio processing. Must be provided through the
103// constructor. It will have no impact if used with
104// AudioProcessing::SetExtraOptions().
105struct AudioProcessing48kHzSupport {
106 AudioProcessing48kHzSupport() : enabled(false) {}
107 explicit AudioProcessing48kHzSupport(bool enabled) : enabled(enabled) {}
108 bool enabled;
109};
110
andrew@webrtc.org40ee3d02014-04-03 21:56:01 +0000111static const int kAudioProcMaxNativeSampleRateHz = 32000;
112
niklase@google.com470e71d2011-07-07 08:21:25 +0000113// The Audio Processing Module (APM) provides a collection of voice processing
114// components designed for real-time communications software.
115//
116// APM operates on two audio streams on a frame-by-frame basis. Frames of the
117// primary stream, on which all processing is applied, are passed to
118// |ProcessStream()|. Frames of the reverse direction stream, which are used for
119// analysis by some components, are passed to |AnalyzeReverseStream()|. On the
120// client-side, this will typically be the near-end (capture) and far-end
121// (render) streams, respectively. APM should be placed in the signal chain as
122// close to the audio hardware abstraction layer (HAL) as possible.
123//
124// On the server-side, the reverse stream will normally not be used, with
125// processing occurring on each incoming stream.
126//
127// Component interfaces follow a similar pattern and are accessed through
128// corresponding getters in APM. All components are disabled at create-time,
129// with default settings that are recommended for most situations. New settings
130// can be applied without enabling a component. Enabling a component triggers
131// memory allocation and initialization to allow it to start processing the
132// streams.
133//
134// Thread safety is provided with the following assumptions to reduce locking
135// overhead:
136// 1. The stream getters and setters are called from the same thread as
137// ProcessStream(). More precisely, stream functions are never called
138// concurrently with ProcessStream().
139// 2. Parameter getters are never called concurrently with the corresponding
140// setter.
141//
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000142// APM accepts only linear PCM audio data in chunks of 10 ms. The int16
143// interfaces use interleaved data, while the float interfaces use deinterleaved
144// data.
niklase@google.com470e71d2011-07-07 08:21:25 +0000145//
146// Usage example, omitting error checking:
147// AudioProcessing* apm = AudioProcessing::Create(0);
niklase@google.com470e71d2011-07-07 08:21:25 +0000148//
149// apm->high_pass_filter()->Enable(true);
150//
151// apm->echo_cancellation()->enable_drift_compensation(false);
152// apm->echo_cancellation()->Enable(true);
153//
154// apm->noise_reduction()->set_level(kHighSuppression);
155// apm->noise_reduction()->Enable(true);
156//
157// apm->gain_control()->set_analog_level_limits(0, 255);
158// apm->gain_control()->set_mode(kAdaptiveAnalog);
159// apm->gain_control()->Enable(true);
160//
161// apm->voice_detection()->Enable(true);
162//
163// // Start a voice call...
164//
165// // ... Render frame arrives bound for the audio HAL ...
166// apm->AnalyzeReverseStream(render_frame);
167//
168// // ... Capture frame arrives from the audio HAL ...
169// // Call required set_stream_ functions.
170// apm->set_stream_delay_ms(delay_ms);
171// apm->gain_control()->set_stream_analog_level(analog_level);
172//
173// apm->ProcessStream(capture_frame);
174//
175// // Call required stream_ functions.
176// analog_level = apm->gain_control()->stream_analog_level();
177// has_voice = apm->stream_has_voice();
178//
179// // Repeate render and capture processing for the duration of the call...
180// // Start a new call...
181// apm->Initialize();
182//
183// // Close the application...
andrew@webrtc.orgf3930e92013-09-18 22:37:32 +0000184// delete apm;
niklase@google.com470e71d2011-07-07 08:21:25 +0000185//
andrew@webrtc.orgf92aaff2014-02-15 04:22:49 +0000186class AudioProcessing {
niklase@google.com470e71d2011-07-07 08:21:25 +0000187 public:
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000188 enum ChannelLayout {
189 kMono,
190 // Left, right.
191 kStereo,
192 // Mono, keyboard mic.
193 kMonoAndKeyboard,
194 // Left, right, keyboard mic.
195 kStereoAndKeyboard
196 };
197
andrew@webrtc.org54744912014-02-05 06:30:29 +0000198 // Creates an APM instance. Use one instance for every primary audio stream
199 // requiring processing. On the client-side, this would typically be one
200 // instance for the near-end stream, and additional instances for each far-end
201 // stream which requires processing. On the server-side, this would typically
202 // be one instance for every incoming stream.
andrew@webrtc.orge84978f2014-01-25 02:09:06 +0000203 static AudioProcessing* Create();
andrew@webrtc.org54744912014-02-05 06:30:29 +0000204 // Allows passing in an optional configuration at create-time.
andrew@webrtc.orge84978f2014-01-25 02:09:06 +0000205 static AudioProcessing* Create(const Config& config);
aluebs@webrtc.orgd82f55d2015-01-15 18:07:21 +0000206 // Only for testing.
mgraczyk@chromium.org0f663de2015-03-13 00:13:32 +0000207 static AudioProcessing* Create(const Config& config,
Michael Graczykdfa36052015-03-25 16:37:27 -0700208 Beamformer<float>* beamformer);
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000209 virtual ~AudioProcessing() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000210
niklase@google.com470e71d2011-07-07 08:21:25 +0000211 // Initializes internal states, while retaining all user settings. This
212 // should be called before beginning to process a new audio stream. However,
213 // it is not necessary to call before processing the first stream after
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000214 // creation.
215 //
216 // It is also not necessary to call if the audio parameters (sample
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000217 // rate and number of channels) have changed. Passing updated parameters
218 // directly to |ProcessStream()| and |AnalyzeReverseStream()| is permissible.
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000219 // If the parameters are known at init-time though, they may be provided.
niklase@google.com470e71d2011-07-07 08:21:25 +0000220 virtual int Initialize() = 0;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000221
222 // The int16 interfaces require:
223 // - only |NativeRate|s be used
224 // - that the input, output and reverse rates must match
225 // - that |output_layout| matches |input_layout|
226 //
227 // The float interfaces accept arbitrary rates and support differing input
228 // and output layouts, but the output may only remove channels, not add.
229 virtual int Initialize(int input_sample_rate_hz,
230 int output_sample_rate_hz,
andrew@webrtc.orga8b97372014-03-10 22:26:12 +0000231 int reverse_sample_rate_hz,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000232 ChannelLayout input_layout,
233 ChannelLayout output_layout,
234 ChannelLayout reverse_layout) = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000235
andrew@webrtc.org61e596f2013-07-25 18:28:29 +0000236 // Pass down additional options which don't have explicit setters. This
237 // ensures the options are applied immediately.
238 virtual void SetExtraOptions(const Config& config) = 0;
239
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000240 // DEPRECATED.
241 // TODO(ajm): Remove after Chromium has upgraded to using Initialize().
niklase@google.com470e71d2011-07-07 08:21:25 +0000242 virtual int set_sample_rate_hz(int rate) = 0;
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000243 // TODO(ajm): Remove after voice engine no longer requires it to resample
244 // the reverse stream to the forward rate.
245 virtual int input_sample_rate_hz() const = 0;
andrew@webrtc.org46b31b12014-04-23 03:33:54 +0000246 // TODO(ajm): Remove after Chromium no longer depends on it.
247 virtual int sample_rate_hz() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000248
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000249 // TODO(ajm): Only intended for internal use. Make private and friend the
250 // necessary classes?
251 virtual int proc_sample_rate_hz() const = 0;
252 virtual int proc_split_sample_rate_hz() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000253 virtual int num_input_channels() const = 0;
254 virtual int num_output_channels() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000255 virtual int num_reverse_channels() const = 0;
256
andrew@webrtc.org17342e52014-02-12 22:28:31 +0000257 // Set to true when the output of AudioProcessing will be muted or in some
258 // other way not used. Ideally, the captured audio would still be processed,
259 // but some components may change behavior based on this information.
260 // Default false.
261 virtual void set_output_will_be_muted(bool muted) = 0;
262 virtual bool output_will_be_muted() const = 0;
263
niklase@google.com470e71d2011-07-07 08:21:25 +0000264 // Processes a 10 ms |frame| of the primary audio stream. On the client-side,
265 // this is the near-end (or captured) audio.
266 //
267 // If needed for enabled functionality, any function with the set_stream_ tag
268 // must be called prior to processing the current frame. Any getter function
269 // with the stream_ tag which is needed should be called after processing.
270 //
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000271 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_|
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000272 // members of |frame| must be valid. If changed from the previous call to this
273 // method, it will trigger an initialization.
niklase@google.com470e71d2011-07-07 08:21:25 +0000274 virtual int ProcessStream(AudioFrame* frame) = 0;
275
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000276 // Accepts deinterleaved float audio with the range [-1, 1]. Each element
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000277 // of |src| points to a channel buffer, arranged according to
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000278 // |input_layout|. At output, the channels will be arranged according to
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000279 // |output_layout| at |output_sample_rate_hz| in |dest|.
280 //
281 // The output layout may only remove channels, not add. |src| and |dest|
282 // may use the same memory, if desired.
283 virtual int ProcessStream(const float* const* src,
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000284 int samples_per_channel,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000285 int input_sample_rate_hz,
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000286 ChannelLayout input_layout,
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000287 int output_sample_rate_hz,
288 ChannelLayout output_layout,
289 float* const* dest) = 0;
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000290
niklase@google.com470e71d2011-07-07 08:21:25 +0000291 // Analyzes a 10 ms |frame| of the reverse direction audio stream. The frame
292 // will not be modified. On the client-side, this is the far-end (or to be
293 // rendered) audio.
294 //
295 // It is only necessary to provide this if echo processing is enabled, as the
296 // reverse stream forms the echo reference signal. It is recommended, but not
297 // necessary, to provide if gain control is enabled. On the server-side this
298 // typically will not be used. If you're not sure what to pass in here,
299 // chances are you don't need to use it.
300 //
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000301 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_|
andrew@webrtc.org60730cf2014-01-07 17:45:09 +0000302 // members of |frame| must be valid. |sample_rate_hz_| must correspond to
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000303 // |input_sample_rate_hz()|
niklase@google.com470e71d2011-07-07 08:21:25 +0000304 //
305 // TODO(ajm): add const to input; requires an implementation fix.
306 virtual int AnalyzeReverseStream(AudioFrame* frame) = 0;
307
andrew@webrtc.org17e40642014-03-04 20:58:13 +0000308 // Accepts deinterleaved float audio with the range [-1, 1]. Each element
309 // of |data| points to a channel buffer, arranged according to |layout|.
310 virtual int AnalyzeReverseStream(const float* const* data,
311 int samples_per_channel,
312 int sample_rate_hz,
313 ChannelLayout layout) = 0;
314
niklase@google.com470e71d2011-07-07 08:21:25 +0000315 // This must be called if and only if echo processing is enabled.
316 //
317 // Sets the |delay| in ms between AnalyzeReverseStream() receiving a far-end
318 // frame and ProcessStream() receiving a near-end frame containing the
319 // corresponding echo. On the client-side this can be expressed as
320 // delay = (t_render - t_analyze) + (t_process - t_capture)
321 // where,
322 // - t_analyze is the time a frame is passed to AnalyzeReverseStream() and
323 // t_render is the time the first sample of the same frame is rendered by
324 // the audio hardware.
325 // - t_capture is the time the first sample of a frame is captured by the
326 // audio hardware and t_pull is the time the same frame is passed to
327 // ProcessStream().
328 virtual int set_stream_delay_ms(int delay) = 0;
329 virtual int stream_delay_ms() const = 0;
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000330 virtual bool was_stream_delay_set() const = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000331
andrew@webrtc.org75dd2882014-02-11 20:52:30 +0000332 // Call to signal that a key press occurred (true) or did not occur (false)
333 // with this chunk of audio.
334 virtual void set_stream_key_pressed(bool key_pressed) = 0;
335 virtual bool stream_key_pressed() const = 0;
336
andrew@webrtc.org6f9f8172012-03-06 19:03:39 +0000337 // Sets a delay |offset| in ms to add to the values passed in through
338 // set_stream_delay_ms(). May be positive or negative.
339 //
340 // Note that this could cause an otherwise valid value passed to
341 // set_stream_delay_ms() to return an error.
342 virtual void set_delay_offset_ms(int offset) = 0;
343 virtual int delay_offset_ms() const = 0;
344
niklase@google.com470e71d2011-07-07 08:21:25 +0000345 // Starts recording debugging information to a file specified by |filename|,
346 // a NULL-terminated string. If there is an ongoing recording, the old file
347 // will be closed, and recording will continue in the newly specified file.
348 // An already existing file will be overwritten without warning.
andrew@webrtc.org5ae19de2011-12-13 22:59:33 +0000349 static const size_t kMaxFilenameSize = 1024;
niklase@google.com470e71d2011-07-07 08:21:25 +0000350 virtual int StartDebugRecording(const char filename[kMaxFilenameSize]) = 0;
351
henrikg@webrtc.org863b5362013-12-06 16:05:17 +0000352 // Same as above but uses an existing file handle. Takes ownership
353 // of |handle| and closes it at StopDebugRecording().
354 virtual int StartDebugRecording(FILE* handle) = 0;
355
xians@webrtc.orge46bc772014-10-10 08:36:56 +0000356 // Same as above but uses an existing PlatformFile handle. Takes ownership
357 // of |handle| and closes it at StopDebugRecording().
358 // TODO(xians): Make this interface pure virtual.
359 virtual int StartDebugRecordingForPlatformFile(rtc::PlatformFile handle) {
360 return -1;
361 }
362
niklase@google.com470e71d2011-07-07 08:21:25 +0000363 // Stops recording debugging information, and closes the file. Recording
364 // cannot be resumed in the same file (without overwriting it).
365 virtual int StopDebugRecording() = 0;
366
367 // These provide access to the component interfaces and should never return
368 // NULL. The pointers will be valid for the lifetime of the APM instance.
369 // The memory for these objects is entirely managed internally.
370 virtual EchoCancellation* echo_cancellation() const = 0;
371 virtual EchoControlMobile* echo_control_mobile() const = 0;
372 virtual GainControl* gain_control() const = 0;
373 virtual HighPassFilter* high_pass_filter() const = 0;
374 virtual LevelEstimator* level_estimator() const = 0;
375 virtual NoiseSuppression* noise_suppression() const = 0;
376 virtual VoiceDetection* voice_detection() const = 0;
377
378 struct Statistic {
379 int instant; // Instantaneous value.
380 int average; // Long-term average.
381 int maximum; // Long-term maximum.
382 int minimum; // Long-term minimum.
383 };
384
andrew@webrtc.org648af742012-02-08 01:57:29 +0000385 enum Error {
386 // Fatal errors.
niklase@google.com470e71d2011-07-07 08:21:25 +0000387 kNoError = 0,
388 kUnspecifiedError = -1,
389 kCreationFailedError = -2,
390 kUnsupportedComponentError = -3,
391 kUnsupportedFunctionError = -4,
392 kNullPointerError = -5,
393 kBadParameterError = -6,
394 kBadSampleRateError = -7,
395 kBadDataLengthError = -8,
396 kBadNumberChannelsError = -9,
397 kFileError = -10,
398 kStreamParameterNotSetError = -11,
andrew@webrtc.org648af742012-02-08 01:57:29 +0000399 kNotEnabledError = -12,
niklase@google.com470e71d2011-07-07 08:21:25 +0000400
andrew@webrtc.org648af742012-02-08 01:57:29 +0000401 // Warnings are non-fatal.
niklase@google.com470e71d2011-07-07 08:21:25 +0000402 // This results when a set_stream_ parameter is out of range. Processing
403 // will continue, but the parameter may have been truncated.
andrew@webrtc.org648af742012-02-08 01:57:29 +0000404 kBadStreamParameterWarning = -13
niklase@google.com470e71d2011-07-07 08:21:25 +0000405 };
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000406
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000407 enum NativeRate {
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000408 kSampleRate8kHz = 8000,
409 kSampleRate16kHz = 16000,
aluebs@webrtc.org087da132014-11-17 23:01:23 +0000410 kSampleRate32kHz = 32000,
411 kSampleRate48kHz = 48000
andrew@webrtc.org56e4a052014-02-27 22:23:17 +0000412 };
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000413
414 static const int kChunkSizeMs = 10;
niklase@google.com470e71d2011-07-07 08:21:25 +0000415};
416
417// The acoustic echo cancellation (AEC) component provides better performance
418// than AECM but also requires more processing power and is dependent on delay
419// stability and reporting accuracy. As such it is well-suited and recommended
420// for PC and IP phone applications.
421//
422// Not recommended to be enabled on the server-side.
423class EchoCancellation {
424 public:
425 // EchoCancellation and EchoControlMobile may not be enabled simultaneously.
426 // Enabling one will disable the other.
427 virtual int Enable(bool enable) = 0;
428 virtual bool is_enabled() const = 0;
429
430 // Differences in clock speed on the primary and reverse streams can impact
431 // the AEC performance. On the client-side, this could be seen when different
432 // render and capture devices are used, particularly with webcams.
433 //
434 // This enables a compensation mechanism, and requires that
andrew@webrtc.orgddbb8a22014-04-22 21:00:04 +0000435 // set_stream_drift_samples() be called.
niklase@google.com470e71d2011-07-07 08:21:25 +0000436 virtual int enable_drift_compensation(bool enable) = 0;
437 virtual bool is_drift_compensation_enabled() const = 0;
438
niklase@google.com470e71d2011-07-07 08:21:25 +0000439 // Sets the difference between the number of samples rendered and captured by
440 // the audio devices since the last call to |ProcessStream()|. Must be called
andrew@webrtc.org6be1e932013-03-01 18:47:28 +0000441 // if drift compensation is enabled, prior to |ProcessStream()|.
442 virtual void set_stream_drift_samples(int drift) = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000443 virtual int stream_drift_samples() const = 0;
444
445 enum SuppressionLevel {
446 kLowSuppression,
447 kModerateSuppression,
448 kHighSuppression
449 };
450
451 // Sets the aggressiveness of the suppressor. A higher level trades off
452 // double-talk performance for increased echo suppression.
453 virtual int set_suppression_level(SuppressionLevel level) = 0;
454 virtual SuppressionLevel suppression_level() const = 0;
455
456 // Returns false if the current frame almost certainly contains no echo
457 // and true if it _might_ contain echo.
458 virtual bool stream_has_echo() const = 0;
459
460 // Enables the computation of various echo metrics. These are obtained
461 // through |GetMetrics()|.
462 virtual int enable_metrics(bool enable) = 0;
463 virtual bool are_metrics_enabled() const = 0;
464
465 // Each statistic is reported in dB.
466 // P_far: Far-end (render) signal power.
467 // P_echo: Near-end (capture) echo signal power.
468 // P_out: Signal power at the output of the AEC.
469 // P_a: Internal signal power at the point before the AEC's non-linear
470 // processor.
471 struct Metrics {
472 // RERL = ERL + ERLE
473 AudioProcessing::Statistic residual_echo_return_loss;
474
475 // ERL = 10log_10(P_far / P_echo)
476 AudioProcessing::Statistic echo_return_loss;
477
478 // ERLE = 10log_10(P_echo / P_out)
479 AudioProcessing::Statistic echo_return_loss_enhancement;
480
481 // (Pre non-linear processing suppression) A_NLP = 10log_10(P_echo / P_a)
482 AudioProcessing::Statistic a_nlp;
483 };
484
485 // TODO(ajm): discuss the metrics update period.
486 virtual int GetMetrics(Metrics* metrics) = 0;
487
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000488 // Enables computation and logging of delay values. Statistics are obtained
489 // through |GetDelayMetrics()|.
490 virtual int enable_delay_logging(bool enable) = 0;
491 virtual bool is_delay_logging_enabled() const = 0;
492
493 // The delay metrics consists of the delay |median| and the delay standard
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000494 // deviation |std|. It also consists of the fraction of delay estimates
495 // |fraction_poor_delays| that can make the echo cancellation perform poorly.
496 // The values are aggregated until the first call to |GetDelayMetrics()| and
497 // afterwards aggregated and updated every second.
498 // Note that if there are several clients pulling metrics from
499 // |GetDelayMetrics()| during a session the first call from any of them will
500 // change to one second aggregation window for all.
501 // TODO(bjornv): Deprecated, remove.
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000502 virtual int GetDelayMetrics(int* median, int* std) = 0;
bjornv@webrtc.orgb1786db2015-02-03 06:06:26 +0000503 virtual int GetDelayMetrics(int* median, int* std,
504 float* fraction_poor_delays) = 0;
bjornv@google.com1ba3dbe2011-10-03 08:18:10 +0000505
bjornv@webrtc.org91d11b32013-03-05 16:53:09 +0000506 // Returns a pointer to the low level AEC component. In case of multiple
507 // channels, the pointer to the first one is returned. A NULL pointer is
508 // returned when the AEC component is disabled or has not been initialized
509 // successfully.
510 virtual struct AecCore* aec_core() const = 0;
511
niklase@google.com470e71d2011-07-07 08:21:25 +0000512 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000513 virtual ~EchoCancellation() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000514};
515
516// The acoustic echo control for mobile (AECM) component is a low complexity
517// robust option intended for use on mobile devices.
518//
519// Not recommended to be enabled on the server-side.
520class EchoControlMobile {
521 public:
522 // EchoCancellation and EchoControlMobile may not be enabled simultaneously.
523 // Enabling one will disable the other.
524 virtual int Enable(bool enable) = 0;
525 virtual bool is_enabled() const = 0;
526
527 // Recommended settings for particular audio routes. In general, the louder
528 // the echo is expected to be, the higher this value should be set. The
529 // preferred setting may vary from device to device.
530 enum RoutingMode {
531 kQuietEarpieceOrHeadset,
532 kEarpiece,
533 kLoudEarpiece,
534 kSpeakerphone,
535 kLoudSpeakerphone
536 };
537
538 // Sets echo control appropriate for the audio routing |mode| on the device.
539 // It can and should be updated during a call if the audio routing changes.
540 virtual int set_routing_mode(RoutingMode mode) = 0;
541 virtual RoutingMode routing_mode() const = 0;
542
543 // Comfort noise replaces suppressed background noise to maintain a
544 // consistent signal level.
545 virtual int enable_comfort_noise(bool enable) = 0;
546 virtual bool is_comfort_noise_enabled() const = 0;
547
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000548 // A typical use case is to initialize the component with an echo path from a
ajm@google.com22e65152011-07-18 18:03:01 +0000549 // previous call. The echo path is retrieved using |GetEchoPath()|, typically
550 // at the end of a call. The data can then be stored for later use as an
551 // initializer before the next call, using |SetEchoPath()|.
552 //
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000553 // Controlling the echo path this way requires the data |size_bytes| to match
554 // the internal echo path size. This size can be acquired using
555 // |echo_path_size_bytes()|. |SetEchoPath()| causes an entire reset, worth
ajm@google.com22e65152011-07-18 18:03:01 +0000556 // noting if it is to be called during an ongoing call.
557 //
558 // It is possible that version incompatibilities may result in a stored echo
559 // path of the incorrect size. In this case, the stored path should be
560 // discarded.
561 virtual int SetEchoPath(const void* echo_path, size_t size_bytes) = 0;
562 virtual int GetEchoPath(void* echo_path, size_t size_bytes) const = 0;
563
564 // The returned path size is guaranteed not to change for the lifetime of
565 // the application.
566 static size_t echo_path_size_bytes();
bjornv@google.comc4b939c2011-07-13 08:09:56 +0000567
niklase@google.com470e71d2011-07-07 08:21:25 +0000568 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000569 virtual ~EchoControlMobile() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000570};
571
572// The automatic gain control (AGC) component brings the signal to an
573// appropriate range. This is done by applying a digital gain directly and, in
574// the analog mode, prescribing an analog gain to be applied at the audio HAL.
575//
576// Recommended to be enabled on the client-side.
577class GainControl {
578 public:
579 virtual int Enable(bool enable) = 0;
580 virtual bool is_enabled() const = 0;
581
582 // When an analog mode is set, this must be called prior to |ProcessStream()|
583 // to pass the current analog level from the audio HAL. Must be within the
584 // range provided to |set_analog_level_limits()|.
585 virtual int set_stream_analog_level(int level) = 0;
586
587 // When an analog mode is set, this should be called after |ProcessStream()|
588 // to obtain the recommended new analog level for the audio HAL. It is the
589 // users responsibility to apply this level.
590 virtual int stream_analog_level() = 0;
591
592 enum Mode {
593 // Adaptive mode intended for use if an analog volume control is available
594 // on the capture device. It will require the user to provide coupling
595 // between the OS mixer controls and AGC through the |stream_analog_level()|
596 // functions.
597 //
598 // It consists of an analog gain prescription for the audio device and a
599 // digital compression stage.
600 kAdaptiveAnalog,
601
602 // Adaptive mode intended for situations in which an analog volume control
603 // is unavailable. It operates in a similar fashion to the adaptive analog
604 // mode, but with scaling instead applied in the digital domain. As with
605 // the analog mode, it additionally uses a digital compression stage.
606 kAdaptiveDigital,
607
608 // Fixed mode which enables only the digital compression stage also used by
609 // the two adaptive modes.
610 //
611 // It is distinguished from the adaptive modes by considering only a
612 // short time-window of the input signal. It applies a fixed gain through
613 // most of the input level range, and compresses (gradually reduces gain
614 // with increasing level) the input signal at higher levels. This mode is
615 // preferred on embedded devices where the capture signal level is
616 // predictable, so that a known gain can be applied.
617 kFixedDigital
618 };
619
620 virtual int set_mode(Mode mode) = 0;
621 virtual Mode mode() const = 0;
622
623 // Sets the target peak |level| (or envelope) of the AGC in dBFs (decibels
624 // from digital full-scale). The convention is to use positive values. For
625 // instance, passing in a value of 3 corresponds to -3 dBFs, or a target
626 // level 3 dB below full-scale. Limited to [0, 31].
627 //
628 // TODO(ajm): use a negative value here instead, if/when VoE will similarly
629 // update its interface.
630 virtual int set_target_level_dbfs(int level) = 0;
631 virtual int target_level_dbfs() const = 0;
632
633 // Sets the maximum |gain| the digital compression stage may apply, in dB. A
634 // higher number corresponds to greater compression, while a value of 0 will
635 // leave the signal uncompressed. Limited to [0, 90].
636 virtual int set_compression_gain_db(int gain) = 0;
637 virtual int compression_gain_db() const = 0;
638
639 // When enabled, the compression stage will hard limit the signal to the
640 // target level. Otherwise, the signal will be compressed but not limited
641 // above the target level.
642 virtual int enable_limiter(bool enable) = 0;
643 virtual bool is_limiter_enabled() const = 0;
644
645 // Sets the |minimum| and |maximum| analog levels of the audio capture device.
646 // Must be set if and only if an analog mode is used. Limited to [0, 65535].
647 virtual int set_analog_level_limits(int minimum,
648 int maximum) = 0;
649 virtual int analog_level_minimum() const = 0;
650 virtual int analog_level_maximum() const = 0;
651
652 // Returns true if the AGC has detected a saturation event (period where the
653 // signal reaches digital full-scale) in the current frame and the analog
654 // level cannot be reduced.
655 //
656 // This could be used as an indicator to reduce or disable analog mic gain at
657 // the audio HAL.
658 virtual bool stream_is_saturated() const = 0;
659
660 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000661 virtual ~GainControl() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000662};
663
664// A filtering component which removes DC offset and low-frequency noise.
665// Recommended to be enabled on the client-side.
666class HighPassFilter {
667 public:
668 virtual int Enable(bool enable) = 0;
669 virtual bool is_enabled() const = 0;
670
671 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000672 virtual ~HighPassFilter() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000673};
674
675// An estimation component used to retrieve level metrics.
676class LevelEstimator {
677 public:
678 virtual int Enable(bool enable) = 0;
679 virtual bool is_enabled() const = 0;
680
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000681 // Returns the root mean square (RMS) level in dBFs (decibels from digital
682 // full-scale), or alternately dBov. It is computed over all primary stream
683 // frames since the last call to RMS(). The returned value is positive but
684 // should be interpreted as negative. It is constrained to [0, 127].
685 //
andrew@webrtc.org382c0c22014-05-05 18:22:21 +0000686 // The computation follows: https://tools.ietf.org/html/rfc6465
andrew@webrtc.org755b04a2011-11-15 16:57:56 +0000687 // with the intent that it can provide the RTP audio level indication.
688 //
689 // Frames passed to ProcessStream() with an |_energy| of zero are considered
690 // to have been muted. The RMS of the frame will be interpreted as -127.
691 virtual int RMS() = 0;
niklase@google.com470e71d2011-07-07 08:21:25 +0000692
693 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000694 virtual ~LevelEstimator() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000695};
696
697// The noise suppression (NS) component attempts to remove noise while
698// retaining speech. Recommended to be enabled on the client-side.
699//
700// Recommended to be enabled on the client-side.
701class NoiseSuppression {
702 public:
703 virtual int Enable(bool enable) = 0;
704 virtual bool is_enabled() const = 0;
705
706 // Determines the aggressiveness of the suppression. Increasing the level
707 // will reduce the noise level at the expense of a higher speech distortion.
708 enum Level {
709 kLow,
710 kModerate,
711 kHigh,
712 kVeryHigh
713 };
714
715 virtual int set_level(Level level) = 0;
716 virtual Level level() const = 0;
717
bjornv@webrtc.org08329f42012-07-12 21:00:43 +0000718 // Returns the internally computed prior speech probability of current frame
719 // averaged over output channels. This is not supported in fixed point, for
720 // which |kUnsupportedFunctionError| is returned.
721 virtual float speech_probability() const = 0;
722
niklase@google.com470e71d2011-07-07 08:21:25 +0000723 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000724 virtual ~NoiseSuppression() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000725};
726
727// The voice activity detection (VAD) component analyzes the stream to
728// determine if voice is present. A facility is also provided to pass in an
729// external VAD decision.
andrew@webrtc.orged083d42011-09-19 15:28:51 +0000730//
731// In addition to |stream_has_voice()| the VAD decision is provided through the
andrew@webrtc.org63a50982012-05-02 23:56:37 +0000732// |AudioFrame| passed to |ProcessStream()|. The |vad_activity_| member will be
andrew@webrtc.orged083d42011-09-19 15:28:51 +0000733// modified to reflect the current decision.
niklase@google.com470e71d2011-07-07 08:21:25 +0000734class VoiceDetection {
735 public:
736 virtual int Enable(bool enable) = 0;
737 virtual bool is_enabled() const = 0;
738
739 // Returns true if voice is detected in the current frame. Should be called
740 // after |ProcessStream()|.
741 virtual bool stream_has_voice() const = 0;
742
743 // Some of the APM functionality requires a VAD decision. In the case that
744 // a decision is externally available for the current frame, it can be passed
745 // in here, before |ProcessStream()| is called.
746 //
747 // VoiceDetection does _not_ need to be enabled to use this. If it happens to
748 // be enabled, detection will be skipped for any frame in which an external
749 // VAD decision is provided.
750 virtual int set_stream_has_voice(bool has_voice) = 0;
751
752 // Specifies the likelihood that a frame will be declared to contain voice.
753 // A higher value makes it more likely that speech will not be clipped, at
754 // the expense of more noise being detected as voice.
755 enum Likelihood {
756 kVeryLowLikelihood,
757 kLowLikelihood,
758 kModerateLikelihood,
759 kHighLikelihood
760 };
761
762 virtual int set_likelihood(Likelihood likelihood) = 0;
763 virtual Likelihood likelihood() const = 0;
764
765 // Sets the |size| of the frames in ms on which the VAD will operate. Larger
766 // frames will improve detection accuracy, but reduce the frequency of
767 // updates.
768 //
769 // This does not impact the size of frames passed to |ProcessStream()|.
770 virtual int set_frame_size_ms(int size) = 0;
771 virtual int frame_size_ms() const = 0;
772
773 protected:
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000774 virtual ~VoiceDetection() {}
niklase@google.com470e71d2011-07-07 08:21:25 +0000775};
776} // namespace webrtc
777
andrew@webrtc.orgd72b3d62012-11-15 21:46:06 +0000778#endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_