1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
/* 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];
}
|
