vtkMRMLModelDisplayNode.h

Go to the documentation of this file.

/*=auto=========================================================================

  Portions (c) Copyright 2005 Brigham and Women's Hospital (BWH) All Rights Reserved.

  See COPYRIGHT.txt
  or http://www.slicer.org/copyright/copyright.txt for details.

  Program:   3D Slicer
  Module:    $RCSfile: vtkMRMLModelDisplayNode.h,v $
  Date:      $Date: 2006/03/19 17:12:28 $
  Version:   $Revision: 1.6 $

=========================================================================auto=*/

#ifndef __vtkMRMLModelDisplayNode_h
#define __vtkMRMLModelDisplayNode_h

// MRML includes
#include "vtkMRMLDisplayNode.h"

// VTK includes
class vtkAlgorithm;
class vtkAlgorithmOutput;
class vtkAssignAttribute;
class vtkGeometryFilter;
class vtkThreshold;
class vtkPassThrough;
class vtkPointSet;
class vtkPolyData;
class vtkUnstructuredGrid;

class VTK_MRML_EXPORT vtkMRMLModelDisplayNode : public vtkMRMLDisplayNode
{
public:
  static vtkMRMLModelDisplayNode *New();
  vtkTypeMacro(vtkMRMLModelDisplayNode,vtkMRMLDisplayNode);
  void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  enum SliceDisplayModeType
    {
    SliceDisplayIntersection, 
    SliceDisplayProjection, 
    SliceDisplayDistanceEncodedProjection, 
    SliceDisplayMode_Last // placeholder after the last valid value, this must be the last in the list of modes
    };

  virtual void ReadXMLAttributes(const char** atts) VTK_OVERRIDE;

  virtual void WriteXML(ostream& of, int indent) VTK_OVERRIDE;

  virtual void Copy(vtkMRMLNode *node) VTK_OVERRIDE;

  virtual vtkMRMLNode* CreateNodeInstance() VTK_OVERRIDE;

  virtual const char* GetNodeTagName() VTK_OVERRIDE {return "ModelDisplay";}

  virtual void SetInputMeshConnection(vtkAlgorithmOutput* meshConnection);
  virtual void SetInputPolyDataConnection(vtkAlgorithmOutput* polyDataConnection);

  virtual vtkAlgorithmOutput* GetInputMeshConnection();
  virtual vtkAlgorithmOutput* GetInputPolyDataConnection();

  virtual vtkPointSet* GetInputMesh();
  virtual vtkPolyData* GetInputPolyData();
  virtual vtkUnstructuredGrid* GetInputUnstructuredGrid();

  virtual vtkPointSet* GetOutputMesh();
  virtual vtkPolyData* GetOutputPolyData();
  virtual vtkUnstructuredGrid* GetOutputUnstructuredGrid();

  virtual vtkAlgorithmOutput* GetOutputMeshConnection();
  virtual vtkAlgorithmOutput* GetOutputPolyDataConnection(); 

  virtual void SetActiveScalarName(const char *scalarName) VTK_OVERRIDE;

  virtual void SetActiveAttributeLocation(int location) VTK_OVERRIDE;

  virtual void SetActiveScalar(const char *scalarName, int location);

  virtual void SetScalarRangeFlag(int flag) VTK_OVERRIDE;

  void SetThresholdEnabled(bool enabled);
  vtkBooleanMacro(ThresholdEnabled,bool);

  vtkGetMacro(ThresholdEnabled,bool);

  void SetThresholdRange(double min, double max);
  void SetThresholdRange(double range[2]);

  void GetThresholdRange(double range[2]);
  double* GetThresholdRange();
  double GetThresholdMin();
  double GetThresholdMax();

  vtkGetMacro(SliceDisplayMode, int);
  vtkSetMacro(SliceDisplayMode, int);
  void SetSliceDisplayModeToIntersection();
  void SetSliceDisplayModeToProjection();
  void SetSliceDisplayModeToDistanceEncodedProjection();

  static const char* GetSliceDisplayModeAsString(int id);
  static int GetSliceDisplayModeFromString(const char* name);

  virtual void SetAndObserveDistanceEncodedProjectionColorNodeID(const char *colorNodeID);
  virtual const char* GetDistanceEncodedProjectionColorNodeID();
  virtual vtkMRMLColorNode* GetDistanceEncodedProjectionColorNode();

  virtual vtkDataArray* GetActiveScalarArray();

protected:
  vtkMRMLModelDisplayNode();
  ~vtkMRMLModelDisplayNode();
  vtkMRMLModelDisplayNode(const vtkMRMLModelDisplayNode&);
  void operator=(const vtkMRMLModelDisplayNode&);

  virtual void ProcessMRMLEvents(vtkObject *caller,
                                 unsigned long event,
                                 void *callData) VTK_OVERRIDE;

  virtual void SetInputToMeshPipeline(vtkAlgorithmOutput* meshConnection);

  virtual void UpdateAssignedAttribute();

  virtual void UpdateScalarRange();

  vtkAssignAttribute* AssignAttribute;

  vtkPassThrough* PassThrough;

  vtkThreshold* ThresholdFilter;

  vtkGeometryFilter* ConvertToPolyDataFilter;

  bool ThresholdEnabled;

  int SliceDisplayMode;

  double ThresholdRangeTemp[2];
};

#endif