eog_regression.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef EOG_REGRESSION_DSP_H
#define EOG_REGRESSION_DSP_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "dsp_global.h"
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QStringList>
#include <QVector>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FIFFLIB { class FiffInfo; }
 
//=============================================================================================================
// DEFINE NAMESPACE UTILSLIB
//=============================================================================================================
 
namespace UTILSLIB {
 
//=============================================================================================================
class DSPSHARED_EXPORT EogRegression
{
public:
    //=========================================================================================================
    EogRegression() = default;
 
    //=========================================================================================================
    void fit(const Eigen::MatrixXd& data,
             const FIFFLIB::FiffInfo& info,
             const QStringList& eogChannels = QStringList());
 
    //=========================================================================================================
    void apply(Eigen::MatrixXd& data,
               const FIFFLIB::FiffInfo& info) const;
 
    //=========================================================================================================
    static void fitApply(Eigen::MatrixXd& data,
                         const FIFFLIB::FiffInfo& info,
                         const QStringList& eogChannels = QStringList());
 
    //=========================================================================================================
    const Eigen::MatrixXd& coefficients() const { return m_matBeta; }
 
    //=========================================================================================================
    bool isFitted() const { return m_bFitted; }
 
private:
    Eigen::MatrixXd m_matBeta;       
    QVector<int> m_vecEogIndices;    
    QVector<int> m_vecTargetIndices; 
    bool m_bFitted = false;          
};
 
} // namespace UTILSLIB
 
#endif // EOG_REGRESSION_DSP_H