1 /*
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/real_fourier.h"
12
13 #include <stdlib.h>
14
15 #include "testing/gtest/include/gtest/gtest.h"
16 #include "webrtc/base/scoped_ptr.h"
17 #include "webrtc/common_audio/real_fourier_openmax.h"
18 #include "webrtc/common_audio/real_fourier_ooura.h"
19
20 namespace webrtc {
21
22 using std::complex;
23
TEST(RealFourierStaticsTest,AllocatorAlignment)24 TEST(RealFourierStaticsTest, AllocatorAlignment) {
25 {
26 RealFourier::fft_real_scoper real;
27 real = RealFourier::AllocRealBuffer(3);
28 ASSERT_TRUE(real.get() != nullptr);
29 uintptr_t ptr_value = reinterpret_cast<uintptr_t>(real.get());
30 EXPECT_EQ(0u, ptr_value % RealFourier::kFftBufferAlignment);
31 }
32 {
33 RealFourier::fft_cplx_scoper cplx;
34 cplx = RealFourier::AllocCplxBuffer(3);
35 ASSERT_TRUE(cplx.get() != nullptr);
36 uintptr_t ptr_value = reinterpret_cast<uintptr_t>(cplx.get());
37 EXPECT_EQ(0u, ptr_value % RealFourier::kFftBufferAlignment);
38 }
39 }
40
TEST(RealFourierStaticsTest,OrderComputation)41 TEST(RealFourierStaticsTest, OrderComputation) {
42 EXPECT_EQ(4, RealFourier::FftOrder(13));
43 EXPECT_EQ(5, RealFourier::FftOrder(32));
44 EXPECT_EQ(1, RealFourier::FftOrder(2));
45 EXPECT_EQ(0, RealFourier::FftOrder(1));
46 }
47
TEST(RealFourierStaticsTest,ComplexLengthComputation)48 TEST(RealFourierStaticsTest, ComplexLengthComputation) {
49 EXPECT_EQ(2U, RealFourier::ComplexLength(1));
50 EXPECT_EQ(3U, RealFourier::ComplexLength(2));
51 EXPECT_EQ(5U, RealFourier::ComplexLength(3));
52 EXPECT_EQ(9U, RealFourier::ComplexLength(4));
53 EXPECT_EQ(17U, RealFourier::ComplexLength(5));
54 EXPECT_EQ(65U, RealFourier::ComplexLength(7));
55 }
56
57 template <typename T>
58 class RealFourierTest : public ::testing::Test {
59 protected:
RealFourierTest()60 RealFourierTest()
61 : rf_(2),
62 real_buffer_(RealFourier::AllocRealBuffer(4)),
63 cplx_buffer_(RealFourier::AllocCplxBuffer(3)) {}
64
~RealFourierTest()65 ~RealFourierTest() {
66 }
67
68 T rf_;
69 const RealFourier::fft_real_scoper real_buffer_;
70 const RealFourier::fft_cplx_scoper cplx_buffer_;
71 };
72
73 using FftTypes = ::testing::Types<
74 #if defined(RTC_USE_OPENMAX_DL)
75 RealFourierOpenmax,
76 #endif
77 RealFourierOoura>;
78 TYPED_TEST_CASE(RealFourierTest, FftTypes);
79
TYPED_TEST(RealFourierTest,SimpleForwardTransform)80 TYPED_TEST(RealFourierTest, SimpleForwardTransform) {
81 this->real_buffer_[0] = 1.0f;
82 this->real_buffer_[1] = 2.0f;
83 this->real_buffer_[2] = 3.0f;
84 this->real_buffer_[3] = 4.0f;
85
86 this->rf_.Forward(this->real_buffer_.get(), this->cplx_buffer_.get());
87
88 EXPECT_NEAR(this->cplx_buffer_[0].real(), 10.0f, 1e-8f);
89 EXPECT_NEAR(this->cplx_buffer_[0].imag(), 0.0f, 1e-8f);
90 EXPECT_NEAR(this->cplx_buffer_[1].real(), -2.0f, 1e-8f);
91 EXPECT_NEAR(this->cplx_buffer_[1].imag(), 2.0f, 1e-8f);
92 EXPECT_NEAR(this->cplx_buffer_[2].real(), -2.0f, 1e-8f);
93 EXPECT_NEAR(this->cplx_buffer_[2].imag(), 0.0f, 1e-8f);
94 }
95
TYPED_TEST(RealFourierTest,SimpleBackwardTransform)96 TYPED_TEST(RealFourierTest, SimpleBackwardTransform) {
97 this->cplx_buffer_[0] = complex<float>(10.0f, 0.0f);
98 this->cplx_buffer_[1] = complex<float>(-2.0f, 2.0f);
99 this->cplx_buffer_[2] = complex<float>(-2.0f, 0.0f);
100
101 this->rf_.Inverse(this->cplx_buffer_.get(), this->real_buffer_.get());
102
103 EXPECT_NEAR(this->real_buffer_[0], 1.0f, 1e-8f);
104 EXPECT_NEAR(this->real_buffer_[1], 2.0f, 1e-8f);
105 EXPECT_NEAR(this->real_buffer_[2], 3.0f, 1e-8f);
106 EXPECT_NEAR(this->real_buffer_[3], 4.0f, 1e-8f);
107 }
108
109 } // namespace webrtc
110
111