24小时热门版块排行榜

返回列表

baobiao007

木虫 (职业作家)

应助: 201 (大学生)
金币: 6482.7
帖子: 3050
在线: 1009.9小时
虫号: 505962

[资源] 【分享】二维傅里叶变换c程序

#include
#include
#include
#include"FFT.h"
//申请二维动态数组的函数
void MySpace(float ***p, int m, int n)
{
int i;
*p=(float **)calloc(m,sizeof(float *));
for(i=0; i (*p)=(float *)calloc(n,sizeof(float));
}
//释放申请的二维动态数组
void FreeMySpace(float ***p, int m)
{
int i;
for(i=0; i free((*p));
free(*p);
}
//矩阵转置a[m][n]->b[n][m]
void Zhuan(float ***a,float ***b,int m,int n)
{
int i,j;
for(i=0; i for(j=0; j (*b)[j]=(*a)[j];
}
//2D-FFT tag=1,正变换;tag=-1,反变换
void FFT2D(float **xr,float **xi,int m,int n,int tag)
{
int i,j,k1,k2;
float **xtr, **xti;
MySpace(&xtr,n,m);
MySpace(&xti,n,m);
k1=(int)(log(n)/log(2)+0.5);
k2=(int)(log(m)/log(2)+0.5);
for(i=0; i fft(xr,xi,k1,tag);

Zhuan(&xr,&xtr,m,n);
Zhuan(&xi,&xti,m,n);
for(i=0; i fft(xtr,xti,k2,tag);
Zhuan(&xtr,&xr,n,m);
Zhuan(&xti,&xi,n,m);
FreeMySpace(&xtr,n);
FreeMySpace(&xti,n);
}

void main()
{
int i,j;
const int M=64;
const int N=256;
float **xr, **xi;
FILE *fp1, *fp2;

MySpace(&xr,M,N);
MySpace(&xi,M,N);
fp1=fopen("2D-test-64-256.dat","rb" );
fp2=fopen("2D-64-256.dat","wb" );
for(i=0; i for(j=0; j fread(&xr[j],sizeof(float),1,fp1);
FFT2D(xr,xi,M,N,1);
for(i=0; i for(j=0; j fwrite(&xr[j],sizeof(float),1,fp2);
FreeMySpace(&xr,M);
FreeMySpace(&xi,M);
fclose(fp1);
fclose(fp2);
}
原始图形：

变换结果：

[ Last edited by baobiao007 on 2011-2-13 at 17:54 ]