niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 1 | /* |
andrew@webrtc.org | 648af74 | 2012-02-08 01:57:29 +0000 | [diff] [blame] | 2 | * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 3 | * |
| 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.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 11 | #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ |
| 12 | #define WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 13 | |
Alejandro Luebs | cb3f9bd | 2015-10-29 18:21:34 -0700 | [diff] [blame] | 14 | // 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.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 18 | #include <stddef.h> // size_t |
henrikg@webrtc.org | 863b536 | 2013-12-06 16:05:17 +0000 | [diff] [blame] | 19 | #include <stdio.h> // FILE |
aluebs@webrtc.org | fb7a039 | 2015-01-05 21:58:58 +0000 | [diff] [blame] | 20 | #include <vector> |
ajm@google.com | 22e6515 | 2011-07-18 18:03:01 +0000 | [diff] [blame] | 21 | |
Alejandro Luebs | cdfe20b | 2015-09-23 12:49:12 -0700 | [diff] [blame] | 22 | #include "webrtc/base/arraysize.h" |
xians@webrtc.org | e46bc77 | 2014-10-10 08:36:56 +0000 | [diff] [blame] | 23 | #include "webrtc/base/platform_file.h" |
andrew@webrtc.org | 61e596f | 2013-07-25 18:28:29 +0000 | [diff] [blame] | 24 | #include "webrtc/common.h" |
aluebs@webrtc.org | 1d88394 | 2015-03-05 20:38:21 +0000 | [diff] [blame] | 25 | #include "webrtc/modules/audio_processing/beamformer/array_util.h" |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 26 | #include "webrtc/typedefs.h" |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 27 | |
bjornv@webrtc.org | 91d11b3 | 2013-03-05 16:53:09 +0000 | [diff] [blame] | 28 | struct AecCore; |
| 29 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 30 | namespace webrtc { |
| 31 | |
| 32 | class AudioFrame; |
Michael Graczyk | dfa3605 | 2015-03-25 16:37:27 -0700 | [diff] [blame] | 33 | |
| 34 | template<typename T> |
| 35 | class Beamformer; |
| 36 | |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 37 | class StreamConfig; |
| 38 | class ProcessingConfig; |
| 39 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 40 | class EchoCancellation; |
| 41 | class EchoControlMobile; |
| 42 | class GainControl; |
| 43 | class HighPassFilter; |
| 44 | class LevelEstimator; |
| 45 | class NoiseSuppression; |
| 46 | class VoiceDetection; |
| 47 | |
Henrik Lundin | 441f634 | 2015-06-09 16:03:13 +0200 | [diff] [blame] | 48 | // 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.org | 6b1e219 | 2013-09-25 23:46:20 +0000 | [diff] [blame] | 52 | // |
| 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 Lundin | 441f634 | 2015-06-09 16:03:13 +0200 | [diff] [blame] | 65 | struct ExtendedFilter { |
| 66 | ExtendedFilter() : enabled(false) {} |
| 67 | explicit ExtendedFilter(bool enabled) : enabled(enabled) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 68 | static const ConfigOptionID identifier = ConfigOptionID::kExtendedFilter; |
Henrik Lundin | 441f634 | 2015-06-09 16:03:13 +0200 | [diff] [blame] | 69 | bool enabled; |
| 70 | }; |
andrew@webrtc.org | 6b1e219 | 2013-09-25 23:46:20 +0000 | [diff] [blame] | 71 | |
henrik.lundin | 366e952 | 2015-07-03 00:50:05 -0700 | [diff] [blame] | 72 | // Enables delay-agnostic echo cancellation. This feature relies on internally |
| 73 | // estimated delays between the process and reverse streams, thus not relying |
| 74 | // on reported system delays. This configuration only applies to |
| 75 | // EchoCancellation and not EchoControlMobile. It can be set in the constructor |
| 76 | // or using AudioProcessing::SetExtraOptions(). |
henrik.lundin | 0f133b9 | 2015-07-02 00:17:55 -0700 | [diff] [blame] | 77 | struct DelayAgnostic { |
| 78 | DelayAgnostic() : enabled(false) {} |
| 79 | explicit DelayAgnostic(bool enabled) : enabled(enabled) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 80 | static const ConfigOptionID identifier = ConfigOptionID::kDelayAgnostic; |
henrik.lundin | 0f133b9 | 2015-07-02 00:17:55 -0700 | [diff] [blame] | 81 | bool enabled; |
| 82 | }; |
bjornv@webrtc.org | 3f83072 | 2014-06-11 04:48:11 +0000 | [diff] [blame] | 83 | |
Bjorn Volcker | adc46c4 | 2015-04-15 11:42:40 +0200 | [diff] [blame] | 84 | // Use to enable experimental gain control (AGC). At startup the experimental |
| 85 | // AGC moves the microphone volume up to |startup_min_volume| if the current |
| 86 | // microphone volume is set too low. The value is clamped to its operating range |
| 87 | // [12, 255]. Here, 255 maps to 100%. |
| 88 | // |
| 89 | // Must be provided through AudioProcessing::Create(Confg&). |
Bjorn Volcker | fb49451 | 2015-04-22 06:39:58 +0200 | [diff] [blame] | 90 | #if defined(WEBRTC_CHROMIUM_BUILD) |
Bjorn Volcker | adc46c4 | 2015-04-15 11:42:40 +0200 | [diff] [blame] | 91 | static const int kAgcStartupMinVolume = 85; |
Bjorn Volcker | fb49451 | 2015-04-22 06:39:58 +0200 | [diff] [blame] | 92 | #else |
| 93 | static const int kAgcStartupMinVolume = 0; |
| 94 | #endif // defined(WEBRTC_CHROMIUM_BUILD) |
andrew@webrtc.org | c7c7a53 | 2014-01-29 04:57:25 +0000 | [diff] [blame] | 95 | struct ExperimentalAgc { |
Bjorn Volcker | adc46c4 | 2015-04-15 11:42:40 +0200 | [diff] [blame] | 96 | ExperimentalAgc() : enabled(true), startup_min_volume(kAgcStartupMinVolume) {} |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 97 | explicit ExperimentalAgc(bool enabled) |
Bjorn Volcker | adc46c4 | 2015-04-15 11:42:40 +0200 | [diff] [blame] | 98 | : enabled(enabled), startup_min_volume(kAgcStartupMinVolume) {} |
| 99 | ExperimentalAgc(bool enabled, int startup_min_volume) |
| 100 | : enabled(enabled), startup_min_volume(startup_min_volume) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 101 | static const ConfigOptionID identifier = ConfigOptionID::kExperimentalAgc; |
andrew@webrtc.org | 6b1e219 | 2013-09-25 23:46:20 +0000 | [diff] [blame] | 102 | bool enabled; |
Bjorn Volcker | adc46c4 | 2015-04-15 11:42:40 +0200 | [diff] [blame] | 103 | int startup_min_volume; |
andrew@webrtc.org | 6b1e219 | 2013-09-25 23:46:20 +0000 | [diff] [blame] | 104 | }; |
| 105 | |
aluebs@webrtc.org | 9825afc | 2014-06-30 17:39:53 +0000 | [diff] [blame] | 106 | // Use to enable experimental noise suppression. It can be set in the |
| 107 | // constructor or using AudioProcessing::SetExtraOptions(). |
| 108 | struct ExperimentalNs { |
| 109 | ExperimentalNs() : enabled(false) {} |
| 110 | explicit ExperimentalNs(bool enabled) : enabled(enabled) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 111 | static const ConfigOptionID identifier = ConfigOptionID::kExperimentalNs; |
aluebs@webrtc.org | 9825afc | 2014-06-30 17:39:53 +0000 | [diff] [blame] | 112 | bool enabled; |
| 113 | }; |
| 114 | |
aluebs@webrtc.org | ae643ce | 2014-12-19 19:57:34 +0000 | [diff] [blame] | 115 | // Use to enable beamforming. Must be provided through the constructor. It will |
| 116 | // have no impact if used with AudioProcessing::SetExtraOptions(). |
| 117 | struct Beamforming { |
eblima | 894ad94 | 2015-07-03 08:34:33 -0700 | [diff] [blame] | 118 | Beamforming() |
| 119 | : enabled(false), |
Alejandro Luebs | cb3f9bd | 2015-10-29 18:21:34 -0700 | [diff] [blame] | 120 | array_geometry(), |
| 121 | target_direction( |
| 122 | SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) {} |
aluebs@webrtc.org | fb7a039 | 2015-01-05 21:58:58 +0000 | [diff] [blame] | 123 | Beamforming(bool enabled, const std::vector<Point>& array_geometry) |
Alejandro Luebs | cb3f9bd | 2015-10-29 18:21:34 -0700 | [diff] [blame] | 124 | : Beamforming(enabled, |
| 125 | array_geometry, |
| 126 | SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) { |
| 127 | } |
| 128 | Beamforming(bool enabled, |
| 129 | const std::vector<Point>& array_geometry, |
| 130 | SphericalPointf target_direction) |
aluebs@webrtc.org | fb7a039 | 2015-01-05 21:58:58 +0000 | [diff] [blame] | 131 | : enabled(enabled), |
Alejandro Luebs | cb3f9bd | 2015-10-29 18:21:34 -0700 | [diff] [blame] | 132 | array_geometry(array_geometry), |
| 133 | target_direction(target_direction) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 134 | static const ConfigOptionID identifier = ConfigOptionID::kBeamforming; |
aluebs@webrtc.org | fb7a039 | 2015-01-05 21:58:58 +0000 | [diff] [blame] | 135 | const bool enabled; |
| 136 | const std::vector<Point> array_geometry; |
Alejandro Luebs | cb3f9bd | 2015-10-29 18:21:34 -0700 | [diff] [blame] | 137 | const SphericalPointf target_direction; |
aluebs@webrtc.org | ae643ce | 2014-12-19 19:57:34 +0000 | [diff] [blame] | 138 | }; |
| 139 | |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 140 | // Use to enable intelligibility enhancer in audio processing. Must be provided |
| 141 | // though the constructor. It will have no impact if used with |
| 142 | // AudioProcessing::SetExtraOptions(). |
| 143 | // |
| 144 | // Note: If enabled and the reverse stream has more than one output channel, |
| 145 | // the reverse stream will become an upmixed mono signal. |
| 146 | struct Intelligibility { |
| 147 | Intelligibility() : enabled(false) {} |
| 148 | explicit Intelligibility(bool enabled) : enabled(enabled) {} |
aluebs | 688e308 | 2016-01-14 04:32:46 -0800 | [diff] [blame^] | 149 | static const ConfigOptionID identifier = ConfigOptionID::kIntelligibility; |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 150 | bool enabled; |
| 151 | }; |
| 152 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 153 | // The Audio Processing Module (APM) provides a collection of voice processing |
| 154 | // components designed for real-time communications software. |
| 155 | // |
| 156 | // APM operates on two audio streams on a frame-by-frame basis. Frames of the |
| 157 | // primary stream, on which all processing is applied, are passed to |
| 158 | // |ProcessStream()|. Frames of the reverse direction stream, which are used for |
| 159 | // analysis by some components, are passed to |AnalyzeReverseStream()|. On the |
| 160 | // client-side, this will typically be the near-end (capture) and far-end |
| 161 | // (render) streams, respectively. APM should be placed in the signal chain as |
| 162 | // close to the audio hardware abstraction layer (HAL) as possible. |
| 163 | // |
| 164 | // On the server-side, the reverse stream will normally not be used, with |
| 165 | // processing occurring on each incoming stream. |
| 166 | // |
| 167 | // Component interfaces follow a similar pattern and are accessed through |
| 168 | // corresponding getters in APM. All components are disabled at create-time, |
| 169 | // with default settings that are recommended for most situations. New settings |
| 170 | // can be applied without enabling a component. Enabling a component triggers |
| 171 | // memory allocation and initialization to allow it to start processing the |
| 172 | // streams. |
| 173 | // |
| 174 | // Thread safety is provided with the following assumptions to reduce locking |
| 175 | // overhead: |
| 176 | // 1. The stream getters and setters are called from the same thread as |
| 177 | // ProcessStream(). More precisely, stream functions are never called |
| 178 | // concurrently with ProcessStream(). |
| 179 | // 2. Parameter getters are never called concurrently with the corresponding |
| 180 | // setter. |
| 181 | // |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 182 | // APM accepts only linear PCM audio data in chunks of 10 ms. The int16 |
| 183 | // interfaces use interleaved data, while the float interfaces use deinterleaved |
| 184 | // data. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 185 | // |
| 186 | // Usage example, omitting error checking: |
| 187 | // AudioProcessing* apm = AudioProcessing::Create(0); |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 188 | // |
| 189 | // apm->high_pass_filter()->Enable(true); |
| 190 | // |
| 191 | // apm->echo_cancellation()->enable_drift_compensation(false); |
| 192 | // apm->echo_cancellation()->Enable(true); |
| 193 | // |
| 194 | // apm->noise_reduction()->set_level(kHighSuppression); |
| 195 | // apm->noise_reduction()->Enable(true); |
| 196 | // |
| 197 | // apm->gain_control()->set_analog_level_limits(0, 255); |
| 198 | // apm->gain_control()->set_mode(kAdaptiveAnalog); |
| 199 | // apm->gain_control()->Enable(true); |
| 200 | // |
| 201 | // apm->voice_detection()->Enable(true); |
| 202 | // |
| 203 | // // Start a voice call... |
| 204 | // |
| 205 | // // ... Render frame arrives bound for the audio HAL ... |
| 206 | // apm->AnalyzeReverseStream(render_frame); |
| 207 | // |
| 208 | // // ... Capture frame arrives from the audio HAL ... |
| 209 | // // Call required set_stream_ functions. |
| 210 | // apm->set_stream_delay_ms(delay_ms); |
| 211 | // apm->gain_control()->set_stream_analog_level(analog_level); |
| 212 | // |
| 213 | // apm->ProcessStream(capture_frame); |
| 214 | // |
| 215 | // // Call required stream_ functions. |
| 216 | // analog_level = apm->gain_control()->stream_analog_level(); |
| 217 | // has_voice = apm->stream_has_voice(); |
| 218 | // |
| 219 | // // Repeate render and capture processing for the duration of the call... |
| 220 | // // Start a new call... |
| 221 | // apm->Initialize(); |
| 222 | // |
| 223 | // // Close the application... |
andrew@webrtc.org | f3930e9 | 2013-09-18 22:37:32 +0000 | [diff] [blame] | 224 | // delete apm; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 225 | // |
andrew@webrtc.org | f92aaff | 2014-02-15 04:22:49 +0000 | [diff] [blame] | 226 | class AudioProcessing { |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 227 | public: |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 228 | // TODO(mgraczyk): Remove once all methods that use ChannelLayout are gone. |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 229 | enum ChannelLayout { |
| 230 | kMono, |
| 231 | // Left, right. |
| 232 | kStereo, |
| 233 | // Mono, keyboard mic. |
| 234 | kMonoAndKeyboard, |
| 235 | // Left, right, keyboard mic. |
| 236 | kStereoAndKeyboard |
| 237 | }; |
| 238 | |
andrew@webrtc.org | 5474491 | 2014-02-05 06:30:29 +0000 | [diff] [blame] | 239 | // Creates an APM instance. Use one instance for every primary audio stream |
| 240 | // requiring processing. On the client-side, this would typically be one |
| 241 | // instance for the near-end stream, and additional instances for each far-end |
| 242 | // stream which requires processing. On the server-side, this would typically |
| 243 | // be one instance for every incoming stream. |
andrew@webrtc.org | e84978f | 2014-01-25 02:09:06 +0000 | [diff] [blame] | 244 | static AudioProcessing* Create(); |
andrew@webrtc.org | 5474491 | 2014-02-05 06:30:29 +0000 | [diff] [blame] | 245 | // Allows passing in an optional configuration at create-time. |
andrew@webrtc.org | e84978f | 2014-01-25 02:09:06 +0000 | [diff] [blame] | 246 | static AudioProcessing* Create(const Config& config); |
aluebs@webrtc.org | d82f55d | 2015-01-15 18:07:21 +0000 | [diff] [blame] | 247 | // Only for testing. |
mgraczyk@chromium.org | 0f663de | 2015-03-13 00:13:32 +0000 | [diff] [blame] | 248 | static AudioProcessing* Create(const Config& config, |
Michael Graczyk | dfa3605 | 2015-03-25 16:37:27 -0700 | [diff] [blame] | 249 | Beamformer<float>* beamformer); |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 250 | virtual ~AudioProcessing() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 251 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 252 | // Initializes internal states, while retaining all user settings. This |
| 253 | // should be called before beginning to process a new audio stream. However, |
| 254 | // it is not necessary to call before processing the first stream after |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 255 | // creation. |
| 256 | // |
| 257 | // It is also not necessary to call if the audio parameters (sample |
andrew@webrtc.org | 60730cf | 2014-01-07 17:45:09 +0000 | [diff] [blame] | 258 | // rate and number of channels) have changed. Passing updated parameters |
| 259 | // directly to |ProcessStream()| and |AnalyzeReverseStream()| is permissible. |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 260 | // If the parameters are known at init-time though, they may be provided. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 261 | virtual int Initialize() = 0; |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 262 | |
| 263 | // The int16 interfaces require: |
| 264 | // - only |NativeRate|s be used |
| 265 | // - that the input, output and reverse rates must match |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 266 | // - that |processing_config.output_stream()| matches |
| 267 | // |processing_config.input_stream()|. |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 268 | // |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 269 | // The float interfaces accept arbitrary rates and support differing input and |
| 270 | // output layouts, but the output must have either one channel or the same |
| 271 | // number of channels as the input. |
| 272 | virtual int Initialize(const ProcessingConfig& processing_config) = 0; |
| 273 | |
| 274 | // Initialize with unpacked parameters. See Initialize() above for details. |
| 275 | // |
| 276 | // TODO(mgraczyk): Remove once clients are updated to use the new interface. |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 277 | virtual int Initialize(int input_sample_rate_hz, |
| 278 | int output_sample_rate_hz, |
andrew@webrtc.org | a8b9737 | 2014-03-10 22:26:12 +0000 | [diff] [blame] | 279 | int reverse_sample_rate_hz, |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 280 | ChannelLayout input_layout, |
| 281 | ChannelLayout output_layout, |
| 282 | ChannelLayout reverse_layout) = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 283 | |
andrew@webrtc.org | 61e596f | 2013-07-25 18:28:29 +0000 | [diff] [blame] | 284 | // Pass down additional options which don't have explicit setters. This |
| 285 | // ensures the options are applied immediately. |
| 286 | virtual void SetExtraOptions(const Config& config) = 0; |
| 287 | |
peah | 66085be | 2015-12-16 02:02:20 -0800 | [diff] [blame] | 288 | // TODO(peah): Remove after voice engine no longer requires it to resample |
| 289 | // the reverse stream to the forward rate. |
| 290 | virtual int input_sample_rate_hz() const = 0; |
| 291 | |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 292 | // TODO(ajm): Only intended for internal use. Make private and friend the |
| 293 | // necessary classes? |
| 294 | virtual int proc_sample_rate_hz() const = 0; |
| 295 | virtual int proc_split_sample_rate_hz() const = 0; |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 296 | virtual size_t num_input_channels() const = 0; |
| 297 | virtual size_t num_proc_channels() const = 0; |
| 298 | virtual size_t num_output_channels() const = 0; |
| 299 | virtual size_t num_reverse_channels() const = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 300 | |
andrew@webrtc.org | 17342e5 | 2014-02-12 22:28:31 +0000 | [diff] [blame] | 301 | // Set to true when the output of AudioProcessing will be muted or in some |
| 302 | // other way not used. Ideally, the captured audio would still be processed, |
| 303 | // but some components may change behavior based on this information. |
| 304 | // Default false. |
| 305 | virtual void set_output_will_be_muted(bool muted) = 0; |
andrew@webrtc.org | 17342e5 | 2014-02-12 22:28:31 +0000 | [diff] [blame] | 306 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 307 | // Processes a 10 ms |frame| of the primary audio stream. On the client-side, |
| 308 | // this is the near-end (or captured) audio. |
| 309 | // |
| 310 | // If needed for enabled functionality, any function with the set_stream_ tag |
| 311 | // must be called prior to processing the current frame. Any getter function |
| 312 | // with the stream_ tag which is needed should be called after processing. |
| 313 | // |
andrew@webrtc.org | 63a5098 | 2012-05-02 23:56:37 +0000 | [diff] [blame] | 314 | // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_| |
andrew@webrtc.org | 60730cf | 2014-01-07 17:45:09 +0000 | [diff] [blame] | 315 | // members of |frame| must be valid. If changed from the previous call to this |
| 316 | // method, it will trigger an initialization. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 317 | virtual int ProcessStream(AudioFrame* frame) = 0; |
| 318 | |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 319 | // Accepts deinterleaved float audio with the range [-1, 1]. Each element |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 320 | // of |src| points to a channel buffer, arranged according to |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 321 | // |input_layout|. At output, the channels will be arranged according to |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 322 | // |output_layout| at |output_sample_rate_hz| in |dest|. |
| 323 | // |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 324 | // The output layout must have one channel or as many channels as the input. |
| 325 | // |src| and |dest| may use the same memory, if desired. |
| 326 | // |
| 327 | // TODO(mgraczyk): Remove once clients are updated to use the new interface. |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 328 | virtual int ProcessStream(const float* const* src, |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 329 | size_t samples_per_channel, |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 330 | int input_sample_rate_hz, |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 331 | ChannelLayout input_layout, |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 332 | int output_sample_rate_hz, |
| 333 | ChannelLayout output_layout, |
| 334 | float* const* dest) = 0; |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 335 | |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 336 | // Accepts deinterleaved float audio with the range [-1, 1]. Each element of |
| 337 | // |src| points to a channel buffer, arranged according to |input_stream|. At |
| 338 | // output, the channels will be arranged according to |output_stream| in |
| 339 | // |dest|. |
| 340 | // |
| 341 | // The output must have one channel or as many channels as the input. |src| |
| 342 | // and |dest| may use the same memory, if desired. |
| 343 | virtual int ProcessStream(const float* const* src, |
| 344 | const StreamConfig& input_config, |
| 345 | const StreamConfig& output_config, |
| 346 | float* const* dest) = 0; |
| 347 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 348 | // Analyzes a 10 ms |frame| of the reverse direction audio stream. The frame |
| 349 | // will not be modified. On the client-side, this is the far-end (or to be |
| 350 | // rendered) audio. |
| 351 | // |
| 352 | // It is only necessary to provide this if echo processing is enabled, as the |
| 353 | // reverse stream forms the echo reference signal. It is recommended, but not |
| 354 | // necessary, to provide if gain control is enabled. On the server-side this |
| 355 | // typically will not be used. If you're not sure what to pass in here, |
| 356 | // chances are you don't need to use it. |
| 357 | // |
andrew@webrtc.org | 63a5098 | 2012-05-02 23:56:37 +0000 | [diff] [blame] | 358 | // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_| |
andrew@webrtc.org | 60730cf | 2014-01-07 17:45:09 +0000 | [diff] [blame] | 359 | // members of |frame| must be valid. |sample_rate_hz_| must correspond to |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 360 | // |input_sample_rate_hz()| |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 361 | // |
| 362 | // TODO(ajm): add const to input; requires an implementation fix. |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 363 | // DEPRECATED: Use |ProcessReverseStream| instead. |
| 364 | // TODO(ekm): Remove once all users have updated to |ProcessReverseStream|. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 365 | virtual int AnalyzeReverseStream(AudioFrame* frame) = 0; |
| 366 | |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 367 | // Same as |AnalyzeReverseStream|, but may modify |frame| if intelligibility |
| 368 | // is enabled. |
| 369 | virtual int ProcessReverseStream(AudioFrame* frame) = 0; |
| 370 | |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 371 | // Accepts deinterleaved float audio with the range [-1, 1]. Each element |
| 372 | // of |data| points to a channel buffer, arranged according to |layout|. |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 373 | // TODO(mgraczyk): Remove once clients are updated to use the new interface. |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 374 | virtual int AnalyzeReverseStream(const float* const* data, |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 375 | size_t samples_per_channel, |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 376 | int rev_sample_rate_hz, |
andrew@webrtc.org | 17e4064 | 2014-03-04 20:58:13 +0000 | [diff] [blame] | 377 | ChannelLayout layout) = 0; |
| 378 | |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 379 | // Accepts deinterleaved float audio with the range [-1, 1]. Each element of |
| 380 | // |data| points to a channel buffer, arranged according to |reverse_config|. |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 381 | virtual int ProcessReverseStream(const float* const* src, |
| 382 | const StreamConfig& reverse_input_config, |
| 383 | const StreamConfig& reverse_output_config, |
| 384 | float* const* dest) = 0; |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 385 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 386 | // This must be called if and only if echo processing is enabled. |
| 387 | // |
| 388 | // Sets the |delay| in ms between AnalyzeReverseStream() receiving a far-end |
| 389 | // frame and ProcessStream() receiving a near-end frame containing the |
| 390 | // corresponding echo. On the client-side this can be expressed as |
| 391 | // delay = (t_render - t_analyze) + (t_process - t_capture) |
| 392 | // where, |
| 393 | // - t_analyze is the time a frame is passed to AnalyzeReverseStream() and |
| 394 | // t_render is the time the first sample of the same frame is rendered by |
| 395 | // the audio hardware. |
| 396 | // - t_capture is the time the first sample of a frame is captured by the |
| 397 | // audio hardware and t_pull is the time the same frame is passed to |
| 398 | // ProcessStream(). |
| 399 | virtual int set_stream_delay_ms(int delay) = 0; |
| 400 | virtual int stream_delay_ms() const = 0; |
andrew@webrtc.org | 56e4a05 | 2014-02-27 22:23:17 +0000 | [diff] [blame] | 401 | virtual bool was_stream_delay_set() const = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 402 | |
andrew@webrtc.org | 75dd288 | 2014-02-11 20:52:30 +0000 | [diff] [blame] | 403 | // Call to signal that a key press occurred (true) or did not occur (false) |
| 404 | // with this chunk of audio. |
| 405 | virtual void set_stream_key_pressed(bool key_pressed) = 0; |
andrew@webrtc.org | 75dd288 | 2014-02-11 20:52:30 +0000 | [diff] [blame] | 406 | |
andrew@webrtc.org | 6f9f817 | 2012-03-06 19:03:39 +0000 | [diff] [blame] | 407 | // Sets a delay |offset| in ms to add to the values passed in through |
| 408 | // set_stream_delay_ms(). May be positive or negative. |
| 409 | // |
| 410 | // Note that this could cause an otherwise valid value passed to |
| 411 | // set_stream_delay_ms() to return an error. |
| 412 | virtual void set_delay_offset_ms(int offset) = 0; |
| 413 | virtual int delay_offset_ms() const = 0; |
| 414 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 415 | // Starts recording debugging information to a file specified by |filename|, |
| 416 | // a NULL-terminated string. If there is an ongoing recording, the old file |
| 417 | // will be closed, and recording will continue in the newly specified file. |
ivoc | a4df27b | 2015-12-19 10:14:10 -0800 | [diff] [blame] | 418 | // An already existing file will be overwritten without warning. |
andrew@webrtc.org | 5ae19de | 2011-12-13 22:59:33 +0000 | [diff] [blame] | 419 | static const size_t kMaxFilenameSize = 1024; |
ivoc | a4df27b | 2015-12-19 10:14:10 -0800 | [diff] [blame] | 420 | virtual int StartDebugRecording(const char filename[kMaxFilenameSize]) = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 421 | |
henrikg@webrtc.org | 863b536 | 2013-12-06 16:05:17 +0000 | [diff] [blame] | 422 | // Same as above but uses an existing file handle. Takes ownership |
| 423 | // of |handle| and closes it at StopDebugRecording(). |
ivoc | a4df27b | 2015-12-19 10:14:10 -0800 | [diff] [blame] | 424 | virtual int StartDebugRecording(FILE* handle) = 0; |
henrikg@webrtc.org | 863b536 | 2013-12-06 16:05:17 +0000 | [diff] [blame] | 425 | |
xians@webrtc.org | e46bc77 | 2014-10-10 08:36:56 +0000 | [diff] [blame] | 426 | // Same as above but uses an existing PlatformFile handle. Takes ownership |
| 427 | // of |handle| and closes it at StopDebugRecording(). |
| 428 | // TODO(xians): Make this interface pure virtual. |
| 429 | virtual int StartDebugRecordingForPlatformFile(rtc::PlatformFile handle) { |
| 430 | return -1; |
| 431 | } |
| 432 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 433 | // Stops recording debugging information, and closes the file. Recording |
| 434 | // cannot be resumed in the same file (without overwriting it). |
| 435 | virtual int StopDebugRecording() = 0; |
| 436 | |
Bjorn Volcker | 4e7aa43 | 2015-07-07 11:50:05 +0200 | [diff] [blame] | 437 | // Use to send UMA histograms at end of a call. Note that all histogram |
| 438 | // specific member variables are reset. |
| 439 | virtual void UpdateHistogramsOnCallEnd() = 0; |
| 440 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 441 | // These provide access to the component interfaces and should never return |
| 442 | // NULL. The pointers will be valid for the lifetime of the APM instance. |
| 443 | // The memory for these objects is entirely managed internally. |
| 444 | virtual EchoCancellation* echo_cancellation() const = 0; |
| 445 | virtual EchoControlMobile* echo_control_mobile() const = 0; |
| 446 | virtual GainControl* gain_control() const = 0; |
| 447 | virtual HighPassFilter* high_pass_filter() const = 0; |
| 448 | virtual LevelEstimator* level_estimator() const = 0; |
| 449 | virtual NoiseSuppression* noise_suppression() const = 0; |
| 450 | virtual VoiceDetection* voice_detection() const = 0; |
| 451 | |
| 452 | struct Statistic { |
| 453 | int instant; // Instantaneous value. |
| 454 | int average; // Long-term average. |
| 455 | int maximum; // Long-term maximum. |
| 456 | int minimum; // Long-term minimum. |
| 457 | }; |
| 458 | |
andrew@webrtc.org | 648af74 | 2012-02-08 01:57:29 +0000 | [diff] [blame] | 459 | enum Error { |
| 460 | // Fatal errors. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 461 | kNoError = 0, |
| 462 | kUnspecifiedError = -1, |
| 463 | kCreationFailedError = -2, |
| 464 | kUnsupportedComponentError = -3, |
| 465 | kUnsupportedFunctionError = -4, |
| 466 | kNullPointerError = -5, |
| 467 | kBadParameterError = -6, |
| 468 | kBadSampleRateError = -7, |
| 469 | kBadDataLengthError = -8, |
| 470 | kBadNumberChannelsError = -9, |
| 471 | kFileError = -10, |
| 472 | kStreamParameterNotSetError = -11, |
andrew@webrtc.org | 648af74 | 2012-02-08 01:57:29 +0000 | [diff] [blame] | 473 | kNotEnabledError = -12, |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 474 | |
andrew@webrtc.org | 648af74 | 2012-02-08 01:57:29 +0000 | [diff] [blame] | 475 | // Warnings are non-fatal. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 476 | // This results when a set_stream_ parameter is out of range. Processing |
| 477 | // will continue, but the parameter may have been truncated. |
andrew@webrtc.org | 648af74 | 2012-02-08 01:57:29 +0000 | [diff] [blame] | 478 | kBadStreamParameterWarning = -13 |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 479 | }; |
andrew@webrtc.org | 56e4a05 | 2014-02-27 22:23:17 +0000 | [diff] [blame] | 480 | |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 481 | enum NativeRate { |
andrew@webrtc.org | 56e4a05 | 2014-02-27 22:23:17 +0000 | [diff] [blame] | 482 | kSampleRate8kHz = 8000, |
| 483 | kSampleRate16kHz = 16000, |
aluebs@webrtc.org | 087da13 | 2014-11-17 23:01:23 +0000 | [diff] [blame] | 484 | kSampleRate32kHz = 32000, |
| 485 | kSampleRate48kHz = 48000 |
andrew@webrtc.org | 56e4a05 | 2014-02-27 22:23:17 +0000 | [diff] [blame] | 486 | }; |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 487 | |
Alejandro Luebs | cdfe20b | 2015-09-23 12:49:12 -0700 | [diff] [blame] | 488 | static const int kNativeSampleRatesHz[]; |
| 489 | static const size_t kNumNativeSampleRates; |
| 490 | static const int kMaxNativeSampleRateHz; |
| 491 | static const int kMaxAECMSampleRateHz; |
| 492 | |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 493 | static const int kChunkSizeMs = 10; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 494 | }; |
| 495 | |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 496 | class StreamConfig { |
| 497 | public: |
| 498 | // sample_rate_hz: The sampling rate of the stream. |
| 499 | // |
| 500 | // num_channels: The number of audio channels in the stream, excluding the |
| 501 | // keyboard channel if it is present. When passing a |
| 502 | // StreamConfig with an array of arrays T*[N], |
| 503 | // |
| 504 | // N == {num_channels + 1 if has_keyboard |
| 505 | // {num_channels if !has_keyboard |
| 506 | // |
| 507 | // has_keyboard: True if the stream has a keyboard channel. When has_keyboard |
| 508 | // is true, the last channel in any corresponding list of |
| 509 | // channels is the keyboard channel. |
| 510 | StreamConfig(int sample_rate_hz = 0, |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 511 | size_t num_channels = 0, |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 512 | bool has_keyboard = false) |
| 513 | : sample_rate_hz_(sample_rate_hz), |
| 514 | num_channels_(num_channels), |
| 515 | has_keyboard_(has_keyboard), |
| 516 | num_frames_(calculate_frames(sample_rate_hz)) {} |
| 517 | |
| 518 | void set_sample_rate_hz(int value) { |
| 519 | sample_rate_hz_ = value; |
| 520 | num_frames_ = calculate_frames(value); |
| 521 | } |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 522 | void set_num_channels(size_t value) { num_channels_ = value; } |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 523 | void set_has_keyboard(bool value) { has_keyboard_ = value; } |
| 524 | |
| 525 | int sample_rate_hz() const { return sample_rate_hz_; } |
| 526 | |
| 527 | // The number of channels in the stream, not including the keyboard channel if |
| 528 | // present. |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 529 | size_t num_channels() const { return num_channels_; } |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 530 | |
| 531 | bool has_keyboard() const { return has_keyboard_; } |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 532 | size_t num_frames() const { return num_frames_; } |
| 533 | size_t num_samples() const { return num_channels_ * num_frames_; } |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 534 | |
| 535 | bool operator==(const StreamConfig& other) const { |
| 536 | return sample_rate_hz_ == other.sample_rate_hz_ && |
| 537 | num_channels_ == other.num_channels_ && |
| 538 | has_keyboard_ == other.has_keyboard_; |
| 539 | } |
| 540 | |
| 541 | bool operator!=(const StreamConfig& other) const { return !(*this == other); } |
| 542 | |
| 543 | private: |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 544 | static size_t calculate_frames(int sample_rate_hz) { |
| 545 | return static_cast<size_t>( |
| 546 | AudioProcessing::kChunkSizeMs * sample_rate_hz / 1000); |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 547 | } |
| 548 | |
| 549 | int sample_rate_hz_; |
Peter Kasting | 6955870 | 2016-01-12 16:26:35 -0800 | [diff] [blame] | 550 | size_t num_channels_; |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 551 | bool has_keyboard_; |
Peter Kasting | dce40cf | 2015-08-24 14:52:23 -0700 | [diff] [blame] | 552 | size_t num_frames_; |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 553 | }; |
| 554 | |
| 555 | class ProcessingConfig { |
| 556 | public: |
| 557 | enum StreamName { |
| 558 | kInputStream, |
| 559 | kOutputStream, |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 560 | kReverseInputStream, |
| 561 | kReverseOutputStream, |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 562 | kNumStreamNames, |
| 563 | }; |
| 564 | |
| 565 | const StreamConfig& input_stream() const { |
| 566 | return streams[StreamName::kInputStream]; |
| 567 | } |
| 568 | const StreamConfig& output_stream() const { |
| 569 | return streams[StreamName::kOutputStream]; |
| 570 | } |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 571 | const StreamConfig& reverse_input_stream() const { |
| 572 | return streams[StreamName::kReverseInputStream]; |
| 573 | } |
| 574 | const StreamConfig& reverse_output_stream() const { |
| 575 | return streams[StreamName::kReverseOutputStream]; |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 576 | } |
| 577 | |
| 578 | StreamConfig& input_stream() { return streams[StreamName::kInputStream]; } |
| 579 | StreamConfig& output_stream() { return streams[StreamName::kOutputStream]; } |
ekmeyerson | 60d9b33 | 2015-08-14 10:35:55 -0700 | [diff] [blame] | 580 | StreamConfig& reverse_input_stream() { |
| 581 | return streams[StreamName::kReverseInputStream]; |
| 582 | } |
| 583 | StreamConfig& reverse_output_stream() { |
| 584 | return streams[StreamName::kReverseOutputStream]; |
| 585 | } |
Michael Graczyk | 86c6d33 | 2015-07-23 11:41:39 -0700 | [diff] [blame] | 586 | |
| 587 | bool operator==(const ProcessingConfig& other) const { |
| 588 | for (int i = 0; i < StreamName::kNumStreamNames; ++i) { |
| 589 | if (this->streams[i] != other.streams[i]) { |
| 590 | return false; |
| 591 | } |
| 592 | } |
| 593 | return true; |
| 594 | } |
| 595 | |
| 596 | bool operator!=(const ProcessingConfig& other) const { |
| 597 | return !(*this == other); |
| 598 | } |
| 599 | |
| 600 | StreamConfig streams[StreamName::kNumStreamNames]; |
| 601 | }; |
| 602 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 603 | // The acoustic echo cancellation (AEC) component provides better performance |
| 604 | // than AECM but also requires more processing power and is dependent on delay |
| 605 | // stability and reporting accuracy. As such it is well-suited and recommended |
| 606 | // for PC and IP phone applications. |
| 607 | // |
| 608 | // Not recommended to be enabled on the server-side. |
| 609 | class EchoCancellation { |
| 610 | public: |
| 611 | // EchoCancellation and EchoControlMobile may not be enabled simultaneously. |
| 612 | // Enabling one will disable the other. |
| 613 | virtual int Enable(bool enable) = 0; |
| 614 | virtual bool is_enabled() const = 0; |
| 615 | |
| 616 | // Differences in clock speed on the primary and reverse streams can impact |
| 617 | // the AEC performance. On the client-side, this could be seen when different |
| 618 | // render and capture devices are used, particularly with webcams. |
| 619 | // |
| 620 | // This enables a compensation mechanism, and requires that |
andrew@webrtc.org | ddbb8a2 | 2014-04-22 21:00:04 +0000 | [diff] [blame] | 621 | // set_stream_drift_samples() be called. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 622 | virtual int enable_drift_compensation(bool enable) = 0; |
| 623 | virtual bool is_drift_compensation_enabled() const = 0; |
| 624 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 625 | // Sets the difference between the number of samples rendered and captured by |
| 626 | // the audio devices since the last call to |ProcessStream()|. Must be called |
andrew@webrtc.org | 6be1e93 | 2013-03-01 18:47:28 +0000 | [diff] [blame] | 627 | // if drift compensation is enabled, prior to |ProcessStream()|. |
| 628 | virtual void set_stream_drift_samples(int drift) = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 629 | virtual int stream_drift_samples() const = 0; |
| 630 | |
| 631 | enum SuppressionLevel { |
| 632 | kLowSuppression, |
| 633 | kModerateSuppression, |
| 634 | kHighSuppression |
| 635 | }; |
| 636 | |
| 637 | // Sets the aggressiveness of the suppressor. A higher level trades off |
| 638 | // double-talk performance for increased echo suppression. |
| 639 | virtual int set_suppression_level(SuppressionLevel level) = 0; |
| 640 | virtual SuppressionLevel suppression_level() const = 0; |
| 641 | |
| 642 | // Returns false if the current frame almost certainly contains no echo |
| 643 | // and true if it _might_ contain echo. |
| 644 | virtual bool stream_has_echo() const = 0; |
| 645 | |
| 646 | // Enables the computation of various echo metrics. These are obtained |
| 647 | // through |GetMetrics()|. |
| 648 | virtual int enable_metrics(bool enable) = 0; |
| 649 | virtual bool are_metrics_enabled() const = 0; |
| 650 | |
| 651 | // Each statistic is reported in dB. |
| 652 | // P_far: Far-end (render) signal power. |
| 653 | // P_echo: Near-end (capture) echo signal power. |
| 654 | // P_out: Signal power at the output of the AEC. |
| 655 | // P_a: Internal signal power at the point before the AEC's non-linear |
| 656 | // processor. |
| 657 | struct Metrics { |
| 658 | // RERL = ERL + ERLE |
| 659 | AudioProcessing::Statistic residual_echo_return_loss; |
| 660 | |
| 661 | // ERL = 10log_10(P_far / P_echo) |
| 662 | AudioProcessing::Statistic echo_return_loss; |
| 663 | |
| 664 | // ERLE = 10log_10(P_echo / P_out) |
| 665 | AudioProcessing::Statistic echo_return_loss_enhancement; |
| 666 | |
| 667 | // (Pre non-linear processing suppression) A_NLP = 10log_10(P_echo / P_a) |
| 668 | AudioProcessing::Statistic a_nlp; |
| 669 | }; |
| 670 | |
| 671 | // TODO(ajm): discuss the metrics update period. |
| 672 | virtual int GetMetrics(Metrics* metrics) = 0; |
| 673 | |
bjornv@google.com | 1ba3dbe | 2011-10-03 08:18:10 +0000 | [diff] [blame] | 674 | // Enables computation and logging of delay values. Statistics are obtained |
| 675 | // through |GetDelayMetrics()|. |
| 676 | virtual int enable_delay_logging(bool enable) = 0; |
| 677 | virtual bool is_delay_logging_enabled() const = 0; |
| 678 | |
| 679 | // The delay metrics consists of the delay |median| and the delay standard |
bjornv@webrtc.org | b1786db | 2015-02-03 06:06:26 +0000 | [diff] [blame] | 680 | // deviation |std|. It also consists of the fraction of delay estimates |
| 681 | // |fraction_poor_delays| that can make the echo cancellation perform poorly. |
| 682 | // The values are aggregated until the first call to |GetDelayMetrics()| and |
| 683 | // afterwards aggregated and updated every second. |
| 684 | // Note that if there are several clients pulling metrics from |
| 685 | // |GetDelayMetrics()| during a session the first call from any of them will |
| 686 | // change to one second aggregation window for all. |
| 687 | // TODO(bjornv): Deprecated, remove. |
bjornv@google.com | 1ba3dbe | 2011-10-03 08:18:10 +0000 | [diff] [blame] | 688 | virtual int GetDelayMetrics(int* median, int* std) = 0; |
bjornv@webrtc.org | b1786db | 2015-02-03 06:06:26 +0000 | [diff] [blame] | 689 | virtual int GetDelayMetrics(int* median, int* std, |
| 690 | float* fraction_poor_delays) = 0; |
bjornv@google.com | 1ba3dbe | 2011-10-03 08:18:10 +0000 | [diff] [blame] | 691 | |
bjornv@webrtc.org | 91d11b3 | 2013-03-05 16:53:09 +0000 | [diff] [blame] | 692 | // Returns a pointer to the low level AEC component. In case of multiple |
| 693 | // channels, the pointer to the first one is returned. A NULL pointer is |
| 694 | // returned when the AEC component is disabled or has not been initialized |
| 695 | // successfully. |
| 696 | virtual struct AecCore* aec_core() const = 0; |
| 697 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 698 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 699 | virtual ~EchoCancellation() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 700 | }; |
| 701 | |
| 702 | // The acoustic echo control for mobile (AECM) component is a low complexity |
| 703 | // robust option intended for use on mobile devices. |
| 704 | // |
| 705 | // Not recommended to be enabled on the server-side. |
| 706 | class EchoControlMobile { |
| 707 | public: |
| 708 | // EchoCancellation and EchoControlMobile may not be enabled simultaneously. |
| 709 | // Enabling one will disable the other. |
| 710 | virtual int Enable(bool enable) = 0; |
| 711 | virtual bool is_enabled() const = 0; |
| 712 | |
| 713 | // Recommended settings for particular audio routes. In general, the louder |
| 714 | // the echo is expected to be, the higher this value should be set. The |
| 715 | // preferred setting may vary from device to device. |
| 716 | enum RoutingMode { |
| 717 | kQuietEarpieceOrHeadset, |
| 718 | kEarpiece, |
| 719 | kLoudEarpiece, |
| 720 | kSpeakerphone, |
| 721 | kLoudSpeakerphone |
| 722 | }; |
| 723 | |
| 724 | // Sets echo control appropriate for the audio routing |mode| on the device. |
| 725 | // It can and should be updated during a call if the audio routing changes. |
| 726 | virtual int set_routing_mode(RoutingMode mode) = 0; |
| 727 | virtual RoutingMode routing_mode() const = 0; |
| 728 | |
| 729 | // Comfort noise replaces suppressed background noise to maintain a |
| 730 | // consistent signal level. |
| 731 | virtual int enable_comfort_noise(bool enable) = 0; |
| 732 | virtual bool is_comfort_noise_enabled() const = 0; |
| 733 | |
bjornv@google.com | c4b939c | 2011-07-13 08:09:56 +0000 | [diff] [blame] | 734 | // A typical use case is to initialize the component with an echo path from a |
ajm@google.com | 22e6515 | 2011-07-18 18:03:01 +0000 | [diff] [blame] | 735 | // previous call. The echo path is retrieved using |GetEchoPath()|, typically |
| 736 | // at the end of a call. The data can then be stored for later use as an |
| 737 | // initializer before the next call, using |SetEchoPath()|. |
| 738 | // |
bjornv@google.com | c4b939c | 2011-07-13 08:09:56 +0000 | [diff] [blame] | 739 | // Controlling the echo path this way requires the data |size_bytes| to match |
| 740 | // the internal echo path size. This size can be acquired using |
| 741 | // |echo_path_size_bytes()|. |SetEchoPath()| causes an entire reset, worth |
ajm@google.com | 22e6515 | 2011-07-18 18:03:01 +0000 | [diff] [blame] | 742 | // noting if it is to be called during an ongoing call. |
| 743 | // |
| 744 | // It is possible that version incompatibilities may result in a stored echo |
| 745 | // path of the incorrect size. In this case, the stored path should be |
| 746 | // discarded. |
| 747 | virtual int SetEchoPath(const void* echo_path, size_t size_bytes) = 0; |
| 748 | virtual int GetEchoPath(void* echo_path, size_t size_bytes) const = 0; |
| 749 | |
| 750 | // The returned path size is guaranteed not to change for the lifetime of |
| 751 | // the application. |
| 752 | static size_t echo_path_size_bytes(); |
bjornv@google.com | c4b939c | 2011-07-13 08:09:56 +0000 | [diff] [blame] | 753 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 754 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 755 | virtual ~EchoControlMobile() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 756 | }; |
| 757 | |
| 758 | // The automatic gain control (AGC) component brings the signal to an |
| 759 | // appropriate range. This is done by applying a digital gain directly and, in |
| 760 | // the analog mode, prescribing an analog gain to be applied at the audio HAL. |
| 761 | // |
| 762 | // Recommended to be enabled on the client-side. |
| 763 | class GainControl { |
| 764 | public: |
| 765 | virtual int Enable(bool enable) = 0; |
| 766 | virtual bool is_enabled() const = 0; |
| 767 | |
| 768 | // When an analog mode is set, this must be called prior to |ProcessStream()| |
| 769 | // to pass the current analog level from the audio HAL. Must be within the |
| 770 | // range provided to |set_analog_level_limits()|. |
| 771 | virtual int set_stream_analog_level(int level) = 0; |
| 772 | |
| 773 | // When an analog mode is set, this should be called after |ProcessStream()| |
| 774 | // to obtain the recommended new analog level for the audio HAL. It is the |
| 775 | // users responsibility to apply this level. |
| 776 | virtual int stream_analog_level() = 0; |
| 777 | |
| 778 | enum Mode { |
| 779 | // Adaptive mode intended for use if an analog volume control is available |
| 780 | // on the capture device. It will require the user to provide coupling |
| 781 | // between the OS mixer controls and AGC through the |stream_analog_level()| |
| 782 | // functions. |
| 783 | // |
| 784 | // It consists of an analog gain prescription for the audio device and a |
| 785 | // digital compression stage. |
| 786 | kAdaptiveAnalog, |
| 787 | |
| 788 | // Adaptive mode intended for situations in which an analog volume control |
| 789 | // is unavailable. It operates in a similar fashion to the adaptive analog |
| 790 | // mode, but with scaling instead applied in the digital domain. As with |
| 791 | // the analog mode, it additionally uses a digital compression stage. |
| 792 | kAdaptiveDigital, |
| 793 | |
| 794 | // Fixed mode which enables only the digital compression stage also used by |
| 795 | // the two adaptive modes. |
| 796 | // |
| 797 | // It is distinguished from the adaptive modes by considering only a |
| 798 | // short time-window of the input signal. It applies a fixed gain through |
| 799 | // most of the input level range, and compresses (gradually reduces gain |
| 800 | // with increasing level) the input signal at higher levels. This mode is |
| 801 | // preferred on embedded devices where the capture signal level is |
| 802 | // predictable, so that a known gain can be applied. |
| 803 | kFixedDigital |
| 804 | }; |
| 805 | |
| 806 | virtual int set_mode(Mode mode) = 0; |
| 807 | virtual Mode mode() const = 0; |
| 808 | |
| 809 | // Sets the target peak |level| (or envelope) of the AGC in dBFs (decibels |
| 810 | // from digital full-scale). The convention is to use positive values. For |
| 811 | // instance, passing in a value of 3 corresponds to -3 dBFs, or a target |
| 812 | // level 3 dB below full-scale. Limited to [0, 31]. |
| 813 | // |
| 814 | // TODO(ajm): use a negative value here instead, if/when VoE will similarly |
| 815 | // update its interface. |
| 816 | virtual int set_target_level_dbfs(int level) = 0; |
| 817 | virtual int target_level_dbfs() const = 0; |
| 818 | |
| 819 | // Sets the maximum |gain| the digital compression stage may apply, in dB. A |
| 820 | // higher number corresponds to greater compression, while a value of 0 will |
| 821 | // leave the signal uncompressed. Limited to [0, 90]. |
| 822 | virtual int set_compression_gain_db(int gain) = 0; |
| 823 | virtual int compression_gain_db() const = 0; |
| 824 | |
| 825 | // When enabled, the compression stage will hard limit the signal to the |
| 826 | // target level. Otherwise, the signal will be compressed but not limited |
| 827 | // above the target level. |
| 828 | virtual int enable_limiter(bool enable) = 0; |
| 829 | virtual bool is_limiter_enabled() const = 0; |
| 830 | |
| 831 | // Sets the |minimum| and |maximum| analog levels of the audio capture device. |
| 832 | // Must be set if and only if an analog mode is used. Limited to [0, 65535]. |
| 833 | virtual int set_analog_level_limits(int minimum, |
| 834 | int maximum) = 0; |
| 835 | virtual int analog_level_minimum() const = 0; |
| 836 | virtual int analog_level_maximum() const = 0; |
| 837 | |
| 838 | // Returns true if the AGC has detected a saturation event (period where the |
| 839 | // signal reaches digital full-scale) in the current frame and the analog |
| 840 | // level cannot be reduced. |
| 841 | // |
| 842 | // This could be used as an indicator to reduce or disable analog mic gain at |
| 843 | // the audio HAL. |
| 844 | virtual bool stream_is_saturated() const = 0; |
| 845 | |
| 846 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 847 | virtual ~GainControl() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 848 | }; |
| 849 | |
| 850 | // A filtering component which removes DC offset and low-frequency noise. |
| 851 | // Recommended to be enabled on the client-side. |
| 852 | class HighPassFilter { |
| 853 | public: |
| 854 | virtual int Enable(bool enable) = 0; |
| 855 | virtual bool is_enabled() const = 0; |
| 856 | |
| 857 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 858 | virtual ~HighPassFilter() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 859 | }; |
| 860 | |
| 861 | // An estimation component used to retrieve level metrics. |
| 862 | class LevelEstimator { |
| 863 | public: |
| 864 | virtual int Enable(bool enable) = 0; |
| 865 | virtual bool is_enabled() const = 0; |
| 866 | |
andrew@webrtc.org | 755b04a | 2011-11-15 16:57:56 +0000 | [diff] [blame] | 867 | // Returns the root mean square (RMS) level in dBFs (decibels from digital |
| 868 | // full-scale), or alternately dBov. It is computed over all primary stream |
| 869 | // frames since the last call to RMS(). The returned value is positive but |
| 870 | // should be interpreted as negative. It is constrained to [0, 127]. |
| 871 | // |
andrew@webrtc.org | 382c0c2 | 2014-05-05 18:22:21 +0000 | [diff] [blame] | 872 | // The computation follows: https://tools.ietf.org/html/rfc6465 |
andrew@webrtc.org | 755b04a | 2011-11-15 16:57:56 +0000 | [diff] [blame] | 873 | // with the intent that it can provide the RTP audio level indication. |
| 874 | // |
| 875 | // Frames passed to ProcessStream() with an |_energy| of zero are considered |
| 876 | // to have been muted. The RMS of the frame will be interpreted as -127. |
| 877 | virtual int RMS() = 0; |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 878 | |
| 879 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 880 | virtual ~LevelEstimator() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 881 | }; |
| 882 | |
| 883 | // The noise suppression (NS) component attempts to remove noise while |
| 884 | // retaining speech. Recommended to be enabled on the client-side. |
| 885 | // |
| 886 | // Recommended to be enabled on the client-side. |
| 887 | class NoiseSuppression { |
| 888 | public: |
| 889 | virtual int Enable(bool enable) = 0; |
| 890 | virtual bool is_enabled() const = 0; |
| 891 | |
| 892 | // Determines the aggressiveness of the suppression. Increasing the level |
| 893 | // will reduce the noise level at the expense of a higher speech distortion. |
| 894 | enum Level { |
| 895 | kLow, |
| 896 | kModerate, |
| 897 | kHigh, |
| 898 | kVeryHigh |
| 899 | }; |
| 900 | |
| 901 | virtual int set_level(Level level) = 0; |
| 902 | virtual Level level() const = 0; |
| 903 | |
bjornv@webrtc.org | 08329f4 | 2012-07-12 21:00:43 +0000 | [diff] [blame] | 904 | // Returns the internally computed prior speech probability of current frame |
| 905 | // averaged over output channels. This is not supported in fixed point, for |
| 906 | // which |kUnsupportedFunctionError| is returned. |
| 907 | virtual float speech_probability() const = 0; |
| 908 | |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 909 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 910 | virtual ~NoiseSuppression() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 911 | }; |
| 912 | |
| 913 | // The voice activity detection (VAD) component analyzes the stream to |
| 914 | // determine if voice is present. A facility is also provided to pass in an |
| 915 | // external VAD decision. |
andrew@webrtc.org | ed083d4 | 2011-09-19 15:28:51 +0000 | [diff] [blame] | 916 | // |
| 917 | // In addition to |stream_has_voice()| the VAD decision is provided through the |
andrew@webrtc.org | 63a5098 | 2012-05-02 23:56:37 +0000 | [diff] [blame] | 918 | // |AudioFrame| passed to |ProcessStream()|. The |vad_activity_| member will be |
andrew@webrtc.org | ed083d4 | 2011-09-19 15:28:51 +0000 | [diff] [blame] | 919 | // modified to reflect the current decision. |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 920 | class VoiceDetection { |
| 921 | public: |
| 922 | virtual int Enable(bool enable) = 0; |
| 923 | virtual bool is_enabled() const = 0; |
| 924 | |
| 925 | // Returns true if voice is detected in the current frame. Should be called |
| 926 | // after |ProcessStream()|. |
| 927 | virtual bool stream_has_voice() const = 0; |
| 928 | |
| 929 | // Some of the APM functionality requires a VAD decision. In the case that |
| 930 | // a decision is externally available for the current frame, it can be passed |
| 931 | // in here, before |ProcessStream()| is called. |
| 932 | // |
| 933 | // VoiceDetection does _not_ need to be enabled to use this. If it happens to |
| 934 | // be enabled, detection will be skipped for any frame in which an external |
| 935 | // VAD decision is provided. |
| 936 | virtual int set_stream_has_voice(bool has_voice) = 0; |
| 937 | |
| 938 | // Specifies the likelihood that a frame will be declared to contain voice. |
| 939 | // A higher value makes it more likely that speech will not be clipped, at |
| 940 | // the expense of more noise being detected as voice. |
| 941 | enum Likelihood { |
| 942 | kVeryLowLikelihood, |
| 943 | kLowLikelihood, |
| 944 | kModerateLikelihood, |
| 945 | kHighLikelihood |
| 946 | }; |
| 947 | |
| 948 | virtual int set_likelihood(Likelihood likelihood) = 0; |
| 949 | virtual Likelihood likelihood() const = 0; |
| 950 | |
| 951 | // Sets the |size| of the frames in ms on which the VAD will operate. Larger |
| 952 | // frames will improve detection accuracy, but reduce the frequency of |
| 953 | // updates. |
| 954 | // |
| 955 | // This does not impact the size of frames passed to |ProcessStream()|. |
| 956 | virtual int set_frame_size_ms(int size) = 0; |
| 957 | virtual int frame_size_ms() const = 0; |
| 958 | |
| 959 | protected: |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 960 | virtual ~VoiceDetection() {} |
niklase@google.com | 470e71d | 2011-07-07 08:21:25 +0000 | [diff] [blame] | 961 | }; |
| 962 | } // namespace webrtc |
| 963 | |
andrew@webrtc.org | d72b3d6 | 2012-11-15 21:46:06 +0000 | [diff] [blame] | 964 | #endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ |