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