mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-25 19:25:43 +00:00
05a4ee9f4a
Setting up the FFT tables turns out to be quite expensive because of the large number of calls to sin() and cos(). In the future we may want to look into having a global cache of FFT tables per size, in order to pay the cost of setting each one of them up only once. --HG-- extra : rebase_source : 871d11e7f74db9e3e6c704da9518d40cba3deb8a
111 lines
2.7 KiB
C++
111 lines
2.7 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim:set ts=2 sw=2 sts=2 et cindent: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef FFTBlock_h_
|
|
#define FFTBlock_h_
|
|
|
|
#include "nsTArray.h"
|
|
#include "AudioNodeEngine.h"
|
|
#include "kiss_fft/kiss_fftr.h"
|
|
|
|
namespace mozilla {
|
|
|
|
// This class defines an FFT block, loosely modeled after Blink's FFTFrame
|
|
// class to make sharing code with Blink easy.
|
|
// Currently it's implemented on top of KissFFT on all platforms.
|
|
class FFTBlock {
|
|
public:
|
|
explicit FFTBlock(uint32_t aFFTSize)
|
|
: mFFT(nullptr)
|
|
, mIFFT(nullptr)
|
|
, mFFTSize(aFFTSize)
|
|
{
|
|
mOutputBuffer.SetLength(aFFTSize / 2 + 1);
|
|
PodZero(mOutputBuffer.Elements(), aFFTSize / 2 + 1);
|
|
}
|
|
~FFTBlock()
|
|
{
|
|
free(mFFT);
|
|
free(mIFFT);
|
|
}
|
|
|
|
void PerformFFT(const float* aData)
|
|
{
|
|
EnsureFFT();
|
|
kiss_fftr(mFFT, aData, mOutputBuffer.Elements());
|
|
}
|
|
void PerformInverseFFT(float* aData)
|
|
{
|
|
EnsureIFFT();
|
|
kiss_fftri(mIFFT, mOutputBuffer.Elements(), aData);
|
|
for (uint32_t i = 0; i < mFFTSize; ++i) {
|
|
aData[i] /= mFFTSize;
|
|
}
|
|
}
|
|
void Multiply(const FFTBlock& aFrame)
|
|
{
|
|
BufferComplexMultiply(reinterpret_cast<const float*>(mOutputBuffer.Elements()),
|
|
reinterpret_cast<const float*>(aFrame.mOutputBuffer.Elements()),
|
|
reinterpret_cast<float*>(mOutputBuffer.Elements()),
|
|
mFFTSize / 2 + 1);
|
|
}
|
|
|
|
void PerformPaddedFFT(const float* aData, size_t dataSize)
|
|
{
|
|
MOZ_ASSERT(dataSize <= FFTSize());
|
|
nsTArray<float> paddedData;
|
|
paddedData.SetLength(FFTSize());
|
|
PodCopy(paddedData.Elements(), aData, dataSize);
|
|
PodZero(paddedData.Elements() + dataSize, mFFTSize - dataSize);
|
|
PerformFFT(paddedData.Elements());
|
|
}
|
|
|
|
void SetFFTSize(uint32_t aSize)
|
|
{
|
|
mFFTSize = aSize;
|
|
mOutputBuffer.SetLength(aSize / 2 + 1);
|
|
PodZero(mOutputBuffer.Elements(), aSize / 2 + 1);
|
|
mFFT = mIFFT = nullptr;
|
|
}
|
|
|
|
uint32_t FFTSize() const
|
|
{
|
|
return mFFTSize;
|
|
}
|
|
float RealData(uint32_t aIndex) const
|
|
{
|
|
return mOutputBuffer[aIndex].r;
|
|
}
|
|
float ImagData(uint32_t aIndex) const
|
|
{
|
|
return mOutputBuffer[aIndex].i;
|
|
}
|
|
|
|
private:
|
|
void EnsureFFT()
|
|
{
|
|
if (!mFFT) {
|
|
mFFT = kiss_fftr_alloc(mFFTSize, 0, nullptr, nullptr);
|
|
}
|
|
}
|
|
void EnsureIFFT()
|
|
{
|
|
if (!mIFFT) {
|
|
mIFFT = kiss_fftr_alloc(mFFTSize, 1, nullptr, nullptr);
|
|
}
|
|
}
|
|
|
|
private:
|
|
kiss_fftr_cfg mFFT, mIFFT;
|
|
nsTArray<kiss_fft_cpx> mOutputBuffer;
|
|
uint32_t mFFTSize;
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|
|
|