blob: 6b4f135affffc95d401d9e9c70f423fe982a8bb0 [file] [log] [blame]
andrew@webrtc.org325cff02014-10-01 17:42:18 +00001/*
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
3 *
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
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11#include "webrtc/common_audio/lapped_transform.h"
12
mgraczyk@chromium.orge5340862015-03-12 23:23:38 +000013#include <algorithm>
andrew@webrtc.org325cff02014-10-01 17:42:18 +000014#include <cstdlib>
15#include <cstring>
16
17#include "webrtc/base/checks.h"
18#include "webrtc/common_audio/real_fourier.h"
19
20namespace webrtc {
21
22void LappedTransform::BlockThunk::ProcessBlock(const float* const* input,
23 int num_frames,
24 int num_input_channels,
25 int num_output_channels,
26 float* const* output) {
27 CHECK_EQ(num_input_channels, parent_->in_channels_);
28 CHECK_EQ(num_output_channels, parent_->out_channels_);
29 CHECK_EQ(parent_->block_length_, num_frames);
30
31 for (int i = 0; i < num_input_channels; ++i) {
32 memcpy(parent_->real_buf_.Row(i), input[i],
33 num_frames * sizeof(*input[0]));
34 parent_->fft_.Forward(parent_->real_buf_.Row(i), parent_->cplx_pre_.Row(i));
35 }
36
37 int block_length = RealFourier::ComplexLength(
38 RealFourier::FftOrder(num_frames));
39 CHECK_EQ(parent_->cplx_length_, block_length);
40 parent_->block_processor_->ProcessAudioBlock(parent_->cplx_pre_.Array(),
41 num_input_channels,
42 parent_->cplx_length_,
43 num_output_channels,
44 parent_->cplx_post_.Array());
45
46 for (int i = 0; i < num_output_channels; ++i) {
47 parent_->fft_.Inverse(parent_->cplx_post_.Row(i),
48 parent_->real_buf_.Row(i));
49 memcpy(output[i], parent_->real_buf_.Row(i),
50 num_frames * sizeof(*input[0]));
51 }
52}
53
54LappedTransform::LappedTransform(int in_channels, int out_channels,
55 int chunk_length, const float* window,
56 int block_length, int shift_amount,
57 Callback* callback)
58 : blocker_callback_(this),
59 in_channels_(in_channels),
60 out_channels_(out_channels),
andrew@webrtc.org325cff02014-10-01 17:42:18 +000061 block_length_(block_length),
62 chunk_length_(chunk_length),
63 block_processor_(callback),
mgraczyk@chromium.orge5340862015-03-12 23:23:38 +000064 blocker_(
65 chunk_length_, block_length_, in_channels_, out_channels_, window,
66 shift_amount, &blocker_callback_),
andrew@webrtc.org325cff02014-10-01 17:42:18 +000067 fft_(RealFourier::FftOrder(block_length_)),
68 cplx_length_(RealFourier::ComplexLength(fft_.order())),
mgraczyk@chromium.orge5340862015-03-12 23:23:38 +000069 real_buf_(in_channels, block_length_, RealFourier::kFftBufferAlignment),
andrew@webrtc.org325cff02014-10-01 17:42:18 +000070 cplx_pre_(in_channels, cplx_length_, RealFourier::kFftBufferAlignment),
71 cplx_post_(out_channels, cplx_length_, RealFourier::kFftBufferAlignment) {
72 CHECK(in_channels_ > 0 && out_channels_ > 0);
73 CHECK_GT(block_length_, 0);
74 CHECK_GT(chunk_length_, 0);
75 CHECK(block_processor_);
andrew@webrtc.org325cff02014-10-01 17:42:18 +000076
mgraczyk@chromium.orge5340862015-03-12 23:23:38 +000077 // block_length_ power of 2?
78 CHECK_EQ(0, block_length_ & (block_length_ - 1));
andrew@webrtc.org325cff02014-10-01 17:42:18 +000079}
80
81void LappedTransform::ProcessChunk(const float* const* in_chunk,
82 float* const* out_chunk) {
mgraczyk@chromium.orge5340862015-03-12 23:23:38 +000083 blocker_.ProcessChunk(in_chunk, chunk_length_, in_channels_, out_channels_,
andrew@webrtc.org325cff02014-10-01 17:42:18 +000084 out_chunk);
85}
86
87} // namespace webrtc
88