rtsourceinterpolationmatworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H
#define BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_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 FsLabel;
}
 
//=============================================================================================================
// DEFINE NAMESPACE
//=============================================================================================================
 
namespace DISP3DLIB
{
 
//=============================================================================================================
class DISP3DSHARED_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::FsLabel>& lLabels,
                               const Eigen::VectorXi& vecVertNo);
 
    //=========================================================================================================
    void setAnnotationInfoRight(const Eigen::VectorXi& vecLabelIds,
                                const QList<FSLIB::FsLabel>& 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::FsLabel>& lLabels,
        const QMap<qint32, qint32>& mapLabelIdSrc,
        const QList<int>& vertNos,
        int nVertices);
 
    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::FsLabel> m_lLabelsLh;          
    QMap<qint32, qint32> m_mapLabelIdSourcesLh; 
    QList<int> m_vertNosLh;                     
    int m_nAnnotationVerticesLh = 0;            
    bool m_bAnnotationLhInit = false;           
 
    // RH annotation data
    QList<FSLIB::FsLabel> m_lLabelsRh;          
    QMap<qint32, qint32> m_mapLabelIdSourcesRh; 
    QList<int> m_vertNosRh;                     
    int m_nAnnotationVerticesRh = 0;            
    bool m_bAnnotationRhInit = false;           
};
 
} // namespace DISP3DLIB
 
#endif // BRAINVIEW_RTSOURCEINTERPOLATIONMATWORKER_H