LVRModelItem.cpp

Go to the documentation of this file.

#include "LVRModelItem.hpp"
#include "LVRPointCloudItem.hpp"
#include "LVRMeshItem.hpp"
#include "LVRItemTypes.hpp"
#include "LVRTextureMeshItem.hpp"

#include <vtkSmartPointer.h>
#include <vtkActor.h>
#include <vtkPolyDataMapper.h>

namespace lvr2
{

LVRModelItem::LVRModelItem(ModelBridgePtr bridge, QString name) :
    QTreeWidgetItem(LVRModelItemType), m_modelBridge(bridge), m_name(name)
{
    // Setup tree widget icon
    QIcon icon;
    icon.addFile(QString::fromUtf8(":/qv_model_tree_icon.png"), QSize(), QIcon::Normal, QIcon::Off);
    setIcon(0, icon);

    // Setup item properties
    setText(0, m_name);
    setCheckState(0, Qt::Checked);

    // Insert sub items
    if(bridge->m_pointBridge->getNumPoints())
    {
        LVRPointCloudItem* pointItem = new LVRPointCloudItem(bridge->m_pointBridge, this);
        addChild(pointItem);
        pointItem->setExpanded(true);
    }

    if(bridge->m_meshBridge->getNumTriangles())
    {
        if(bridge->m_meshBridge->hasTextures())
        {
                LVRTextureMeshItem* texItem = new LVRTextureMeshItem(bridge->m_meshBridge, this);
                addChild(texItem);
                texItem->setExpanded(true);
        }
        else
        {
                LVRMeshItem* meshItem = new LVRMeshItem(bridge->m_meshBridge, this);
                addChild(meshItem);
                meshItem->setExpanded(true);
        }
    }

    // Setup Pose
    m_poseItem = new LVRPoseItem(bridge, this);
    addChild(m_poseItem);
}

LVRModelItem::LVRModelItem(const LVRModelItem& item)
{
    m_modelBridge   = item.m_modelBridge;
    m_name          = item.m_name;
    m_poseItem      = item.m_poseItem;
}

Pose LVRModelItem::getPose()
{
    return m_poseItem->getPose();
}

void LVRModelItem::setPose(const Pose& pose)
{
    // Update vtk representation
    m_modelBridge->setPose(pose);

    // Update pose item
    m_poseItem->setPose(pose);
}

QString LVRModelItem::getName()
{
    return m_name;
}

void LVRModelItem::setName(QString name)
{
    m_name = name;
    setText(0, m_name);
}

ModelBridgePtr LVRModelItem::getModelBridge()
{
        return m_modelBridge;
}

bool LVRModelItem::isEnabled()
{
    return this->checkState(0);
}

void LVRModelItem::setVisibility(bool visible)
{
        m_modelBridge->setVisibility(visible);
}

void LVRModelItem::setModelVisibility(int column, bool globalValue)
{
        if(checkState(column) == globalValue || globalValue == true)
        {
            setVisibility(checkState(column));
        }
}

LVRModelItem::~LVRModelItem()
{
    // TODO Auto-generated destructor stub
}

} /* namespace lvr2 */