2015-08-20 148 views
0

使用此鏈接的答案:Spectrogram C++ library我寫了一個代碼來計算正弦信號的光譜圖: 1-創建了一個正弦信號。 2-我申請了Hann窗口。 3使用FFTW。 4-計算頻率係數的對數幅度。 這裏是腳本:繪製光譜圖

#include <stdlib.h> 
#include <stdio.h> 
#include <time.h> 
#include <fftw3.h> 
#include <iostream> 
#include <cmath> 

using namespace std; 

int main(void) 
{ 
    int i; 
    double y; 
    int N=256; 
    double Fs=30000;//sampling frequency 
    double T=1/Fs;//sample time 
    double f=5000;//frequency 
    double *in; 
    fftw_complex *out; 
    double t[N-1];//time vector 
    fftw_plan plan_forward; 

    in = (double*) fftw_malloc(sizeof(double) * N); 
    out = (fftw_complex*) fftw_malloc(sizeof(fftw_complex) * N); 
    for (int i=0; i< N;i++) 
    { 
     t[i]=i*T; 
     in[i] =0.7 *sin(2*M_PI*f*t[i]);// generate sine waveform 
     double multiplier = 0.5 * (1 - cos(2*M_PI*i/(N-1)));//Hanning Window 
     in[i] = multiplier * in[i]; 
    } 

    plan_forward = fftw_plan_dft_r2c_1d (N, in, out, FFTW_ESTIMATE); 

    printf ("\n"); 
    printf (" Input Data:\n"); 
    printf ("\n"); 

    for (i = 0; i < N; i++) 
    { 
     printf (" %4d %12f\n", i, in[i]); 
    } 

    fftw_execute (plan_forward); 

    printf ("\n"); 
    printf (" log magnitude of frequency domain components :\n"); 
    printf ("\n"); 

    for (i = 0; i < N; i++) 
    { 
     cout << log(sqrt(out[i][0]*out[i][0]+ out[i][1]*out[i][1])) ; 

    } 
    fftw_destroy_plan (plan_forward); 
    fftw_free (in); 
    fftw_free (out); 
    return 0; 
} 

問題是我該如何從這裏出發?我應該使用哪個庫來繪製譜圖?有什麼建議麼?謝謝。

回答