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

//=============================================================================================================
// 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