Question

我正在尝试使用FFTW后端使用Eigen不支持的FFT库。具体来说，我想做一个2D FFT。这是我的代码：

void fft2(Eigen::MatrixXf * matIn,Eigen::MatrixXcf * matOut)
{
    const int nRows = matIn->rows();
    const int nCols = matIn->cols();

    Eigen::FFT< float > fft;

    for (int k = 0; k < nRows; ++k) {
        Eigen::VectorXcf tmpOut(nRows);
        fft.fwd(tmpOut, matIn->row(k));
        matOut->row(k) = tmpOut;
    }

    for (int k = 0; k < nCols; ++k) {
        Eigen::VectorXcf tmpOut(nCols);
        fft.fwd(tmpOut, matOut->col(k));
        matOut->col(k) = tmpOut;
    }

}

我有两个问题：

首先，在某些矩阵上使用此代码时出现分段错误。所有矩阵都不会发生此错误。我猜它与对齐错误有关。我按以下方式使用这些函数：

Eigen :: MatrixXcf matFFT（mat.rows（），mat.cols（））; FFT2（安培; matFloat，＆安培; matFFT）;

其中mat可以是任何矩阵。有趣的是，代码仅在我在第二维上计算FFT时才会种植，而从不在第一维上。 kissFFT后端不会发生这种情况。

当函数工作时，我没有得到与Matlab（使用FFTW）相同的结果。例如：

输入矩阵：

[2, 1, 2]
[3, 2, 1]
[1, 2, 3]

Eigen给出：

[           (0,5),    (0.5,0.86603),          (0,0.5)]
[  (-4.3301,-2.5),     (-1,-1.7321), (0.31699,-1.549)]
[ (-1.5,-0.86603),       (2,3.4641),       (2,3.4641)]

Matlab给出：

   17 +          0i          0.5 +    0.86603i          0.5 -    0.86603i
   -1 +          0i           -1 -     1.7321i            2 -     3.4641i
   -1 +          0i            2 +     3.4641i           -1 +     1.7321i

只有中心部分是相同的。

欢迎任何帮助。

Answer 1

我在第一个解决方案中未能激活EIGEN_FFTW_DEFAULT，激活它会在Eigen的fftw-support实现中显示错误。以下作品：

#define EIGEN_FFTW_DEFAULT
#include <iostream>
#include <unsupported/Eigen/FFT>

int main(int argc, char *argv[])
{
    Eigen::MatrixXf A(3,3);
    A << 2,1,2,  3,2,1,  1,2,3;
    const int nRows = A.rows();
    const int nCols = A.cols();

    std::cout << A << "\n\n";

    Eigen::MatrixXcf B(3,3);

    Eigen::FFT< float > fft;

    for (int k = 0; k < nRows; ++k) {
        Eigen::VectorXcf tmpOut(nRows);
        fft.fwd(tmpOut, A.row(k));
        B.row(k) = tmpOut;
    }
    std::cout << B << "\n\n";
    Eigen::FFT< float > fft2;  // Workaround: Using the same FFT object for a real and a complex FFT seems not to work with FFTW
    for (int k = 0; k < nCols; ++k) {
        Eigen::VectorXcf tmpOut(nCols);
        fft2.fwd(tmpOut, B.col(k));
        B.col(k) = tmpOut;
    }
    std::cout << B << '\n';
}

我得到了这个输出：

2 1 2
3 2 1
1 2 3

     (17,0)  (0.5,0.866025) (0.5,-0.866025)
     (-1,0)   (-1,-1.73205)     (2,-3.4641)
     (-1,0)      (2,3.4641)    (-1,1.73205)

这与您的Matlab结果相同。

N.B。：FFTW似乎本身支持2D实 - >复FFT（不使用单独的FFT）。这可能更有效率。

fftwf_plan fftwf_plan_dft_r2c_2d(int n0, int n1,               
                                 float *in, fftwf_complex *out, unsigned flags);

特征FFT库

1 个答案: