sourceestimatemanager.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef SOURCEESTIMATEMANAGER_H
#define SOURCEESTIMATEMANAGER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_global.h"
 
#include <QObject>
#include <QVector>
#include <QMap>
#include <QString>
#include <Eigen/Core>
#include <memory>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class QThread;
class BrainSurface;
class SourceEstimateOverlay;
class StcLoadingWorker;
class RtSourceDataController;
struct SubView;
 
//=============================================================================================================
class DISP3DSHARED_EXPORT SourceEstimateManager : public QObject
{
    Q_OBJECT
 
public:
    //=========================================================================================================
    explicit SourceEstimateManager(QObject *parent = nullptr);
 
    //=========================================================================================================
    ~SourceEstimateManager() override;
 
    // ── Loading ────────────────────────────────────────────────────────

    bool load(const QString &lhPath, const QString &rhPath,
              const QMap<QString, std::shared_ptr<BrainSurface>> &surfaces,
              const QString &activeSurfaceType);

    void cancelLoading();

    bool isLoading() const { return m_isLoading; }

    bool isLoaded() const;
 
    // ── Time-point navigation ──────────────────────────────────────────

    void setTimePoint(int index,
                      const QMap<QString, std::shared_ptr<BrainSurface>> &surfaces,
                      const SubView &singleView,
                      const QVector<SubView> &subViews);

    int currentTimePoint() const { return m_currentTimePoint; }

    float tstep() const;

    float tmin() const;

    int numTimePoints() const;

    int closestIndex(float timeSec) const;
 
    // ── Colormap / thresholds ──────────────────────────────────────────

    void setColormap(const QString &name);

    void setThresholds(float min, float mid, float max);
 
    // ── Real-time streaming ────────────────────────────────────────────

    void startStreaming(const QMap<QString, std::shared_ptr<BrainSurface>> &surfaces,
                        const SubView &singleView,
                        const QVector<SubView> &subViews);

    void stopStreaming();

    bool isStreaming() const { return m_isStreaming; }

    void pushData(const Eigen::VectorXd &data);

    void setInterval(int msec);

    void setLooping(bool enabled);
 
    // ── Accessors ──────────────────────────────────────────────────────

    const SourceEstimateOverlay *overlay() const;
 
signals:
    //=========================================================================================================
    void loaded(int numTimePoints);

    void thresholdsUpdated(float min, float mid, float max);

    void timePointChanged(int index, float time);

    void loadingProgress(int percent, const QString &message);

    void realtimeColorsAvailable(const QVector<uint32_t> &colorsLh,
                                  const QVector<uint32_t> &colorsRh);
 
private slots:
    void onStcLoadingFinished(bool success);
 
private:
    std::unique_ptr<SourceEstimateOverlay> m_overlay;       
    std::unique_ptr<RtSourceDataController> m_rtController; 
    QThread *m_loadingThread = nullptr;                     
    StcLoadingWorker *m_stcWorker = nullptr;                
    int m_currentTimePoint = 0;                             
    bool m_isLoading = false;                               
    bool m_isStreaming = false;                              
};
 
#endif // SOURCEESTIMATEMANAGER_H