vtkMRMLInteractionWidgetRepresentation.h

Go to the documentation of this file.

/*==============================================================================
 
  Copyright (c) Laboratory for Percutaneous Surgery (PerkLab)
  Queen's University, Kingston, ON, Canada. All Rights Reserved.
 
  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.
 
  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
 
  This file was originally developed by Kyle Sunderland, PerkLab, Queen's University
  and was supported through CANARIE's Research Software Program, Cancer
  Care Ontario, OpenAnatomy, and Brigham and Women's Hospital through NIH grant R01MH112748.
 
==============================================================================*/

 
#ifndef vtkSlicerInteractionRepresentation_h
#define vtkSlicerInteractionRepresentation_h
 
// MRMLDM includes
#include "vtkMRMLDisplayableManagerExport.h"
#include "vtkMRMLAbstractWidgetRepresentation.h"
 
// MRML includes
#include <vtkMRMLSliceNode.h>
 
// VTK includes
#include <vtkActor.h>
#include <vtkActor2D.h>
#include <vtkAppendPolyData.h>
#include <vtkArrayCalculator.h>
#include <vtkGlyph3D.h>
#include <vtkLookupTable.h>
#include <vtkPlane.h>
#include <vtkPointPlacer.h>
#include <vtkPointSetToLabelHierarchy.h>
#include <vtkPolyDataMapper.h>
#include <vtkPolyDataMapper2D.h>
#include <vtkProperty.h>
#include <vtkProperty2D.h>
#include <vtkSmartPointer.h>
#include <vtkSphereSource.h>
#include <vtkTextActor.h>
#include <vtkTextProperty.h>
#include <vtkTensorGlyph.h>
#include <vtkTransform.h>
#include <vtkTransformPolyDataFilter.h>
 
class vtkMRMLInteractionEventData;
class vtkMRMLDisplayNode;
class vtkMRMLDisplayableNode;
 
class VTK_MRML_DISPLAYABLEMANAGER_EXPORT vtkMRMLInteractionWidgetRepresentation : public vtkMRMLAbstractWidgetRepresentation
{
public:
  vtkTypeMacro(vtkMRMLInteractionWidgetRepresentation, vtkMRMLAbstractWidgetRepresentation);
  void PrintSelf(ostream& os, vtkIndent indent) override;

  void UpdateFromMRML(vtkMRMLNode* caller, unsigned long event, void* callData = nullptr) override;

  void GetActors(vtkPropCollection*) override;
  void ReleaseGraphicsResources(vtkWindow*) override;
  int RenderOverlay(vtkViewport* viewport) override;
  int RenderOpaqueGeometry(vtkViewport* viewport) override;
  int RenderTranslucentPolygonalGeometry(vtkViewport* viewport) override;
  vtkTypeBool HasTranslucentPolygonalGeometry() override;

  virtual bool GetTransformationReferencePoint(double referencePointWorld[3]);

  virtual void CanInteract(vtkMRMLInteractionEventData* interactionEventData, int& foundComponentType, int& foundComponentIndex, double& closestDistance2);
 
  virtual vtkPointPlacer* GetPointPlacer();

  vtkProp* GetInteractionActor();

  virtual bool IsDisplayable() = 0;
 
  virtual vtkMRMLSliceNode* GetSliceNode();
  virtual void GetSliceToWorldCoordinates(const double slicePos[2], double worldPos[3]);
 
  virtual void UpdateSlicePlaneFromSliceNode();

  virtual void UpdateInteractionPipeline();
 
  virtual void UpdateHandlePolyData();

  virtual void GetInteractionHandleAxisLocal(int type, int index, double axis[3]);

  virtual void GetInteractionHandleAxisWorld(int type, int index, double axis[3]);
  virtual void GetInteractionHandleOriginWorld(double origin[3]);
 
  enum
  {
    InteractionNone,
    InteractionTranslationHandle,
    InteractionRotationHandle,
    InteractionScaleHandle,
    Interaction_Last
  };
 
  enum
  {
    GlyphArrow,
    GlyphCircle,
    GlyphRing
  };
 
  virtual int GetActiveComponentType() = 0;
  virtual void SetActiveComponentType(int type) = 0;
 
  virtual int GetActiveComponentIndex() = 0;
  virtual void SetActiveComponentIndex(int index) = 0;
 
  virtual double GetMaximumHandlePickingDistance2();
 
  virtual vtkTransform* GetHandleToWorldTransform();
 
  virtual int GetNumberOfHandles(int type);
  virtual int GetNumberOfHandles();
 
  vtkSetMacro(Interacting, bool);
  vtkGetMacro(Interacting, bool);
  vtkBooleanMacro(Interacting, bool);

  virtual void GetHandleColor(int type, int index, double color[4]);
  virtual double GetHandleOpacity(int type, int index);
  virtual bool GetHandleVisibility(int type, int index);
  virtual int GetHandleGlyphType(int type, int index);
  virtual bool GetApplyScaleToPosition(int type, int index);
  virtual void GetInteractionHandlePositionLocal(int type, int index, double position[3]);
  virtual void GetInteractionHandlePositionWorld(int type, int index, double position[3]);

  virtual void UpdateHandleToWorldTransform();
  virtual void UpdateHandleToWorldTransform(vtkTransform* handleToWorldTransform) = 0;
 
protected:
  vtkMRMLInteractionWidgetRepresentation();
  ~vtkMRMLInteractionWidgetRepresentation() override;
 
  class VTK_MRML_DISPLAYABLEMANAGER_EXPORT InteractionPipeline
  {
  public:
    InteractionPipeline();
    virtual ~InteractionPipeline();
 
    vtkSmartPointer<vtkPolyData> ArrowPolyData;
    vtkSmartPointer<vtkPolyData> CirclePolyData;
    vtkSmartPointer<vtkPolyData> RingPolyData;
    vtkSmartPointer<vtkPolyData> ArrowOutlinePolyData;
    vtkSmartPointer<vtkPolyData> CircleOutlinePolyData;
    vtkSmartPointer<vtkPolyData> RingOutlinePolyData;
 
    std::map<std::pair<int, int>, vtkSmartPointer<vtkPolyData>> HandleGlyphPolyDataMap;
    std::map<std::pair<int, int>, vtkSmartPointer<vtkPolyData>> HandleOutlineGlyphPolyDataMap;
 
    vtkSmartPointer<vtkPolyData> RotationHandlePoints;
    vtkSmartPointer<vtkPolyData> TranslationHandlePoints;
    vtkSmartPointer<vtkPolyData> ScaleHandlePoints;
 
    vtkSmartPointer<vtkAppendPolyData> Append;
    vtkSmartPointer<vtkTransformPolyDataFilter> HandleToWorldTransformFilter;
    vtkSmartPointer<vtkTransform> HandleToWorldTransform;
    vtkSmartPointer<vtkLookupTable> ColorTable;
 
    vtkSmartPointer<vtkPolyDataMapper> Mapper3D;
    vtkSmartPointer<vtkProperty> Property3D;
    vtkSmartPointer<vtkActor> Actor3D;
 
    vtkSmartPointer<vtkTransform> WorldToSliceTransform;
    vtkSmartPointer<vtkTransformPolyDataFilter> WorldToSliceTransformFilter;
  };
 
  struct HandleInfo
  {
    HandleInfo(int index, int componentType, double positionWorld[3], double positionLocal[3], double color[4], int glyphType, bool applyScaleToPosition)
      : Index(index)
      , ComponentType(componentType)
      , GlyphType(glyphType)
      , ApplyScaleToPosition(applyScaleToPosition)
    {
      for (int i = 0; i < 3; ++i)
      {
        this->PositionWorld[i] = positionWorld[i];
      }
      this->PositionWorld[3] = 1.0;
      for (int i = 0; i < 3; ++i)
      {
        this->PositionLocal[i] = positionLocal[i];
      }
      this->PositionLocal[3] = 1.0;
      for (int i = 0; i < 4; ++i)
      {
        this->Color[i] = color[i];
      }
    }
    int Index;
    int ComponentType;
    int GlyphType;
    double PositionLocal[4];
    double PositionWorld[4];
    double Color[4];
    bool ApplyScaleToPosition{ true };
    bool IsVisible()
    {
      double epsilon = 0.001;
      return this->Color[3] > epsilon;
    }
  };

  typedef std::vector<HandleInfo> HandleInfoList;
  virtual HandleInfoList GetHandleInfoList();
  virtual HandleInfo GetHandleInfo(int type, int index);
 
  virtual void InitializePipeline();
  virtual void CreateRotationHandles();
  virtual void CreateTranslationHandles();
  virtual void CreateScaleHandles();
  virtual void UpdateHandleColors();
  virtual int UpdateHandleColors(int type, int startIndex);
  virtual vtkPolyData* GetHandlePolydata(int type);
 
  virtual void UpdateHandleOrientation();
  virtual void UpdateTranslationHandleOrientation();
  virtual void UpdateScaleHandleOrientation();
  virtual void UpdateRotationHandleOrientation();

  virtual void SetWidgetScale(double scale);

  virtual void GetHandleToCameraVectorWorld(double handlePosition_World[3], double normal_World[3]);

  virtual void OrthoganalizeTransform(vtkTransform* transform);

  virtual void UpdateViewScaleFactor();
 
  virtual void UpdateHandleSize();
  virtual double GetInteractionScalePercent();
  virtual double GetInteractionSizeMm();
  virtual bool GetInteractionSizeAbsolute();
 
  virtual void CanInteractWithCircleHandle(vtkMRMLInteractionEventData* interactionEventData,
                                           int& foundComponentType,
                                           int& foundComponentIndex,
                                           double& closestDistance2,
                                           HandleInfo& handleInfo);
  virtual void CanInteractWithArrowHandle(vtkMRMLInteractionEventData* interactionEventData,
                                          int& foundComponentType,
                                          int& foundComponentIndex,
                                          double& closestDistance2,
                                          HandleInfo& handleInfo);
  virtual void CanInteractWithRingHandle(vtkMRMLInteractionEventData* interactionEventData,
                                         int& foundComponentType,
                                         int& foundComponentIndex,
                                         double& closestDistance2,
                                         HandleInfo& handleInfo);

  double StartFadeAngleDegrees{ 10.0 };
  double EndFadeAngleDegrees{ 8.0 };
 
  double ViewScaleFactorMmPerPixel;
  double ScreenSizePixel; // diagonal size of the screen
 
  vtkSmartPointer<vtkPlane> SlicePlane;

  double InteractionSize{ 1.0 };
 
  double WidgetScale{ 1.0 };
 
  bool Interacting{ false };
 
  vtkSmartPointer<vtkPointPlacer> PointPlacer;
 
  virtual void SetupInteractionPipeline();
  InteractionPipeline* Pipeline;
 
private:
  vtkMRMLInteractionWidgetRepresentation(const vtkMRMLInteractionWidgetRepresentation&) = delete;
  void operator=(const vtkMRMLInteractionWidgetRepresentation&) = delete;
};
 
#endif