xdawn.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef XDAWN_DSP_H
#define XDAWN_DSP_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
#include <mne/mne_epoch_data.h>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT XdawnResult
{
    Eigen::MatrixXd matFilters;      
    Eigen::MatrixXd matPatterns;     
    Eigen::MatrixXd matSignalCov;    
    Eigen::MatrixXd matNoiseCov;     
    Eigen::MatrixXd matTargetEvoked; 
    int iTargetEvent = 1;            
    bool bValid = false;             
};
 
//=============================================================================================================
class DSPSHARED_EXPORT Xdawn
{
public:
    //=========================================================================================================
    static XdawnResult fit(const QVector<MNELIB::MNEEpochData>& epochs,
                           int iTargetEvent = 1,
                           int nComponents = 2,
                           double dReg = 1e-6);
 
    //=========================================================================================================
    static Eigen::MatrixXd apply(const Eigen::MatrixXd& matEpoch,
                                 const XdawnResult& result);
 
    //=========================================================================================================
    static Eigen::MatrixXd denoise(const Eigen::MatrixXd& matEpoch,
                                   const XdawnResult& result,
                                   int nComponents = -1);
 
    //=========================================================================================================
    static QVector<MNELIB::MNEEpochData> denoiseEpochs(const QVector<MNELIB::MNEEpochData>& epochs,
                                                       const XdawnResult& result,
                                                       int nComponents = -1);
};
 
} // namespace UTILSLIB
 
#endif // XDAWN_DSP_H