dce40cf804
use of int16_t/uint16_t. This is the upshot of a recommendation by henrik.lundin and kwiberg on an original small change ( https://webrtc-codereview.appspot.com/42569004/#ps1 ) to stop using int16_t just because values could fit in it, and is similar in nature to a previous "mass change to use size_t more" ( https://webrtc-codereview.appspot.com/23129004/ ) which also needed to be split up for review but to land all at once, since, like adding "const", such changes tend to cause a lot of transitive effects. This was be reviewed and approved in pieces: https://codereview.webrtc.org/1224093003 https://codereview.webrtc.org/1224123002 https://codereview.webrtc.org/1224163002 https://codereview.webrtc.org/1225133003 https://codereview.webrtc.org/1225173002 https://codereview.webrtc.org/1227163003 https://codereview.webrtc.org/1227203003 https://codereview.webrtc.org/1227213002 https://codereview.webrtc.org/1227893002 https://codereview.webrtc.org/1228793004 https://codereview.webrtc.org/1228803003 https://codereview.webrtc.org/1228823002 https://codereview.webrtc.org/1228823003 https://codereview.webrtc.org/1228843002 https://codereview.webrtc.org/1230693002 https://codereview.webrtc.org/1231713002 The change is being landed as TBR to all the folks who reviewed the above. BUG=chromium:81439 TEST=none R=andrew@webrtc.org, pbos@webrtc.org TBR=aluebs, andrew, asapersson, henrika, hlundin, jan.skoglund, kwiberg, minyue, pbos, pthatcher Review URL: https://codereview.webrtc.org/1230503003 . Cr-Commit-Position: refs/heads/master@{#9768}
102 lines
3.9 KiB
C++
102 lines
3.9 KiB
C++
/*
|
|
* 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/lapped_transform.h"
|
|
|
|
#include <algorithm>
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
|
|
#include "webrtc/base/checks.h"
|
|
#include "webrtc/common_audio/real_fourier.h"
|
|
|
|
namespace webrtc {
|
|
|
|
void LappedTransform::BlockThunk::ProcessBlock(const float* const* input,
|
|
size_t num_frames,
|
|
int num_input_channels,
|
|
int num_output_channels,
|
|
float* const* output) {
|
|
CHECK_EQ(num_input_channels, parent_->num_in_channels_);
|
|
CHECK_EQ(num_output_channels, parent_->num_out_channels_);
|
|
CHECK_EQ(parent_->block_length_, num_frames);
|
|
|
|
for (int i = 0; i < num_input_channels; ++i) {
|
|
memcpy(parent_->real_buf_.Row(i), input[i],
|
|
num_frames * sizeof(*input[0]));
|
|
parent_->fft_->Forward(parent_->real_buf_.Row(i),
|
|
parent_->cplx_pre_.Row(i));
|
|
}
|
|
|
|
size_t block_length = RealFourier::ComplexLength(
|
|
RealFourier::FftOrder(num_frames));
|
|
CHECK_EQ(parent_->cplx_length_, block_length);
|
|
parent_->block_processor_->ProcessAudioBlock(parent_->cplx_pre_.Array(),
|
|
num_input_channels,
|
|
parent_->cplx_length_,
|
|
num_output_channels,
|
|
parent_->cplx_post_.Array());
|
|
|
|
for (int i = 0; i < num_output_channels; ++i) {
|
|
parent_->fft_->Inverse(parent_->cplx_post_.Row(i),
|
|
parent_->real_buf_.Row(i));
|
|
memcpy(output[i], parent_->real_buf_.Row(i),
|
|
num_frames * sizeof(*input[0]));
|
|
}
|
|
}
|
|
|
|
LappedTransform::LappedTransform(int num_in_channels,
|
|
int num_out_channels,
|
|
size_t chunk_length,
|
|
const float* window,
|
|
size_t block_length,
|
|
size_t shift_amount,
|
|
Callback* callback)
|
|
: blocker_callback_(this),
|
|
num_in_channels_(num_in_channels),
|
|
num_out_channels_(num_out_channels),
|
|
block_length_(block_length),
|
|
chunk_length_(chunk_length),
|
|
block_processor_(callback),
|
|
blocker_(chunk_length_,
|
|
block_length_,
|
|
num_in_channels_,
|
|
num_out_channels_,
|
|
window,
|
|
shift_amount,
|
|
&blocker_callback_),
|
|
fft_(RealFourier::Create(RealFourier::FftOrder(block_length_))),
|
|
cplx_length_(RealFourier::ComplexLength(fft_->order())),
|
|
real_buf_(num_in_channels,
|
|
block_length_,
|
|
RealFourier::kFftBufferAlignment),
|
|
cplx_pre_(num_in_channels,
|
|
cplx_length_,
|
|
RealFourier::kFftBufferAlignment),
|
|
cplx_post_(num_out_channels,
|
|
cplx_length_,
|
|
RealFourier::kFftBufferAlignment) {
|
|
CHECK(num_in_channels_ > 0 && num_out_channels_ > 0);
|
|
CHECK_GT(block_length_, 0u);
|
|
CHECK_GT(chunk_length_, 0u);
|
|
CHECK(block_processor_);
|
|
|
|
// block_length_ power of 2?
|
|
CHECK_EQ(0u, block_length_ & (block_length_ - 1));
|
|
}
|
|
|
|
void LappedTransform::ProcessChunk(const float* const* in_chunk,
|
|
float* const* out_chunk) {
|
|
blocker_.ProcessChunk(in_chunk, chunk_length_, num_in_channels_,
|
|
num_out_channels_, out_chunk);
|
|
}
|
|
|
|
} // namespace webrtc
|