#include "uart.h"#include "fft.h" void main(){ u8 Freq=0; //峰值对应频率 //从y=200*sin(2*pi*40*t)中取得数据点 s16 code DATA[128]={0,118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 ,-118 , 0 , 118 , -190 , 190 , -118, 0 , 118 ,-190 , 190 , -118 , 0 , 118 , -190 , 190 , -118,0 ,118 , -190 , 190 , -118, 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118, 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118 , 0 , 118 , -190 , 190 , -118, 0 , 118 , -190, 190 , -118, 0 , 118 , -190 }; u8 ii,Max_Place=0; //用Max_Place保存频谱中最大峰值在数组中的位置 float Out_abs[NN/2]; //用于保存模值,因为频谱图关于Fs/2对称,所以只求一半 complex y[NN]; for (ii = 0; ii < NN; ++ii) { y[ii].real = DATA[ii]; y[ii].imag = 0; } fft(NN, y); //FFT运算 c_abs(y, Out_abs, NN/2); //求模 Max_Place=Mag(Out_abs); //求最大模所在位置 Freq=(Max_Place-1)*Fs/(NN-1); //得到最大模对应频率 UartInit(); printf("Freq: %bd,max_place: %bd\r\n",Freq,Max_Place); while(1) { }}
#ifndef __FFT_H__ #define __FFT_H__ typedef struct complex //复数类型 { float real; //实部 float imag; //虚部 }complex; #define PI 3.1415926535 #define NN 16 //采样点数#define Fs 100 //采样频率#define u8 unsigned char#define s8 signed char#define u16 unsigned int#define s16 signed int///////////////////////////////////////////In file fft.c 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 fft(int N,complex f[]);//傅立叶变换 输出也存在数组f中 void c_abs(complex f[],float out[],int n);//复数数组取模u16 Mag(float outRes[]); //求最大值所在位置//////////////////////////////////////////// #endif
#include "math.h" #include "fft.h" //精度0.0001弧度 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 Wn_i(int n,int i,complex *Wn) { Wn->real = cos(2*PI*i/n); 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);//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 } } } } u16 Mag(float outRes[]){ u8 i=1; u8 max,place; max=outRes[i]; place=i; for(i=2;i<NN/2;i++) { if(outRes[i]>max) { max=outRes[i]; place=i; } } return place; }