LVRLabelDialog.hpp

Go to the documentation of this file.

 
#ifndef LVRLABELDIALOG_HPP_
#define LVRLABELDIALOG_HPP_
 
#include "ui_LVRLabelDialogUI.h"
#include <QTreeWidget>
 
#include <lvr2/geometry/Matrix4.hpp>
#include <lvr2/geometry/BaseVector.hpp>
#include <lvr2/registration/EigenSVDPointAlign.hpp>
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
 
#include <boost/shared_array.hpp>
 
#include <iostream>
 
#define LABEL_NAME_COLUMN 0
#define LABELED_POINT_COLUMN 1
#define LABEL_VISIBLE_COLUMN 2
#define LABEL_ID_COLUMN 3
 
using namespace std;
 
namespace lvr2
{
 
 
class LVRLabelDialog : public QObject
{
 
    Q_OBJECT
public:
 
    LVRLabelDialog(QTreeWidget* parent);
    virtual ~LVRLabelDialog();
    void showEvent();
    void setPoints(const std::string, const vtkSmartPointer<vtkPolyData> points);
 
public Q_SLOTS:
    void addNewLabel();
    void addNewInstance();
    void responseLabels(std::vector<uint16_t>);
    void updatePointCount(uint16_t, int);
    void cellSelected(QTreeWidgetItem*, int);
    void comboBoxIndexChanged(int index);
    void visibilityChanged(QTreeWidgetItem*, int);
    void loadLabels();
 
Q_SIGNALS:
    void labelRemoved(QPair<int, QColor>);
    void labelAdded(QTreeWidgetItem*);
    void labelChanged(uint16_t);
    void labelLoaded(int, std::vector<int>);
    void hidePoints(int, bool);
 
public:
    Ui_LabelDialog*    m_ui;
    QDialog*                    m_dialog;
 
private:
    QTreeWidget*                m_treeWidget;
    QColor                      m_dataSelectionColor;
    QColor                      m_modelSelectionColor;
    int                         m_id_hack = 1;
    std::map<std::string, vtkSmartPointer<vtkPolyData>> m_points;
};
 
} /* namespace lvr2 */
 
#endif /* LVRLABELDIALOG_HPP_ */