vtkSlicerMarkupsWidgetRepresentation.h

Go to the documentation of this file.

/*=========================================================================
 
 Copyright (c) ProxSim ltd., Kwun Tong, Hong Kong. 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 Davide Punzo, punzodavide@hotmail.it,
 and development was supported by ProxSim ltd.
 
=========================================================================*/

 
#ifndef vtkSlicerMarkupsWidgetRepresentation_h
#define vtkSlicerMarkupsWidgetRepresentation_h
 
#include "vtkSlicerMarkupsModuleVTKWidgetsExport.h"
 
#include "vtkMRMLAbstractWidgetRepresentation.h"
#include "vtkMRMLMarkupsDisplayNode.h"
#include "vtkMRMLMarkupsNode.h"
 
#include "vtkActor2D.h"
#include "vtkAppendPolyData.h"
#include "vtkArcSource.h"
#include "vtkArrowSource.h"
#include "vtkGlyph3D.h"
#include "vtkLookupTable.h"
#include "vtkMarkupsGlyphSource2D.h"
#include "vtkPointPlacer.h"
#include "vtkPointSetToLabelHierarchy.h"
#include "vtkPolyDataMapper2D.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"
#include "vtkTubeFilter.h"
 
class vtkMRMLInteractionEventData;
 
class VTK_SLICER_MARKUPS_MODULE_VTKWIDGETS_EXPORT vtkSlicerMarkupsWidgetRepresentation : public vtkMRMLAbstractWidgetRepresentation
{
public:
  enum
  {
    Unselected,
    Selected,
    Active,
    Project,
    ProjectBack,
    NumberOfControlPointTypes
  };

  vtkTypeMacro(vtkSlicerMarkupsWidgetRepresentation, vtkMRMLAbstractWidgetRepresentation);
  void PrintSelf(ostream& os, vtkIndent indent) override;

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

  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 int GetNumberOfControlPoints();

  virtual int GetNthControlPointDisplayPosition(int n, double pos[2]);

  virtual vtkMRMLMarkupsNode::ControlPoint* GetNthControlPoint(int n);

  virtual void SetMarkupsDisplayNode(vtkMRMLMarkupsDisplayNode* markupsDisplayNode);
  virtual vtkMRMLMarkupsDisplayNode* GetMarkupsDisplayNode();
  virtual vtkMRMLMarkupsNode* GetMarkupsNode();

  virtual void UpdateCenterOfRotation();

  virtual bool GetTransformationReferencePoint(double referencePointWorld[3]);

  virtual void CanInteract(vtkMRMLInteractionEventData* interactionEventData, int& foundComponentType, int& foundComponentIndex, double& closestDistance2);
 
  virtual int FindClosestPointOnWidget(const int displayPos[2], double worldPos[3], int* idx);
 
  virtual vtkPointPlacer* GetPointPlacer();

  virtual vtkPolyData* GetControlPointsPolyData(int controlPointType);
  virtual vtkPolyData* GetLabelControlPointsPolyData(int pipeline);
  virtual vtkStringArray* GetLabels(int pipeline);


  bool IsDisplayable();
 
protected:
  vtkSlicerMarkupsWidgetRepresentation();
  ~vtkSlicerMarkupsWidgetRepresentation() override;
 
  // Convert glyph types from display node enums to 2D glyph source enums
  static int GetGlyphTypeSourceFromDisplay(int glyphTypeDisplay);
 
  class ControlPointsPipeline
  {
  public:
    ControlPointsPipeline();
    virtual ~ControlPointsPipeline();

    // vtkSmartPointer<vtkPolyData> PointMarkerShape;
    vtkSmartPointer<vtkMarkupsGlyphSource2D> GlyphSource2D;
    vtkSmartPointer<vtkSphereSource> GlyphSourceSphere;
 
    vtkSmartPointer<vtkPolyData> ControlPointsPolyData;
    vtkSmartPointer<vtkPoints> ControlPoints;
    vtkSmartPointer<vtkPolyData> LabelControlPointsPolyData;
    vtkSmartPointer<vtkPoints> LabelControlPoints;
    vtkSmartPointer<vtkPointSetToLabelHierarchy> PointSetToLabelHierarchyFilter;
    vtkSmartPointer<vtkStringArray> Labels;
    vtkSmartPointer<vtkStringArray> LabelsPriority;
    vtkSmartPointer<vtkTextProperty> TextProperty;
  };
 
  // Calculate view size and scale factor
  virtual void UpdateViewScaleFactor() = 0;
 
  virtual void UpdateControlPointSize() = 0;
 
  double ViewScaleFactorMmPerPixel;
  double ScreenSizePixel; // diagonal size of the screen
 
  // Control point size, specified in renderer world coordinate system.
  // For slice views, renderer world coordinate system is the display coordinate system, so it is measured in pixels.
  // For 3D views, renderer world coordinate system is the Slicer world coordinate system, so it is measured in the
  // scene length unit (typically millimeters).
  double ControlPointSize;
 
  virtual void SetMarkupsNode(vtkMRMLMarkupsNode* markupsNode);
 
  vtkWeakPointer<vtkMRMLMarkupsDisplayNode> MarkupsDisplayNode;
  vtkWeakPointer<vtkMRMLMarkupsNode> MarkupsNode;
 
  vtkSmartPointer<vtkPointPlacer> PointPlacer;
 
  vtkSmartPointer<vtkTextActor> TextActor;
 
  vtkTypeBool CurveClosed;

  virtual bool GetAllControlPointsVisible();

  bool GetAllControlPointsSelected();
 
  // Utility function to build straight lines between control points.
  // If displayPosition is true then positions will be computed in display coordinate system,
  // otherwise in world coordinate system.
  // displayPosition is normally set to true in 2D, and to false in 3D representations.
  void BuildLine(vtkPolyData* linePolyData, bool displayPosition);
 
  vtkTimeStamp MarkupsTransformModifiedTime;
 
  double* GetWidgetColor(int controlPointType) VTK_SIZEHINT(3);
 
  ControlPointsPipeline* ControlPoints[NumberOfControlPointTypes]; // Unselected, Selected, Active, Project, ProjectBehind
 
private:
  vtkSlicerMarkupsWidgetRepresentation(const vtkSlicerMarkupsWidgetRepresentation&) = delete;
  void operator=(const vtkSlicerMarkupsWidgetRepresentation&) = delete;
};
 
#endif