data3Dtreemodel.h

Go to the documentation of this file.

//=============================================================================================================
#ifndef DISP3DLIB_DATA3DTREEMODEL_H
#define DISP3DLIB_DATA3DTREEMODEL_H

//=============================================================================================================
// INCLUDES
//=============================================================================================================

#include "../../disp3D_global.h"

#include "items/common/types.h"

#include <mne/mne_forwardsolution.h>
#include <mne/mne_bem.h>

#include <connectivity/network/network.h>

//=============================================================================================================
// QT INCLUDES
//=============================================================================================================

#include <QStandardItemModel>
#include <QPointer>

//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================

//=============================================================================================================
// FORWARD DECLARATIONS
//=============================================================================================================

namespace Qt3DCore {
    class QEntity;
}

class QSurfaceFormat;

namespace FSLIB {
    class SurfaceSet;
    class AnnotationSet;
    class Annotation;
    class Surface;
}

namespace MNELIB {
    class MNESourceSpace;
    class MNEBem;
    class MNESourceEstimate;
    class MNEBemSurface;
}

namespace FIFFLIB{
    class FiffEvoked;
    class FiffDigPointSet;
}

namespace INVERSELIB{
    class ECDSet;
}

//=============================================================================================================
// DEFINE NAMESPACE DISP3DLIB
//=============================================================================================================

namespace DISP3DLIB
{

//=============================================================================================================
// DISP3DLIB FORWARD DECLARATIONS
//=============================================================================================================

class MneDataTreeItem;
class NetworkTreeItem;
class EcdDataTreeItem;
class FsSurfaceTreeItem;
class SourceSpaceTreeItem;
class BemTreeItem;
class SensorSetTreeItem;
class DigitizerSetTreeItem;
class SubjectTreeItem;
class MeasurementTreeItem;
class SensorDataTreeItem;

//=============================================================================================================
class DISP3DSHARED_EXPORT Data3DTreeModel : public QStandardItemModel
{
    Q_OBJECT

public:
    typedef QSharedPointer<Data3DTreeModel> SPtr;             
    typedef QSharedPointer<const Data3DTreeModel> ConstSPtr;  
    //=========================================================================================================
    explicit Data3DTreeModel(QObject *parent = 0);

    //=========================================================================================================
    QVariant data(const QModelIndex& index, int role = Qt::DisplayRole) const;
    int columnCount(const QModelIndex &parent) const;
    QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;
    Qt::ItemFlags flags(const QModelIndex &index) const;

    //=========================================================================================================
    QList<FsSurfaceTreeItem*> addSurfaceSet(const QString& sSubject,
                                            const QString& sMriSetName,
                                            const FSLIB::SurfaceSet& surfaceSet,
                                            const FSLIB::AnnotationSet& annotationSet = FSLIB::AnnotationSet());

    //=========================================================================================================
    FsSurfaceTreeItem* addSurface(const QString& sSubject,
                                  const QString& sSet,
                                  const FSLIB::Surface& surface,
                                  const FSLIB::Annotation& annotation = FSLIB::Annotation());

    //=========================================================================================================
    QList<SourceSpaceTreeItem*> addSourceSpace(const QString& sSubject,
                                               const QString& sMeasurementSetName,
                                               const MNELIB::MNESourceSpace& sourceSpace);

    //=========================================================================================================
    QList<SourceSpaceTreeItem*> addForwardSolution(const QString& sSubject,
                                                   const QString& sMeasurementSetName,
                                                   const MNELIB::MNEForwardSolution& forwardSolution);

    //=========================================================================================================
    MneDataTreeItem* addSourceData(const QString& sSubject,
                                   const QString& sMeasurementSetName,
                                   const MNELIB::MNESourceEstimate& tSourceEstimate,
                                   const MNELIB::MNEForwardSolution& tForwardSolution,
                                   const FSLIB::SurfaceSet& tSurfSet,
                                   const FSLIB::AnnotationSet& tAnnotSet);

    //=========================================================================================================
    EcdDataTreeItem* addDipoleFitData(const QString& sSubject,
                                      const QString& sSet,
                                      const INVERSELIB::ECDSet& ecdSet);

    //=========================================================================================================
    QList<NetworkTreeItem*> addConnectivityData(const QString& sSubject,
                                                const QString& sMeasurementSetName,
                                                const QList<CONNECTIVITYLIB::Network>& networkData);

    //=========================================================================================================
    NetworkTreeItem* addConnectivityData(const QString& sSubject,
                                         const QString& sMeasurementSetName,
                                         const CONNECTIVITYLIB::Network& networkData);

    //=========================================================================================================
    BemTreeItem* addBemData(const QString& sSubject,
                            const QString& sBemSetName,
                            const MNELIB::MNEBem& bem);

    //=========================================================================================================
    SensorSetTreeItem* addMegSensorInfo(const QString& sSubject,
                                        const QString& sSensorSetName,
                                        const QList<FIFFLIB::FiffChInfo>& lChInfo,
                                        const MNELIB::MNEBem& sensor = MNELIB::MNEBem(),
                                        const QStringList &bads = QStringList());

    //=========================================================================================================
    SensorSetTreeItem* addEegSensorInfo(const QString& sSubject,
                                        const QString& sSensorSetName,
                                        const QList<FIFFLIB::FiffChInfo>& lChInfo,
                                        const QStringList &bads = QStringList());

    //=========================================================================================================
    DigitizerSetTreeItem* addDigitizerData(const QString& sSubject,
                                           const QString& sMeasurementSetName,
                                           const FIFFLIB::FiffDigPointSet &digitizer);

    //=========================================================================================================
    SensorDataTreeItem* addSensorData(const QString& sSubject,
                                      const QString& sMeasurementSetName,
                                      const Eigen::MatrixXd& matSensorData,
                                      const MNELIB::MNEBemSurface& tBemSurface,
                                      const FIFFLIB::FiffInfo &fiffInfo,
                                      const QString &sDataType);

    //=========================================================================================================
    QPointer<Qt3DCore::QEntity> getRootEntity();

protected:
    //=========================================================================================================
    SubjectTreeItem* addSubject(const QString& sSubject);

    //=========================================================================================================
    SensorDataTreeItem *addCpuSensorData(const QString& sSubject,
                                         const QString& sMeasurementSetName,
                                         const Eigen::MatrixXd& matSensorData,
                                         const MNELIB::MNEBemSurface& tBemSurface,
                                         const FIFFLIB::FiffInfo &fiffInfo,
                                         const QString &sDataType,
                                         const double dCancelDist,
                                         const QString &sInterpolationFunction);

    //=========================================================================================================
    SensorDataTreeItem *addGpuSensorData(const QString& sSubject,
                                         const QString& sMeasurementSetName,
                                         const Eigen::MatrixXd& matSensorData,
                                         const MNELIB::MNEBemSurface& tBemSurface,
                                         const FIFFLIB::FiffInfo &fiffInfo,
                                         const QString &sDataType,
                                         const double dCancelDist,
                                         const QString &sInterpolationFunction);

    //=========================================================================================================
    void initMetatypes();

    QStandardItem*                   m_pRootItem;            
    QPointer<Qt3DCore::QEntity>      m_pModelEntity;         
};
} // NAMESPACE

#endif // DISP3DLIB_DATA3DTREEMODEL_H