PowerSpectrumGetter.cpp

Go to the documentation of this file.

/*  SPDX-License-Identifier: GPL-2.0-or-later */
/*!********************************************************************
 
  Audacity: A Digital Audio Editor
 
  PowerSpectrumGetter.cpp
 
  Matthieu Hodgkinson
 
**********************************************************************/
#include "PowerSpectrumGetter.h"
 
#include <cassert>
#include <pffft.h>
 
void PffftSetupDeleter::Pffft_destroy_setup(PFFFT_Setup *p)
{
   pffft_destroy_setup(p);
}
 
void *PffftAllocatorBase::Pffft_aligned_malloc(size_t nb_bytes)
{
   return pffft_aligned_malloc(nb_bytes);
}
 
void PffftAllocatorBase::Pffft_aligned_free(void *p)
{
   pffft_aligned_free(p);
}
 
PffftFloats PffftFloatVector::aligned(PffftAlignedCount c)
{
   return PffftFloats{ data() + c };
}
 
PffftConstFloats PffftFloatVector::aligned(PffftAlignedCount c) const
{
   return PffftConstFloats{ data() + c };
}
 
PowerSpectrumGetter::PowerSpectrumGetter(int fftSize)
    : mFftSize { fftSize }
    , mSetup { pffft_new_setup(fftSize, PFFFT_REAL) }
    , mWork(fftSize)
{
}
 
PowerSpectrumGetter::~PowerSpectrumGetter()
{
}
 
void PowerSpectrumGetter::operator()(
   PffftFloats alignedBuffer, PffftFloats alignedOutput)
{
   const auto buffer = alignedBuffer.get();
   const auto output = alignedOutput.get();
   pffft_transform_ordered(mSetup.get(),
      buffer, buffer, mWork.data(), PFFFT_FORWARD);
   output[0] = buffer[0] * buffer[0];
   for (auto i = 1; i < mFftSize / 2; ++i)
      output[i] =
         buffer[i * 2] * buffer[i * 2] + buffer[i * 2 + 1] * buffer[i * 2 + 1];
   output[mFftSize / 2] = buffer[1] * buffer[1];
}