rtsourcedataworker.h

Go to the documentation of this file.

//=============================================================================================================
#ifndef DISP3DLIB_RTSOURCEDATAWORKER_H
#define DISP3DLIB_RTSOURCEDATAWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../../../../disp3D_global.h"
 
#include <disp/plots/helpers/colormap.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QRgb>
#include <QSharedPointer>
#include <QObject>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/SparseCore>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================
 
namespace DISP3DLIB
{
 
struct VisualizationInfo {
    double                      dThresholdX;
    double                      dThresholdZ;
 
    Eigen::VectorXd             vecSensorValues;
    Eigen::MatrixX4f            matOriginalVertColor;
    Eigen::MatrixX4f            matFinalVertColor;
 
    QSharedPointer<Eigen::SparseMatrix<float> >  pMatInterpolationMatrix;         
    QString sColormapType;
    QRgb (*functionHandlerColorMap)(double v, const QString& sColorMap) = DISPLIB::ColorMap::valueToColor;
}; 
struct ColorComputationInfo {
    double                      dThresholdX;
    double                      dThresholdZ;
    int                         iFinalMatSize;
 
    Eigen::VectorXf             vecData;
 
    QRgb (*functionHandlerColorMap)(double v);
};
 
//=============================================================================================================
// DISP3DLIB FORWARD DECLARATIONS
//=============================================================================================================
 
//=============================================================================================================
class DISP3DSHARED_EXPORT RtSourceDataWorker : public QObject
{
    Q_OBJECT
 
public:
    typedef QSharedPointer<RtSourceDataWorker> SPtr;            
    typedef QSharedPointer<const RtSourceDataWorker> ConstSPtr; 
    //=========================================================================================================
    explicit RtSourceDataWorker();
 
    //=========================================================================================================
    void addData(const Eigen::MatrixXd& data);
 
    //=========================================================================================================
    void clear();
 
    //=========================================================================================================
    void setSurfaceColor(const Eigen::MatrixX4f &matColorLeft,
                         const Eigen::MatrixX4f &matColorRight);
 
    //=========================================================================================================
    void setNumberAverages(int iNumAvr);
 
    //=========================================================================================================
    void setStreamSmoothedData(bool bStreamSmoothedData);
 
    //=========================================================================================================
    void setColormapType(const QString& sColormapType);
 
    //=========================================================================================================
    void setThresholds(const QVector3D &vecThresholds);
    
    //=========================================================================================================
    void setLoopState(bool bLoopState);
 
    //=========================================================================================================
    void setSFreq(const double dSFreq);
 
    //=========================================================================================================
    void setInterpolationMatrixLeft(QSharedPointer<Eigen::SparseMatrix<float> > pMatInterpolationMatrixLeft);
 
    //=========================================================================================================
    void setInterpolationMatrixRight(QSharedPointer<Eigen::SparseMatrix<float> > pMatInterpolationMatrixRight);
 
    //=========================================================================================================
    void streamData();
 
protected:
    //=========================================================================================================
    static void normalizeAndTransformToColor(const Eigen::VectorXf& vecData,
                                             Eigen::MatrixX4f &matFinalVertColor,
                                             double dThresholdX,
                                             double dThresholdZ,
                                             QRgb (*functionHandlerColorMap)(double v, const QString& sColorMap),
                                             const QString& sColorMap);
 
    //=========================================================================================================
    static void generateColorsFromSensorValues(VisualizationInfo &visualizationInfoHemi);
 
    QList<Eigen::VectorXd>                              m_lDataQ;                           
    QList<Eigen::VectorXd>                              m_lDataLoopQ;                       
    Eigen::VectorXd                                     m_vecAverage;                       
    bool                                                m_bIsLooping;                       
    bool                                                m_bStreamSmoothedData;              
    int                                                 m_iCurrentSample;                   
    int                                                 m_iAverageSamples;                  
    int                                                 m_iSampleCtr;                       
    double                                              m_dSFreq;                           
    QList<VisualizationInfo>                            m_lHemiVisualizationInfo;           
signals:
    //=========================================================================================================
    void newRtRawData(const Eigen::VectorXd &vecDataVectorLeftHemi,
                      const Eigen::VectorXd &vecDataVectorRightHemi);
 
    //=========================================================================================================
    void newRtSmoothedData(const Eigen::MatrixX4f &matColorMatrixLeftHemi,
                           const Eigen::MatrixX4f &matColorMatrixRightHemi);
};
} // NAMESPACE
 
#endif //DISP3DLIB_RTSOURCEDATAWORKER_H