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, 23dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t num_frames, 246955870806624479723addfae6dcf5d13968796cPeter Kasting size_t num_input_channels, 256955870806624479723addfae6dcf5d13968796cPeter Kasting size_t num_output_channels, 26325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org float* const* output) { 2791d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_EQ(num_input_channels, parent_->num_in_channels_); 2891d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_EQ(num_output_channels, parent_->num_out_channels_); 2991d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_EQ(parent_->block_length_, num_frames); 30325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 316955870806624479723addfae6dcf5d13968796cPeter Kasting for (size_t 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 38dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t block_length = RealFourier::ComplexLength( 39325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org RealFourier::FftOrder(num_frames)); 4091d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_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 476955870806624479723addfae6dcf5d13968796cPeter Kasting for (size_t 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 556955870806624479723addfae6dcf5d13968796cPeter KastingLappedTransform::LappedTransform(size_t num_in_channels, 566955870806624479723addfae6dcf5d13968796cPeter Kasting size_t num_out_channels, 57dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t chunk_length, 58cc84649389eff5ac5b25c7a19a5825abf8e6aa9eMichael Graczyk const float* window, 59dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t block_length, 60dce40cf804019a9898b6ab8d8262466b697c56e0Peter Kasting size_t 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) { 8691d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK(num_in_channels_ > 0 && num_out_channels_ > 0); 8791d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_GT(block_length_, 0u); 8891d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_GT(chunk_length_, 0u); 8991d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK(block_processor_); 90325cff01b42064313b221f2c8819ce01218f5ca4andrew@webrtc.org 91e534086492e92c45d74b176f3c8be4addb69713fmgraczyk@chromium.org // block_length_ power of 2? 9291d6edef35e7275879c30ce16ecb8b6dc73c6e4ahenrikg RTC_CHECK_EQ(0u, 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