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