c++ 傅里叶变换

最新推荐文章于 2024-08-20 18:18:10 发布

renbaifen

最新推荐文章于 2024-08-20 18:18:10 发布

阅读量2k

点赞数

分类专栏： c c++

本文链接：https://blog.csdn.net/renbaifen/article/details/97366035

版权

c 同时被 2 个专栏收录

15 篇文章 0 订阅

订阅专栏

c++

14 篇文章 0 订阅

订阅专栏

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <math.h>
#include<iostream>
#include "fftw3.h"
using namespace std;
#pragma comment(lib, "libfftw3-3.lib") // double版本
//#pragma comment(lib, "libfftw3f-3.lib")// float版本
// #pragma comment(lib, "libfftw3l-3.lib")// long float版本

int main(void)
{
   int i, n= 10;
   fftw_complex *in;
   in= (fftw_complex *)fftw_malloc(sizeof(fftw_complex) * n);
   for (int i = 0; i < n; i++)
   {
       in[i][0] = i + 1;
       in[i][1] = in[i][0];
   }
   fftw_complex *out = (fftw_complex *)fftw_malloc(sizeof(fftw_complex) * n);

fftw_plan plan_forward = fftw_plan_dft_1d(n, in, out, FFTW_FORWARD, FFTW_ESTIMATE);

   fftw_execute(plan_forward);

   fftw_complex *in2 = (fftw_complex *)fftw_malloc(sizeof(fftw_complex) * n);
   fftw_plan plan_backward = fftw_plan_dft_1d(n, out, in2, FFTW_BACKWARD, FFTW_ESTIMATE);

   fftw_execute(plan_backward);
   for (i = 0; i<n; i++)
   {
       printf("%f\t%f\t%f\t%f\n", in[i][0], in[i][1], (in2[i][0])/n, (in2[i][1])/n);//需要做归一化处理
       //printf("%f\t%f\t%f\t%f\n", in[i][0],in[i][1], out[i][0],out[i][1] );
   }

fftw_destroy_plan(plan_forward);
fftw_destroy_plan(plan_backward);

fftw_free(in);
fftw_free(in2);

   fftw_free(out);

   return 0;
}