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