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(),
                                  bool bRemoveDc = true);
 
    //=========================================================================================================
    static Eigen::RowVectorXd computeVector(const Eigen::RowVectorXd& vecData,
                                            double dSFreq,
                                            int iNfft = 256,
                                            double dOverlap = 0.5,
                                            WindowType window = Hann,
                                            bool bRemoveDc = true);
 
    //=========================================================================================================
    static Eigen::RowVectorXd freqAxis(int iNfft, double dSFreq);
 
private:
    static Eigen::VectorXd buildWindow(int iN, WindowType window);
};
 
} // namespace UTILSLIB
 
#endif // WELCH_PSD_H