LVRCorrespondanceDialog.hpp

Go to the documentation of this file.

#ifndef LVRCORRESPONDANCEDIALOG_HPP_
#define LVRCORRESPONDANCEDIALOG_HPP_

#include "ui_LVRRegistrationPickCorrespondancesDialogUI.h"
#include "../vtkBridge/LVRVtkArrow.hpp"

#include "lvr2/types/MatrixTypes.hpp"
#include "lvr2/registration/EigenSVDPointAlign.hpp"

#include <iostream>
using namespace std;

#include <boost/optional.hpp>
#include <Eigen/Dense>

namespace lvr2
{

class LVRCorrespondanceDialog : public QObject
{
    Q_OBJECT
public:

    LVRCorrespondanceDialog(QTreeWidget* parent);
    virtual ~LVRCorrespondanceDialog();
    void fillComboBoxes();
    boost::optional<Transformf> getTransformation();
    QString  getModelName();
    QString  getDataName();

    bool    doICP();
    double  getEpsilon();
    double  getMaxDistance();
    int     getMaxIterations();

public Q_SLOTS:
    void updateModelSelection(int);
    void updateDataSelection(int);
    void firstPointPicked(double*);
    void secondPointPicked(double*);
    void insertNewItem();
    void deleteItem();
    void swapItemPositions();
    void saveCorrespondences();
    void loadCorrespondences();
    void treeItemSelected(QTreeWidgetItem*, QTreeWidgetItem*);
    void clearAllItems();


Q_SIGNALS:
    void render();
    void removeArrow(LVRVtkArrow *);
    void addArrow(LVRVtkArrow *);
    void disableCorrespondenceSearch();
    void enableCorrespondenceSearch();

public:
    Ui_CorrespondenceDialog*    m_ui;
    QDialog*                    m_dialog;

private:
    QTreeWidget*                m_treeWidget;
    QColor                      m_dataSelectionColor;
    QColor                      m_modelSelectionColor;
    QColor                      m_defaultColor;
    QString                     m_dataSelection;
    QString                     m_modelSelection;
};

} /* namespace lvr2 */

#endif /* LVRCORRESPONDANCEDIALOG_HPP_ */