braintreemodel.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef BRAINTREEMODEL_H
#define BRAINTREEMODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "../disp3D_rhi_global.h"
 
#include <QStandardItemModel>
#include <fs/surface.h>
#include <fs/annotation.h>
#include <mne/mne_bem_surface.h>
#include <mne/mne_source_spaces.h>
#include <fiff/fiff_dig_point.h>
#include <connectivity/network/network.h>
 
namespace INVERSELIB {
class ECDSet;
}
 
class NetworkTreeItem;
 
class SurfaceTreeItem;
class BemTreeItem;

class DISP3DRHISHARED_EXPORT BrainTreeModel : public QStandardItemModel
{
    Q_OBJECT
public:
    explicit BrainTreeModel(QObject *parent = nullptr);
    ~BrainTreeModel() override = default;
 
    // Helper functions to populate the tree
    SurfaceTreeItem* addSurface(const QString &subject, const QString &hemi, const QString &surfType, const FSLIB::Surface &surface);
    
    // Add annotation to an existing surface item? Or separate item?
    // Usually annotation is property of a surface or child of surface.
    // Let's attach it to the surface item for now or find the surface item and update it.
    bool addAnnotation(const QString &subject, const QString &hemi, const FSLIB::Annotation &annotation);
 
    // Add BEM surface
    BemTreeItem* addBemSurface(const QString &subject, const QString &bemName, const MNELIB::MNEBemSurface &bemSurf);
 
    // Helpers to populate the tree
    void addSensors(const QString &type, const QList<QStandardItem*> &items);
    void addDipoles(const INVERSELIB::ECDSet &set);
 
    // Add digitizer points with proper categorization (Cardinal, HPI, EEG, Extra)
    void addDigitizerData(const QList<FIFFLIB::FiffDigPoint> &digitizerPoints);
 
    // Add source space points
    void addSourceSpace(const MNELIB::MNESourceSpaces &srcSpace);
 
    // Add connectivity network
    NetworkTreeItem* addNetwork(const CONNECTIVITYLIB::Network &network, const QString &name = "Network");
 
private:
   // Helpers to find specific items
   QStandardItem* getSubjectItem(const QString &subject);
};
 
#endif // BRAINTREEMODEL_H