stcloadingworker.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef STCLOADINGWORKER_H
#define STCLOADINGWORKER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <inv/inv_source_estimate.h>
#include <QObject>
#include <QString>
#include <QSharedPointer>
#include <Eigen/Sparse>
#include <atomic>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class BrainSurface;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT StcLoadingWorker : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    StcLoadingWorker(const QString &lhPath,
                     const QString &rhPath,
                     BrainSurface *lhSurface,
                     BrainSurface *rhSurface,
                     double cancelDist = 0.05,
                     QObject *parent = nullptr);
 
    //=========================================================================================================
    ~StcLoadingWorker();
 
    //=========================================================================================================
    const INVLIB::InvSourceEstimate& stcLh() const { return m_stcLh; }
 
    //=========================================================================================================
    const INVLIB::InvSourceEstimate& stcRh() const { return m_stcRh; }
 
    //=========================================================================================================
    QSharedPointer<Eigen::SparseMatrix<float>> interpolationMatLh() const { return m_interpMatLh; }
 
    //=========================================================================================================
    QSharedPointer<Eigen::SparseMatrix<float>> interpolationMatRh() const { return m_interpMatRh; }
 
    //=========================================================================================================
    bool hasLh() const { return m_hasLh; }
 
    //=========================================================================================================
    bool hasRh() const { return m_hasRh; }
 
    //=========================================================================================================
    void requestCancel() { m_cancelled.store(true, std::memory_order_relaxed); }

    bool isCancelled() const { return m_cancelled.load(std::memory_order_relaxed); }
 
public slots:
    //=========================================================================================================
    void process();
 
signals:
    //=========================================================================================================
    void progress(int percent, const QString &message);
 
    //=========================================================================================================
    void finished(bool success);
 
    //=========================================================================================================
    void error(const QString &message);
 
private:
    QSharedPointer<Eigen::SparseMatrix<float>> computeInterpolationMatrix(
        const Eigen::MatrixX3f &matVertices,
        Eigen::VectorXi &vecSourceVertices,
        double cancelDist,
        const QString &hemiLabel,
        int progressStart,
        int progressEnd);
 
    QString m_lhPath;                                          
    QString m_rhPath;                                          
    BrainSurface *m_lhSurface;                                 
    BrainSurface *m_rhSurface;                                 
    double m_cancelDist;                                       
 
    INVLIB::InvSourceEstimate m_stcLh;                         
    INVLIB::InvSourceEstimate m_stcRh;                         
    bool m_hasLh = false;                                      
    bool m_hasRh = false;                                      
 
    QSharedPointer<Eigen::SparseMatrix<float>> m_interpMatLh;  
    QSharedPointer<Eigen::SparseMatrix<float>> m_interpMatRh;  
    std::atomic<bool> m_cancelled{false};                       
};
 
#endif // STCLOADINGWORKER_H