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>
#include <fiff/fiff_proj.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 = 35.0;   
    double dQrsThreshold = -1.0; 
    double dTStart = 0.0;        
};
 
//=============================================================================================================
struct DSPSHARED_EXPORT ArtifactDetectEogParams
{
    double dFilterLow = 1.0;                                      
    double dFilterHigh = 10.0;                                    
    double dThreshold = std::numeric_limits<double>::quiet_NaN(); 
    double dTStart = 0.0;                                         
};
 
//=============================================================================================================
struct DSPSHARED_EXPORT ArtifactDetectProjParams
{
    double dTmin = -0.2;           
    double dTmax = 0.4;            
    int nGrad = 2;                 
    int nMag = 2;                  
    int nEeg = 2;                  
    double dFilterLow = 1.0;       
    double dFilterHigh = 35.0;     
    double dGradReject = 2000e-13; 
    double dMagReject = 3000e-15;  
    double dEegReject = 50e-6;     
    double dEogReject = 250e-6;    

    static ArtifactDetectProjParams ecg()
    {
        return {};
    }

    static ArtifactDetectProjParams eog()
    {
        ArtifactDetectProjParams p;
        p.dTmax = 0.2;
        p.dEegReject = 500e-6;
        p.dEogReject = 0.0;
        return p;
    }
};
 
//=============================================================================================================
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());
 
    using ProjParams = ArtifactDetectProjParams; 
 
    //=========================================================================================================
    static QList<FIFFLIB::FiffProj> computeProjEcg(const Eigen::MatrixXd& matData,
                                                   const FIFFLIB::FiffInfo& fiffInfo,
                                                   double dSFreq,
                                                   const ProjParams& proj = ProjParams::ecg(),
                                                   const EcgParams& detect = EcgParams());
 
    //=========================================================================================================
    static QList<FIFFLIB::FiffProj> computeProjEog(const Eigen::MatrixXd& matData,
                                                   const FIFFLIB::FiffInfo& fiffInfo,
                                                   double dSFreq,
                                                   const ProjParams& proj = ProjParams::eog(),
                                                   const EogParams& detect = EogParams());
 
private:
    static QVector<int> qrsDetector(const Eigen::RowVectorXd& filtered, double dSFreq, double dThreshold, double dTStart);
};
 
} // namespace UTILSLIB
 
#endif // ARTIFACT_DETECT_DSP_H