Home | History | Annotate | Download | only in common_audio 
      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 
     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 
     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 
     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:
     60   RealFourierTest()
     61       : rf_(2),
     62         real_buffer_(RealFourier::AllocRealBuffer(4)),
     63         cplx_buffer_(RealFourier::AllocCplxBuffer(3)) {}
     64 
     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 
     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 
     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