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