rtsensordataworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSENSORDATAWORKER_H
#define BRAINVIEW_RTSENSORDATAWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <QObject>
#include <QList>
#include <QMutex>
#include <QVector>
#include <Eigen/Core>
#include <memory>
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_EXPORT RtSensorDataWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit RtSensorDataWorker(QObject* parent = nullptr);
 
    //=========================================================================================================
    void addData(const Eigen::VectorXf& data);
 
    //=========================================================================================================
    void clear();
 
    //=========================================================================================================
    void setMappingMatrix(const QString& surfaceKey, std::shared_ptr<Eigen::MatrixXf> mat);
 
    //=========================================================================================================
    void setNumberAverages(int numAvr);
 
    //=========================================================================================================
    void setColormapType(const QString& name);
 
    //=========================================================================================================
    void setThresholds(double min, double max);
 
    //=========================================================================================================
    void setLoopState(bool enabled);
 
    //=========================================================================================================
    void setSFreq(double sFreq);
 
    //=========================================================================================================
    void setStreamSmoothedData(bool bStreamSmoothedData);
 
public slots:
    //=========================================================================================================
    void streamData();
 
signals:
    //=========================================================================================================
    void newRtSensorColors(const QString& surfaceKey,
                           const QVector<uint32_t>& colors);
 
    //=========================================================================================================
    void newRtRawSensorData(const Eigen::VectorXf& data);
 
private:
    //=========================================================================================================
    QVector<uint32_t> computeSurfaceColors(const Eigen::VectorXf& sensorData) const;
 
    mutable QMutex m_mutex; 
 
    QList<Eigen::VectorXf> m_lDataQ;     
    QList<Eigen::VectorXf> m_lDataLoopQ; 
    Eigen::VectorXf m_vecAverage;        
    int m_iSampleCtr = 0;                
    int m_iCurrentSample = 0;            
 
    std::shared_ptr<Eigen::MatrixXf> m_mappingMat; 
    QString m_sSurfaceKey;                         
 
    int m_iNumAverages = 1;            
    bool m_bIsLooping = true;          
    bool m_bStreamSmoothedData = true; 
    double m_dSFreq = 1000.0;          
 
    QString m_sColormapType = QStringLiteral("MNE"); 
    double m_dThreshMin = 0.0;                       
    double m_dThreshMax = 0.0;                       
    bool m_bUseAutoNorm = true;                      
};
 
} // namespace DISP3DLIB
 
#endif // BRAINVIEW_RTSENSORDATAWORKER_H