sourcespacetreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "sourcespacetreeitem.h"
#include "../common/metatreeitem.h"
#include "../../3dhelpers/renderable3Dentity.h"
#include "../../3dhelpers/custommesh.h"
#include "../../3dhelpers/geometrymultiplier.h"
#include "../../materials/geometrymultipliermaterial.h"
 
#include <mne/mne_hemisphere.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <Qt3DExtras/QSphereGeometry>
#include <QMatrix4x4>
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace Eigen;
using namespace MNELIB;
using namespace DISP3DLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
SourceSpaceTreeItem::SourceSpaceTreeItem(Qt3DCore::QEntity *p3DEntityParent, int iType, const QString& text)
: AbstractMeshTreeItem(p3DEntityParent, iType, text)
{
    initItem();
}
 
//=============================================================================================================
 
void SourceSpaceTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Checked);
    this->setToolTip("Source space item");
}
 
//=============================================================================================================
 
void SourceSpaceTreeItem::addData(const MNEHemisphere& tHemisphere)
{
    //Create color from curvature information with default gyri and sulcus colors
    MatrixX4f matVertColor = createVertColor(tHemisphere.rr.rows());
 
    //Set renderable 3D entity mesh and color data
    m_pCustomMesh->setMeshData(tHemisphere.rr,
                               tHemisphere.nn,
                               tHemisphere.tris,
                               matVertColor,
                               Qt3DRender::QGeometryRenderer::Triangles);
 
    //Add data which is held by this SourceSpaceTreeItem
    QVariant data;
 
    data.setValue(tHemisphere.rr.rows());
    this->setData(data, Data3DTreeModelItemRoles::NumberVertices);
 
    plotSources(tHemisphere);
 
    this->setAlpha(1.0f);
}
 
//=============================================================================================================
 
void SourceSpaceTreeItem::plotSources(const MNEHemisphere& tHemisphere)
{
    //use QEntity instead of Renderable3DEntity to avoid a double transformation of digitizers
    //from Abstract3DMeshTree and the child Renderable3DEntity
    QEntity* pSourceSphereEntity = new QEntity(this);
 
    //create geometry
    QSharedPointer<Qt3DExtras::QSphereGeometry> pSourceSphereGeometry = QSharedPointer<Qt3DExtras::QSphereGeometry>::create();
    pSourceSphereGeometry->setRadius(0.00075f);
    //create instanced renderer
    GeometryMultiplier *pSphereMesh = new GeometryMultiplier(pSourceSphereGeometry);
 
    //Create transform matrix for each sphere instance
    QVector<QMatrix4x4> vTransforms;
    QVector3D tempPos;
 
    if(tHemisphere.isClustered())
    {
        vTransforms.reserve(tHemisphere.cluster_info.centroidVertno.size());
 
        for(int i = 0; i < tHemisphere.cluster_info.centroidVertno.size(); i++)
        {
            QMatrix4x4 tempTransform;
            const RowVector3f& sourcePos = tHemisphere.rr.row(tHemisphere.cluster_info.centroidVertno.at(i));
 
            tempPos.setX(sourcePos(0));
            tempPos.setY(sourcePos(1));
            tempPos.setZ(sourcePos(2));
 
            //Set position
            tempTransform.translate(tempPos);
            vTransforms.push_back(tempTransform);
        }
    }
    else
    {
        vTransforms.reserve(tHemisphere.vertno.rows());
 
        for(int i = 0; i < tHemisphere.vertno.rows(); i++)
        {
            QMatrix4x4 tempTransform;
            const RowVector3f& sourcePos = tHemisphere.rr.row(tHemisphere.vertno(i));
 
            tempPos.setX(sourcePos(0));
            tempPos.setY(sourcePos(1));
            tempPos.setZ(sourcePos(2));
 
            //Set position
            tempTransform.translate(tempPos);
            vTransforms.push_back(tempTransform);
        }
    }
    //Set instance Transform
    pSphereMesh->setTransforms(vTransforms);
 
    pSourceSphereEntity->addComponent(pSphereMesh);
 
    //Add material
    GeometryMultiplierMaterial* pMaterial = new GeometryMultiplierMaterial;
    QColor defaultColor(212, 28, 92);
    pMaterial->setAmbient(defaultColor);
    pMaterial->setAlpha(1.0f);
 
    pSourceSphereEntity->addComponent(pMaterial);
}