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
//=============================================================================================================
 
namespace DISP3DLIB
{
 
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};                     
};
 
} // namespace DISP3DLIB
 
#endif // STCLOADINGWORKER_H