Построение частотного спектра с помощью c++

пожалуйста, смотрите изменения в ответе ниже этого вопроса.

Я написал сценарий для построения частотного спектра синусоидального сигнала с помощью c++. Вот шаги

  1. применение Hanning window
  2. применить FFT с помощью библиотеки fftw3

У меня есть три графика: сигнал, сигнал, когда умножается на функцию Ханнинга, и частотный спектр. Частотный спектр выглядит неправильно. Он должен иметь пик 50 Гц. Любой буду признателен за предложение. Вот код:

#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <fftw3.h>
#include <iostream>
#include <cmath>
#include <fstream>
using namespace std;

int main()
{
int i;
double y;
int N=50;
double Fs=1000;//sampling frequency
double  T=1/Fs;//sample time 
double f=50;//frequency
double *in;
fftw_complex *out;
double t[N];//time vector 
double ff[N];
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;
    ff[i]=1/t[i];
    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 );

  fftw_execute ( plan_forward );

  double v[N];

  for (int i = 0; i < N; i++)
    {

    v[i]=20*log(sqrt(out[i][0]*out[i][0]+ out[i][1]*out[i][1])/N/2);//Here I have calculated the y axis of the spectrum in dB

    }

   fstream myfile;

   myfile.open("example2.txt",fstream::out);

   myfile << "plot '-' using 1:2" << std::endl;

   for(i = 0; i < N; ++i)

    { 

      myfile << ff[i]<< " " << v[i]<< std::endl;

    }

 myfile.close();

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

Я должен добавить, что я построил графики с помощью gnuplot после вставки результатов в example2.формат txt. Поэтому ff [i] vs v[i] должен дать мне частотный спектр.

вот графики:enter image description here частотный спектр и синусоидальное временное окно соответственно: enter image description here

3 ответов


enter image description hereмои частотные интервалы были совершенно неправильными. Согласно http://www.ni.com/white-paper/3995/en/#toc1; диапазон изменения частот и разрешение на x-оси зависит от частоты дискретизации и N. Последняя точка на оси частот должна быть Fs/2-Fs / N и разрешение dF=FS / N.Поэтому я изменил свой скрипт на: (так как разрешение частоты Fs / N по мере увеличения количества образцы изделия N (или уменьшить частоту дискретизации Fs) вы получаете меньшее разрешение частоты и лучшие результаты.)

#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <fftw3.h>
#include <iostream>
#include <cmath>
#include <fstream>
using namespace std;

int main()
{
int i;
double y;
int N=550;//Number of points acquired inside the window
double Fs=200;//sampling frequency
double dF=Fs/N;
double  T=1/Fs;//sample time 
double f=50;//frequency
double *in;
fftw_complex *out;
double t[N];//time vector 
double ff[N];
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];
 }

 for (int i=0; i<= ((N/2)-1);i++)
{ff[i]=Fs*i/N;
}
plan_forward = fftw_plan_dft_r2c_1d ( N, in, out, FFTW_ESTIMATE );

fftw_execute ( plan_forward );

double v[N];

for (int i = 0; i<= ((N/2)-1); i++)
{

v[i]=(20*log(sqrt(out[i][0]*out[i][0]+ out[i][1]*out[i][1])))/N;  //Here   I  have calculated the y axis of the spectrum in dB

   }

fstream myfile;

myfile.open("example2.txt",fstream::out);

myfile << "plot '-' using 1:2" << std::endl;

for(i = 0;i< ((N/2)-1); i++)

{ 

myfile << ff[i]<< " " << v[i]<< std::endl;

}

 myfile.close();

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

Я думаю, что у вас может быть недостаточно образцов, в частности, ссылаться на эту электронику.StackExhcange пост: https://electronics.stackexchange.com/q/12407/84272.

вы пробуете для 50 образцов, так что 25 БПФ бункеров. Вы пробуете на 1000 Гц, поэтому 1000 / 2 / 25 = = 250 Гц на бункеры БПФ. Разрешение bin слишком низкое.

Я думаю, вам нужно снизить частоту дискретизации или увеличить количество образцов.


поскольку ваш вопрос в SO, ваш код может использовать некоторые отступы и улучшение стиля, чтобы сделать его легче читать.

#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <fftw3.h>
#include <iostream>
#include <cmath>
#include <fstream>
using namespace std;

int main(){
    // use meaningful names for all the variables
    int i;  
    double y;
    int N = 550; // number of points acquired inside the window
    double Fs = 200; // sampling frequency
    double dF = Fs / N;
    double  T = 1 / Fs; // sample time 
    double f = 50; // frequency
    double *in;
    fftw_complex *out;
    double t[N]; // time vector 
    double ff[N];
    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];
    }

    for(int i = 0; i <= ((N/2)-1); i++){
        ff[i] = (Fs * i) / N;
    }

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

    fftw_execute(plan_forward);

    double v[N];
    // Here I have calculated the y axis of the spectrum in dB
    for(int i = 0; i <= ((N/2)-1); i++){
        v[i] = (20 * log(sqrt(out[i][0] * out[i][0] + out[i][1] * out[i][1]))) / N;  
    }

    fstream myfile;
    myfile.open("example2.txt", fstream::out);
    myfile << "plot '-' using 1:2" << std::endl;

    for(i = 0; i < ((N/2)-1); i++){ 
        myfile << ff[i] << " " << v[i] << std::endl;
    }
    myfile.close();

    fftw_destroy_plan(plan_forward);
    fftw_free(in);
    fftw_free(out);

    return 0;
    }

ваш код может использовать больше комментариев, особенно перед циклами или вызовами функций, чтобы указать их входное значение (цель) и/или возвращаемое значение (результат).