LVRPickingInteractor.hpp

Go to the documentation of this file.

#ifndef LVRPICKINGINTERACTOR_HPP_
#define LVRPICKINGINTERACTOR_HPP_

#include <QObject>
#include <QMessageBox>

#include <vtkIdTypeArray.h>
#include <vtkTextActor.h>
#include <QTreeWidgetItem>
#include <vtkInteractorStyleTrackballCamera.h>
#include <vtkInteractorStyleRubberBandPick.h>
#include <vtkSmartPointer.h>
#include <vtkRenderer.h>
#include <vtkMath.h>
#include <vtkDataSetMapper.h>

#include <boost/shared_array.hpp>

#include <memory>
#include "LVRInteractorStylePolygonPick.hpp"
#include <map>

namespace lvr2
{


class LVRPickingInteractor : public QObject, public LVRInteractorStylePolygonPick
{
    Q_OBJECT
public:
    static LVRPickingInteractor* New();
    LVRPickingInteractor();
    vtkTypeMacro(LVRPickingInteractor, LVRInteractorStylePolygonPick);
    void setRenderer(vtkSmartPointer<vtkRenderer> renderer);
    void removeLabel(const int&);
    //LVRPickingInteractor(vtkSmartPointer<vtkRenderer> renderer);
    virtual ~LVRPickingInteractor();

    virtual void OnLeftButtonDown();
    virtual void OnLeftButtonUp();
    virtual void OnMouseMove();
    virtual void OnMiddleButtonUp();
    virtual void OnMiddleButtonDown();
    virtual void OnRightButtonUp();
    virtual void OnRightButtonDown();
    virtual void OnMouseWheelBackward();
    virtual void OnMouseWheelForward();

    virtual void OnChar();
    virtual void OnTimer();

    virtual void OnKeyPress();

    virtual void OnKeyRelease();

    virtual void OnKeyDown();

    virtual void Dolly();
    virtual void Dolly(double factor);
    virtual void Pan();
    virtual void Spin();
    virtual void Zoom();
    virtual void Rotate();


    vtkSmartPointer<vtkTextActor>   getTextActor(){ return m_textActor; }

    void updateFocalPoint();
    void setPoints(vtkSmartPointer<vtkPolyData> points);
    vtkSmartPointer<vtkPolyData> getPoints();

    std::vector<uint16_t>& getLabeles();
public Q_SLOTS:
    void correspondenceSearchOn();
    void correspondenceSearchOff();

    void labelingOn();
    void labelingOff();
    void setLabeledPointVisibility(int, bool);
    void setLabel(int, std::vector<int>);
    void requestLabels();

    void labelModeChanged(bool);
    void newLabel(QTreeWidgetItem*);
    void setLassoTool(bool);
    void labelSelected(uint16_t);

    void setMotionFactor(double factor);
    void setRotationFactor(double factor);

    void setFocalPointRendering(int state);
    void setStereoMode(int state);

    void pickFocalPoint();

    void modeTerrain();
    void modeTrackball();
    void modeShooter();

    void resetCamera();

Q_SIGNALS:
    void clusterSelected(double*);
    void firstPointPicked(double*);
    void secondPointPicked(double*);
    void pointSelected(vtkActor*, int);
    void pointsLabeled(uint16_t, int);
    void responseLabels(std::vector<uint16_t>);
    void labelingStarted(bool);

private:

    enum InteractorMode {TRACKBALL, SHOOTER, TERRAIN};
    enum ShooterMode {LOOK, HOVER};
    enum PickMode {None, PickPoint, PickFirst, PickSecond, PickFocal, PickLabel};

    void handlePicking();

    // ------------------------- TRACKBALL

    void dollyTrackball();
    void dollyTrackball(double factor);
    void panTrackball();
    void spinTrackball();
    void zoomTrackball();
    void rotateTrackball();

    void onLeftButtonDownTrackball();
    void onLeftButtonUpTrackball();
    void onMouseMoveTrackball();
    void onMiddleButtonUpTrackball();
    void onMiddleButtonDownTrackball();
    void onRightButtonUpTrackball();
    void onRightButtonDownTrackball();
    void onMouseWheelBackwardTrackball();
    void onMouseWheelForwardTrackball();

    // ------------------------ TERRAIN

    void dollyTerrain();
    void dollyTerrain(double factor);
    void panTerrain();
    void spinTerrain();
    void zoomTerrain();
    void rotateTerrain();

    void onLeftButtonDownTerrain();
    void onLeftButtonUpTerrain();
    void onMouseMoveTerrain();
    void onMiddleButtonUpTerrain();
    void onMiddleButtonDownTerrain();
    void onRightButtonUpTerrain();
    void onRightButtonDownTerrain();
    void onMouseWheelBackwardTerrain();
    void onMouseWheelForwardTerrain();

    // ------------------------ SHOOTER

    void dollyShooter();
    void dollyShooter(double factor);
    void panShooter();
    void spinShooter();
    void zoomShooter();
    void rotateShooter();
    void hoverShooter();
    void resetViewUpShooter();
    void strafeShooter(double factor);

    void onLeftButtonDownShooter();
    void onLeftButtonUpShooter();
    void onMouseMoveShooter();
    void onMiddleButtonUpShooter();
    void onMiddleButtonDownShooter();
    void onRightButtonUpShooter();
    void onRightButtonDownShooter();
    void onMouseWheelBackwardShooter();
    void onMouseWheelForwardShooter();

    //Labeling
    bool isInside(std::vector<vtkVector2i>* polygon, int& pX, int& pY);
    void calculateSelection(bool select);
    void saveCurrentLabelSelection();
    void discardChanges();
    void updateActor(int);

    PickMode            m_pickMode;

    vtkSmartPointer<vtkTextActor>     m_textActor;
    vtkSmartPointer<vtkActor>         m_sphereActor;
    vtkSmartPointer<vtkActor>         m_cubeActor;
    vtkSmartPointer<vtkActor>         m_polyActor;
    std::vector<bool>                 m_selectedPoints;
    std::map<uint16_t, vtkSmartPointer<vtkActor>> m_labelActors;
    vtkSmartPointer<vtkActor> m_selectedActor;
    std::vector<uint16_t>              m_pointLabels;
    vtkSmartPointer<vtkPolyData>      m_points;
    vtkSmartPointer<vtkDataSetMapper> m_selectedMapper;
    vtkSmartPointer<vtkIdTypeArray> m_selectedIds; 

    vtkSmartPointer<vtkRenderer>    m_renderer;

    bool                            m_correspondenceMode;
    bool                            m_labelingMode;
    bool                            m_modified;

    unsigned int                    m_numberOfClicks;
    int                             m_previousPosition[2];
    int                             m_startCameraMovePosition[2];
    int                             m_selectedLabel;


    double                          m_viewUp[3];

    float                           m_motionFactor;
    float                           m_rotationFactor;

    InteractorMode                  m_interactorMode;
    ShooterMode                     m_shooterMode;

    std::map<uint16_t, QColor>      m_labelColors;


};

} /* namespace lvr2 */

#endif /* LVRPICKINGINTERACTOR_HPP_ */