rt_hpis.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef RT_HPIS_RTPROCESSING_H
#define RT_HPIS_RTPROCESSING_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../dsp_global.h"
#include <inv/hpi/inv_sensor_set.h>
#include <inv/hpi/inv_signal_model.h>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QThread>
#include <QMutex>
#include <QSharedPointer>
#include <QVector>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FIFFLIB
{
class FiffInfo;
}
 
namespace INVLIB
{
class InvSensorSet;
class InvHpiFit;
struct HpiFitResult;
class InvHpiModelParameters;
}
 
//=============================================================================================================
// DEFINE NAMESPACE RTPROCESSINGLIB
//=============================================================================================================
 
namespace RTPROCESSINGLIB
{
 
//=============================================================================================================
class DSPSHARED_EXPORT RtHpiWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit RtHpiWorker(const INVLIB::InvSensorSet sensorSet);
 
    //=========================================================================================================
    void doWork(const Eigen::MatrixXd& matData,
                const Eigen::MatrixXd& matProjectors,
                const INVLIB::InvHpiModelParameters& hpiModelParameters,
                const Eigen::MatrixXd& matCoilsHead);
 
protected:
    //=========================================================================================================
    QSharedPointer<INVLIB::InvHpiFit> m_pHpiFit; 
 
signals:
    void resultReady(const INVLIB::HpiFitResult& fitResult);
};
 
//=============================================================================================================
class DSPSHARED_EXPORT RtHpi : public QObject
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<RtHpi> SPtr;            
    typedef QSharedPointer<const RtHpi> ConstSPtr; 
 
    //=========================================================================================================
    explicit RtHpi(const INVLIB::InvSensorSet sensorSet,
                   QObject* parent = 0);
 
    //=========================================================================================================
    ~RtHpi();
 
    //=========================================================================================================
    void append(const Eigen::MatrixXd& data);
 
    //=========================================================================================================
    void setModelParameters(INVLIB::InvHpiModelParameters hpiModelParameters);
 
    //=========================================================================================================
    void setProjectionMatrix(const Eigen::MatrixXd& matProjectors);
 
    //=========================================================================================================
    void setHpiDigitizer(const Eigen::MatrixXd& matCoilsHead);
 
    //=========================================================================================================
    void restart();
 
    //=========================================================================================================
    void stop();
 
protected:
    //=========================================================================================================
    void handleResults(const INVLIB::HpiFitResult& fitResult);
 
    QThread m_workerThread;          
    Eigen::MatrixXd m_matCoilsHead;  
    Eigen::MatrixXd m_matProjectors; 
    INVLIB::InvSensorSet m_sensorSet;
    INVLIB::InvHpiModelParameters m_modelParameters;
 
signals:
    void newHpiFitResultAvailable(const INVLIB::HpiFitResult& fitResult);
    void operate(const Eigen::MatrixXd& matData,
                 const Eigen::MatrixXd& matProjectors,
                 const INVLIB::InvHpiModelParameters& hpiModelParameters,
                 const Eigen::MatrixXd& matCoilsHead);
};
 
//=============================================================================================================
// INLINE DEFINITIONS
//=============================================================================================================
} // NAMESPACE
 
#endif // RT_HPIS_RTPROCESSING_H