325cff01b4
Add code for doing block-based frequency domain processing. Developed and reviewed in isolation. Corresponding export CL: https://chromereviews.googleplex.com/95187013/ R=bercic@google.com, kjellander@webrtc.org, turaj@webrtc.org Review URL: https://webrtc-codereview.appspot.com/31539004 git-svn-id: http://webrtc.googlecode.com/svn/trunk@7359 4adac7df-926f-26a2-2b94-8c16560cd09d
89 lines
2.4 KiB
C++
89 lines
2.4 KiB
C++
/*
|
|
* Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license
|
|
* that can be found in the LICENSE file in the root of the source
|
|
* tree. An additional intellectual property rights grant can be found
|
|
* in the file PATENTS. All contributing project authors may
|
|
* be found in the AUTHORS file in the root of the source tree.
|
|
*/
|
|
|
|
#include "webrtc/common_audio/real_fourier.h"
|
|
|
|
#include <cstdlib>
|
|
|
|
#include "third_party/openmax_dl/dl/sp/api/omxSP.h"
|
|
#include "webrtc/base/checks.h"
|
|
|
|
namespace webrtc {
|
|
|
|
using std::complex;
|
|
|
|
// The omx implementation uses this macro to check order validity.
|
|
const int RealFourier::kMaxFftOrder = TWIDDLE_TABLE_ORDER;
|
|
const int RealFourier::kFftBufferAlignment = 32;
|
|
|
|
RealFourier::RealFourier(int fft_order)
|
|
: order_(fft_order),
|
|
omx_spec_(nullptr) {
|
|
CHECK_GE(order_, 1);
|
|
CHECK_LE(order_, kMaxFftOrder);
|
|
|
|
OMX_INT buffer_size;
|
|
OMXResult r;
|
|
|
|
r = omxSP_FFTGetBufSize_R_F32(order_, &buffer_size);
|
|
CHECK_EQ(r, OMX_Sts_NoErr);
|
|
|
|
omx_spec_ = malloc(buffer_size);
|
|
DCHECK(omx_spec_);
|
|
|
|
r = omxSP_FFTInit_R_F32(omx_spec_, order_);
|
|
CHECK_EQ(r, OMX_Sts_NoErr);
|
|
}
|
|
|
|
RealFourier::~RealFourier() {
|
|
free(omx_spec_);
|
|
}
|
|
|
|
int RealFourier::FftOrder(int length) {
|
|
for (int order = 0; order <= kMaxFftOrder; order++) {
|
|
if ((1 << order) >= length) {
|
|
return order;
|
|
}
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
int RealFourier::ComplexLength(int order) {
|
|
CHECK_LE(order, kMaxFftOrder);
|
|
CHECK_GT(order, 0);
|
|
return (1 << order) / 2 + 1;
|
|
}
|
|
|
|
RealFourier::fft_real_scoper RealFourier::AllocRealBuffer(int count) {
|
|
return fft_real_scoper(static_cast<float*>(
|
|
AlignedMalloc(sizeof(float) * count, kFftBufferAlignment)));
|
|
}
|
|
|
|
RealFourier::fft_cplx_scoper RealFourier::AllocCplxBuffer(int count) {
|
|
return fft_cplx_scoper(static_cast<complex<float>*>(
|
|
AlignedMalloc(sizeof(complex<float>) * count, kFftBufferAlignment)));
|
|
}
|
|
|
|
void RealFourier::Forward(const float* src, complex<float>* dest) const {
|
|
OMXResult r;
|
|
r = omxSP_FFTFwd_RToCCS_F32(src, reinterpret_cast<OMX_F32*>(dest), omx_spec_);
|
|
CHECK_EQ(r, OMX_Sts_NoErr);
|
|
}
|
|
|
|
void RealFourier::Inverse(const complex<float>* src, float* dest) const {
|
|
OMXResult r;
|
|
r = omxSP_FFTInv_CCSToR_F32(reinterpret_cast<const OMX_F32*>(src), dest,
|
|
omx_spec_);
|
|
CHECK_EQ(r, OMX_Sts_NoErr);
|
|
}
|
|
|
|
} // namespace webrtc
|
|
|