welch_psd.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef WELCH_PSD_H
#define WELCH_PSD_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT WelchPsdResult
{
    Eigen::MatrixXd    matPsd;    
    Eigen::RowVectorXd vecFreqs;  
};
 
//=============================================================================================================
class DSPSHARED_EXPORT WelchPsd
{
public:
    enum WindowType { Hann=0, Hamming=1, Blackman=2, FlatTop=3 };
 
    //=========================================================================================================
    static WelchPsdResult compute(const Eigen::MatrixXd&    matData,
                                   double                    dSFreq,
                                   int                       iNfft    = 256,
                                   double                    dOverlap = 0.5,
                                   WindowType                window   = Hann,
                                   const Eigen::RowVectorXi& vecPicks = Eigen::RowVectorXi());
 
    //=========================================================================================================
    static Eigen::RowVectorXd computeVector(const Eigen::RowVectorXd& vecData,
                                             double     dSFreq,
                                             int        iNfft    = 256,
                                             double     dOverlap = 0.5,
                                             WindowType window   = Hann);
 
    //=========================================================================================================
    static Eigen::RowVectorXd freqAxis(int iNfft, double dSFreq);
 
private:
    static Eigen::VectorXd buildWindow(int iN, WindowType window);
};
 
} // namespace UTILSLIB
 
#endif // WELCH_PSD_H