FFTW复数数组映射为Eigen复数矩阵时构造报错问题求解

问题背景 #

在C++项目中同时使用Eigen与FFTW库时，由于Eigen自带FFT属于非官方支持模块，缺少部分高级特性，因此选择直接调用FFTW库做傅里叶变换。需求是将FFTW输出的复数数组直接映射为Eigen复数矩阵，避免数据拷贝，方便后续线性代数运算。

原始实现代码如下：

#include <Eigen/Dense>
#include <fftw3.h>

struct cplx_buffer
{
  fftw_complex* a;
  int rows;
  int cols;

  fftw_complex& operator()(int i, int j) const { return a[i * cols + j]; }
};
struct real_buffer
{
  double* a;
  int rows;
  int cols;

  double& operator()(int i, int j) const { return a[i * cols + j]; }
};

// 假设已实现FFTW内存分配函数
cplx_buffer my_fftw_allocate_cplx(int rows, int cols);
real_buffer my_fftw_allocate_real(int rows, int cols);

int main(){ 
    static const int nx = 10;
    static const int ny = 10; 
    static const int nyk = ny/2 + 1;
    static const int mm = nx* 3/2;

    cplx_buffer outW = my_fftw_allocate_cplx((ny+1), mm); 
    real_buffer inW = my_fftw_allocate_real((ny+1), mm);

    //初始化输入
    for (int i = 0; i < ny+1; i++){ 
        for (int j = 0; j < mm; j++){ 
            inW(i,j) = 0; // 替换为实际赋值表达式
        }
    }
    // 对每行做一维实序列FFT
    fftw_execute(fftw_plan_many_dft_r2c(
        1, &nx, inW.rows, 
        inW.a, &inW.cols, 1, inW.cols, 
        outW.a, &inW.cols, 1, outW.cols, 
        FFTW_ESTIMATE
    ));

    // 此处映射代码编译报错
    Eigen::Map<Eigen::MatrixXcd, Eigen::Unaligned> invnek(*reinterpret_cast<fftw_complex*>(&outW),(ny+1),mm);
    return 0;
}

编译抛出如下错误：

no instance of constructor "Eigen::Map<PlainObjectType, MapOptions, StrideType>::Map [with PlainObjectType=Eigen::MatrixXcd, MapOptions=0, StrideType=Eigen::Stride<0, 0>]" matches the argument list

错误原因 #

• 传入Eigen::Map构造函数的指针完全错误：代码对cplx_buffer类型的结构体实例outW取地址强转，传入的是结构体对象本身的地址，而非结构体内部存储FFTW输出数据的数组成员a的地址，参数类型和Eigen::Map要求的元素指针类型不匹配，直接触发构造函数找不到的编译错误。
• 维度参数错误：调用fftw_plan_many_dft_r2c做一维实序列FFT时，每行输出的有效复数长度为nx/2 + 1，而非输入实序列的列数mm，传入错误维度会导致后续内存访问越界。
• 对齐参数多余：FFTW分配的内存默认满足SIMD对齐要求，不需要指定Eigen::Unaligned参数，指定未对齐反而会损失运算性能。

正确实现 #

fftw_complex类型本质是长度为2的double数组（依次存储实部、虚部），和C++标准的std::complex<double>（即Eigen::MatrixXcd的元素类型）内存布局完全兼容，可以直接安全转换指针。

修正后的映射代码如下：

// r2c变换每行输出有效复数长度
const int fft_out_cols = nx / 2 + 1;
// 直接取缓冲区结构体内部的数组首地址做转换，映射为Eigen矩阵
Eigen::Map<Eigen::MatrixXcd> invnek(
    reinterpret_cast<std::complex<double>*>(outW.a),
    ny + 1,
    fft_out_cols
);

注意事项 #

• 映射后的invnek是Eigen的Map对象，不会拷贝数据，直接操作FFTW分配的原始内存，注意内存生命周期不要提前释放FFTW缓冲区。
• 如果后续需要把映射的矩阵赋值给普通Eigen::MatrixXcd对象做独立运算，直接赋值即可，赋值操作会自动完成数据拷贝：Eigen::MatrixXcd mat = invnek;
• 只有当映射的内存不是按16/32字节对齐分配时，才需要添加Eigen::Unaligned模板参数，FFTW官方分配接口返回的内存默认是对齐的，无需添加该参数。

内容的提问来源于stack exchange，提问作者Jamie