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(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