| Index: webrtc/modules/audio_processing/include/aec_dump.h
|
| diff --git a/webrtc/modules/audio_processing/include/aec_dump.h b/webrtc/modules/audio_processing/include/aec_dump.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..eba2fbb043a6eb1fdeaeca8781d87249770de91e
|
| --- /dev/null
|
| +++ b/webrtc/modules/audio_processing/include/aec_dump.h
|
| @@ -0,0 +1,145 @@
|
| +/*
|
| + * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
|
| + *
|
| + * Use of this source code is governed by a BSD-style license
|
| + * that can be found in the LICENSE file in the root of the source
|
| + * tree. An additional intellectual property rights grant can be found
|
| + * in the file PATENTS. All contributing project authors may
|
| + * be found in the AUTHORS file in the root of the source tree.
|
| + */
|
| +
|
| +#ifndef WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AEC_DUMP_H_
|
| +#define WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AEC_DUMP_H_
|
| +
|
| +#include <memory>
|
| +#include <string>
|
| +#include <vector>
|
| +
|
| +#include "webrtc/base/array_view.h"
|
| +#include "webrtc/modules/audio_processing/include/audio_processing.h"
|
| +
|
| +namespace audioproc {
|
| +class Event;
|
| +} // namespace audioproc
|
| +
|
| +namespace webrtc {
|
| +
|
| +class AudioFrame;
|
| +
|
| +// Struct for passing current config from APM without having to
|
| +// include protobuf headers.
|
| +struct InternalAPMConfig {
|
| + InternalAPMConfig();
|
| +
|
| + InternalAPMConfig(const InternalAPMConfig&) = delete;
|
| + InternalAPMConfig(const InternalAPMConfig&&) = delete;
|
| + InternalAPMConfig& operator=(const InternalAPMConfig&) = delete;
|
| + InternalAPMConfig& operator=(const InternalAPMConfig&&) = delete;
|
| +
|
| + bool aec_enabled = false;
|
| + bool aec_delay_agnostic_enabled = false;
|
| + bool aec_drift_compensation_enabled = false;
|
| + bool aec_extended_filter_enabled = false;
|
| + int aec_suppression_level = 0;
|
| + bool aecm_enabled = false;
|
| + bool aecm_comfort_noise_enabled = false;
|
| + int aecm_routing_mode = 0;
|
| + bool agc_enabled = false;
|
| + int agc_mode = 0;
|
| + bool agc_limiter_enabled = false;
|
| + bool hpf_enabled = false;
|
| + bool ns_enabled = false;
|
| + int ns_level = 0;
|
| + bool transient_suppression_enabled = false;
|
| + bool intelligibility_enhancer_enabled = false;
|
| + bool noise_robust_agc_enabled = false;
|
| + std::string experiments_description = "";
|
| +};
|
| +
|
| +class AecDump {
|
| + public:
|
| + // A capture stream frame is logged before and after processing in
|
| + // the same protobuf message. To facilitate that, a CaptureStreamInfo
|
| + // instance is first filled with Input, then Output.
|
| + //
|
| + // To log an input/output pair, first call
|
| + // AecDump::GetCaptureStreamInfo. Add the input and output to
|
| + // it. Then call AecDump::WriteCaptureStreamMessage.
|
| + class CaptureStreamInfo {
|
| + public:
|
| + virtual ~CaptureStreamInfo() = default;
|
| + virtual void AddInput(
|
| + const std::vector<rtc::ArrayView<const float>>& src) = 0;
|
| + virtual void AddOutput(
|
| + const std::vector<rtc::ArrayView<const float>>& src) = 0;
|
| +
|
| + virtual void AddInput(const AudioFrame& frame) = 0;
|
| + virtual void AddOutput(const AudioFrame& frame) = 0;
|
| +
|
| + virtual void set_delay(int delay) = 0;
|
| + virtual void set_drift(int drift) = 0;
|
| + virtual void set_level(int level) = 0;
|
| + virtual void set_keypress(bool keypress) = 0;
|
| + };
|
| +
|
| + virtual ~AecDump() = default;
|
| +
|
| + static std::unique_ptr<AecDump> CreateNullDump();
|
| +
|
| + virtual std::unique_ptr<CaptureStreamInfo> GetCaptureStreamInfo() = 0;
|
| +
|
| + // The Write* methods are always safe to call. If no recording is in
|
| + // progress, calls will have no effect. Messages are written to file
|
| + // in a 'best effort' manner. If the AecDump can't keep up with
|
| + // the flow of messages, some will be silently dropped.
|
| + virtual void WriteInitMessage(const ProcessingConfig& api_format) = 0;
|
| +
|
| + virtual void WriteRenderStreamMessage(const AudioFrame& frame) = 0;
|
| +
|
| + virtual void WriteRenderStreamMessage(
|
| + std::vector<rtc::ArrayView<const float>> src) = 0;
|
| +
|
| + virtual void WriteCaptureStreamMessage(
|
| + std::unique_ptr<CaptureStreamInfo> stream_info) = 0;
|
| +
|
| + // If not |forced|, only writes the current config if it is
|
| + // different from the last saved one; if |forced|, writes the config
|
| + // regardless of the last saved.
|
| + virtual void WriteConfig(const InternalAPMConfig& config, bool forced) = 0;
|
| +};
|
| +
|
| +class NullCaptureStreamInfo final : public AecDump::CaptureStreamInfo {
|
| + void AddInput(const std::vector<rtc::ArrayView<const float>>& src) override{};
|
| + void AddOutput(
|
| + const std::vector<rtc::ArrayView<const float>>& src) override{};
|
| +
|
| + void AddInput(const AudioFrame& frame) override{};
|
| + void AddOutput(const AudioFrame& frame) override{};
|
| + void set_delay(int delay) override{};
|
| + void set_drift(int drift) override{};
|
| + void set_level(int level) override{};
|
| + void set_keypress(bool keypress) override{};
|
| +};
|
| +
|
| +class NullAecDump final : public AecDump {
|
| + public:
|
| + ~NullAecDump() override = default;
|
| +
|
| + std::unique_ptr<CaptureStreamInfo> GetCaptureStreamInfo() override;
|
| +
|
| + // TODO(aleloi): doc
|
| + void WriteInitMessage(const ProcessingConfig& api_format) override{};
|
| +
|
| + void WriteRenderStreamMessage(const AudioFrame& frame) override{};
|
| +
|
| + void WriteRenderStreamMessage(
|
| + std::vector<rtc::ArrayView<const float>> src) override{};
|
| +
|
| + void WriteCaptureStreamMessage(
|
| + std::unique_ptr<CaptureStreamInfo> stream_info) override{};
|
| +
|
| + void WriteConfig(const InternalAPMConfig& config, bool forced) override{};
|
| +};
|
| +} // namespace webrtc
|
| +
|
| +#endif // WEBRTC_MODULES_AUDIO_PROCESSING_INCLUDE_AEC_DUMP_H_
|
|
|