mitkGizmoInteractor.h

Go to the documentation of this file.

/*===================================================================

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
All rights reserved.

This software is distributed WITHOUT ANY WARRANTY; without
even the implied warranty of MERCHANTABILITY or FITNESS FOR
A PARTICULAR PURPOSE.

See LICENSE.txt or http://www.mitk.org for details.

===================================================================*/

#ifndef mitkGizmoInteractor_h
#define mitkGizmoInteractor_h

#include "mitkGizmo.h"

// MITK includes
#include <mitkDataInteractor.h>
#include <mitkDataNode.h>
#include <mitkGeometry3D.h>

// VTK includes
#include <vtkCellPicker.h>
#include <vtkSmartPointer.h>

// System includes
#include <memory>

#include "MitkGizmoExports.h"

namespace mitk
{
  class InteractionPositionEvent;

  class MITKGIZMO_EXPORT GizmoInteractor : public DataInteractor
  {
  public:
    mitkClassMacro(GizmoInteractor, DataInteractor);
    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      
      void SetGizmoNode(DataNode *node);

    void SetManipulatedObjectNode(DataNode *node);

  private:
    GizmoInteractor();
    virtual ~GizmoInteractor();

    virtual void ConnectActionsAndFunctions() override;

    bool HasPickedHandle(const InteractionEvent *);

    void DecideInteraction(StateMachineAction *, InteractionEvent *interactionEvent);

    void MoveAlongAxis(StateMachineAction *, InteractionEvent *interactionEvent);

    void ScaleEqually(StateMachineAction *, InteractionEvent *interactionEvent);

    void RotateAroundAxis(StateMachineAction *, InteractionEvent *interactionEvent);

    void MoveFreely(StateMachineAction *, InteractionEvent *interactionEvent);

    void FeedUndoStack(StateMachineAction *, InteractionEvent *interactionEvent);

    void ApplyTranslationToManipulatedObject(const Vector3D &projectedMovement);

    void ApplyEqualScalingToManipulatedObject(double scalingFactor);

    void ApplyRotationToManipulatedObject(double angle_deg);

    Gizmo::HandleType PickFrom2D(const InteractionPositionEvent *positionEvent);

    Gizmo::HandleType PickFrom3D(const InteractionPositionEvent *positionEvent);

    Gizmo::Pointer m_Gizmo;

    BaseGeometry::Pointer m_ManipulatedObjectGeometry;

    std::map<BaseRenderer *, vtkSmartPointer<vtkCellPicker>> m_Picker;

    Gizmo::HandleType m_PickedHandle;

    Point2D m_InitialClickPosition2D; //< Initial screen click position
    double m_InitialClickPosition2DZ; //< Z value of the initial screen click position
    Point3D m_InitialClickPosition3D; //< Initial 3D click position

    Point2D m_InitialGizmoCenter2D; //< Initial position of the gizmo's center in screen coordinates
    Point3D m_InitialGizmoCenter3D; //< Initial 3D position of the gizmo's center

    Vector3D m_AxisOfMovement; //< Axis along which we move when translating
    Vector3D m_AxisOfRotation; //< Axis around which we turn when rotating

    std::unique_ptr<Operation> m_FinalDoOperation;   //< Operation for the undo-stack
    std::unique_ptr<Operation> m_FinalUndoOperation; //< Operation for the undo-stack

    BaseGeometry::Pointer m_InitialManipulatedObjectGeometry;
  };
}
#endif