gpt4 book ai didi

c++ - 如何正确使用带有 C++ vector 的 kissFFT?

转载 作者:行者123 更新时间:2023-11-28 06:05:49 25 4
gpt4 key购买 nike

我想在 C++ 项目中使用 kissFFT std::vector轻松处理任意长度的信号。为此,我编辑了 this SO answer来自 kissFFT 的作者。我换了

std::complex<float> x[nfft];

vector<std::complex<float>> x(nfft, 0.0);

并替换了函数调用

kiss_fft(fwd,(kiss_fft_cpx*)x,(kiss_fft_cpx*)fx)

kiss_fft(fwd,(kiss_fft_cpx*)&x[0],(kiss_fft_cpx*)&fx[0]);

这似乎工作得很好,但这是正确的方法吗?在 kissFFT 的“test”文件夹中,我发现了“testcpp.cc”,它似乎使用了一些花哨的 C++ 模板功能(不幸的是,对我来说太花哨了;我无法编译它)。我的解决方案有什么缺点吗?


完整代码

#include "kiss_fft.h"
#include <complex>
#include <iostream>
#include <vector>
using namespace std;

int main()
{
const int nfft=256;
kiss_fft_cfg fwd = kiss_fft_alloc(nfft,0,NULL,NULL);
kiss_fft_cfg inv = kiss_fft_alloc(nfft,1,NULL,NULL);

vector<std::complex<float>> x(nfft, 0.0);
vector<std::complex<float>> fx(nfft, 0.0);

x[0] = 1;
x[1] = std::complex<float>(0,3);

kiss_fft(fwd,(kiss_fft_cpx*)&x[0],(kiss_fft_cpx*)&fx[0]);
for (int k=0;k<nfft;++k) {
fx[k] = fx[k] * conj(fx[k]);
fx[k] *= 1./nfft;
}
kiss_fft(inv,(kiss_fft_cpx*)&fx[0],(kiss_fft_cpx*)&x[0]);
cout << "the circular correlation of [1, 3i, 0 0 ....] with itself = ";
cout
<< x[0] << ","
<< x[1] << ","
<< x[2] << ","
<< x[3] << " ... " << endl;
kiss_fft_free(fwd);
kiss_fft_free(inv);
return 0;
}

最佳答案

我倾向于使用 std::vector<kiss_fft_cpx>我自己,但除此之外,我的代码与您的代码几乎相同。 (嗯,kiss_fft_allockiss_fft_free 进入 ctor/dtor 以避免内存泄漏,但这是一种风格。)

再往下一点,fx[k] * conj(fx[k])是一个典型的数学表达式。使用fx[k].norm()效率更高.

关于c++ - 如何正确使用带有 C++ vector 的 kissFFT?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/32443748/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com