blob: dec2be6d60b8e70e63f82b6362051e9b99fa626b [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/real_fourier.h"
12
andrew@webrtc.org325cff02014-10-01 17:42:18 +000013#include "webrtc/base/checks.h"
andrew@webrtc.org04c50982015-03-19 20:06:29 +000014#include "webrtc/common_audio/real_fourier_ooura.h"
15#include "webrtc/common_audio/real_fourier_openmax.h"
16#include "webrtc/common_audio/signal_processing/include/spl_inl.h"
andrew@webrtc.org325cff02014-10-01 17:42:18 +000017
18namespace webrtc {
19
20using std::complex;
21
andrew@webrtc.org325cff02014-10-01 17:42:18 +000022const int RealFourier::kFftBufferAlignment = 32;
23
andrew@webrtc.org04c50982015-03-19 20:06:29 +000024rtc::scoped_ptr<RealFourier> RealFourier::Create(int fft_order) {
25#if defined(RTC_USE_OPENMAX_DL)
26 return rtc::scoped_ptr<RealFourier>(new RealFourierOpenmax(fft_order));
27#else
28 return rtc::scoped_ptr<RealFourier>(new RealFourierOoura(fft_order));
29#endif
andrew@webrtc.org325cff02014-10-01 17:42:18 +000030}
31
32int RealFourier::FftOrder(int length) {
andrew@webrtc.org04c50982015-03-19 20:06:29 +000033 CHECK_GT(length, 0);
34 return WebRtcSpl_GetSizeInBits(length - 1);
35}
36
37int RealFourier::FftLength(int order) {
38 CHECK_GE(order, 0);
39 return 1 << order;
andrew@webrtc.org325cff02014-10-01 17:42:18 +000040}
41
42int RealFourier::ComplexLength(int order) {
andrew@webrtc.org04c50982015-03-19 20:06:29 +000043 CHECK_GE(order, 0);
andrew@webrtc.org325cff02014-10-01 17:42:18 +000044 return (1 << order) / 2 + 1;
45}
46
47RealFourier::fft_real_scoper RealFourier::AllocRealBuffer(int count) {
48 return fft_real_scoper(static_cast<float*>(
49 AlignedMalloc(sizeof(float) * count, kFftBufferAlignment)));
50}
51
52RealFourier::fft_cplx_scoper RealFourier::AllocCplxBuffer(int count) {
53 return fft_cplx_scoper(static_cast<complex<float>*>(
54 AlignedMalloc(sizeof(complex<float>) * count, kFftBufferAlignment)));
55}
56
andrew@webrtc.org325cff02014-10-01 17:42:18 +000057} // namespace webrtc
58