rtsourceinterpolationmatworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H
#define BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_rhi_global.h"
 
#include <QObject>
#include <QMutex>
#include <QSharedPointer>
#include <QVector>
#include <QMap>
#include <QList>
#include <QString>
#include <vector>
#include <Eigen/Core>
#include <Eigen/SparseCore>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace FSLIB {
    class Label;
}
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DRHILIB {
 
//=============================================================================================================
class DISP3DRHISHARED_EXPORT RtSourceInterpolationMatWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    enum VisualizationType {
        InterpolationBased = 0,   
        AnnotationBased    = 1    
    };
 
    //=========================================================================================================
    explicit RtSourceInterpolationMatWorker(QObject *parent = nullptr);
 
    //=========================================================================================================
    void setInterpolationFunction(const QString &sInterpolationFunction);
 
    //=========================================================================================================
    void setCancelDistance(double dCancelDist);
 
    //=========================================================================================================
    void setInterpolationInfoLeft(const Eigen::MatrixX3f &matVertices,
                                    const std::vector<Eigen::VectorXi> &vecNeighborVertices,
                                    const Eigen::VectorXi &vecSourceVertices);
    //=========================================================================================================
    void setInterpolationInfoRight(const Eigen::MatrixX3f &matVertices,
                                     const std::vector<Eigen::VectorXi> &vecNeighborVertices,
                                     const Eigen::VectorXi &vecSourceVertices);
    //=========================================================================================================
    void setVisualizationType(int iVisType);
 
    //=========================================================================================================
    void setAnnotationInfoLeft(const Eigen::VectorXi &vecLabelIds,
                               const QList<FSLIB::Label> &lLabels,
                               const Eigen::VectorXi &vecVertNo);
 
    //=========================================================================================================
    void setAnnotationInfoRight(const Eigen::VectorXi &vecLabelIds,
                                const QList<FSLIB::Label> &lLabels,
                                const Eigen::VectorXi &vecVertNo);
 
public slots:
    //=========================================================================================================
    void computeInterpolationMatrix();
 
signals:
    //=========================================================================================================
    void newInterpolationMatrixLeftAvailable(QSharedPointer<Eigen::SparseMatrix<float>> interpMat);
 
    //=========================================================================================================
    void newInterpolationMatrixRightAvailable(QSharedPointer<Eigen::SparseMatrix<float>> interpMat);
 
private:
    //=========================================================================================================
    static double (*resolveInterpolationFunction(const QString &name))(double);
 
    //=========================================================================================================
    static QSharedPointer<Eigen::SparseMatrix<float>> computeHemi(
        const Eigen::MatrixX3f &matVertices,
        const std::vector<Eigen::VectorXi> &vecNeighborVertices,
        Eigen::VectorXi vecSourceVertices,
        double dCancelDist,
        double (*interpFunc)(double));
 
    //=========================================================================================================
    static QSharedPointer<Eigen::SparseMatrix<float>> computeAnnotationOperator(
        const QList<FSLIB::Label> &lLabels,
        const QMap<qint32, qint32> &mapLabelIdSrc,
        const QList<int> &vertNos);
 
    mutable QMutex m_mutex;                                      
 
    // LH geometry
    Eigen::MatrixX3f m_matVerticesLh;                            
    std::vector<Eigen::VectorXi> m_vecNeighborsLh;               
    Eigen::VectorXi m_vecSourceVerticesLh;                        
    bool m_hasLh = false;                                        
 
    // RH geometry
    Eigen::MatrixX3f m_matVerticesRh;                            
    std::vector<Eigen::VectorXi> m_vecNeighborsRh;               
    Eigen::VectorXi m_vecSourceVerticesRh;                        
    bool m_hasRh = false;                                        
 
    // Interpolation parameters
    double m_dCancelDist = 0.05;                                 
    QString m_sInterpolationFunction = QStringLiteral("cubic");  
 
    // Visualization type
    int m_iVisualizationType = InterpolationBased;               
 
    // LH annotation data
    QList<FSLIB::Label> m_lLabelsLh;                             
    QMap<qint32, qint32> m_mapLabelIdSourcesLh;                  
    QList<int> m_vertNosLh;                                      
    bool m_bAnnotationLhInit = false;                             
 
    // RH annotation data
    QList<FSLIB::Label> m_lLabelsRh;                             
    QMap<qint32, qint32> m_mapLabelIdSourcesRh;                  
    QList<int> m_vertNosRh;                                      
    bool m_bAnnotationRhInit = false;                             
};
 
} // namespace DISP3DRHILIB
 
#endif // BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H