LVRPointCloudItem.cpp

Go to the documentation of this file.

#include "LVRPointCloudItem.hpp"
#include "LVRItemTypes.hpp"

namespace lvr2
{

LVRPointCloudItem::LVRPointCloudItem(PointBufferBridgePtr ptr, QTreeWidgetItem* item) :
       QTreeWidgetItem(item, LVRPointCloudItemType), m_parent(item), m_pointBridge(ptr)
{
    // Setup tree widget icon
    QIcon icon;
    icon.addFile(QString::fromUtf8(":/qv_pc_tree_icon.png"), QSize(), QIcon::Normal, QIcon::Off);
    setIcon(0, icon);
    setText(0, "Point Cloud");
    setExpanded(true);

    // Setup Infos
    m_numItem = new QTreeWidgetItem(this);
    QString num;
    m_numItem->setText(0, "Num Points:");
    m_numItem->setText(1, num.setNum(ptr->getNumPoints()));
    addChild(m_numItem);

    m_normalItem = new QTreeWidgetItem(this);
    m_normalItem->setText(0, "Has Normals:");
    if(ptr->hasNormals())
    {
        m_normalItem->setText(1, "yes");
    }
    else
    {
        m_normalItem->setText(1, "no");
    }
    addChild(m_normalItem);

    m_colorItem = new QTreeWidgetItem(this);
    m_colorItem->setText(0, "Has Colors:");
    if(ptr->hasColors())
    {
        m_colorItem->setText(1, "yes");
    }
    else
    {
        m_colorItem->setText(1, "no");
    }
    addChild(m_colorItem);

    m_specItem = new QTreeWidgetItem(this);
    m_specItem->setText(0, "Has Spectraldata:");
    if(ptr->getPointBuffer()->getUCharChannel("spectral_channels"))
    {
        m_specItem->setText(1, "yes");
    }
    else
    {
        m_specItem->setText(1, "no");
    }
    addChild(m_specItem);

    // set initial values
    m_opacity = 1;
    m_pointSize = 1;
    m_color = QColor(255,255,255);
    m_visible = true;
}

void LVRPointCloudItem::update()
{
    m_pointBridge.reset(new LVRPointBufferBridge(m_pointBridge->getPointBuffer()));

    QString num;
    m_numItem->setText(1, num.setNum(m_pointBridge->getNumPoints()));

    // normals update
    if(m_pointBridge->hasNormals()) {
        m_normalItem->setText(1, "yes");
    } else {
        m_normalItem->setText(1, "no");
    }

    // color update
    if(m_pointBridge->hasColors())
    {
        m_colorItem->setText(1, "yes");
    } else {
        m_colorItem->setText(1, "no");
    }

    // hyperspectral update
    if(m_pointBridge->getPointBuffer()->getUCharChannel("spectral_channels")) {
        m_specItem->setText(1, "yes");
    } else {
        m_specItem->setText(1, "no");
    }

}

QColor LVRPointCloudItem::getColor()
{
        return m_color;
}

void LVRPointCloudItem::setColor(QColor &c)
{
    m_pointBridge->setBaseColor(c.redF(), c.greenF(), c.blueF());
    m_color = c;
}

void LVRPointCloudItem::setSelectionColor(QColor& c)
{
    m_pointBridge->setBaseColor(c.redF(), c.greenF(), c.blueF());
}

void LVRPointCloudItem::resetColor()
{
    m_pointBridge->setBaseColor(m_color.redF(), m_color.greenF(), m_color.blueF());
}

int LVRPointCloudItem::getPointSize()
{
        return m_pointSize;
}

void LVRPointCloudItem::setPointSize(int &pointSize)
{
    m_pointBridge->setPointSize(pointSize);
    m_pointSize = pointSize;
}

float LVRPointCloudItem::getOpacity()
{
        return m_opacity;
}

void LVRPointCloudItem::setOpacity(float &opacity)
{
    m_pointBridge->setOpacity(opacity);
    m_opacity = opacity;
}

bool LVRPointCloudItem::getVisibility()
{
        return m_visible;
}

void LVRPointCloudItem::setVisibility(bool &visibility)
{
        m_pointBridge->setVisibility(visibility);
        m_visible = visibility;
}

size_t LVRPointCloudItem::getNumPoints()
{
    return m_pointBridge->getNumPoints();
}

PointBufferPtr LVRPointCloudItem::getPointBuffer()
{
    return m_pointBridge->getPointBuffer();
}

PointBufferBridgePtr LVRPointCloudItem::getPointBufferBridge()
{
    return m_pointBridge;
}

vtkSmartPointer<vtkActor> LVRPointCloudItem::getActor()
{
        return m_pointBridge->getPointCloudActor();
}

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

} /* namespace lvr2 */