| Index: webrtc/modules/audio_processing/aec3/echo_canceller3.cc
|
| diff --git a/webrtc/modules/audio_processing/aec3/echo_canceller3.cc b/webrtc/modules/audio_processing/aec3/echo_canceller3.cc
|
| index b409e60350e2c2b6c45b2395d8a42d6285f92410..e69ccdcbc63fe068a09058e97e32518d6050af05 100644
|
| --- a/webrtc/modules/audio_processing/aec3/echo_canceller3.cc
|
| +++ b/webrtc/modules/audio_processing/aec3/echo_canceller3.cc
|
| @@ -9,312 +9,36 @@
|
| */
|
| #include "webrtc/modules/audio_processing/aec3/echo_canceller3.h"
|
|
|
| -#include <sstream>
|
| -
|
| #include "webrtc/base/atomicops.h"
|
| -#include "webrtc/modules/audio_processing/logging/apm_data_dumper.h"
|
| +#include "webrtc/system_wrappers/include/logging.h"
|
|
|
| namespace webrtc {
|
|
|
| -namespace {
|
| -
|
| -bool DetectSaturation(rtc::ArrayView<const float> y) {
|
| - for (auto y_k : y) {
|
| - if (y_k >= 32767.0f || y_k <= -32768.0f) {
|
| - return true;
|
| - }
|
| - }
|
| - return false;
|
| -}
|
| -
|
| -void FillSubFrameView(AudioBuffer* frame,
|
| - size_t sub_frame_index,
|
| - std::vector<rtc::ArrayView<float>>* sub_frame_view) {
|
| - RTC_DCHECK_GE(1, sub_frame_index);
|
| - RTC_DCHECK_LE(0, sub_frame_index);
|
| - RTC_DCHECK_EQ(frame->num_bands(), sub_frame_view->size());
|
| - for (size_t k = 0; k < sub_frame_view->size(); ++k) {
|
| - (*sub_frame_view)[k] = rtc::ArrayView<float>(
|
| - &frame->split_bands_f(0)[k][sub_frame_index * kSubFrameLength],
|
| - kSubFrameLength);
|
| - }
|
| -}
|
| -
|
| -void FillSubFrameView(std::vector<std::vector<float>>* frame,
|
| - size_t sub_frame_index,
|
| - std::vector<rtc::ArrayView<float>>* sub_frame_view) {
|
| - RTC_DCHECK_GE(1, sub_frame_index);
|
| - RTC_DCHECK_EQ(frame->size(), sub_frame_view->size());
|
| - for (size_t k = 0; k < frame->size(); ++k) {
|
| - (*sub_frame_view)[k] = rtc::ArrayView<float>(
|
| - &(*frame)[k][sub_frame_index * kSubFrameLength], kSubFrameLength);
|
| - }
|
| -}
|
| -
|
| -void ProcessCaptureFrameContent(
|
| - AudioBuffer* capture,
|
| - bool known_echo_path_change,
|
| - bool saturated_microphone_signal,
|
| - size_t sub_frame_index,
|
| - FrameBlocker* capture_blocker,
|
| - BlockFramer* output_framer,
|
| - BlockProcessor* block_processor,
|
| - std::vector<std::vector<float>>* block,
|
| - std::vector<rtc::ArrayView<float>>* sub_frame_view) {
|
| - FillSubFrameView(capture, sub_frame_index, sub_frame_view);
|
| - capture_blocker->InsertSubFrameAndExtractBlock(*sub_frame_view, block);
|
| - block_processor->ProcessCapture(known_echo_path_change,
|
| - saturated_microphone_signal, block);
|
| - output_framer->InsertBlockAndExtractSubFrame(*block, sub_frame_view);
|
| -}
|
| -
|
| -void ProcessRemainingCaptureFrameContent(
|
| - bool known_echo_path_change,
|
| - bool saturated_microphone_signal,
|
| - FrameBlocker* capture_blocker,
|
| - BlockFramer* output_framer,
|
| - BlockProcessor* block_processor,
|
| - std::vector<std::vector<float>>* block) {
|
| - if (!capture_blocker->IsBlockAvailable()) {
|
| - return;
|
| - }
|
| -
|
| - capture_blocker->ExtractBlock(block);
|
| - block_processor->ProcessCapture(known_echo_path_change,
|
| - saturated_microphone_signal, block);
|
| - output_framer->InsertBlock(*block);
|
| -}
|
| -
|
| -bool BufferRenderFrameContent(
|
| - std::vector<std::vector<float>>* render_frame,
|
| - size_t sub_frame_index,
|
| - FrameBlocker* render_blocker,
|
| - BlockProcessor* block_processor,
|
| - std::vector<std::vector<float>>* block,
|
| - std::vector<rtc::ArrayView<float>>* sub_frame_view) {
|
| - FillSubFrameView(render_frame, sub_frame_index, sub_frame_view);
|
| - render_blocker->InsertSubFrameAndExtractBlock(*sub_frame_view, block);
|
| - return block_processor->BufferRender(block);
|
| -}
|
| -
|
| -bool BufferRemainingRenderFrameContent(FrameBlocker* render_blocker,
|
| - BlockProcessor* block_processor,
|
| - std::vector<std::vector<float>>* block) {
|
| - if (!render_blocker->IsBlockAvailable()) {
|
| - return false;
|
| - }
|
| - render_blocker->ExtractBlock(block);
|
| - return block_processor->BufferRender(block);
|
| -}
|
| -
|
| -void CopyAudioBufferIntoFrame(AudioBuffer* buffer,
|
| - size_t num_bands,
|
| - size_t frame_length,
|
| - std::vector<std::vector<float>>* frame) {
|
| - RTC_DCHECK_EQ(num_bands, frame->size());
|
| - for (size_t i = 0; i < num_bands; ++i) {
|
| - rtc::ArrayView<float> buffer_view(&buffer->split_bands_f(0)[i][0],
|
| - frame_length);
|
| - std::copy(buffer_view.begin(), buffer_view.end(), (*frame)[i].begin());
|
| - }
|
| -}
|
| -
|
| -// [B,A] = butter(2,100/4000,'high')
|
| -const CascadedBiQuadFilter::BiQuadCoefficients
|
| - kHighPassFilterCoefficients_8kHz = {{0.94598f, -1.89195f, 0.94598f},
|
| - {-1.88903f, 0.89487f}};
|
| -const int kNumberOfHighPassBiQuads_8kHz = 1;
|
| -
|
| -// [B,A] = butter(2,100/8000,'high')
|
| -const CascadedBiQuadFilter::BiQuadCoefficients
|
| - kHighPassFilterCoefficients_16kHz = {{0.97261f, -1.94523f, 0.97261f},
|
| - {-1.94448f, 0.94598f}};
|
| -const int kNumberOfHighPassBiQuads_16kHz = 1;
|
| -
|
| -static constexpr size_t kRenderTransferQueueSize = 30;
|
| -
|
| -} // namespace
|
| -
|
| -class EchoCanceller3::RenderWriter {
|
| - public:
|
| - RenderWriter(ApmDataDumper* data_dumper,
|
| - SwapQueue<std::vector<std::vector<float>>,
|
| - Aec3RenderQueueItemVerifier>* render_transfer_queue,
|
| - std::unique_ptr<CascadedBiQuadFilter> render_highpass_filter,
|
| - int sample_rate_hz,
|
| - int frame_length,
|
| - int num_bands);
|
| - ~RenderWriter();
|
| - bool Insert(AudioBuffer* render);
|
| -
|
| - private:
|
| - ApmDataDumper* data_dumper_;
|
| - const int sample_rate_hz_;
|
| - const size_t frame_length_;
|
| - const int num_bands_;
|
| - std::unique_ptr<CascadedBiQuadFilter> render_highpass_filter_;
|
| - std::vector<std::vector<float>> render_queue_input_frame_;
|
| - SwapQueue<std::vector<std::vector<float>>, Aec3RenderQueueItemVerifier>*
|
| - render_transfer_queue_;
|
| - RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(RenderWriter);
|
| -};
|
| -
|
| -EchoCanceller3::RenderWriter::RenderWriter(
|
| - ApmDataDumper* data_dumper,
|
| - SwapQueue<std::vector<std::vector<float>>, Aec3RenderQueueItemVerifier>*
|
| - render_transfer_queue,
|
| - std::unique_ptr<CascadedBiQuadFilter> render_highpass_filter,
|
| - int sample_rate_hz,
|
| - int frame_length,
|
| - int num_bands)
|
| - : data_dumper_(data_dumper),
|
| - sample_rate_hz_(sample_rate_hz),
|
| - frame_length_(frame_length),
|
| - num_bands_(num_bands),
|
| - render_highpass_filter_(std::move(render_highpass_filter)),
|
| - render_queue_input_frame_(num_bands_,
|
| - std::vector<float>(frame_length_, 0.f)),
|
| - render_transfer_queue_(render_transfer_queue) {
|
| - RTC_DCHECK(data_dumper);
|
| -}
|
| -
|
| -EchoCanceller3::RenderWriter::~RenderWriter() = default;
|
| -
|
| -bool EchoCanceller3::RenderWriter::Insert(AudioBuffer* input) {
|
| - RTC_DCHECK_EQ(1, input->num_channels());
|
| - RTC_DCHECK_EQ(num_bands_, input->num_bands());
|
| - RTC_DCHECK_EQ(frame_length_, input->num_frames_per_band());
|
| - data_dumper_->DumpWav("aec3_render_input", frame_length_,
|
| - &input->split_bands_f(0)[0][0],
|
| - LowestBandRate(sample_rate_hz_), 1);
|
| -
|
| - CopyAudioBufferIntoFrame(input, num_bands_, frame_length_,
|
| - &render_queue_input_frame_);
|
| -
|
| - if (render_highpass_filter_) {
|
| - render_highpass_filter_->Process(render_queue_input_frame_[0]);
|
| - }
|
| -
|
| - return render_transfer_queue_->Insert(&render_queue_input_frame_);
|
| -}
|
| -
|
| int EchoCanceller3::instance_count_ = 0;
|
|
|
| -EchoCanceller3::EchoCanceller3(int sample_rate_hz, bool use_highpass_filter)
|
| - : EchoCanceller3(sample_rate_hz,
|
| - use_highpass_filter,
|
| - std::unique_ptr<BlockProcessor>(
|
| - BlockProcessor::Create(sample_rate_hz))) {}
|
| -EchoCanceller3::EchoCanceller3(int sample_rate_hz,
|
| - bool use_highpass_filter,
|
| - std::unique_ptr<BlockProcessor> block_processor)
|
| - : data_dumper_(
|
| - new ApmDataDumper(rtc::AtomicOps::Increment(&instance_count_))),
|
| - sample_rate_hz_(sample_rate_hz),
|
| - num_bands_(NumBandsForRate(sample_rate_hz_)),
|
| - frame_length_(rtc::CheckedDivExact(LowestBandRate(sample_rate_hz_), 100)),
|
| - output_framer_(num_bands_),
|
| - capture_blocker_(num_bands_),
|
| - render_blocker_(num_bands_),
|
| - render_transfer_queue_(
|
| - kRenderTransferQueueSize,
|
| - std::vector<std::vector<float>>(
|
| - num_bands_,
|
| - std::vector<float>(frame_length_, 0.f)),
|
| - Aec3RenderQueueItemVerifier(num_bands_, frame_length_)),
|
| - block_processor_(std::move(block_processor)),
|
| - render_queue_output_frame_(num_bands_,
|
| - std::vector<float>(frame_length_, 0.f)),
|
| - block_(num_bands_, std::vector<float>(kBlockSize, 0.f)),
|
| - sub_frame_view_(num_bands_) {
|
| - std::unique_ptr<CascadedBiQuadFilter> render_highpass_filter;
|
| - if (use_highpass_filter) {
|
| - render_highpass_filter.reset(new CascadedBiQuadFilter(
|
| - sample_rate_hz_ == 8000 ? kHighPassFilterCoefficients_8kHz
|
| - : kHighPassFilterCoefficients_16kHz,
|
| - sample_rate_hz_ == 8000 ? kNumberOfHighPassBiQuads_8kHz
|
| - : kNumberOfHighPassBiQuads_16kHz));
|
| - capture_highpass_filter_.reset(new CascadedBiQuadFilter(
|
| - sample_rate_hz_ == 8000 ? kHighPassFilterCoefficients_8kHz
|
| - : kHighPassFilterCoefficients_16kHz,
|
| - sample_rate_hz_ == 8000 ? kNumberOfHighPassBiQuads_8kHz
|
| - : kNumberOfHighPassBiQuads_16kHz));
|
| - }
|
| +EchoCanceller3::EchoCanceller3(int sample_rate_hz, bool use_anti_hum_filter) {
|
| + int band_sample_rate_hz = (sample_rate_hz == 8000 ? sample_rate_hz : 16000);
|
| + frame_length_ = rtc::CheckedDivExact(band_sample_rate_hz, 100);
|
|
|
| - render_writer_.reset(
|
| - new RenderWriter(data_dumper_.get(), &render_transfer_queue_,
|
| - std::move(render_highpass_filter), sample_rate_hz_,
|
| - frame_length_, num_bands_));
|
| -
|
| - RTC_DCHECK_EQ(num_bands_, std::max(sample_rate_hz_, 16000) / 16000);
|
| - RTC_DCHECK_GE(kMaxNumBands, num_bands_);
|
| + LOG(LS_INFO) << "AEC3 created : "
|
| + << "{ instance_count: " << instance_count_ << "}";
|
| + instance_count_ = rtc::AtomicOps::Increment(&instance_count_);
|
| }
|
|
|
| EchoCanceller3::~EchoCanceller3() = default;
|
|
|
| bool EchoCanceller3::AnalyzeRender(AudioBuffer* render) {
|
| - RTC_DCHECK_RUNS_SERIALIZED(&render_race_checker_);
|
| - RTC_DCHECK(render);
|
| - return render_writer_->Insert(render);
|
| + RTC_DCHECK_EQ(1u, render->num_channels());
|
| + RTC_DCHECK_EQ(frame_length_, render->num_frames_per_band());
|
| + return true;
|
| }
|
|
|
| -void EchoCanceller3::AnalyzeCapture(AudioBuffer* capture) {
|
| - RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
| - RTC_DCHECK(capture);
|
| - data_dumper_->DumpWav("aec3_capture_analyze_input", frame_length_,
|
| - capture->channels_f()[0], sample_rate_hz_, 1);
|
| -
|
| - saturated_microphone_signal_ = false;
|
| - for (size_t k = 0; k < capture->num_channels(); ++k) {
|
| - saturated_microphone_signal_ |=
|
| - DetectSaturation(rtc::ArrayView<const float>(capture->channels_f()[k],
|
| - capture->num_frames()));
|
| - if (saturated_microphone_signal_) {
|
| - break;
|
| - }
|
| - }
|
| -}
|
| +void EchoCanceller3::AnalyzeCapture(AudioBuffer* capture) {}
|
|
|
| void EchoCanceller3::ProcessCapture(AudioBuffer* capture,
|
| bool known_echo_path_change) {
|
| - RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
| - RTC_DCHECK(capture);
|
| RTC_DCHECK_EQ(1u, capture->num_channels());
|
| - RTC_DCHECK_EQ(num_bands_, capture->num_bands());
|
| RTC_DCHECK_EQ(frame_length_, capture->num_frames_per_band());
|
| -
|
| - rtc::ArrayView<float> capture_lower_band =
|
| - rtc::ArrayView<float>(&capture->split_bands_f(0)[0][0], frame_length_);
|
| -
|
| - data_dumper_->DumpWav("aec3_capture_input", capture_lower_band,
|
| - LowestBandRate(sample_rate_hz_), 1);
|
| -
|
| - const bool render_buffer_overrun = EmptyRenderQueue();
|
| - RTC_DCHECK(!render_buffer_overrun);
|
| -
|
| - if (capture_highpass_filter_) {
|
| - capture_highpass_filter_->Process(capture_lower_band);
|
| - }
|
| -
|
| - ProcessCaptureFrameContent(capture, known_echo_path_change,
|
| - saturated_microphone_signal_, 0, &capture_blocker_,
|
| - &output_framer_, block_processor_.get(), &block_,
|
| - &sub_frame_view_);
|
| -
|
| - if (sample_rate_hz_ != 8000) {
|
| - ProcessCaptureFrameContent(
|
| - capture, known_echo_path_change, saturated_microphone_signal_, 1,
|
| - &capture_blocker_, &output_framer_, block_processor_.get(), &block_,
|
| - &sub_frame_view_);
|
| - }
|
| -
|
| - ProcessRemainingCaptureFrameContent(
|
| - known_echo_path_change, saturated_microphone_signal_, &capture_blocker_,
|
| - &output_framer_, block_processor_.get(), &block_);
|
| -
|
| - data_dumper_->DumpWav("aec3_capture_output", frame_length_,
|
| - &capture->split_bands_f(0)[0][0],
|
| - LowestBandRate(sample_rate_hz_), 1);
|
| }
|
|
|
| std::string EchoCanceller3::ToString(
|
| @@ -330,29 +54,4 @@
|
| return true;
|
| }
|
|
|
| -bool EchoCanceller3::EmptyRenderQueue() {
|
| - RTC_DCHECK_RUNS_SERIALIZED(&capture_race_checker_);
|
| - bool render_buffer_overrun = false;
|
| - bool frame_to_buffer =
|
| - render_transfer_queue_.Remove(&render_queue_output_frame_);
|
| - while (frame_to_buffer) {
|
| - render_buffer_overrun |= BufferRenderFrameContent(
|
| - &render_queue_output_frame_, 0, &render_blocker_,
|
| - block_processor_.get(), &block_, &sub_frame_view_);
|
| -
|
| - if (sample_rate_hz_ != 8000) {
|
| - render_buffer_overrun |= BufferRenderFrameContent(
|
| - &render_queue_output_frame_, 1, &render_blocker_,
|
| - block_processor_.get(), &block_, &sub_frame_view_);
|
| - }
|
| -
|
| - render_buffer_overrun |= BufferRemainingRenderFrameContent(
|
| - &render_blocker_, block_processor_.get(), &block_);
|
| -
|
| - frame_to_buffer =
|
| - render_transfer_queue_.Remove(&render_queue_output_frame_);
|
| - }
|
| - return render_buffer_overrun;
|
| -}
|
| -
|
| } // namespace webrtc
|
|
|