blob: e721346f31e6bc2a4a2025083db37a7781acedc8 [file] [log] [blame]
andrew@webrtc.org325cff02014-10-01 17:42:181/*
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:1813#include "webrtc/base/checks.h"
andrew@webrtc.org04c50982015-03-19 20:06:2914#include "webrtc/common_audio/real_fourier_ooura.h"
15#include "webrtc/common_audio/real_fourier_openmax.h"
kwiberg729b21f2016-06-02 11:02:1216#include "webrtc/common_audio/signal_processing/include/signal_processing_library.h"
andrew@webrtc.org325cff02014-10-01 17:42:1817
18namespace webrtc {
19
20using std::complex;
21
pkasting25702cb2016-01-08 21:50:2722const size_t RealFourier::kFftBufferAlignment = 32;
andrew@webrtc.org325cff02014-10-01 17:42:1823
kwibergbfefb032016-05-01 21:53:4624std::unique_ptr<RealFourier> RealFourier::Create(int fft_order) {
andrew@webrtc.org04c50982015-03-19 20:06:2925#if defined(RTC_USE_OPENMAX_DL)
kwibergbfefb032016-05-01 21:53:4626 return std::unique_ptr<RealFourier>(new RealFourierOpenmax(fft_order));
andrew@webrtc.org04c50982015-03-19 20:06:2927#else
kwibergbfefb032016-05-01 21:53:4628 return std::unique_ptr<RealFourier>(new RealFourierOoura(fft_order));
andrew@webrtc.org04c50982015-03-19 20:06:2929#endif
andrew@webrtc.org325cff02014-10-01 17:42:1830}
31
Peter Kastingdce40cf2015-08-24 21:52:2332int RealFourier::FftOrder(size_t length) {
henrikg91d6ede2015-09-17 07:24:3433 RTC_CHECK_GT(length, 0U);
Peter Kastingb7e50542015-06-11 19:55:5034 return WebRtcSpl_GetSizeInBits(static_cast<uint32_t>(length - 1));
andrew@webrtc.org04c50982015-03-19 20:06:2935}
36
Peter Kastingdce40cf2015-08-24 21:52:2337size_t RealFourier::FftLength(int order) {
henrikg91d6ede2015-09-17 07:24:3438 RTC_CHECK_GE(order, 0);
Peter Kastingdce40cf2015-08-24 21:52:2339 return static_cast<size_t>(1 << order);
andrew@webrtc.org325cff02014-10-01 17:42:1840}
41
Peter Kastingdce40cf2015-08-24 21:52:2342size_t RealFourier::ComplexLength(int order) {
Peter Kastingf045e4d2015-06-11 04:15:3843 return FftLength(order) / 2 + 1;
andrew@webrtc.org325cff02014-10-01 17:42:1844}
45
46RealFourier::fft_real_scoper RealFourier::AllocRealBuffer(int count) {
47 return fft_real_scoper(static_cast<float*>(
48 AlignedMalloc(sizeof(float) * count, kFftBufferAlignment)));
49}
50
51RealFourier::fft_cplx_scoper RealFourier::AllocCplxBuffer(int count) {
52 return fft_cplx_scoper(static_cast<complex<float>*>(
53 AlignedMalloc(sizeof(complex<float>) * count, kFftBufferAlignment)));
54}
55
andrew@webrtc.org325cff02014-10-01 17:42:1856} // namespace webrtc
57