digitizertreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "digitizertreeitem.h"
#include "../common/metatreeitem.h"
#include "../../3dhelpers/geometrymultiplier.h"
#include "../../materials/geometrymultipliermaterial.h"
 
#include <fiff/fiff_constants.h>
#include <fiff/fiff_dig_point.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <Qt3DExtras/QSphereGeometry>
#include <QMatrix4x4>
#include <Qt3DCore/QTransform>
 
//=============================================================================================================
// EIGEN INCLUDES
//============================================================================= ================================
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace DISP3DLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
DigitizerTreeItem::DigitizerTreeItem(Qt3DCore::QEntity *p3DEntityParent, int iType, const QString& text)
: Abstract3DTreeItem(p3DEntityParent, iType, text)
{
    initItem();
}
 
//=============================================================================================================
 
void DigitizerTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Checked);
    this->setToolTip(this->text());
}
 
//=============================================================================================================
 
void DigitizerTreeItem::addData(const QList<FIFFLIB::FiffDigPoint>& tDigitizer, const float tSphereRadius, const QColor &tSphereColor)
{
    //create instanced renderer if there is none
    if(!m_pSphereMesh && !tDigitizer.isEmpty())
    {
        QSharedPointer<Qt3DExtras::QSphereGeometry> pSourceSphereGeometry = QSharedPointer<Qt3DExtras::QSphereGeometry>::create();
        pSourceSphereGeometry->setRadius(tSphereRadius);
 
        m_pSphereMesh = new GeometryMultiplier(pSourceSphereGeometry);
 
        //Set sphere positions
        this->addComponent(m_pSphereMesh);
 
        //Add material
        GeometryMultiplierMaterial* pMaterial = new GeometryMultiplierMaterial;
        pMaterial->setAmbient(tSphereColor);
        this->addComponent(pMaterial);
    }
 
    //Set transforms
    if(!tDigitizer.isEmpty())
    {
        QVector<QColor> vColorsEdges;
 
        QVector<QMatrix4x4> vTransforms;
        vTransforms.reserve(tDigitizer.size());
 
        QVector3D tempPos;
 
        for(int i = 0; i < tDigitizer.size(); ++i) {
            QMatrix4x4 tempTransform;
 
            tempPos.setX(tDigitizer[i].r[0]);
            tempPos.setY(tDigitizer[i].r[1]);
            tempPos.setZ(tDigitizer[i].r[2]);
 
            tempTransform.translate(tempPos);
            vTransforms.push_back(tempTransform);
 
            vColorsEdges.push_back(tSphereColor);
        }
 
        //Set instance Transform
        m_pSphereMesh->setTransforms(vTransforms);
        m_pSphereMesh->setColors(vColorsEdges);
    }
 
    //Update alpha
    float alpha = 1.0;
 
    QList<QStandardItem*> items = m_pItemAppearanceOptions->findChildren(MetaTreeItemTypes::AlphaValue);
 
    for(int i = 0; i < items.size(); ++i) {
        if(MetaTreeItem* item = dynamic_cast<MetaTreeItem*>(items.at(i))) {
            QVariant data;
            data.setValue(alpha);
            item->setData(data, MetaTreeItemRoles::AlphaValue);
            item->setData(data, Qt::DecorationRole);
        }
    }
 
    //Update colors in color item
    items = m_pItemAppearanceOptions->findChildren(MetaTreeItemTypes::Color);
 
    for(int i = 0; i < items.size(); ++i) {
        if(MetaTreeItem* item = dynamic_cast<MetaTreeItem*>(items.at(i))) {
            QVariant data;
            data.setValue(tSphereColor);
            item->setData(data, MetaTreeItemRoles::Color);
            item->setData(data, Qt::DecorationRole);
        }
    }
}