c代码实现fftw3图片傅里叶变换

以下是使用FFTW3库进行图片二维傅里叶变换的C代码示例：

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <fftw3.h>

#define WIDTH 512
#define HEIGHT 512

int main()
{
    // 读取图片数据
    FILE *fp = fopen("lena.raw", "rb");
    unsigned char *img = (unsigned char *)malloc(WIDTH * HEIGHT * sizeof(unsigned char));
    fread(img, sizeof(unsigned char), WIDTH * HEIGHT, fp);
    fclose(fp);

    // 创建输入和输出数组
    fftw_complex *in = (fftw_complex *)fftw_malloc(sizeof(fftw_complex) * WIDTH * HEIGHT);
    fftw_complex *out = (fftw_complex *)fftw_malloc(sizeof(fftw_complex) * WIDTH * HEIGHT);

    // 将图片数据复制到输入数组中
    for (int i = 0; i < WIDTH * HEIGHT; i++)
    {
        in[i][0] = img[i];
        in[i][1] = 0.0;
    }

    // 创建傅里叶变换计算计划
    fftw_plan plan = fftw_plan_dft_2d(WIDTH, HEIGHT, in, out, FFTW_FORWARD, FFTW_ESTIMATE);

    // 执行傅里叶变换
    fftw_execute(plan);

    // 将输出数组中的数据转换为幅度谱
    double *amp = (double *)malloc(WIDTH * HEIGHT * sizeof(double));
    for (int i = 0; i < WIDTH * HEIGHT; i++)
    {
        amp[i] = sqrt(out[i][0] * out[i][0] + out[i][1] * out[i][1]);
    }

    // 将幅度谱数据写入文件
    fp = fopen("lena_amp.raw", "wb");
    fwrite(amp, sizeof(double), WIDTH * HEIGHT, fp);
    fclose(fp);

    // 释放内存
    free(img);
    fftw_free(in);
    fftw_free(out);
    free(amp);
    fftw_destroy_plan(plan);

    return 0;
}

该代码使用FFTW3库进行二维傅里叶变换，并将变换结果转换为幅度谱，最后将幅度谱数据写入文件。你可以将代码中的lena.raw替换为你自己的图片文件名，并将输出文件名lena_amp.raw替换为你想要的文件名。