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