lapped_transform.cc revision cc84649389eff5ac5b25c7a19a5825abf8e6aa9e
1325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org/* 2325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. 3325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * 4325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * Use of this source code is governed by a BSD-style license 5325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * that can be found in the LICENSE file in the root of the source 6325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * tree. An additional intellectual property rights grant can be found 7325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * in the file PATENTS. All contributing project authors may 8325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org * be found in the AUTHORS file in the root of the source tree. 9325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org */ 10325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 11325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org#include "webrtc/common_audio/lapped_transform.h" 12325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 13e534086492e92c45d74b176f3c8be4addb69713fmgraczyk@chromium.org#include <algorithm> 14325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org#include <cstdlib> 15325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org#include <cstring> 16325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 17325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org#include "webrtc/base/checks.h" 18325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org#include "webrtc/common_audio/real_fourier.h" 19325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 20325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.orgnamespace webrtc { 21325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 22325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.orgvoid LappedTransform::BlockThunk::ProcessBlock(const float* const* input, 23325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org int num_frames, 24325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org int num_input_channels, 25325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org int num_output_channels, 26325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org float* const* output) { 27cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk CHECK_EQ(num_input_channels, parent_->num_in_channels_); 28cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk CHECK_EQ(num_output_channels, parent_->num_out_channels_); 29325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org CHECK_EQ(parent_->block_length_, num_frames); 30325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 31325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org for (int i = 0; i < num_input_channels; ++i) { 32325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org memcpy(parent_->real_buf_.Row(i), input[i], 33325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org num_frames * sizeof(*input[0])); 3404c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org parent_->fft_->Forward(parent_->real_buf_.Row(i), 3504c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org parent_->cplx_pre_.Row(i)); 36325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org } 37325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 38325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org int block_length = RealFourier::ComplexLength( 39325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org RealFourier::FftOrder(num_frames)); 40325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org CHECK_EQ(parent_->cplx_length_, block_length); 41325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org parent_->block_processor_->ProcessAudioBlock(parent_->cplx_pre_.Array(), 42325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org num_input_channels, 43325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org parent_->cplx_length_, 44325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org num_output_channels, 45325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org parent_->cplx_post_.Array()); 46325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 47325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org for (int i = 0; i < num_output_channels; ++i) { 4804c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org parent_->fft_->Inverse(parent_->cplx_post_.Row(i), 4904c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org parent_->real_buf_.Row(i)); 50325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org memcpy(output[i], parent_->real_buf_.Row(i), 51325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org num_frames * sizeof(*input[0])); 52325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org } 53325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org} 54325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 55cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael GraczykLappedTransform::LappedTransform(int num_in_channels, 56cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk int num_out_channels, 57cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk int chunk_length, 58cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk const float* window, 59cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk int block_length, 60cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk int shift_amount, 61325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org Callback* callback) 62325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org : blocker_callback_(this), 63cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk num_in_channels_(num_in_channels), 64cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk num_out_channels_(num_out_channels), 65325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org block_length_(block_length), 66325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org chunk_length_(chunk_length), 67325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org block_processor_(callback), 68cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk blocker_(chunk_length_, 69cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk block_length_, 70cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk num_in_channels_, 71cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk num_out_channels_, 72cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk window, 73cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk shift_amount, 74cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk &blocker_callback_), 7504c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org fft_(RealFourier::Create(RealFourier::FftOrder(block_length_))), 7604c50981f8574b4ef3f8751ac37d0eb50a569a2bandrew@webrtc.org cplx_length_(RealFourier::ComplexLength(fft_->order())), 77cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk real_buf_(num_in_channels, 78cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk block_length_, 79cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk RealFourier::kFftBufferAlignment), 80cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk cplx_pre_(num_in_channels, 81cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk cplx_length_, 82cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk RealFourier::kFftBufferAlignment), 83cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk cplx_post_(num_out_channels, 84cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk cplx_length_, 85cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk RealFourier::kFftBufferAlignment) { 86cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk CHECK(num_in_channels_ > 0 && num_out_channels_ > 0); 87325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org CHECK_GT(block_length_, 0); 88325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org CHECK_GT(chunk_length_, 0); 89325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org CHECK(block_processor_); 90325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 91e534086492e92c45d74b176f3c8be4addb69713fmgraczyk@chromium.org // block_length_ power of 2? 92e534086492e92c45d74b176f3c8be4addb69713fmgraczyk@chromium.org CHECK_EQ(0, block_length_ & (block_length_ - 1)); 93325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org} 94325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 95325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.orgvoid LappedTransform::ProcessChunk(const float* const* in_chunk, 96325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org float* const* out_chunk) { 97cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk blocker_.ProcessChunk(in_chunk, chunk_length_, num_in_channels_, 98cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk num_out_channels_, out_chunk); 99325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org} 100325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 101325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org} // namespace webrtc 102