1 /* 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. 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 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ 13 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> 18 #include <stddef.h> // size_t 19 #include <stdio.h> // FILE 20 #include <vector> 21 22 #include "webrtc/base/arraysize.h" 23 #include "webrtc/base/platform_file.h" 24 #include "webrtc/common.h" 25 #include "webrtc/modules/audio_processing/beamformer/array_util.h" 26 #include "webrtc/typedefs.h" 27 28 struct AecCore; 29 30 namespace webrtc { 31 32 class AudioFrame; 33 34 template<typename T> 35 class Beamformer; 36 37 class StreamConfig; 38 class ProcessingConfig; 39 40 class EchoCancellation; 41 class EchoControlMobile; 42 class GainControl; 43 class HighPassFilter; 44 class LevelEstimator; 45 class NoiseSuppression; 46 class VoiceDetection; 47 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. 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. 65 struct ExtendedFilter { ExtendedFilterExtendedFilter66 ExtendedFilter() : enabled(false) {} ExtendedFilterExtendedFilter67 explicit ExtendedFilter(bool enabled) : enabled(enabled) {} 68 static const ConfigOptionID identifier = ConfigOptionID::kExtendedFilter; 69 bool enabled; 70 }; 71 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(). 77 struct DelayAgnostic { DelayAgnosticDelayAgnostic78 DelayAgnostic() : enabled(false) {} DelayAgnosticDelayAgnostic79 explicit DelayAgnostic(bool enabled) : enabled(enabled) {} 80 static const ConfigOptionID identifier = ConfigOptionID::kDelayAgnostic; 81 bool enabled; 82 }; 83 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&). 90 #if defined(WEBRTC_CHROMIUM_BUILD) 91 static const int kAgcStartupMinVolume = 85; 92 #else 93 static const int kAgcStartupMinVolume = 0; 94 #endif // defined(WEBRTC_CHROMIUM_BUILD) 95 struct ExperimentalAgc { ExperimentalAgcExperimentalAgc96 ExperimentalAgc() : enabled(true), startup_min_volume(kAgcStartupMinVolume) {} ExperimentalAgcExperimentalAgc97 explicit ExperimentalAgc(bool enabled) 98 : enabled(enabled), startup_min_volume(kAgcStartupMinVolume) {} ExperimentalAgcExperimentalAgc99 ExperimentalAgc(bool enabled, int startup_min_volume) 100 : enabled(enabled), startup_min_volume(startup_min_volume) {} 101 static const ConfigOptionID identifier = ConfigOptionID::kExperimentalAgc; 102 bool enabled; 103 int startup_min_volume; 104 }; 105 106 // Use to enable experimental noise suppression. It can be set in the 107 // constructor or using AudioProcessing::SetExtraOptions(). 108 struct ExperimentalNs { ExperimentalNsExperimentalNs109 ExperimentalNs() : enabled(false) {} ExperimentalNsExperimentalNs110 explicit ExperimentalNs(bool enabled) : enabled(enabled) {} 111 static const ConfigOptionID identifier = ConfigOptionID::kExperimentalNs; 112 bool enabled; 113 }; 114 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 { BeamformingBeamforming118 Beamforming() 119 : enabled(false), 120 array_geometry(), 121 target_direction( 122 SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) {} BeamformingBeamforming123 Beamforming(bool enabled, const std::vector<Point>& array_geometry) 124 : Beamforming(enabled, 125 array_geometry, 126 SphericalPointf(static_cast<float>(M_PI) / 2.f, 0.f, 1.f)) { 127 } BeamformingBeamforming128 Beamforming(bool enabled, 129 const std::vector<Point>& array_geometry, 130 SphericalPointf target_direction) 131 : enabled(enabled), 132 array_geometry(array_geometry), 133 target_direction(target_direction) {} 134 static const ConfigOptionID identifier = ConfigOptionID::kBeamforming; 135 const bool enabled; 136 const std::vector<Point> array_geometry; 137 const SphericalPointf target_direction; 138 }; 139 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 { IntelligibilityIntelligibility147 Intelligibility() : enabled(false) {} IntelligibilityIntelligibility148 explicit Intelligibility(bool enabled) : enabled(enabled) {} 149 static const ConfigOptionID identifier = ConfigOptionID::kIntelligibility; 150 bool enabled; 151 }; 152 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 // 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. 185 // 186 // Usage example, omitting error checking: 187 // AudioProcessing* apm = AudioProcessing::Create(0); 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... 224 // delete apm; 225 // 226 class AudioProcessing { 227 public: 228 // TODO(mgraczyk): Remove once all methods that use ChannelLayout are gone. 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 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. 244 static AudioProcessing* Create(); 245 // Allows passing in an optional configuration at create-time. 246 static AudioProcessing* Create(const Config& config); 247 // Only for testing. 248 static AudioProcessing* Create(const Config& config, 249 Beamformer<float>* beamformer); ~AudioProcessing()250 virtual ~AudioProcessing() {} 251 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 255 // creation. 256 // 257 // It is also not necessary to call if the audio parameters (sample 258 // rate and number of channels) have changed. Passing updated parameters 259 // directly to |ProcessStream()| and |AnalyzeReverseStream()| is permissible. 260 // If the parameters are known at init-time though, they may be provided. 261 virtual int Initialize() = 0; 262 263 // The int16 interfaces require: 264 // - only |NativeRate|s be used 265 // - that the input, output and reverse rates must match 266 // - that |processing_config.output_stream()| matches 267 // |processing_config.input_stream()|. 268 // 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. 277 virtual int Initialize(int input_sample_rate_hz, 278 int output_sample_rate_hz, 279 int reverse_sample_rate_hz, 280 ChannelLayout input_layout, 281 ChannelLayout output_layout, 282 ChannelLayout reverse_layout) = 0; 283 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 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 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; 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; 300 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; 306 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 // 314 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_| 315 // members of |frame| must be valid. If changed from the previous call to this 316 // method, it will trigger an initialization. 317 virtual int ProcessStream(AudioFrame* frame) = 0; 318 319 // Accepts deinterleaved float audio with the range [-1, 1]. Each element 320 // of |src| points to a channel buffer, arranged according to 321 // |input_layout|. At output, the channels will be arranged according to 322 // |output_layout| at |output_sample_rate_hz| in |dest|. 323 // 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. 328 virtual int ProcessStream(const float* const* src, 329 size_t samples_per_channel, 330 int input_sample_rate_hz, 331 ChannelLayout input_layout, 332 int output_sample_rate_hz, 333 ChannelLayout output_layout, 334 float* const* dest) = 0; 335 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 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 // 358 // The |sample_rate_hz_|, |num_channels_|, and |samples_per_channel_| 359 // members of |frame| must be valid. |sample_rate_hz_| must correspond to 360 // |input_sample_rate_hz()| 361 // 362 // TODO(ajm): add const to input; requires an implementation fix. 363 // DEPRECATED: Use |ProcessReverseStream| instead. 364 // TODO(ekm): Remove once all users have updated to |ProcessReverseStream|. 365 virtual int AnalyzeReverseStream(AudioFrame* frame) = 0; 366 367 // Same as |AnalyzeReverseStream|, but may modify |frame| if intelligibility 368 // is enabled. 369 virtual int ProcessReverseStream(AudioFrame* frame) = 0; 370 371 // Accepts deinterleaved float audio with the range [-1, 1]. Each element 372 // of |data| points to a channel buffer, arranged according to |layout|. 373 // TODO(mgraczyk): Remove once clients are updated to use the new interface. 374 virtual int AnalyzeReverseStream(const float* const* data, 375 size_t samples_per_channel, 376 int rev_sample_rate_hz, 377 ChannelLayout layout) = 0; 378 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|. 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; 385 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; 401 virtual bool was_stream_delay_set() const = 0; 402 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; 406 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 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. 418 // An already existing file will be overwritten without warning. 419 static const size_t kMaxFilenameSize = 1024; 420 virtual int StartDebugRecording(const char filename[kMaxFilenameSize]) = 0; 421 422 // Same as above but uses an existing file handle. Takes ownership 423 // of |handle| and closes it at StopDebugRecording(). 424 virtual int StartDebugRecording(FILE* handle) = 0; 425 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. StartDebugRecordingForPlatformFile(rtc::PlatformFile)429 virtual int StartDebugRecordingForPlatformFile(rtc::PlatformFile /*handle*/) { 430 return -1; 431 } 432 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 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 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 459 enum Error { 460 // Fatal errors. 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, 473 kNotEnabledError = -12, 474 475 // Warnings are non-fatal. 476 // This results when a set_stream_ parameter is out of range. Processing 477 // will continue, but the parameter may have been truncated. 478 kBadStreamParameterWarning = -13 479 }; 480 481 enum NativeRate { 482 kSampleRate8kHz = 8000, 483 kSampleRate16kHz = 16000, 484 kSampleRate32kHz = 32000, 485 kSampleRate48kHz = 48000 486 }; 487 488 static const int kNativeSampleRatesHz[]; 489 static const size_t kNumNativeSampleRates; 490 static const int kMaxNativeSampleRateHz; 491 static const int kMaxAECMSampleRateHz; 492 493 static const int kChunkSizeMs = 10; 494 }; 495 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, 511 size_t num_channels = 0, 512 bool has_keyboard = false) sample_rate_hz_(sample_rate_hz)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 set_sample_rate_hz(int value)518 void set_sample_rate_hz(int value) { 519 sample_rate_hz_ = value; 520 num_frames_ = calculate_frames(value); 521 } set_num_channels(size_t value)522 void set_num_channels(size_t value) { num_channels_ = value; } set_has_keyboard(bool value)523 void set_has_keyboard(bool value) { has_keyboard_ = value; } 524 sample_rate_hz()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. num_channels()529 size_t num_channels() const { return num_channels_; } 530 has_keyboard()531 bool has_keyboard() const { return has_keyboard_; } num_frames()532 size_t num_frames() const { return num_frames_; } num_samples()533 size_t num_samples() const { return num_channels_ * num_frames_; } 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: calculate_frames(int sample_rate_hz)544 static size_t calculate_frames(int sample_rate_hz) { 545 return static_cast<size_t>( 546 AudioProcessing::kChunkSizeMs * sample_rate_hz / 1000); 547 } 548 549 int sample_rate_hz_; 550 size_t num_channels_; 551 bool has_keyboard_; 552 size_t num_frames_; 553 }; 554 555 class ProcessingConfig { 556 public: 557 enum StreamName { 558 kInputStream, 559 kOutputStream, 560 kReverseInputStream, 561 kReverseOutputStream, 562 kNumStreamNames, 563 }; 564 input_stream()565 const StreamConfig& input_stream() const { 566 return streams[StreamName::kInputStream]; 567 } output_stream()568 const StreamConfig& output_stream() const { 569 return streams[StreamName::kOutputStream]; 570 } reverse_input_stream()571 const StreamConfig& reverse_input_stream() const { 572 return streams[StreamName::kReverseInputStream]; 573 } reverse_output_stream()574 const StreamConfig& reverse_output_stream() const { 575 return streams[StreamName::kReverseOutputStream]; 576 } 577 input_stream()578 StreamConfig& input_stream() { return streams[StreamName::kInputStream]; } output_stream()579 StreamConfig& output_stream() { return streams[StreamName::kOutputStream]; } reverse_input_stream()580 StreamConfig& reverse_input_stream() { 581 return streams[StreamName::kReverseInputStream]; 582 } reverse_output_stream()583 StreamConfig& reverse_output_stream() { 584 return streams[StreamName::kReverseOutputStream]; 585 } 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 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 621 // set_stream_drift_samples() be called. 622 virtual int enable_drift_compensation(bool enable) = 0; 623 virtual bool is_drift_compensation_enabled() const = 0; 624 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 627 // if drift compensation is enabled, prior to |ProcessStream()|. 628 virtual void set_stream_drift_samples(int drift) = 0; 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 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 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. 688 virtual int GetDelayMetrics(int* median, int* std) = 0; 689 virtual int GetDelayMetrics(int* median, int* std, 690 float* fraction_poor_delays) = 0; 691 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 698 protected: ~EchoCancellation()699 virtual ~EchoCancellation() {} 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 734 // A typical use case is to initialize the component with an echo path from a 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 // 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 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(); 753 754 protected: ~EchoControlMobile()755 virtual ~EchoControlMobile() {} 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: ~GainControl()847 virtual ~GainControl() {} 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: ~HighPassFilter()858 virtual ~HighPassFilter() {} 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 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 // 872 // The computation follows: https://tools.ietf.org/html/rfc6465 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; 878 879 protected: ~LevelEstimator()880 virtual ~LevelEstimator() {} 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 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 909 protected: ~NoiseSuppression()910 virtual ~NoiseSuppression() {} 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. 916 // 917 // In addition to |stream_has_voice()| the VAD decision is provided through the 918 // |AudioFrame| passed to |ProcessStream()|. The |vad_activity_| member will be 919 // modified to reflect the current decision. 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: ~VoiceDetection()960 virtual ~VoiceDetection() {} 961 }; 962 } // namespace webrtc 963 964 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AUDIO_PROCESSING_H_ 965