fft.cpp

基于vc编写的正弦信号的傅立叶变换程序

//#include "stdafx.h"
#include <math.h>
#include < stdio.h>
#define pi 3.14159265358979323846

typedef struct COMPX
{
	float real;
	float imag;
}COMPX;

COMPX EE(COMPX b1,COMPX b2)
{
	COMPX b3;h
	b3.real=b1.real*b2.real-b1.imag*b2.imag;
	b3.imag=b1.real*b2. imag +b1.imag*b2.real;
	return(b3);
}

void FFT(COMPX *xin,int N)
{
	int m,LH,nm,I,k,J,M,K;
	float p,ps;
	int B,N1;
	COMPX w,T;
	M=log10(N)/log10(2);
	LH=N/2;
	J=LH;
	N1=N-2;
	for(I=1;I<=N1;I++)
	{
		if(I<J)
		{
			T=xin[I];
			xin[I]= xin[J]; 
			xin[J]=T;
		}
		
		K=LH;
		while(J>=K)
		{
				J=J-K;
				K=K/2;
		}
		J=J+K;
	}
	
	for(m=1;m<=M;m++)
	{
		B=pow(2,m-1);
		for(J=0;J<=B-1;J++)
		{
			p=pow(2,M-m)*J;
			ps=2*pi/N*p;
			w.real=cos(ps);
			w.imag=-sin(ps);
			
			for(k=J;k<=N-1;k=k+pow(2,m))
			{
				T=EE(xin[k+B],w);
				xin[k+B].real=xin[k].real-T.real;
				xin[k+B].imag=xin[k]. imag -T.imag;
				xin[k].real=xin[k].real+T.real;
				xin[k].imag=xin[k]. imag +T.imag;
			}
		}
	}
}

void sine_generate(COMPX *xin)
{	float sum=0;
	for(int i=0;i<64;i++)
		xin[i].imag=0;
	for(i=0;i<64;i++)
		xin[i].real=sin(4*pi*i/64);
	for(i=0;i<64;i++)
	{ sum+=xin[i].real*xin[i].real;
	  sum=sum/64;
    }
    printf("power=%f\n",sum);
}

void main()
{

	COMPX pXIN[64];
	sine_generate(pXIN);	
	FFT(pXIN,64);
    for(int i=0;i<64;i++)
	{
		if(i%2!=0) printf("%10.7f+j%10.7f\n",pXIN[i].real,pXIN[i].imag);
		else printf("%10.7f+j%10.7f\t",pXIN[i].real,pXIN[i].imag); 
	}
   
}