decoding_ica_label.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DECODING_ICA_LABEL_H
#define DECODING_ICA_LABEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "decoding_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QList>
#include <QString>
#include <QPair>
 
//=============================================================================================================
// DEFINE NAMESPACE DECODINGLIB
//=============================================================================================================
 
namespace DECODINGLIB
{
 
//=============================================================================================================
enum class IcaComponentLabel
{
    Brain,
    Eog,
    Ecg,
    Muscle,
    Other
};
 
//=============================================================================================================
struct DECODINGSHARED_EXPORT IcaLabelResult
{
    int componentIndex;         
    IcaComponentLabel label;    
    double confidence;          
 
    static QString labelToString(IcaComponentLabel label);
};
 
//=============================================================================================================
class DECODINGSHARED_EXPORT MlIcaLabel
{
public:
    //=========================================================================================================
    static QList<IcaLabelResult> classify(const Eigen::MatrixXd& matSources,
                                           const Eigen::MatrixXd& matEog,
                                           const Eigen::MatrixXd& matEcg,
                                           double dSFreq,
                                           double dEogThresh = 0.3,
                                           double dEcgThresh = 0.3);
 
    //=========================================================================================================
    static QVector<int> findArtifactComponents(const QList<IcaLabelResult>& labels);
 
    //=========================================================================================================
    static double maxAbsCorrelation(const Eigen::VectorXd& source,
                                     const Eigen::MatrixXd& matRef);
 
    //=========================================================================================================
    static double muscleScore(const Eigen::VectorXd& source, double dSFreq);
 
private:
    MlIcaLabel() = delete;
};
 
} // namespace DECODINGLIB
 
#endif // DECODING_ICA_LABEL_H