我使用 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);