FFT for real data using decimation-in-frequency algorithm. 実データに対するFFT.周波数間引きアルゴリズムを使用. このライブラリを登録した際のプログラム: Demo_FFT_IFFT

Dependents:   UIT2_SpectrumAnalyzer F746_SpectralAnalysis_NoPhoto F746_FFT_Speed F746_RealtimeSpectrumAnalyzer ... more

fftReal.cpp

Committer:
MikamiUitOpen
Date:
2020-12-12
Revision:
4:0b4975fffc90
Parent:
3:dc123081d491

File content as of revision 4:0b4975fffc90:

//------------------------------------------------------------------------------
//  データが実数の場合の FFT class
//      FFT     複素共役になる部分は計算しない
//      IFFT    複素共役の部分を除いた値から計算する
//
//  2020/12/12, Copyright (c) 2020 MIKAMI, Naoki
//------------------------------------------------------------------------------

#include "fftReal.hpp"

namespace Mikami
{
    // コンストラクタ
    FftReal::FftReal(int16_t n)
            : N_FFT_(n), N_INV_(1.0f/n), wTable_(n/2), bTable_(n), u_(n)
    {
        // __clz(): リーディング 0 の数を取得する命令に対応
        uint32_t shifted = n << (__clz(n) + 1);
        MBED_ASSERT(shifted == 0);  // n が 2 のべき乗であることをチェック

        // 回転子の表を作成
        Complex arg = Complex(0, -6.283185f/N_FFT_);
        for (int k=0; k<N_FFT_/2; k++)
            wTable_[k] = exp(arg*(float)k);

        // ビット逆順のための表を作成
        uint16_t nShift = __clz(n) + 1;
        for (int k=0; k<n; k++)
            // __rbit(k): ビットの並びを逆にする命令に対応
            bTable_[k] = __rbit(k) >> nShift;
    }

    // FFT の実行
    //      有効な部分 y[0] ~ y[N_FFT_/2]
    //      それ以外の部分は複素共役になるので,計算をしていない
    void FftReal::Execute(const float x[], Complex y[])
    {
        for (int n=0; n<N_FFT_; n++) u_[n] = x[n];

        // 最終ステージを除いた部分
        ExcludeLastStage();

        // 最終ステージ
        y[0] = u_[0] + u_[1];
        y[N_FFT_/2] = u_[0] - u_[1];
        for (int k=2; k<N_FFT_; k+=2) u_[k] = u_[k] + u_[k+1];

        // ビット逆順の並べ替え
        for (int k=1; k<N_FFT_/2; k++) y[k] = u_[bTable_[k]];
    }

    // IFFT の実行
    //      このクラスで計算された FFT の結果の IFFT を計算する
    //      実行結果    x[0] ~ x[N_FFT_-1]
    void FftReal::ExecuteIfft(const Complex y[], float x[])
    {
        int half = N_FFT_/2;

        for (int n=0; n<=half; n++) u_[n] = y[n];
        for (int n=half+1; n<N_FFT_; n++)
            u_[n] = conj(y[N_FFT_-n]);      // 後半は複素共役になっているものとする

        // 最終ステージを除いた部分
        ExcludeLastStage();

        // 最終ステージとビット逆順の並べ替え処理
        x[0] = N_INV_*(u_[0].real() + u_[1].real());
        x[half] = N_INV_*(u_[0].real() - u_[1].real());

        for (int n=2; n<N_FFT_; n+=2)
        {
            float un  = u_[n].real();
            float un1 = u_[n+1].real();
            x[Index(n)]   = N_INV_*(un + un1);
            x[Index(n+1)] = N_INV_*(un - un1);
        }
    }

    // 最終ステージを除いた処理
    void FftReal::ExcludeLastStage()
    {
        uint16_t nHalf = N_FFT_/2;
        for (int stg=1; stg<N_FFT_/2; stg*=2)
        {
            uint16_t nHalf2 = nHalf*2;
            for (int kp=0; kp<N_FFT_; kp+=nHalf2)
            {
                uint16_t kx = 0;
                for (int k=kp; k<kp+nHalf; k++)
                {
                    // バタフライ演算
                    Complex uTmp = u_[k+nHalf];
                    u_[k+nHalf] = (u_[k] - uTmp)*wTable_[kx];
                    u_[k] = u_[k] + uTmp;
                    kx = kx + stg;
                }
            }
            nHalf = nHalf/2;
        }
    }
}