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