如何正确初始化r2cfft_t结构体并传入指定函数？

首先先修正你提供的r2cfft_initialize函数里的几个错误：

1. 用memset(ctx, 42, sizeof(r2cfft_t))完全错误，42是随机值，会把指针成员搞成野指针，应该用memset(ctx, 0, sizeof(r2cfft_t))清零结构体，避免未定义行为。
2. buffer_out分配时的nyk是笔误，应该改成ny。
3. 原初始化失败的判断逻辑!((ctx->nx = nx) && (ctx->ny = ny))可以调整为直接检查nx/ny是否为0，逻辑更清晰。

接下来讲两种正确初始化r2cfft_t结构体并传入函数的方式：

方式1：栈上分配结构体 #

栈上分配是最简单的方式，结构体在栈上创建，不需要手动释放结构体本身，适合在函数内部短期使用：

#include <cstring>
#include <cstdio>
#include <fftw3.h>

static const int nx = 128;
static const int ny = 128;
static const int ncomp = 2;

typedef struct r2cfft_t
{
    std::size_t nx, ny;
    fftw_plan plan;
    double *buffer_in;
    fftw_complex *buffer_out;
}r2cfft_t;

int r2cfft_initialize(r2cfft_t *const ctx, const std::size_t nx, const std::size_t ny);

int main(){ 
    // 栈上声明结构体并清零，避免野指针
    r2cfft_t fft_ctx;
    memset(&fft_ctx, 0, sizeof(r2cfft_t));

    // 调用初始化函数，传入结构体地址和参数
    if (!r2cfft_initialize(&fft_ctx, nx, ny)) {
        fprintf(stderr, "FFTW初始化失败\n");
        return 1;
    }

    // 这里可以使用fft_ctx执行FFT操作，例如：
    // fftw_execute(fft_ctx.plan);

    // 使用完毕必须释放FFTW资源，避免内存泄漏
    fftw_destroy_plan(fft_ctx.plan);
    fftw_free(fft_ctx.buffer_in);
    fftw_free(fft_ctx.buffer_out);

    return 0;
}

// 修正后的初始化函数
int r2cfft_initialize(r2cfft_t *const ctx, const std::size_t nx, const std::size_t ny)
{
    // 先清零结构体，确保所有成员初始状态合法
    memset(ctx, 0, sizeof(r2cfft_t));
    
    // 检查输入参数是否合法
    if (nx == 0 || ny == 0)
    {
        return 0;
    }
    ctx->nx = nx;
    ctx->ny = ny;

    ctx->buffer_in = fftw_alloc_real(nx * ny);
    if (!ctx->buffer_in)
    {
        return 0;
    }

    ctx->buffer_out = fftw_alloc_complex(nx * ny); // 修正笔误nyk为ny
    if (!ctx->buffer_out)
    {
        fftw_free(ctx->buffer_in); // 分配失败时释放已申请的资源
        return 0;
    }

    ctx->plan = fftw_plan_dft_r2c_2d(nx, ny, ctx->buffer_in, ctx->buffer_out, FFTW_PATIENT);
    if (!ctx->plan)
    {
        fftw_free(ctx->buffer_in);
        fftw_free(ctx->buffer_out);
        return 0;
    }

    return 1;
}

方式2：堆上分配结构体 #

如果结构体需要跨函数使用或者长期存在，可以用堆分配，记得最后要释放结构体本身：

int main(){ 
    // 堆上分配结构体内存，再清零
    r2cfft_t *fft_ctx = (r2cfft_t*)malloc(sizeof(r2cfft_t));
    if (!fft_ctx) {
        fprintf(stderr, "结构体内存分配失败\n");
        return 1;
    }
    memset(fft_ctx, 0, sizeof(r2cfft_t));

    // 调用初始化函数
    if (!r2cfft_initialize(fft_ctx, nx, ny)) {
        fprintf(stderr, "FFTW初始化失败\n");
        free(fft_ctx); // 初始化失败时释放结构体内存
        return 1;
    }

    // 使用FFT资源...

    // 清理资源：先释放FFTW相关，再释放结构体
    fftw_destroy_plan(fft_ctx->plan);
    fftw_free(fft_ctx->buffer_in);
    fftw_free(fft_ctx->buffer_out);
    free(fft_ctx);

    return 0;
}

关键注意事项 #

• 无论哪种方式，初始化结构体前必须清零，避免指针成员指向随机内存导致崩溃。
• FFTW的plan和缓冲区必须手动释放，否则会造成内存泄漏。
• 初始化失败时，要检查是否有已分配的资源，及时释放避免泄漏。

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