rtsourcedataworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSOURCEDATAWORKER_H
#define BRAINVIEW_RTSOURCEDATAWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <QObject>
#include <QList>
#include <QMutex>
#include <QVector>
#include <QSharedPointer>
#include <Eigen/Core>
#include <Eigen/SparseCore>
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_EXPORT RtSourceDataWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit RtSourceDataWorker(QObject* parent = nullptr);
 
    //=========================================================================================================
    void addData(const Eigen::VectorXd& data);
 
    //=========================================================================================================
    void clear();
 
    //=========================================================================================================
    void setInterpolationMatrixLeft(QSharedPointer<Eigen::SparseMatrix<float>> mat);
 
    //=========================================================================================================
    void setInterpolationMatrixRight(QSharedPointer<Eigen::SparseMatrix<float>> mat);
 
    //=========================================================================================================
    void setNumberAverages(int numAvr);
 
    //=========================================================================================================
    void setColormapType(const QString& name);
 
    //=========================================================================================================
    void setThresholds(double min, double mid, double max);
 
    //=========================================================================================================
    void setLoopState(bool enabled);
 
    //=========================================================================================================
    void setSFreq(double sFreq);
 
    //=========================================================================================================
    void setStreamSmoothedData(bool bStreamSmoothedData);
 
    //=========================================================================================================
    void setSurfaceColor(const QVector<uint32_t>& baseColorsLh,
                         const QVector<uint32_t>& baseColorsRh);
 
public slots:
    //=========================================================================================================
    void streamData();
 
signals:
    //=========================================================================================================
    void newRtSmoothedData(const QVector<uint32_t>& colorsLh,
                           const QVector<uint32_t>& colorsRh);
 
    //=========================================================================================================
    void newRtRawData(const Eigen::VectorXd& dataLh,
                      const Eigen::VectorXd& dataRh);
 
private:
    //=========================================================================================================
    uint32_t valueToColor(double value, uint8_t alpha = 255) const;
 
    //=========================================================================================================
    QVector<uint32_t> computeHemiColors(const Eigen::VectorXf& sourceData,
                                        const QSharedPointer<Eigen::SparseMatrix<float>>& interpMat,
                                        const QVector<uint32_t>& baseColors) const;
 
    mutable QMutex m_mutex; 
 
    QList<Eigen::VectorXd> m_lDataQ;     
    QList<Eigen::VectorXd> m_lDataLoopQ; 
    Eigen::VectorXd m_vecAverage;        
    int m_iSampleCtr = 0;                
    int m_iCurrentSample = 0;            
 
    QSharedPointer<Eigen::SparseMatrix<float>> m_interpMatLh; 
    QSharedPointer<Eigen::SparseMatrix<float>> m_interpMatRh; 
 
    int m_iNumAverages = 1;            
    bool m_bIsLooping = true;          
    bool m_bStreamSmoothedData = true; 
    double m_dSFreq = 1000.0;          
 
    QString m_sColormapType = QStringLiteral("Hot"); 
    double m_dThreshMin = 0.0;                       
    double m_dThreshMid = 0.5;                       
    double m_dThreshMax = 1.0;                       
 
    QVector<uint32_t> m_baseColorsLh; 
    QVector<uint32_t> m_baseColorsRh; 
};
 
} // namespace DISP3DLIB
 
#endif // BRAINVIEW_RTSOURCEDATAWORKER_H