| Index: webrtc/common_audio/blocker.cc
|
| diff --git a/webrtc/common_audio/blocker.cc b/webrtc/common_audio/blocker.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..13432f2e7a89fd990debd0f59cd12b4528c79c75
|
| --- /dev/null
|
| +++ b/webrtc/common_audio/blocker.cc
|
| @@ -0,0 +1,236 @@
|
| +/*
|
| + * Copyright (c) 2014 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.
|
| + */
|
| +
|
| +#include "webrtc/common_audio/blocker.h"
|
| +
|
| +#include <string.h>
|
| +
|
| +#include "webrtc/base/checks.h"
|
| +
|
| +namespace {
|
| +
|
| +// Adds |a| and |b| frame by frame into |result| (basically matrix addition).
|
| +void AddFrames(const float* const* a,
|
| + size_t a_start_index,
|
| + const float* const* b,
|
| + int b_start_index,
|
| + size_t num_frames,
|
| + size_t num_channels,
|
| + float* const* result,
|
| + size_t result_start_index) {
|
| + for (size_t i = 0; i < num_channels; ++i) {
|
| + for (size_t j = 0; j < num_frames; ++j) {
|
| + result[i][j + result_start_index] =
|
| + a[i][j + a_start_index] + b[i][j + b_start_index];
|
| + }
|
| + }
|
| +}
|
| +
|
| +// Copies |src| into |dst| channel by channel.
|
| +void CopyFrames(const float* const* src,
|
| + size_t src_start_index,
|
| + size_t num_frames,
|
| + size_t num_channels,
|
| + float* const* dst,
|
| + size_t dst_start_index) {
|
| + for (size_t i = 0; i < num_channels; ++i) {
|
| + memcpy(&dst[i][dst_start_index],
|
| + &src[i][src_start_index],
|
| + num_frames * sizeof(dst[i][dst_start_index]));
|
| + }
|
| +}
|
| +
|
| +// Moves |src| into |dst| channel by channel.
|
| +void MoveFrames(const float* const* src,
|
| + size_t src_start_index,
|
| + size_t num_frames,
|
| + size_t num_channels,
|
| + float* const* dst,
|
| + size_t dst_start_index) {
|
| + for (size_t i = 0; i < num_channels; ++i) {
|
| + memmove(&dst[i][dst_start_index],
|
| + &src[i][src_start_index],
|
| + num_frames * sizeof(dst[i][dst_start_index]));
|
| + }
|
| +}
|
| +
|
| +void ZeroOut(float* const* buffer,
|
| + size_t starting_idx,
|
| + size_t num_frames,
|
| + size_t num_channels) {
|
| + for (size_t i = 0; i < num_channels; ++i) {
|
| + memset(&buffer[i][starting_idx], 0,
|
| + num_frames * sizeof(buffer[i][starting_idx]));
|
| + }
|
| +}
|
| +
|
| +// Pointwise multiplies each channel of |frames| with |window|. Results are
|
| +// stored in |frames|.
|
| +void ApplyWindow(const float* window,
|
| + size_t num_frames,
|
| + size_t num_channels,
|
| + float* const* frames) {
|
| + for (size_t i = 0; i < num_channels; ++i) {
|
| + for (size_t j = 0; j < num_frames; ++j) {
|
| + frames[i][j] = frames[i][j] * window[j];
|
| + }
|
| + }
|
| +}
|
| +
|
| +size_t gcd(size_t a, size_t b) {
|
| + size_t tmp;
|
| + while (b) {
|
| + tmp = a;
|
| + a = b;
|
| + b = tmp % b;
|
| + }
|
| + return a;
|
| +}
|
| +
|
| +} // namespace
|
| +
|
| +namespace webrtc {
|
| +
|
| +Blocker::Blocker(size_t chunk_size,
|
| + size_t block_size,
|
| + size_t num_input_channels,
|
| + size_t num_output_channels,
|
| + const float* window,
|
| + size_t shift_amount,
|
| + BlockerCallback* callback)
|
| + : chunk_size_(chunk_size),
|
| + block_size_(block_size),
|
| + num_input_channels_(num_input_channels),
|
| + num_output_channels_(num_output_channels),
|
| + initial_delay_(block_size_ - gcd(chunk_size, shift_amount)),
|
| + frame_offset_(0),
|
| + input_buffer_(num_input_channels_, chunk_size_ + initial_delay_),
|
| + output_buffer_(chunk_size_ + initial_delay_, num_output_channels_),
|
| + input_block_(block_size_, num_input_channels_),
|
| + output_block_(block_size_, num_output_channels_),
|
| + window_(new float[block_size_]),
|
| + shift_amount_(shift_amount),
|
| + callback_(callback) {
|
| + RTC_CHECK_LE(num_output_channels_, num_input_channels_);
|
| + RTC_CHECK_LE(shift_amount_, block_size_);
|
| +
|
| + memcpy(window_.get(), window, block_size_ * sizeof(*window_.get()));
|
| + input_buffer_.MoveReadPositionBackward(initial_delay_);
|
| +}
|
| +
|
| +// When block_size < chunk_size the input and output buffers look like this:
|
| +//
|
| +// delay* chunk_size chunk_size + delay*
|
| +// buffer: <-------------|---------------------|---------------|>
|
| +// _a_ _b_ _c_
|
| +//
|
| +// On each call to ProcessChunk():
|
| +// 1. New input gets read into sections _b_ and _c_ of the input buffer.
|
| +// 2. We block starting from frame_offset.
|
| +// 3. We block until we reach a block |bl| that doesn't contain any frames
|
| +// from sections _a_ or _b_ of the input buffer.
|
| +// 4. We window the current block, fire the callback for processing, window
|
| +// again, and overlap/add to the output buffer.
|
| +// 5. We copy sections _a_ and _b_ of the output buffer into output.
|
| +// 6. For both the input and the output buffers, we copy section _c_ into
|
| +// section _a_.
|
| +// 7. We set the new frame_offset to be the difference between the first frame
|
| +// of |bl| and the border between sections _b_ and _c_.
|
| +//
|
| +// When block_size > chunk_size the input and output buffers look like this:
|
| +//
|
| +// chunk_size delay* chunk_size + delay*
|
| +// buffer: <-------------|---------------------|---------------|>
|
| +// _a_ _b_ _c_
|
| +//
|
| +// On each call to ProcessChunk():
|
| +// The procedure is the same as above, except for:
|
| +// 1. New input gets read into section _c_ of the input buffer.
|
| +// 3. We block until we reach a block |bl| that doesn't contain any frames
|
| +// from section _a_ of the input buffer.
|
| +// 5. We copy section _a_ of the output buffer into output.
|
| +// 6. For both the input and the output buffers, we copy sections _b_ and _c_
|
| +// into section _a_ and _b_.
|
| +// 7. We set the new frame_offset to be the difference between the first frame
|
| +// of |bl| and the border between sections _a_ and _b_.
|
| +//
|
| +// * delay here refers to inintial_delay_
|
| +//
|
| +// TODO(claguna): Look at using ring buffers to eliminate some copies.
|
| +void Blocker::ProcessChunk(const float* const* input,
|
| + size_t chunk_size,
|
| + size_t num_input_channels,
|
| + size_t num_output_channels,
|
| + float* const* output) {
|
| + RTC_CHECK_EQ(chunk_size, chunk_size_);
|
| + RTC_CHECK_EQ(num_input_channels, num_input_channels_);
|
| + RTC_CHECK_EQ(num_output_channels, num_output_channels_);
|
| +
|
| + input_buffer_.Write(input, num_input_channels, chunk_size_);
|
| + size_t first_frame_in_block = frame_offset_;
|
| +
|
| + // Loop through blocks.
|
| + while (first_frame_in_block < chunk_size_) {
|
| + input_buffer_.Read(input_block_.channels(), num_input_channels,
|
| + block_size_);
|
| + input_buffer_.MoveReadPositionBackward(block_size_ - shift_amount_);
|
| +
|
| + ApplyWindow(window_.get(),
|
| + block_size_,
|
| + num_input_channels_,
|
| + input_block_.channels());
|
| + callback_->ProcessBlock(input_block_.channels(),
|
| + block_size_,
|
| + num_input_channels_,
|
| + num_output_channels_,
|
| + output_block_.channels());
|
| + ApplyWindow(window_.get(),
|
| + block_size_,
|
| + num_output_channels_,
|
| + output_block_.channels());
|
| +
|
| + AddFrames(output_buffer_.channels(),
|
| + first_frame_in_block,
|
| + output_block_.channels(),
|
| + 0,
|
| + block_size_,
|
| + num_output_channels_,
|
| + output_buffer_.channels(),
|
| + first_frame_in_block);
|
| +
|
| + first_frame_in_block += shift_amount_;
|
| + }
|
| +
|
| + // Copy output buffer to output
|
| + CopyFrames(output_buffer_.channels(),
|
| + 0,
|
| + chunk_size_,
|
| + num_output_channels_,
|
| + output,
|
| + 0);
|
| +
|
| + // Copy output buffer [chunk_size_, chunk_size_ + initial_delay]
|
| + // to output buffer [0, initial_delay], zero the rest.
|
| + MoveFrames(output_buffer_.channels(),
|
| + chunk_size,
|
| + initial_delay_,
|
| + num_output_channels_,
|
| + output_buffer_.channels(),
|
| + 0);
|
| + ZeroOut(output_buffer_.channels(),
|
| + initial_delay_,
|
| + chunk_size_,
|
| + num_output_channels_);
|
| +
|
| + // Calculate new starting frames.
|
| + frame_offset_ = first_frame_in_block - chunk_size_;
|
| +}
|
| +
|
| +} // namespace webrtc
|
|
|