Chromium Code Reviews| Index: webrtc/modules/audio_processing/echo_cancellation_impl.cc |
| diff --git a/webrtc/modules/audio_processing/echo_cancellation_impl.cc b/webrtc/modules/audio_processing/echo_cancellation_impl.cc |
| index bdcfa2ad075877d6c772b2b6492c781d151150b9..05f53f836e3b49112d1d742153a2fb6b42bcf174 100644 |
| --- a/webrtc/modules/audio_processing/echo_cancellation_impl.cc |
| +++ b/webrtc/modules/audio_processing/echo_cancellation_impl.cc |
| @@ -68,7 +68,15 @@ EchoCancellationImpl::EchoCancellationImpl(const AudioProcessing* apm, |
| stream_has_echo_(false), |
| delay_logging_enabled_(false), |
| extended_filter_enabled_(false), |
| - delay_agnostic_enabled_(false) { |
| + delay_agnostic_enabled_(false), |
| + render_queue_buffer_(kMaxNumSamplesPerFrameToBuffer), |
| + capture_queue_buffer_(kMaxNumSamplesPerFrameToBuffer) { |
| + std::vector<float> template_queue_element(kMaxNumSamplesPerFrameToBuffer); |
| + |
| + render_signal_queue_.reset( |
| + new SwapQueue<std::vector<float>, &RenderQueueItemVerifier>( |
| + (kMaxNumFramesToBuffer * kMaxNumChannelsPerFrameToBuffer), |
| + template_queue_element)); |
| } |
| EchoCancellationImpl::~EchoCancellationImpl() {} |
| @@ -88,15 +96,27 @@ int EchoCancellationImpl::ProcessRenderAudio(const AudioBuffer* audio) { |
| for (int i = 0; i < apm_->num_output_channels(); i++) { |
| for (int j = 0; j < audio->num_channels(); j++) { |
| Handle* my_handle = static_cast<Handle*>(handle(handle_index)); |
| - err = WebRtcAec_BufferFarend( |
| - my_handle, |
| - audio->split_bands_const_f(j)[kBand0To8kHz], |
| + // Retrieve any error code produced by the buffering of the farend |
| + // signal |
| + err = WebRtcAec_GetBufferFarendError( |
| + my_handle, audio->split_bands_const_f(j)[kBand0To8kHz], |
| audio->num_frames_per_band()); |
| if (err != apm_->kNoError) { |
| return MapError(err); // TODO(ajm): warning possible? |
| } |
| + // Buffer the samples in the render queue. |
| + memcpy(&render_queue_buffer_[0], |
| + audio->split_bands_const_f(j)[kBand0To8kHz], |
| + (audio->num_frames_per_band() * |
| + sizeof(*audio->split_bands_const_f(j)[kBand0To8kHz]))); |
|
kwiberg-webrtc
2015/10/17 07:50:07
Maybe I just haven't stared long enough, but where
peah-webrtc
2015/10/26 09:03:48
I rewrote this quite a lot. Could you please let m
|
| + render_queue_buffer_.resize(audio->num_frames_per_band()); |
|
kwiberg-webrtc
2015/10/17 07:50:07
Oh, here. This is illegal if it ends up increasing
peah-webrtc
2015/10/26 09:03:48
Agree. It is now rewritten. Could you please reche
|
| + // TODO(peah): Refactor so that it is possible to check the |
| + // return value of Insert. Currently, that is not possible |
| + // due to the code design when the capture thread is never |
| + // started. |
|
kwiberg-webrtc
2015/10/17 07:50:07
Yeah, I guess that has to be done before this CL c
peah-webrtc
2015/10/26 09:03:48
This is also now redesigned. It should be fine now
|
| + render_signal_queue_->Insert(&render_queue_buffer_); |
| handle_index++; |
| } |
| } |
| @@ -104,6 +124,25 @@ int EchoCancellationImpl::ProcessRenderAudio(const AudioBuffer* audio) { |
| return apm_->kNoError; |
| } |
| +// Read chunks of data that were received and queued on the render side from |
| +// a queue. All the data chunks are buffered into the farend signal of the AEC. |
| +void EchoCancellationImpl::ReadQueuedRenderData() { |
| + if (!is_component_enabled()) { |
| + return; |
| + } |
| + |
| + while (render_signal_queue_->Remove(&capture_queue_buffer_)) { |
| + size_t handle_index = 0; |
| + for (int i = 0; i < apm_->num_output_channels(); i++) { |
| + for (int j = 0; j < apm_->num_reverse_channels(); j++) { |
| + Handle* my_handle = static_cast<Handle*>(handle(handle_index)); |
| + (void)WebRtcAec_BufferFarend(my_handle, &capture_queue_buffer_[0], |
| + capture_queue_buffer_.size()); |
| + } |
| + } |
| + } |
| +} |
| + |
| int EchoCancellationImpl::ProcessCaptureAudio(AudioBuffer* audio) { |
| if (!is_component_enabled()) { |
| return apm_->kNoError; |
| @@ -328,6 +367,7 @@ struct AecCore* EchoCancellationImpl::aec_core() const { |
| } |
| int EchoCancellationImpl::Initialize() { |
| + render_signal_queue_->Clear(); |
| int err = ProcessingComponent::Initialize(); |
| if (err != apm_->kNoError || !is_component_enabled()) { |
| return err; |
| @@ -368,7 +408,6 @@ int EchoCancellationImpl::ConfigureHandle(void* handle) const { |
| config.nlpMode = MapSetting(suppression_level_); |
| config.skewMode = drift_compensation_enabled_; |
| config.delay_logging = delay_logging_enabled_; |
| - |
| WebRtcAec_enable_extended_filter( |
| WebRtcAec_aec_core(static_cast<Handle*>(handle)), |
| extended_filter_enabled_ ? 1 : 0); |
| @@ -387,4 +426,11 @@ int EchoCancellationImpl::GetHandleError(void* handle) const { |
| assert(handle != NULL); |
| return AudioProcessing::kUnspecifiedError; |
| } |
| + |
| +bool EchoCancellationImpl::RenderQueueItemVerifier( |
| + const std::vector<float>& v) { |
| + return ((v.size() == kMinNumSamplesPerFrameToBuffer) || |
| + (v.size() == kMaxNumSamplesPerFrameToBuffer)); |
|
kwiberg-webrtc
2015/10/17 07:50:07
From the names of these constants, I would've expe
peah-webrtc
2015/10/26 09:03:48
I totally agree.
Regarding the names I renamed th
|
| +} |
| + |
| } // namespace webrtc |