braintreemodel.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "braintreemodel.h"
#include "model/items/surfacetreeitem.h"
#include "model/items/bemtreeitem.h"
#include "model/items/sensortreeitem.h"
#include "model/items/dipoletreeitem.h"
#include "model/items/sourcespacetreeitem.h"
#include "model/items/digitizersettreeitem.h"
#include "model/items/networktreeitem.h"
#include <inv/dipole_fit/inv_ecd_set.h>
#include <mne/mne_hemisphere.h>
 
namespace DISP3DLIB
{
 
BrainTreeModel::BrainTreeModel(QObject* parent)
: QStandardItemModel(parent)
{
    // Set headers
    setHorizontalHeaderLabels(QStringList() << "Data" << "Description");
}
 
SurfaceTreeItem* BrainTreeModel::addSurface(const QString& subject, const QString& hemi, const QString& surfType, const FSLIB::FsSurface& surface)
{
    QStandardItem* subjectItem = getSubjectItem(subject);
 
    // Find or create Hemi item
    QStandardItem* hemiItem = nullptr;
    for (int i = 0; i < subjectItem->rowCount(); ++i) {
        if (subjectItem->child(i)->text() == hemi) {
            hemiItem = subjectItem->child(i);
            break;
        }
    }
 
    if (!hemiItem) {
        hemiItem = new QStandardItem(hemi);
        subjectItem->appendRow(hemiItem);
    }
 
    // Create FsSurface Item
    SurfaceTreeItem* surfItem = new SurfaceTreeItem(surfType);
    surfItem->setSurfaceData(surface);
 
    hemiItem->appendRow(surfItem);
    return surfItem;
}
 
bool BrainTreeModel::addAnnotation(const QString& subject, const QString& hemi, const FSLIB::FsAnnotation& annotation)
{
    QStandardItem* subjectItem = getSubjectItem(subject);
    if (!subjectItem)
        return false;
 
    QStandardItem* hemiItem = nullptr;
    for (int i = 0; i < subjectItem->rowCount(); ++i) {
        if (subjectItem->child(i)->text() == hemi) {
            hemiItem = subjectItem->child(i);
            break;
        }
    }
 
    if (!hemiItem)
        return false;
 
    // Apply annotation to ALL surfaces in this hemi? Or just one?
    // Ex_brain_view logic applies atlas to the loaded surfaces of that hemi.
    // So we iterate children of hemiItem and if they are SurfaceTreeItems, set annotation.
    bool applied = false;
    for (int i = 0; i < hemiItem->rowCount(); ++i) {
        SurfaceTreeItem* item = dynamic_cast<SurfaceTreeItem*>(hemiItem->child(i));
        if (item) {
            item->setAnnotationData(annotation);
            applied = true;
        }
    }
    return applied;
}
 
QStandardItem* BrainTreeModel::getSubjectItem(const QString& subject)
{
    // Search top level
    QList<QStandardItem*> items = findItems(subject);
    if (!items.isEmpty()) {
        return items.first();
    }
 
    // Create new
    QStandardItem* item = new QStandardItem(subject);
    appendRow(item);
    return item;
}
 
BemTreeItem* BrainTreeModel::addBemSurface(const QString& subject, const QString& bemName, const MNELIB::MNEBemSurface& bemSurf)
{
    QStandardItem* subjItem = getSubjectItem(subject);
 
    // Check if BEM item already exists to avoid duplicates?
    // Usually BEM is added once.
 
    BemTreeItem* bemItem = new BemTreeItem(bemName, bemSurf);
    bemItem->setVisible(true); // Default to visible
 
    // Set default colors based on type/name
    // Set colors based on ID (matching zeiss-intent)
    // 4=Head (Reddish), 3=OuterSkull (Greenish), 1=InnerSkull (Blueish)
    if (bemSurf.id == 4) {
        bemItem->setColor(QColor(128, 77, 77)); // Reddish
    } else if (bemSurf.id == 3) {
        bemItem->setColor(QColor(77, 128, 77)); // Greenish
    } else if (bemSurf.id == 1) {
        bemItem->setColor(QColor(77, 77, 128)); // Blueish
    } else {
        bemItem->setColor(QColor(100, 100, 100)); // Grey default
    }
 
    subjItem->appendRow(bemItem);
 
    return bemItem;
}
 
//=============================================================================================================
 
void BrainTreeModel::addSensors(const QString& type, const QList<QStandardItem*>& items)
{
    QStandardItem* parentItem = new QStandardItem(type);
    parentItem->setCheckable(true);
    parentItem->setCheckState(Qt::Checked);
 
    for (auto* item : items) {
        parentItem->appendRow(item);
    }
 
    this->appendRow(parentItem);
}
 
//=============================================================================================================
 
void BrainTreeModel::addDipoles(const INVLIB::InvEcdSet& set)
{
    DipoleTreeItem* item = new DipoleTreeItem("Dipoles", set);
    item->setCheckable(true);
    item->setCheckState(Qt::Checked);
    this->appendRow(item);
}
 
//=============================================================================================================
 
void BrainTreeModel::addSourceSpace(const MNELIB::MNESourceSpaces& srcSpace)
{
    QStandardItem* parentItem = new QStandardItem("Source Space");
    parentItem->setCheckable(true);
    parentItem->setCheckState(Qt::Checked);
 
    // Color: pinkish-red matching disp3D's source space rendering
    QColor srcColor(212, 28, 92);
 
    for (int h = 0; h < srcSpace.size(); ++h) {
        const auto& hemi = srcSpace[h];
        QString hemiLabel = (h == 0) ? "LH" : "RH";
 
        // Collect all source point positions for this hemisphere
        QVector<QVector3D> positions;
        positions.reserve(hemi.vertno.size());
        for (int i = 0; i < hemi.vertno.size(); ++i) {
            int vIdx = hemi.vertno(i);
            if (vIdx < 0 || vIdx >= hemi.rr.rows())
                continue;
            positions.append(QVector3D(hemi.rr(vIdx, 0), hemi.rr(vIdx, 1), hemi.rr(vIdx, 2)));
        }
 
        // One item per hemisphere with all positions batched (0.00075 = 0.75mm radius, same as disp3D)
        parentItem->appendRow(new SourceSpaceTreeItem(hemiLabel, positions, srcColor, 0.00075f));
        qDebug() << "BrainTreeModel: Source space" << hemiLabel
                 << "- points:" << positions.size()
                 << "coord_frame:" << hemi.coord_frame;
        if (!positions.isEmpty()) {
            qDebug() << "  First point:" << positions.first()
                     << "  Last point:" << positions.last();
        }
    }
 
    this->appendRow(parentItem);
}
 
//=============================================================================================================
 
void BrainTreeModel::addDigitizerData(const QList<FIFFLIB::FiffDigPoint>& digitizerPoints)
{
    if (digitizerPoints.isEmpty())
        return;
 
    auto* setItem = new DigitizerSetTreeItem("Digitizer", digitizerPoints);
    this->appendRow(setItem);
 
    qInfo() << "BrainTreeModel: Added digitizer set with"
            << setItem->totalPointCount() << "points in"
            << setItem->rowCount() << "categories";
}
 
//=============================================================================================================
 
NetworkTreeItem* BrainTreeModel::addNetwork(const CONNECTIVITYLIB::Network& network, const QString& name)
{
    QString displayName = name;
    if (displayName.isEmpty()) {
        displayName = network.getConnectivityMethod();
        if (displayName.isEmpty())
            displayName = "Network";
    }
 
    QString objectKey = "net_" + displayName.toLower().replace(" ", "_");
 
    auto* item = new NetworkTreeItem(displayName, objectKey);
    item->setCheckable(true);
    item->setCheckState(Qt::Checked);
 
    this->appendRow(item);
 
    qDebug() << "BrainTreeModel: Added network" << displayName
             << "with" << network.getNodes().size() << "nodes and"
             << network.getFullEdges().size() << "edges";
 
    return item;
}
 
} // namespace DISP3DLIB