dataloader.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DATALOADER_H
#define DATALOADER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_rhi_global.h"
 
#include <QStringList>
#include <QList>
#include <QStandardItem>
#include <QVector3D>
#include <QMatrix4x4>
#include <memory>
 
#include <fiff/fiff_info.h>
#include <fiff/fiff_evoked.h>
#include <fiff/fiff_coord_trans.h>
#include <fiff/fiff_dig_point.h>
#include <mne/mne_source_spaces.h>
#include <inverse/dipoleFit/ecd_set.h>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
class BrainSurface;
class SensorTreeItem;
 
//=============================================================================================================
class DISP3DRHISHARED_EXPORT DataLoader
{
public:
    DataLoader() = delete; // all-static class
 
    // ── Result structs ────────────────────────────────────────────────

    struct SensorLoadResult
    {
        bool                        hasInfo     = false;
        bool                        hasDigitizer = false;
 
        FIFFLIB::FiffInfo           info;          
        QList<QStandardItem*>       megGradItems;  
        QList<QStandardItem*>       megMagItems;
        QList<QStandardItem*>       eegItems;
        QList<FIFFLIB::FiffDigPoint> digitizerPoints;
 
        std::shared_ptr<BrainSurface> helmetSurface; 
 
        QMatrix4x4                  devHeadTrans;  
        bool                        hasDevHead = false; 
    };
 
    // ── Static I/O methods ────────────────────────────────────────────

    static SensorLoadResult loadSensors(const QString &fifPath,
                                        const QString &megHelmetOverridePath = {});

    static std::shared_ptr<BrainSurface> loadHelmetSurface(
        const QString &helmetFilePath,
        const QMatrix4x4 &devHeadTrans = QMatrix4x4(),
        bool applyTrans = false);

    static INVERSELIB::ECDSet loadDipoles(const QString &dipPath);

    static MNELIB::MNESourceSpaces loadSourceSpace(const QString &fwdPath);

    static bool loadHeadToMriTransform(const QString &transPath,
                                       FIFFLIB::FiffCoordTrans &trans);

    static FIFFLIB::FiffEvoked loadEvoked(const QString &evokedPath, int aveIndex = 0);

    static QStringList probeEvokedSets(const QString &evokedPath);
};
 
#endif // DATALOADER_H