1 /*
2 * Copyright (c) 2015 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_openmax.h"
12
13 #include <cstdlib>
14
15 #include "dl/sp/api/omxSP.h"
16 #include "webrtc/base/checks.h"
17
18 namespace webrtc {
19
20 using std::complex;
21
22 namespace {
23
24 // Creates and initializes the Openmax state. Transfers ownership to caller.
CreateOpenmaxState(int order)25 OMXFFTSpec_R_F32* CreateOpenmaxState(int order) {
26 RTC_CHECK_GE(order, 1);
27 // The omx implementation uses this macro to check order validity.
28 RTC_CHECK_LE(order, TWIDDLE_TABLE_ORDER);
29
30 OMX_INT buffer_size;
31 OMXResult r = omxSP_FFTGetBufSize_R_F32(order, &buffer_size);
32 RTC_CHECK_EQ(r, OMX_Sts_NoErr);
33
34 OMXFFTSpec_R_F32* omx_spec = malloc(buffer_size);
35 RTC_DCHECK(omx_spec);
36
37 r = omxSP_FFTInit_R_F32(omx_spec, order);
38 RTC_CHECK_EQ(r, OMX_Sts_NoErr);
39 return omx_spec;
40 }
41
42 } // namespace
43
RealFourierOpenmax(int fft_order)44 RealFourierOpenmax::RealFourierOpenmax(int fft_order)
45 : order_(fft_order),
46 omx_spec_(CreateOpenmaxState(order_)) {
47 }
48
~RealFourierOpenmax()49 RealFourierOpenmax::~RealFourierOpenmax() {
50 free(omx_spec_);
51 }
52
Forward(const float * src,complex<float> * dest) const53 void RealFourierOpenmax::Forward(const float* src, complex<float>* dest) const {
54 // This cast is well-defined since C++11. See "Non-static data members" at:
55 // http://en.cppreference.com/w/cpp/numeric/complex
56 OMXResult r =
57 omxSP_FFTFwd_RToCCS_F32(src, reinterpret_cast<OMX_F32*>(dest), omx_spec_);
58 RTC_CHECK_EQ(r, OMX_Sts_NoErr);
59 }
60
Inverse(const complex<float> * src,float * dest) const61 void RealFourierOpenmax::Inverse(const complex<float>* src, float* dest) const {
62 OMXResult r =
63 omxSP_FFTInv_CCSToR_F32(reinterpret_cast<const OMX_F32*>(src), dest,
64 omx_spec_);
65 RTC_CHECK_EQ(r, OMX_Sts_NoErr);
66 }
67
68 } // namespace webrtc
69
70