c語(yǔ)言的fft和ifft

#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include "fft_soft.h"

#define PI 3.14159265358979323846
complex add(complex a, complex b)
{
    complex ret = {a.real + b.real, a.imag + b.imag};
    return ret;
}

complex sub(complex a, complex b)
{
    complex ret = {a.real - b.real, a.imag - b.imag};
    return ret;
}

complex mul(complex a, complex b)
{
    complex ret = {a.real * b.real - a.imag * b.imag, a.real * b.imag + a.imag * b.real};
    return ret;
}

void bitrev(complex *data, int n)
{
    int j = 0;
    int m = 0;
    for (int i = 0; i < n; i++)
    {
        if (j > i)
            SWAP(data[i], data[j]);
        m = n / 2;
        while (j >= m && m != 0)
        {
            j -= m;
            m >>= 1;
        }
        j += m;
    }
}

void fft_soft(complex *data, int n)
{
    int M = 0;
    for (int i = n; i > 1; i = i >> 1, M++);

    bitrev(data, n);

    for (int m = 0; m < M; m++)
    {
        int K = n >> (m + 1);
        for (int k = 0; k < K; k++)
        {
            int J = 2 << m;
            int base = k * J;
            for (int j = 0; j < J / 2; j++)
            {
                int t = base + j;
                complex w = {cos(-2 * PI * j * K / n), sin(-2 * PI * j * K / n)};
                complex wn = mul(data[t + J / 2], w);
                complex temp = data[t];
                data[t] = add(data[t], wn);
                data[t + J / 2] = sub(temp, wn); 
            }
        }
    }
}

void ifft_soft(complex *data, int n)
{
    int M = 0;
    for (int i = n; i > 1; i = i >> 1, M++);

    bitrev(data, n);

    for (int m = 0; m < M; m++)
    {
        int K = n >> (m + 1);
        for (int k = 0; k < K; k++)
        {
            int J = 2 << m;
            int base = k * J;
            for (int j = 0; j < J / 2; j++)
            {
                int t = base + j;
                complex w = {cos(2 * PI * j * K / n), sin(2 * PI * j * K / n)};
                complex wn = mul(data[t + J / 2], w);
                complex temp = data[t];
                data[t] = add(data[t], wn);
                data[t + J / 2] = sub(temp, wn); 
            }
        }
    }

    for (int i = 0; i < n; i++)
    {
        data[i].real /= n;
        data[i].imag /= n;
    }
}
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末诊杆,一起剝皮案震驚了整個(gè)濱河市召边,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖举农,帶你破解...
    沈念sama閱讀 207,113評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡路星,警方通過(guò)查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,644評(píng)論 2 381
  • 文/潘曉璐 我一進(jìn)店門诱桂,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)洋丐,“玉大人,你說(shuō)我怎么就攤上這事挥等∮丫” “怎么了?”我有些...
    開(kāi)封第一講書人閱讀 153,340評(píng)論 0 344
  • 文/不壞的土叔 我叫張陵肝劲,是天一觀的道長(zhǎng)迁客。 經(jīng)常有香客問(wèn)我,道長(zhǎng)辞槐,這世上最難降的妖魔是什么掷漱? 我笑而不...
    開(kāi)封第一講書人閱讀 55,449評(píng)論 1 279
  • 正文 為了忘掉前任,我火速辦了婚禮榄檬,結(jié)果婚禮上卜范,老公的妹妹穿的比我還像新娘。我一直安慰自己鹿榜,他們只是感情好海雪,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,445評(píng)論 5 374
  • 文/花漫 我一把揭開(kāi)白布。 她就那樣靜靜地躺著舱殿,像睡著了一般奥裸。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上沪袭,一...
    開(kāi)封第一講書人閱讀 49,166評(píng)論 1 284
  • 那天湾宙,我揣著相機(jī)與錄音,去河邊找鬼冈绊。 笑死创倔,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的焚碌。 我是一名探鬼主播畦攘,決...
    沈念sama閱讀 38,442評(píng)論 3 401
  • 文/蒼蘭香墨 我猛地睜開(kāi)眼,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼十电!你這毒婦竟也來(lái)了知押?” 一聲冷哼從身側(cè)響起叹螟,我...
    開(kāi)封第一講書人閱讀 37,105評(píng)論 0 261
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎台盯,沒(méi)想到半個(gè)月后罢绽,有當(dāng)?shù)厝嗽跇?shù)林里發(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 43,601評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡静盅,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,066評(píng)論 2 325
  • 正文 我和宋清朗相戀三年良价,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片蒿叠。...
    茶點(diǎn)故事閱讀 38,161評(píng)論 1 334
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡明垢,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出市咽,到底是詐尸還是另有隱情痊银,我是刑警寧澤,帶...
    沈念sama閱讀 33,792評(píng)論 4 323
  • 正文 年R本政府宣布施绎,位于F島的核電站溯革,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏谷醉。R本人自食惡果不足惜致稀,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,351評(píng)論 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望俱尼。 院中可真熱鬧抖单,春花似錦、人聲如沸号显。這莊子的主人今日做“春日...
    開(kāi)封第一講書人閱讀 30,352評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)押蚤。三九已至,卻和暖如春羹应,著一層夾襖步出監(jiān)牢的瞬間揽碘,已是汗流浹背。 一陣腳步聲響...
    開(kāi)封第一講書人閱讀 31,584評(píng)論 1 261
  • 我被黑心中介騙來(lái)泰國(guó)打工园匹, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留雳刺,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 45,618評(píng)論 2 355
  • 正文 我出身青樓裸违,卻偏偏與公主長(zhǎng)得像掖桦,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子供汛,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,916評(píng)論 2 344

推薦閱讀更多精彩內(nèi)容