rtsourcedataworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSOURCEDATAWORKER_H
#define BRAINVIEW_RTSOURCEDATAWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_rhi_global.h"
 
#include <QObject>
#include <QList>
#include <QMutex>
#include <QVector>
#include <QSharedPointer>
#include <Eigen/Core>
#include <Eigen/SparseCore>
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DRHILIB {
 
//=============================================================================================================
class DISP3DRHISHARED_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 DISP3DRHILIB
 
#endif // BRAINVIEW_RTSOURCEDATAWORKER_H