草庐IT

快速傅里叶变换(FFT)c语言实现

AlepH_Sin 2023-07-21 原文

基本原理在这里就不多讲了,可以看看其他高浏览量的博文,这篇文章针对c语言的实现

复数运算算子

        我们都知道C语言本身是没有复数运算的,很多DSP、单片机要用到也没有开源库可以使用复数运算,针对FFT在硬件上运行只能手动从底层开始

定义复数类型

        这里用最简单高效的方法——结构体

struct complex
{
    double real;
    double image;
};

复数加法

struct complex complex_add(struct complex c1,struct complex c2)     //复数加法
{
    struct complex p;

    p.real = c1.real + c2.real;
    p.image = c1.image + c2.image;

    return p;

}

复数减法

struct complex complex_sub(struct complex c1,struct complex c2)     //复数减
{
    struct complex p;

    p.real = c1.real - c2.real;
    p.image = c1.image - c2.image;

    return p;

}

复数乘法

struct complex complex_multi(struct complex c1,struct complex c2)  //复数乘法
{
    struct complex c3;
    c3.real=c1.real*c2.real - c1.image *c2.image;
    c3.image = c2.real* c1.image + c1.real*c2.image;

    return c3;
};

复数取模

double mold_length(struct complex c)        //幅度
{
    return sqrt(c.real * c.real + c.image * c.image);
};

旋转因子

    教科书上的旋转因子一般长成这样把它变成可以运算的形式   

struct complex rotation_factor(int N,int n,int k)       //旋转因子
{
    struct complex w;

    w.real = cos(2* PI * n * k /N);
    w.image = - sin(2* PI * n * k /N);        //这里的PI是宏定义

    return w;
}

反位序操作

        我用的方法是基于时间抽取的基2FFT (DIT-FFT)所以开始进入FFT之前要先把序列顺序取反位序

int reverse_num(int l,int oringin_num)          //反位序
{
    int q=0,m=0;

    for(int k=l-1;k>=0;k--)
    {
        q = oringin_num % 2;
        m += q*(1<<k);
        oringin_num = oringin_num / 2;
    }

    return m;
}

解释一下这个代码含义,用取余操作取出最低位,把最低位放在m中最高位置,取出后除2操作把每一位右移,直到为0

FFT

我们首先拿出祖传的蝶形运算信号流图,

 在其中抽取一个蝶形详细分析,注意这其中上半部分是相加【+】下半部分是相减【-】,而且下半部分都会带有一个旋转因子,先记住这个特征,接下来编程会利用这个特征

 为了方便编程,把它表达成数学式子,这其中前一级与后一级的关系用等式表达

现在问题就是k和j的确定,重新观察完整的蝶形图在第【1】级两两运算的节点序号是【0、1】【2、3】【4、5】【6、7】;在第【2】级两两运算的节点序号是【0、2】【1、3】【4、6】【5、7】;第【3】级运算的节点是【0、4】【1、5】【2、6】【3、7】。注意这里不要用图上的反位置序,而是反位后的顺序,也就是下图红色序号

首先【j】与【k】的间隔可以容易观察出规律,每增加一级,间隔就会扩大两倍,换句话说【j】与【k】的间搁可以表达为下面的式子

好了,这里我们破解了第一条规律,急需确定的是这个旋转因子当中的,这里我没有观察到规律,只好直接引用教材上的解释

 我们可以简单验证一下:

比如第【3】级 【k=2】的情况

   

第【2】级【k=5】的情况

   

两次验证都符合实际FFT蝶形图的结果。

接下来看似没有急需破解的规律了,但是真正编程时候还是存在一个问题,【k】的取值不是从上到下按顺序的例如第一级的【k】取值为【0,2,4,6】第二级的取值为【0,1,4,5】第三级的取值为【0,1,2,3】。

这样如果按找从头到尾的顺序确实找不到什么规律,但是整体没规律,部分总会有规律吧,注意看我笔迹画的线为啥颜色是这样分的呢,没有发现很有规律吗?我们按找相同颜色分组,第一级的第一组两个蝶形运算的【k】之间相差距离刚好是【dist】的两倍,第二级的第一组两个蝶形运算的【k】之间相差的距离也是【dist】的两倍,但是到了第三级, 【dist】的两倍是【8】如果【k=0】 那么【k+8】就会超出最大序号【7】,那么考虑第三级分成【4】组,每一组只有一个蝶形运算,有别于其他两个,可以用for循环这样写

for(k=0;k<len;k+=(dist<<1))    //在二进制左移一位相当于乘2 ,这里len是FFT的点数N

同时考虑每一级内又有很多组,可以用for循环嵌套

for(j=0;j<dist;j++)
    for(k=0;k<len;k+=(dist<<1))    

最终把分析的结果整合到一起,写成单独的函数 ,注意一下,程序中的一些标号和前面分析的会有点不同

void fft(int len, struct complex in_x[],struct complex out_y[])
{

    /*
        param len 序列长度,目前只能是2的指数
        param in_x输入的序列
        param out_y输出的序列

    */


    int l,k,r,z,dist,q,j;       //l是级
    struct complex w,tmp;
    struct complex in_x_mem[len];

    l = log2(len);

    for(k=0;k<len;k++)
    {
        in_x_mem[k] = in_x[reverse_num(l,k)];       //反位序号操作
    }

    for(r = 0;r<l;r++)      //遍历每一级蝶形运算
    {

        dist = 1<<r;        //提前计算每一级的间隔距离

        for( j=0;j<dist;j++ )      //计算策略是拆成上下两组,先上计算,后下计算,j是计算的起始序号
        {

            for(k=j;k<len;k+=(dist<<1))     //不好解释,得看画图理解
            {
                q = k+dist; //q同一组蝶形运算第二个序号
                z = k << (l - r -1);    //确定旋转因子的指数


                w = rotation_factor(len,1,z);
                //由于不是并行计算,必须先缓存
                tmp = in_x_mem[k];

                in_x_mem[k] = complex_add( in_x_mem[k] , complex_multi(w, in_x_mem[q]) );
                in_x_mem[q] = complex_sub(tmp , complex_multi(w, in_x_mem[q]) );

                

            }


        }
    }
    
    memcpy(out_y,in_x_mem,len*sizeof(struct complex));    //拷贝到输出的数组
}

结果测试

这里我用序列输入到FFT,输出应该是

 对照结果,和计算一样,程序成功

完整程序

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>


#define PI 3.1415926
struct complex
{
    double real;
    double image;
};
struct complex complex_add(struct complex c1,struct complex c2);
struct complex complex_sub(struct complex c1,struct complex c2);
struct complex complex_multi(struct complex c1,struct complex c2);
struct complex rotation_factor(int N,int n,int k);
double mold_length(struct complex c);
void fft(int len, struct complex in_x[],struct complex out_y[]);




int main()
{
    int sam[8] = {1,-1,1,-1,1,-1,1,-1};
    int jhg[8] = {0,0,0,0,0,0,0,0};
    struct complex x[8];
    struct complex y[8];

    for(int i=0;i<8;i++)
    {
        x[i].real = sam[i];
        x[i].image = jhg[i];
    }
    printf("时域序列\n");

    for(int i=0;i<8;i++)
    {
        printf("(%.2f, %.2fi) \n",x[i].real,x[i].image);
    }

    fft(8,x,y);

    printf("频域序列\n");

    for(int i=0;i<8;i++)
    {
        printf("(%.2f, %.2fi)\n",y[i].real,y[i].image);
    }

    

    return 0;
}

struct complex complex_add(struct complex c1,struct complex c2)     //复数加法
{
    struct complex p;

    p.real = c1.real + c2.real;
    p.image = c1.image + c2.image;

    return p;

}

struct complex complex_sub(struct complex c1,struct complex c2)     //复数减
{
    struct complex p;

    p.real = c1.real - c2.real;
    p.image = c1.image - c2.image;

    return p;

}

struct complex complex_multi(struct complex c1,struct complex c2)  //复数乘法
{
    struct complex c3;
    c3.real=c1.real*c2.real - c1.image *c2.image;
    c3.image = c2.real* c1.image + c1.real*c2.image;

    return c3;
};



struct complex rotation_factor(int N,int n,int k)       //旋转因子
{
    struct complex w;

    w.real = cos(2* PI * n * k /N);
    w.image = - sin(2* PI * n * k /N);

    return w;
}

double mold_length(struct complex c)        //幅度
{
    return sqrt(c.real * c.real + c.image * c.image);
};


int reverse_num(int l,int oringin_num)          //反位序
{
    int q=0,m=0;

    for(int k=l-1;k>=0;k--)
    {
        q = oringin_num % 2;
        m += q*(1<<k);
        oringin_num = oringin_num / 2;
    }

    return m;
}


void fft(int len, struct complex in_x[],struct complex out_y[])
{
    /*
        param len 序列长度,目前只能是2的指数
        param in_x输入的序列
        param out_y输出的序列

    */

    int l,k,r,z,dist,q,j;       //l是级
    struct complex w,tmp;
    struct complex in_x_mem[len];

    l = log2(len);

    for(k=0;k<len;k++)
    {
        in_x_mem[k] = in_x[reverse_num(l,k)];       //反位序号操作
    }

    for(r = 0;r<l;r++)      //遍历每一级蝶形运算
    {

        dist = 1<<r;        //提前计算每一级的间隔距离

        for( j=0;j<dist;j++ )      //计算策略是拆成上下两组,先上计算,后下计算,j是计算的起始序号
        {
            for(k=j;k<len;k+=(dist<<1))     //不好解释,得画图理解
            {
                q = k+dist; //q同一组蝶形运算第二个序号
                z = k << (l - r -1);    //确定旋转因子的指数

                w = rotation_factor(len,1,z);
                //由于不是并行计算,必须先缓存
                tmp = in_x_mem[k];

                in_x_mem[k] = complex_add( in_x_mem[k] , complex_multi(w, in_x_mem[q]) );
                in_x_mem[q] = complex_sub(tmp , complex_multi(w, in_x_mem[q]) );

            }
        }
    }
    memcpy(out_y,in_x_mem,len*sizeof(struct complex));
}



有关快速傅里叶变换(FFT)c语言实现的更多相关文章

  1. ruby - 如何将脚本文件的末尾读取为数据文件(Perl 或任何其他语言) - 2

    我正在寻找执行以下操作的正确语法(在Perl、Shell或Ruby中):#variabletoaccessthedatalinesappendedasafileEND_OF_SCRIPT_MARKERrawdatastartshereanditcontinues. 最佳答案 Perl用__DATA__做这个:#!/usr/bin/perlusestrict;usewarnings;while(){print;}__DATA__Texttoprintgoeshere 关于ruby-如何将脚

  2. ruby - 如何根据特征实现 FactoryGirl 的条件行为 - 2

    我有一个用户工厂。我希望默认情况下确认用户。但是鉴于unconfirmed特征,我不希望它们被确认。虽然我有一个基于实现细节而不是抽象的工作实现,但我想知道如何正确地做到这一点。factory:userdoafter(:create)do|user,evaluator|#unwantedimplementationdetailshereunlessFactoryGirl.factories[:user].defined_traits.map(&:name).include?(:unconfirmed)user.confirm!endendtrait:unconfirmeddoenden

  3. ruby - 寻找通过阅读代码确定编程语言的ruby gem? - 2

    几个月前,我读了一篇关于ruby​​gem的博客文章,它可以通过阅读代码本身来确定编程语言。对于我的生活,我不记得博客或gem的名称。谷歌搜索“ruby编程语言猜测”及其变体也无济于事。有人碰巧知道相关gem的名称吗? 最佳答案 是这个吗:http://github.com/chrislo/sourceclassifier/tree/master 关于ruby-寻找通过阅读代码确定编程语言的rubygem?,我们在StackOverflow上找到一个类似的问题:

  4. 华为OD机试用Python实现 -【明明的随机数】 2023Q1A - 2

    华为OD机试题本篇题目:明明的随机数题目输入描述输出描述:示例1输入输出说明代码编写思路最近更新的博客华为od2023|什么是华为od,od薪资待遇,od机试题清单华为OD机试真题大全,用Python解华为机试题|机试宝典【华为OD机试】全流程解析+经验分享,题型分享,防作弊指南华为o

  5. Unity 热更新技术 | (三) Lua语言基本介绍及下载安装 - 2

    ?博客主页:https://xiaoy.blog.csdn.net?本文由呆呆敲代码的小Y原创,首发于CSDN??学习专栏推荐:Unity系统学习专栏?游戏制作专栏推荐:游戏制作?Unity实战100例专栏推荐:Unity实战100例教程?欢迎点赞?收藏⭐留言?如有错误敬请指正!?未来很长,值得我们全力奔赴更美好的生活✨------------------❤️分割线❤️-------------------------

  6. 7个大一C语言必学的程序 / C语言经典代码大全 - 2

    嗨~大家好,这里是可莉!今天给大家带来的是7个C语言的经典基础代码~那一起往下看下去把【程序一】打印100到200之间的素数#includeintmain(){ inti; for(i=100;i 【程序二】输出乘法口诀表#includeintmain(){inti;for(i=1;i 【程序三】判断1000年---2000年之间的闰年#includeintmain(){intyear;for(year=1000;year 【程序四】给定两个整形变量的值,将两个值的内容进行交换。这里提供两种方法来进行交换,第一种为创建临时变量来进行交换,第二种是不创建临时变量而直接进行交换。1.创建临时变量来

  7. 基于C#实现简易绘图工具【100010177】 - 2

    C#实现简易绘图工具一.引言实验目的:通过制作窗体应用程序(C#画图软件),熟悉基本的窗体设计过程以及控件设计,事件处理等,熟悉使用C#的winform窗体进行绘图的基本步骤,对于面向对象编程有更加深刻的体会.Tutorial任务设计一个具有基本功能的画图软件**·包括简单的新建文件,保存,重新绘图等功能**·实现一些基本图形的绘制,包括铅笔和基本形状等,学习橡皮工具的创建**·设计一个合理舒适的UI界面**注明:你可能需要先了解一些关于winform窗体应用程序绘图的基本知识,以及关于GDI+类和结构的知识二.实验环境Windows系统下的visualstudio2017C#窗体应用程序三.

  8. MIMO-OFDM无线通信技术及MATLAB实现(1)无线信道:传播和衰落 - 2

     MIMO技术的优缺点优点通过下面三个增益来总体概括:阵列增益。阵列增益是指由于接收机通过对接收信号的相干合并而活得的平均SNR的提高。在发射机不知道信道信息的情况下,MIMO系统可以获得的阵列增益与接收天线数成正比复用增益。在采用空间复用方案的MIMO系统中,可以获得复用增益,即信道容量成倍增加。信道容量的增加与min(Nt,Nr)成正比分集增益。在采用空间分集方案的MIMO系统中,可以获得分集增益,即可靠性性能的改善。分集增益用独立衰落支路数来描述,即分集指数。在使用了空时编码的MIMO系统中,由于接收天线或发射天线之间的间距较远,可认为它们各自的大尺度衰落是相互独立的,因此分布式MIMO

  9. 【Java入门】使用Java实现文件夹的遍历 - 2

    遍历文件夹我们通常是使用递归进行操作,这种方式比较简单,也比较容易理解。本文为大家介绍另一种不使用递归的方式,由于没有使用递归,只用到了循环和集合,所以效率更高一些!一、使用递归遍历文件夹整体思路1、使用File封装初始目录,2、打印这个目录3、获取这个目录下所有的子文件和子目录的数组。4、遍历这个数组,取出每个File对象4-1、如果File是否是一个文件,打印4-2、否则就是一个目录,递归调用代码实现publicclassSearchFile{publicstaticvoidmain(String[]args){//初始目录Filedir=newFile("d:/Dev");Datebeg

  10. ruby - Arrays Sets 和 SortedSets 在 Ruby 中是如何实现的 - 2

    通常,数组被实现为内存块,集合被实现为HashMap,有序集合被实现为跳跃列表。在Ruby中也是如此吗?我正在尝试从性能和内存占用方面评估Ruby中不同容器的使用情况 最佳答案 数组是Ruby核心库的一部分。每个Ruby实现都有自己的数组实现。Ruby语言规范只规定了Ruby数组的行为,并没有规定任何特定的实现策略。它甚至没有指定任何会强制或至少建议特定实现策略的性能约束。然而,大多数Rubyist对数组的性能特征有一些期望,这会迫使不符合它们的实现变得默默无闻,因为实际上没有人会使用它:插入、前置或追加以及删除元素的最坏情况步骤复

随机推荐