abstractmeshtreeitem.cpp

Go to the documentation of this file.

//=============================================================================================================
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "abstractmeshtreeitem.h"
#include "../common/metatreeitem.h"
#include "../../materials/gpuinterpolationmaterial.h"
#include "../../materials/pervertexphongalphamaterial.h"
#include "../../materials/pervertextessphongalphamaterial.h"
#include "../../materials/shownormalsmaterial.h"
#include "../../3dhelpers/custommesh.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
//=============================================================================================================
// EIGEN INCLUDES
//=============================================================================================================
 
#include <Eigen/Core>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace DISP3DLIB;
using namespace Eigen;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
AbstractMeshTreeItem::AbstractMeshTreeItem(QEntity* p3DEntityParent, int iType, const QString& text)
: Abstract3DTreeItem(p3DEntityParent, iType, text)
, m_pMaterial(new PerVertexPhongAlphaMaterial)
, m_pCustomMesh(new CustomMesh())
{
    initItem();
}
 
//=============================================================================================================
 
QPointer<CustomMesh> AbstractMeshTreeItem::getCustomMesh()
{
    return m_pCustomMesh;
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::setVertColor(const MatrixX4f& vertColor)
{
    QVariant data;
    data.setValue(vertColor);
 
    this->setData(data, Data3DTreeModelItemRoles::SurfaceCurrentColorVert);
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::initItem()
{
    this->setEditable(false);
    this->setCheckable(true);
    this->setCheckState(Qt::Checked);
    this->setToolTip("Abstract Mesh Tree Item");
 
    //Add surface meta information as item children
    QList<QStandardItem*> list;
    QVariant data;
 
    //Material options
    MetaTreeItem* pItemMaterialOptions = new MetaTreeItem(MetaTreeItemTypes::UnknownItem, "Material");
    pItemMaterialOptions->setEditable(false);
    list.clear();
    list << pItemMaterialOptions;
    list << new QStandardItem("The material options");
    this->appendRow(list);
 
    QString surfaceType("Phong Alpha");
    MetaTreeItem* pItemMaterialType = new MetaTreeItem(MetaTreeItemTypes::MaterialType, surfaceType);
    connect(pItemMaterialType, &MetaTreeItem::dataChanged,
            this, &AbstractMeshTreeItem::onSurfaceMaterialChanged);
    list.clear();
    list << pItemMaterialType;
    list << new QStandardItem(pItemMaterialType->toolTip());
    pItemMaterialOptions->appendRow(list);
    data.setValue(QString(surfaceType));
    pItemMaterialType->setData(data, MetaTreeItemRoles::SurfaceMaterial);
    pItemMaterialType->setData(data, Qt::DecorationRole);
 
    float fTessInner = 1.0;
    MetaTreeItem *itemTessInner = new MetaTreeItem(MetaTreeItemTypes::SurfaceTessInner, QString("%1").arg(fTessInner));
    connect(itemTessInner, &MetaTreeItem::dataChanged,
            this, &AbstractMeshTreeItem::onSurfaceTessInnerChanged);
    list.clear();
    list << itemTessInner;
    list << new QStandardItem(itemTessInner->toolTip());
    pItemMaterialOptions->appendRow(list);
    data.setValue(fTessInner);
    itemTessInner->setData(data, MetaTreeItemRoles::SurfaceTessInner);
 
    float fTessOuter = 1.0;
    MetaTreeItem *itemTessOuter = new MetaTreeItem(MetaTreeItemTypes::SurfaceTessOuter, QString("%1").arg(fTessOuter));
    connect(itemTessOuter, &MetaTreeItem::dataChanged,
            this, &AbstractMeshTreeItem::onSurfaceTessOuterChanged);
    list.clear();
    list << itemTessOuter;
    list << new QStandardItem(itemTessOuter->toolTip());
    pItemMaterialOptions->appendRow(list);
    data.setValue(fTessOuter);
    itemTessOuter->setData(data, MetaTreeItemRoles::SurfaceTessOuter);
 
    float fTriangleScale = 1.0;
    MetaTreeItem *itemTriangleScale = new MetaTreeItem(MetaTreeItemTypes::SurfaceTriangleScale, QString("%1").arg(fTriangleScale));
    connect(itemTriangleScale, &MetaTreeItem::dataChanged,
            this, &AbstractMeshTreeItem::onSurfaceTriangleScaleChanged);
    list.clear();
    list << itemTriangleScale;
    list << new QStandardItem(itemTriangleScale->toolTip());
    pItemMaterialOptions->appendRow(list);
    data.setValue(fTriangleScale);
    itemTriangleScale->setData(data, MetaTreeItemRoles::SurfaceTriangleScale);
 
    //Init materials
    this->onSurfaceMaterialChanged("Phong Alpha");
 
    //Init custom mesh
    this->addComponent(m_pCustomMesh);
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::setData(const QVariant& value, int role)
{
    Abstract3DTreeItem::setData(value, role);
 
    switch(role) {
        case Data3DTreeModelItemRoles::SurfaceCurrentColorVert:
            if(m_pCustomMesh) {
                m_pCustomMesh->setColor(value.value<MatrixX4f>());
            }
            break;
 
        default: // do nothing;
            break;
    }
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::setMaterial(Qt3DRender::QMaterial* pMaterial)
{
    if(!pMaterial) {
        return;
    }
 
    //Remove material here. This should also call delete since we did not specify a parent for the material
    //before adding it as a component. Hence, QEntity should take care of delete.
    if(m_pMaterial) {
        this->removeComponent(m_pMaterial);
    }
 
    //Set new material
    m_pMaterial = pMaterial;
    this->addComponent(m_pMaterial);
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::setMeshData(const MatrixX3f& tMatVert,
                                       const MatrixX3f& tMatNorm,
                                       const MatrixXi& tMatTris,
                                       const MatrixX4f& tMatColors,
                                       Qt3DRender::QGeometryRenderer::PrimitiveType primitiveType)
{
    if(m_pCustomMesh) {
        m_pCustomMesh->setMeshData(tMatVert,
                                   tMatNorm,
                                   tMatTris,
                                   tMatColors,
                                   primitiveType);
 
        int iNumVerts = tMatVert.rows();
 
        this->setData(QVariant(iNumVerts), Data3DTreeModelItemRoles::NumberVertices);
    }
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::onSurfaceTessInnerChanged(const QVariant& fTessInner)
{
    this->setMaterialParameter(fTessInner.toFloat(), "innerTess");
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::onSurfaceTessOuterChanged(const QVariant& fTessOuter)
{
    this->setMaterialParameter(fTessOuter.toFloat(), "outerTess");
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::onSurfaceTriangleScaleChanged(const QVariant& fTriangleScale)
{
    this->setMaterialParameter(fTriangleScale.toFloat(), "triangleScale");
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::onColorChanged(const QVariant& color)
{
    QVariant data;
    MatrixX4f matNewVertColor = createVertColor(this->data(Data3DTreeModelItemRoles::NumberVertices).toInt(),
                                                color.value<QColor>());
 
    data.setValue(matNewVertColor);
    this->setData(data, Data3DTreeModelItemRoles::SurfaceCurrentColorVert);
}
 
//=============================================================================================================
 
void AbstractMeshTreeItem::onSurfaceMaterialChanged(const QVariant& sMaterial)
{
    QPointer<Qt3DRender::QMaterial> pMaterial;
 
    if(sMaterial.toString() == "Phong Alpha") {
        pMaterial = new PerVertexPhongAlphaMaterial();
        if(m_pCustomMesh) {
            m_pCustomMesh->setPrimitiveType(Qt3DRender::QGeometryRenderer::Triangles);
        }
    } else if(sMaterial.toString() == "Phong Alpha Tesselation") {
        pMaterial = new PerVertexTessPhongAlphaMaterial();
        if(m_pCustomMesh) {
            m_pCustomMesh->setPrimitiveType(Qt3DRender::QGeometryRenderer::Patches);
        }
    } else if(sMaterial.toString() == "Show normals") {
        pMaterial = new ShowNormalsMaterial();
        if(m_pCustomMesh) {
            m_pCustomMesh->setPrimitiveType(Qt3DRender::QGeometryRenderer::Triangles);
        }
    } else if(sMaterial.toString() == "GPU Interpolation") {
        pMaterial = new GpuInterpolationMaterial();
        if(m_pCustomMesh) {
            m_pCustomMesh->setPrimitiveType(Qt3DRender::QGeometryRenderer::Triangles);
        }
    }
 
    this->setMaterial(pMaterial);
}