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(const QString& surfaceKey, 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; 
};
 
} // namespace DISP3DLIB
 
#endif // BRAINVIEW_RTSENSORDATACONTROLLER_H