mitkGizmoInteractor.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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

#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();
    ~GizmoInteractor() override;

    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);

    void UpdateHandleHighlight();

    Gizmo::Pointer m_Gizmo;

    BaseGeometry::Pointer m_ManipulatedObjectGeometry;

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

    Gizmo::HandleType m_PickedHandle = Gizmo::NoHandle;

    Gizmo::HandleType m_HighlightedHandle = Gizmo::NoHandle;

    double m_ColorForHighlight[4];

    double m_ColorReplacedByHighlight[4];

    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