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