OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #include "webrtc/common_audio/lapped_transform.h" | 11 #include "webrtc/modules/audio_processing/utility/lapped_transform.h" |
12 | 12 |
13 #include <algorithm> | 13 #include <algorithm> |
14 #include <cstdlib> | 14 #include <cstdlib> |
15 #include <cstring> | 15 #include <cstring> |
16 | 16 |
17 #include "webrtc/base/checks.h" | 17 #include "webrtc/base/checks.h" |
18 #include "webrtc/common_audio/real_fourier.h" | 18 #include "webrtc/common_audio/real_fourier.h" |
19 | 19 |
20 namespace webrtc { | 20 namespace webrtc { |
21 | 21 |
22 void LappedTransform::BlockThunk::ProcessBlock(const float* const* input, | 22 void LappedTransform::BlockThunk::ProcessBlock(const float* const* input, |
23 size_t num_frames, | 23 size_t num_frames, |
24 size_t num_input_channels, | 24 size_t num_input_channels, |
25 size_t num_output_channels, | 25 size_t num_output_channels, |
26 float* const* output) { | 26 float* const* output) { |
27 RTC_CHECK_EQ(num_input_channels, parent_->num_in_channels_); | 27 RTC_CHECK_EQ(num_input_channels, parent_->num_in_channels_); |
28 RTC_CHECK_EQ(num_output_channels, parent_->num_out_channels_); | 28 RTC_CHECK_EQ(num_output_channels, parent_->num_out_channels_); |
29 RTC_CHECK_EQ(parent_->block_length_, num_frames); | 29 RTC_CHECK_EQ(parent_->block_length_, num_frames); |
30 | 30 |
31 for (size_t i = 0; i < num_input_channels; ++i) { | 31 for (size_t i = 0; i < num_input_channels; ++i) { |
32 memcpy(parent_->real_buf_.Row(i), input[i], | 32 memcpy(parent_->real_buf_.Row(i), input[i], num_frames * sizeof(*input[0])); |
33 num_frames * sizeof(*input[0])); | |
34 parent_->fft_->Forward(parent_->real_buf_.Row(i), | 33 parent_->fft_->Forward(parent_->real_buf_.Row(i), |
35 parent_->cplx_pre_.Row(i)); | 34 parent_->cplx_pre_.Row(i)); |
36 } | 35 } |
37 | 36 |
38 size_t block_length = RealFourier::ComplexLength( | 37 size_t block_length = |
39 RealFourier::FftOrder(num_frames)); | 38 RealFourier::ComplexLength(RealFourier::FftOrder(num_frames)); |
40 RTC_CHECK_EQ(parent_->cplx_length_, block_length); | 39 RTC_CHECK_EQ(parent_->cplx_length_, block_length); |
41 parent_->block_processor_->ProcessAudioBlock(parent_->cplx_pre_.Array(), | 40 parent_->block_processor_->ProcessAudioBlock( |
42 num_input_channels, | 41 parent_->cplx_pre_.Array(), num_input_channels, parent_->cplx_length_, |
43 parent_->cplx_length_, | 42 num_output_channels, parent_->cplx_post_.Array()); |
44 num_output_channels, | |
45 parent_->cplx_post_.Array()); | |
46 | 43 |
47 for (size_t i = 0; i < num_output_channels; ++i) { | 44 for (size_t i = 0; i < num_output_channels; ++i) { |
48 parent_->fft_->Inverse(parent_->cplx_post_.Row(i), | 45 parent_->fft_->Inverse(parent_->cplx_post_.Row(i), |
49 parent_->real_buf_.Row(i)); | 46 parent_->real_buf_.Row(i)); |
50 memcpy(output[i], parent_->real_buf_.Row(i), | 47 memcpy(output[i], parent_->real_buf_.Row(i), |
51 num_frames * sizeof(*input[0])); | 48 num_frames * sizeof(*input[0])); |
52 } | 49 } |
53 } | 50 } |
54 | 51 |
55 LappedTransform::LappedTransform(size_t num_in_channels, | 52 LappedTransform::LappedTransform(size_t num_in_channels, |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
93 RTC_CHECK_EQ(0u, block_length_ & (block_length_ - 1)); | 90 RTC_CHECK_EQ(0u, block_length_ & (block_length_ - 1)); |
94 } | 91 } |
95 | 92 |
96 void LappedTransform::ProcessChunk(const float* const* in_chunk, | 93 void LappedTransform::ProcessChunk(const float* const* in_chunk, |
97 float* const* out_chunk) { | 94 float* const* out_chunk) { |
98 blocker_.ProcessChunk(in_chunk, chunk_length_, num_in_channels_, | 95 blocker_.ProcessChunk(in_chunk, chunk_length_, num_in_channels_, |
99 num_out_channels_, out_chunk); | 96 num_out_channels_, out_chunk); |
100 } | 97 } |
101 | 98 |
102 } // namespace webrtc | 99 } // namespace webrtc |
OLD | NEW |