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_ */