artifact_detect.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ARTIFACT_DETECT_DSP_H
#define ARTIFACT_DETECT_DSP_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
#include <fiff/fiff_info.h>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QVector>
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB
{
 
//=============================================================================================================
struct DSPSHARED_EXPORT ArtifactDetectEcgParams
{
    double dFilterLow    =  5.0;    
    double dFilterHigh   = 40.0;    
    int    iFilterOrder  =  4;      
    double dThreshFactor =  0.5;    
    double dMinRRSec     =  0.35;   
};
 
//=============================================================================================================
struct DSPSHARED_EXPORT ArtifactDetectEogParams
{
    double dFilterHigh   = 10.0;    
    int    iFilterOrder  =  4;      
    double dThresholdV   = 150e-6;  
    double dMinGapSec    =  0.3;    
};
 
//=============================================================================================================
class DSPSHARED_EXPORT ArtifactDetect
{
public:
    using EcgParams = ArtifactDetectEcgParams; 
    using EogParams = ArtifactDetectEogParams; 
 
    //=========================================================================================================
    static QVector<int> detectEcg(const Eigen::MatrixXd&  matData,
                                   const FIFFLIB::FiffInfo& fiffInfo,
                                   double                   dSFreq,
                                   const EcgParams&         params = EcgParams());
 
    //=========================================================================================================
    static QVector<int> detectEog(const Eigen::MatrixXd&  matData,
                                   const FIFFLIB::FiffInfo& fiffInfo,
                                   double                   dSFreq,
                                   const EogParams&         params = EogParams());
 
private:
    //=========================================================================================================
    static Eigen::RowVectorXd bandpassFilter(const Eigen::RowVectorXd& vecSignal,
                                              double                    dSFreq,
                                              double                    dLow,
                                              double                    dHigh,
                                              int                       iOrder);
 
    //=========================================================================================================
    static QVector<int> findPeaks(const Eigen::RowVectorXd& vecSignal,
                                   double                    dThreshold,
                                   int                       iMinDist);
};
 
} // namespace UTILSLIB
 
#endif // ARTIFACT_DETECT_DSP_H