fsannotationtreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "fsannotationtreeitem.h"
#include "../common/metatreeitem.h"
 
#include <fs/label.h>
#include <fs/surface.h>
#include <fs/annotation.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace DISP3DLIB;
using namespace FSLIB;
using namespace Eigen;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
FsAnnotationTreeItem::FsAnnotationTreeItem(int iType, const QString & text)
: AbstractTreeItem(iType, text)
{
    initItem();
}
 
//=============================================================================================================
 
void FsAnnotationTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Unchecked);
    this->setToolTip("Freesurfer annotation item");
}
 
//=============================================================================================================
 
void FsAnnotationTreeItem::addData(const Surface& tSurface, const Annotation& tAnnotation)
{
    //Create color from annotation data if annotation is not empty
    if(!tAnnotation.isEmpty()) {
        MatrixX4f matAnnotColors(tAnnotation.getVertices().rows(), 4);
 
        QList<FSLIB::Label> qListLabels;
        QList<RowVector4i> qListLabelRGBAs;
 
        tAnnotation.toLabels(tSurface, qListLabels, qListLabelRGBAs);
 
        for(int i = 0; i < qListLabels.size(); ++i) {
            FSLIB::Label label = qListLabels.at(i);
            for(int j = 0; j<label.vertices.rows(); j++) {
                QColor patchColor;
                patchColor.setRed(qListLabelRGBAs.at(i)(0));
                patchColor.setGreen(qListLabelRGBAs.at(i)(1));
                patchColor.setBlue(qListLabelRGBAs.at(i)(2));
 
                patchColor = patchColor.darker(200);
 
                if(label.vertices(j) < matAnnotColors.rows()) {
                    matAnnotColors(label.vertices(j),0) = patchColor.redF();
                    matAnnotColors(label.vertices(j),1) = patchColor.greenF();
                    matAnnotColors(label.vertices(j),2) = patchColor.blueF();
                    matAnnotColors(label.vertices(j),3) = patchColor.alphaF();
                }
            }
        }
 
        //Add data which is held by this FsAnnotationTreeItem
        QVariant data;
        data.setValue(matAnnotColors);
        this->setData(data, Data3DTreeModelItemRoles::AnnotColors);
 
        //Add annotation meta information as item children
        QList<QStandardItem*> list;
 
        MetaTreeItem *itemAnnotFileName = new MetaTreeItem(MetaTreeItemTypes::FileName, tAnnotation.fileName());
        itemAnnotFileName->setEditable(false);
        list << itemAnnotFileName;
        list << new QStandardItem(itemAnnotFileName->toolTip());
        this->appendRow(list);
        data.setValue(tAnnotation.fileName());
        itemAnnotFileName->setData(data, Data3DTreeModelItemRoles::FileName);
 
        list.clear();
        MetaTreeItem *itemAnnotPath = new MetaTreeItem(MetaTreeItemTypes::FilePath, tAnnotation.filePath());
        itemAnnotPath->setEditable(false);
        list << itemAnnotPath;
        list << new QStandardItem(itemAnnotPath->toolTip());
        this->appendRow(list);
        data.setValue(tAnnotation.filePath());
        itemAnnotFileName->setData(data, Data3DTreeModelItemRoles::FilePath);
    }
}
 
//=============================================================================================================
 
void FsAnnotationTreeItem::onCheckStateChanged(const Qt::CheckState& checkState)
{
    emit annotationVisibiltyChanged(checkState==Qt::Unchecked ? false : true);
}