FFT的c语言实现与物理意义

fft.h

#ifndef __FFT_H__
#define __FFT_H__

typedef struct complex //复数类型
{
  float real;		//实部
  float imag;		//虚部
}complex;

#define PI 3.1415926535897932384626433832795028841971


///
void conjugate_complex(int n,complex in[],complex out[]);
void c_plus(complex a,complex b,complex *c);//复数加
void c_mul(complex a,complex b,complex *c) ;//复数乘
void c_sub(complex a,complex b,complex *c);	//复数减法
void c_div(complex a,complex b,complex *c);	//复数除法
void fft(int N,complex f[]);//傅立叶变换 输出也存在数组f中
void ifft(int N,complex f[]); // 傅里叶逆变换
void c_abs(complex f[],float out[],int n);//复数数组取模

#endif
fft.c

#include "math.h"
#include "fft.h"

/*******************************************************************************
* 功能: 共轭复数 
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void conjugate_complex(int n,complex in[],complex out[])
{
  int i = 0;
  for(i=0;i<n;i++)
  {
    out[i].imag = -in[i].imag;
    out[i].real = in[i].real;
  }	
}
/*******************************************************************************
* 功能: 复数的模 
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void c_abs(complex f[],float out[],int n)
{
  int i = 0;
  float t;
  for(i=0;i<n;i++)
  {
    t = f[i].real * f[i].real + f[i].imag * f[i].imag;
    out[i] = sqrt(t);
  }	
}
/*******************************************************************************
* 功能: 复数加法
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void c_plus(complex a,complex b,complex *c)
{
  c->real = a.real + b.real;
  c->imag = a.imag + b.imag;
}
/*******************************************************************************
* 功能: 复数减法
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void c_sub(complex a,complex b,complex *c)
{
  c->real = a.real - b.real;
  c->imag = a.imag - b.imag;	
}
/*******************************************************************************
* 功能: 复数乘法
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void c_mul(complex a,complex b,complex *c)
{
  c->real = a.real * b.real - a.imag * b.imag;
  c->imag = a.real * b.imag + a.imag * b.real;	
}
/*******************************************************************************
* 功能: 复数除法
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void c_div(complex a,complex b,complex *c)
{
  c->real = (a.real * b.real + a.imag * b.imag)/(b.real * b.real +b.imag * b.imag);
  c->imag = (a.imag * b.real - a.real * b.imag)/(b.real * b.real +b.imag * b.imag);
}

void Wn_i(int n,int i,complex *Wn,char flag)
{
  Wn->real = cos(2*PI*i/n);
  if(flag == 1)
  Wn->imag = -sin(2*PI*i/n);
  else if(flag == 0)
  Wn->imag = -sin(2*PI*i/n);
}
/*******************************************************************************
* 功能: 傅里叶变化
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void fft(int N,complex f[])
{
  complex t,wn;//中间变量
  int i,j,k,m,n,l,r,M;
  int la,lb,lc;
  /*----计算分解的级数M=log2(N)----*/
  for(i=N,M=1;(i=i/2)!=1;M++); 
  /*----按照倒位序重新排列原信号----*/
  for(i=1,j=N/2;i<=N-2;i++)
  {
    if(i<j)
    {
      t=f[j];
      f[j]=f[i];
      f[i]=t;
    }
    k=N/2;
    while(k<=j)
    {
      j=j-k;
      k=k/2;
    }
    j=j+k;
  }

  /*----FFT算法----*/
  for(m=1;m<=M;m++)
  {
    la=pow(2,m); //la=2^m代表第m级每个分组所含节点数		
    lb=la/2;    //lb代表第m级每个分组所含碟形单元数
                 //同时它也表示每个碟形单元上下节点之间的距离
    /*----碟形运算----*/
    for(l=1;l<=lb;l++)
    {
      r=(l-1)*pow(2,M-m);	
      for(n=l-1;n<N-1;n=n+la) //遍历每个分组,分组总数为N/la
      {
        lc=n+lb;  //n,lc分别代表一个碟形单元的上、下节点编号     
        Wn_i(N,r,&wn,1);//wn=Wnr
        c_mul(f[lc],wn,&t);//t = f[lc] * wn复数运算
        c_sub(f[n],t,&(f[lc]));//f[lc] = f[n] - f[lc] * Wnr
        c_plus(f[n],t,&(f[n]));//f[n] = f[n] + f[lc] * Wnr
      }
    }
  }
}
/*******************************************************************************
* 功能: 傅里叶逆变换
* 形参: 
* 返回: 无
* 说明: 
*******************************************************************************/
void ifft(int N,complex f[])
{
  int i=0;
  conjugate_complex(N,f,f);
  fft(N,f);
  conjugate_complex(N,f,f);
  for(i=0;i<N;i++)
  {
    f[i].imag = (f[i].imag)/N;
    f[i].real = (f[i].real)/N;
  }
}
main.c

#include <stdio.h>
#include "fft.h"

#define  FFT_NPT 256

/*******************************************************************************
* 功能: fft测试函数1 
* 形参: 无 
* 返回: 无
* 说明: 模拟220V市电f(t)=220cos(2PI*50t)的ADC采样并进行快速傅里叶变换
*******************************************************************************/
void fftTest1(void)
{
	int i;
	complex s[FFT_NPT];
	float r;
	for(i=0;i<FFT_NPT;i++)                 //256Hz的采样频率采样256个点 
	{
		s[i].real=220*cos(2*PI*50*i/FFT_NPT);
		s[i].imag=0;                                //虚部为0
	}
	fft(FFT_NPT, s);		//进行FFT处理
	for(i=0;i<FFT_NPT;i++) 
	{
		r=sqrt(s[i].real*s[i].real+s[i].imag*s[i].imag);
		printf("s[%d].real=%f,imag=%f,r=%f\n",i,s[i].real,s[i].imag,r); 
	}
}
/*******************************************************************************
* 功能: fft测试函数2 
* 形参: 无 
* 返回: 无
* 说明:对f(t)=2+3*cos(2*pi*50*t-pi*30/180)+1.5*cos(2*pi*75*t+pi*90/180)采样并FFT 
*******************************************************************************/
void fftTest2(void)
{
	int i;
	complex s[FFT_NPT];
	float r;
	for(i=0;i<FFT_NPT;i++)                 //256Hz的采样频率采样256个点 
	{
		s[i].real=2+3*cos(2*PI*50*i/FFT_NPT-PI*30/180)+1.5*cos(2*PI*75*i/FFT_NPT+PI*90/180);
		s[i].imag=0;                                //虚部为0
	}
	fft(FFT_NPT, s);		//进行FFT处理
	for(i=0;i<FFT_NPT;i++) 
	{
		r=sqrt(s[i].real*s[i].real+s[i].imag*s[i].imag);
		printf("s[%d].real=%f,imag=%f,r=%f\n",i,s[i].real,s[i].imag,r); 
	}
}
/*******************************************************************************
* 功能: 主函数 
* 形参: 
* 返回: 
* 说明: 
*******************************************************************************/
int main()
{
	printf("fftTest1\n");
	fftTest1();
	printf("fftTest2\n");
	fftTest2();		
	return 0;
}

由于FFT结果的对称性,通常我们只使用前半部分的结果,即小于采样频率一半的结果进行分析

1.运行fft测试t函数1,可得到如下结果(忽略值近似为0 的点和对称点):

s[50].real=28160.000000,imag=-0.001160,r=28160.000000

分析:因为以256Hz的频率采样256个点,每两个点之间的间距就是1Hz,所以该点表示50Hz的信号

    幅度:28160/(N/2)=28160/(256/2)=220

    相位:arctan(0.001160/28160) = 0 °

2.运行fft测试t函数2,可得到如下结果(忽略值近似为0 的点和对称点):

s[0].real=512.000000,imag=0.000000,r=512.000000

s[50].real=332.553772,imag=-191.999985,r=384.000000

s[75].real=0.000002,imag=192.000000,r=192.000000

结论:(1)直流幅度:512/N = 512/256 = 2,相位:无

    (2)50Hz幅度:384/(N/2)=384/(256/2)=3,相位:arctan(-191.999985/332.553772) =  -30 ° 

            (3)75Hz幅度:191.999985/(N/2)=192/(256/2)=1.5,相位:arctan(192.000000/0.000002) = 90 ° 

参看文章:http://blog.sina.com.cn/s/blog_640029b301010xkv.html

  • 4
    点赞
  • 15
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值