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