epoch_extractor.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef EPOCH_EXTRACTOR_DSP_H
#define EPOCH_EXTRACTOR_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 EpochExtractorParams
{
    double dTmin      = -0.2;   
    double dTmax      =  0.5;   
    double dBaseMin   = -0.2;   
    double dBaseMax   =  0.0;   
    double dThreshold =  0.0;   
    bool   bApplyBaseline = true; 
};
 
//=============================================================================================================
class DSPSHARED_EXPORT EpochExtractor
{
public:
    using Params = EpochExtractorParams; 
 
    //=========================================================================================================
    static QVector<MNELIB::MNEEpochData> extract(const Eigen::MatrixXd&  matData,
                                                   const QVector<int>&     eventSamples,
                                                   double                  dSFreq,
                                                   const Params&           params     = Params(),
                                                   const QVector<int>&     eventCodes = QVector<int>());
 
    //=========================================================================================================
    static Eigen::MatrixXd average(const QVector<MNELIB::MNEEpochData>& epochs);
 
    //=========================================================================================================
    static QVector<MNELIB::MNEEpochData> rejectMarked(const QVector<MNELIB::MNEEpochData>& epochs);
 
private:
    //=========================================================================================================
    static void applyBaseline(Eigen::MatrixXd& matEpoch, int iBase0, int iBase1);
};
 
} // namespace UTILSLIB
 
#endif // EPOCH_EXTRACTOR_DSP_H