mritreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "mritreeitem.h"
#include "../freesurfer/fsannotationtreeitem.h"
#include "../freesurfer/fssurfacetreeitem.h"
#include "../hemisphere/hemispheretreeitem.h"
 
#include <fs/annotationset.h>
#include <fs/surfaceset.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace FSLIB;
using namespace DISP3DLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
MriTreeItem::MriTreeItem(int iType,
                         const QString& text)
: AbstractTreeItem(iType, text)
{
    initItem();
}
 
//=============================================================================================================
 
void MriTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Checked);
    this->setToolTip("MRI item");
}
 
//=============================================================================================================
 
QList<FsSurfaceTreeItem*> MriTreeItem::addData(const SurfaceSet& tSurfaceSet,
                                               const AnnotationSet& tAnnotationSet,
                                               Qt3DCore::QEntity* p3DEntityParent)
{
    //Generate child items based on surface set input parameters
    QList<FsSurfaceTreeItem*> returnItemList;
 
    //Search for already created hemi items and add source space data respectivley
    for(int i = 0; i < tSurfaceSet.size(); ++i) {
        if(i < tAnnotationSet.size()) {
            returnItemList.append(addData(tSurfaceSet[i], tAnnotationSet[i], p3DEntityParent));
        } else {
            returnItemList.append(addData(tSurfaceSet[i], Annotation(), p3DEntityParent));
        }
    }
 
    return returnItemList;
}
 
//=============================================================================================================
 
FsSurfaceTreeItem* MriTreeItem::addData(const Surface& tSurface,
                                        const Annotation& tAnnotation,
                                        Qt3DCore::QEntity* p3DEntityParent)
{
    //Generate child items based on surface set input parameters
    FsSurfaceTreeItem* pReturnItem = Q_NULLPTR;
 
    QList<QStandardItem*> itemList = this->findChildren(Data3DTreeModelItemTypes::HemisphereItem);
 
    bool hemiItemFound = false;
 
    //Search for already created hemi items and add data respectivley
    for(int j = 0; j < itemList.size(); ++j) {
        if(HemisphereTreeItem* pHemiItem = dynamic_cast<HemisphereTreeItem*>(itemList.at(j))) {
            if(pHemiItem->data(Data3DTreeModelItemRoles::SurfaceHemi).toInt() == tSurface.hemi()) {
                hemiItemFound = true;
 
                if(tAnnotation.hemi() == tSurface.hemi()) {
                    pReturnItem = pHemiItem->addData(tSurface, tAnnotation, p3DEntityParent);
                } else {
                    pReturnItem = pHemiItem->addData(tSurface, Annotation(), p3DEntityParent);
                }
            }
        }
    }
 
    //If item does not exist yet, create it here.
    if(!hemiItemFound) {
        HemisphereTreeItem* pHemiItem = new HemisphereTreeItem(Data3DTreeModelItemTypes::HemisphereItem);
 
        if(tAnnotation.hemi() == tSurface.hemi()) {
            pReturnItem = pHemiItem->addData(tSurface, tAnnotation, p3DEntityParent);
        } else {
            pReturnItem = pHemiItem->addData(tSurface, Annotation(), p3DEntityParent);
        }
 
        QList<QStandardItem*> list;
        list << pHemiItem;
        list << new QStandardItem(pHemiItem->toolTip());
        this->appendRow(list);
    }
 
    return pReturnItem;
}