dataloader.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef DATALOADER_H
#define DATALOADER_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_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 <inv/dipole_fit/inv_ecd_set.h>
 
//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================
 
namespace DISP3DLIB
{
 
class BrainSurface;
class SensorTreeItem;
 
//=============================================================================================================
class DISP3DSHARED_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 INVLIB::InvEcdSet 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);
};
 
} // namespace DISP3DLIB
 
#endif // DATALOADER_H