1

我使用 fftwpp 将我的数据和卷积核都转换为傅立叶空间,将它们相乘,就像在标量积中一样,然后将它们转换回真实空间。当我第一次运行程序时,它会创建一个完全用零填充的数组。当我再次运行它时,它给了我想要的结果。

在运行它时,wisdom3.txt会创建一个。如果我删除它,程序需要很长时间才能再次创建一个零填充数组。

我的代码有什么问题?

// sx, sy and sz are the dimensions of my data

int szp = sz / 2 + 1;
size_t align = sizeof(Complex);

// creates arrays to store the data in, the double one is for the real data
// the Complex one for the fourier data
array3<double> f(sx, sy, sz, align);
array3<Complex> g(sx, sy, szp, align);

// copying data into double array
for(int k = 0; k < sz; k++)
    for(int j = 0; j < sy; j++)
        for(int i = 0; i < sx; i++)
            f(i, j, k) = data[i + sx * j + sx * sy * k];

// transforming data into fourier space
rcfft3d Forward3(sz, f, g);
Forward3.fft(f, g);


// generate the kernel
array3<double> kernel(sx, sy, sz);
array3<Complex> kernel2(sx, sy, szp, align);
// more code to create the kernel left out ...


// transform the kernel into the fourier space
rcfft3d ForwardKernel3(sz, kernel, kernel2);
ForwardKernel3.fft(kernel, kernel2);


// multiplying data and kernel in fourier space together
for(int k = 0; k < szp; k++)
    for(int j = 0; j < sy; j++)
        for(int i = 0; i < sx; i++)
            g(i, j, k) = g(i, j, k) * kernel2(i, j, k);


// transform back to normal space
crfft3d Backward3(sz, g, f);
Backward3.fftNormalized(g, f);


// putting everything in the results array and normalize
for(int k = 0; k < sz; k++)
    for(int j = 0; j < sy; j++)
        for(int i = 0; i < sx; i++)
            result[i + sx * j + sx * sy * k] =
                (f(i, j, k) >= thresholdValue ? f(i, j, k) : 0);
4

1 回答 1

3

FFTW++ 是 FFTW 的包装器。FFTW 需要生成一个执行计划来有效地处理您的数据。一旦生成了计划,就可以重复使用它来处理各种数据集。您看到的文件wisdom3.txt是有关您的程序在第一次运行时生成的计划的信息。一旦存在,它会在连续运行时加载,并允许您的程序快速运行。如果你删除它,FFTW 必须重新生成它,让你的程序再次运行缓慢。

至于为什么你的输出在第一次运行时为零,是因为这个计划生成步骤。FFTW 常见问题解答中对此进行了说明。

总而言之,由于我没有使用 FFTW++,我不知道您应该调用的确切方法来确保在执行 FT 之前生成计划。但是你应该在定义你的输入/输出数组之后和用你的数据初始化它们之前立即调用它。如果你需要多次运行你的代码,你应该保留智慧文件。

于 2011-03-23T20:49:04.407 回答