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