rtsensordatacontroller.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSENSORDATACONTROLLER_H
#define BRAINVIEW_RTSENSORDATACONTROLLER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <fiff/fiff_evoked.h>
#include <fiff/fiff_coord_trans.h>
 
#include <QObject>
#include <QVector>
#include <QString>
#include <Eigen/Core>
#include <memory>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class QThread;
class QTimer;
 
namespace DISP3DLIB {
    class RtSensorDataWorker;
    class RtSensorInterpolationMatWorker;
}
 
//=============================================================================================================
class DISP3DSHARED_EXPORT RtSensorDataController : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit RtSensorDataController(QObject *parent = nullptr);
 
    //=========================================================================================================
    ~RtSensorDataController() override;
 
    //=========================================================================================================
    void addData(const Eigen::VectorXf &data);
 
    //=========================================================================================================
    void setMappingMatrix(std::shared_ptr<Eigen::MatrixXf> mat);
 
    //=========================================================================================================
    void setStreamingState(bool state);
 
    //=========================================================================================================
    bool isStreaming() const;
 
    //=========================================================================================================
    void setTimeInterval(int msec);
 
    //=========================================================================================================
    void setNumberAverages(int numAvr);
 
    //=========================================================================================================
    void setColormapType(const QString &name);
 
    //=========================================================================================================
    void setThresholds(double min, double max);
 
    //=========================================================================================================
    void setLoopState(bool enabled);
 
    //=========================================================================================================
    void setSFreq(double sFreq);
 
    //=========================================================================================================
    void clearData();
 
    //=========================================================================================================
    void setStreamSmoothedData(bool bStreamSmoothedData);
 
    //=========================================================================================================
    // ── On-the-fly interpolation matrix computation ─────────────────────
    //=========================================================================================================

    void setEvoked(const FIFFLIB::FiffEvoked &evoked);
 
    //=========================================================================================================
    void setTransform(const FIFFLIB::FiffCoordTrans &trans, bool applySensorTrans);
 
    //=========================================================================================================
    void setMegFieldMapOnHead(bool onHead);
 
    //=========================================================================================================
    void setMegSurface(const QString &surfaceKey,
                       const Eigen::MatrixX3f &vertices,
                       const Eigen::MatrixX3f &normals,
                       const Eigen::MatrixX3i &triangles);
 
    //=========================================================================================================
    void setEegSurface(const QString &surfaceKey,
                       const Eigen::MatrixX3f &vertices);
 
    //=========================================================================================================
    void setBadChannels(const QStringList &bads);
 
    //=========================================================================================================
    void recomputeMapping();
 
signals:
    //=========================================================================================================
    void newSensorColorsAvailable(const QString &surfaceKey,
                                  const QVector<uint32_t> &colors);
 
    //=========================================================================================================
    void newRawSensorDataAvailable(const Eigen::VectorXf &data);
 
    //=========================================================================================================
    void newMegMappingAvailable(const QString &surfaceKey,
                                std::shared_ptr<Eigen::MatrixXf> mappingMat,
                                const QVector<int> &pick);
 
    //=========================================================================================================
    void newEegMappingAvailable(const QString &surfaceKey,
                                std::shared_ptr<Eigen::MatrixXf> mappingMat,
                                const QVector<int> &pick);
 
private slots:
    void onNewMegMapping(const QString &surfaceKey,
                         std::shared_ptr<Eigen::MatrixXf> mappingMat,
                         const QVector<int> &pick);
    void onNewEegMapping(const QString &surfaceKey,
                         std::shared_ptr<Eigen::MatrixXf> mappingMat,
                         const QVector<int> &pick);
 
private:
    QThread *m_pWorkerThread = nullptr;                         
    DISP3DLIB::RtSensorDataWorker *m_pWorker = nullptr;      
    QTimer *m_pTimer = nullptr;                                 
    bool m_bIsStreaming = false;                                 
    int m_iTimeInterval = 17;                                   
 
    QThread *m_pInterpThread = nullptr;                          
    DISP3DLIB::RtSensorInterpolationMatWorker *m_pInterpWorker = nullptr; 
};
 
#endif // BRAINVIEW_RTSENSORDATACONTROLLER_H