morlet_tfr.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef MORLET_TFR_H
#define MORLET_TFR_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT MorletTfrResult
{
    Eigen::MatrixXd matPower;    
    Eigen::RowVectorXd vecFreqs; 
};
 
//=============================================================================================================
class DSPSHARED_EXPORT MorletTfr
{
public:
    //=========================================================================================================
    static MorletTfrResult compute(const Eigen::RowVectorXd& vecData,
                                   double dSFreq,
                                   const Eigen::RowVectorXd& vecFreqs,
                                   double dNCycles = 7.0);
 
    //=========================================================================================================
    static QVector<MorletTfrResult> computeMultiChannel(
        const Eigen::MatrixXd& matData,
        double dSFreq,
        const Eigen::RowVectorXd& vecFreqs,
        double dNCycles = 7.0,
        const Eigen::RowVectorXi& vecPicks = Eigen::RowVectorXi());
 
private:
    //=========================================================================================================
    static Eigen::VectorXcd buildWavelet(double dFreq, double dSFreq, double dNCycles, int& halfLen);

    static int nextPow2(int n);
};
 
} // namespace UTILSLIB
 
#endif // MORLET_TFR_H