vtkMRMLSegmentationDisplayNode.h

Go to the documentation of this file.

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

  Program: 3D Slicer

  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 Csaba Pinter, PerkLab, Queen's University
  and was supported through the Applied Cancer Research Unit program of Cancer Care
  Ontario with funds provided by the Ontario Ministry of Health and Long-Term Care

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

#ifndef __vtkMRMLSegmentationDisplayNode_h
#define __vtkMRMLSegmentationDisplayNode_h

#include "vtkMRMLDisplayNode.h"

#include <set>

class vtkMRMLColorTableNode;
class vtkSegmentation;
class vtkStringArray;
class vtkVector3d;


class VTK_MRML_EXPORT vtkMRMLSegmentationDisplayNode : public vtkMRMLDisplayNode
{
public:
  // Define constants
  static const double SEGMENT_COLOR_NO_OVERRIDE;

  struct SegmentDisplayProperties
    {
    double OverrideColor[3];
    bool Visible; // Turns visibility on/off in general. Useful for allowing the user to show/hide a segment without changing any detailed visibility options.
    bool Visible3D;
    bool Visible2DFill; // This one is used for labelmap volume related operations (color table, merged labelmap)
    bool Visible2DOutline;
    double Opacity3D;
    double Opacity2DFill; // This one is used for labelmap volume related operations (color table, merged labelmap)
    double Opacity2DOutline;

    // Initialize with default values
    SegmentDisplayProperties()
    : Visible(true)
    , Visible3D(true)
    , Visible2DFill(true)
    , Visible2DOutline(true)
    , Opacity3D(1.0)
    , Opacity2DFill(1.0) // Default is 1, because these are relative values. Half transparency is default for the whole segmentation
    , Opacity2DOutline(1.0)
      {
      OverrideColor[0] = SEGMENT_COLOR_NO_OVERRIDE;
      OverrideColor[1] = SEGMENT_COLOR_NO_OVERRIDE;
      OverrideColor[2] = SEGMENT_COLOR_NO_OVERRIDE;
      }

    // Automatically generated operator= and copy constructor work
    // correctly for these members, so there is no need to define them.
    };

  typedef std::map<std::string, SegmentDisplayProperties> SegmentDisplayPropertiesMap;

public:
  static vtkMRMLSegmentationDisplayNode *New();
  vtkTypeMacro(vtkMRMLSegmentationDisplayNode,vtkMRMLDisplayNode);
  void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  virtual vtkMRMLNode* CreateNodeInstance() VTK_OVERRIDE;

  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 const char* GetNodeTagName() VTK_OVERRIDE { return "SegmentationDisplay"; }

public:
  vtkGetStringMacro(PreferredDisplayRepresentationName2D);
  vtkSetStringMacro(PreferredDisplayRepresentationName2D);
  vtkGetStringMacro(PreferredDisplayRepresentationName3D);
  vtkSetStringMacro(PreferredDisplayRepresentationName3D);

  vtkGetMacro(Visibility3D, bool);
  vtkSetMacro(Visibility3D, bool);
  vtkBooleanMacro(Visibility3D, bool);
  vtkGetMacro(Visibility2DFill, bool);
  vtkSetMacro(Visibility2DFill, bool);
  vtkBooleanMacro(Visibility2DFill, bool);
  vtkGetMacro(Visibility2DOutline, bool);
  vtkSetMacro(Visibility2DOutline, bool);
  vtkBooleanMacro(Visibility2DOutline, bool);

  vtkGetMacro(Opacity3D, double);
  vtkSetMacro(Opacity3D, double);
  vtkGetMacro(Opacity2DFill, double);
  vtkSetMacro(Opacity2DFill, double);
  vtkGetMacro(Opacity2DOutline, double);
  vtkSetMacro(Opacity2DOutline, double);

public:
  bool GetSegmentDisplayProperties(std::string segmentID, SegmentDisplayProperties &properties);

  void SetSegmentDisplayProperties(std::string segmentID, SegmentDisplayProperties &properties);

  void SetSegmentDisplayPropertiesToDefault(const std::string& segmentId);

  void RemoveSegmentDisplayProperties(std::string segmentID);

  void ClearSegmentDisplayProperties();

  bool CalculateAutoOpacitiesForSegments();

  void GenerateSegmentColor(double color[3], int colorNumber=0);
  void GenerateSegmentColor(double &r, double &g, double &b);

  void GetPolyDataRepresentationNames(std::set<std::string> &representationNames);

  std::string GetDisplayRepresentationName3D();

  std::string GetDisplayRepresentationName2D();

// Convenience and python compatibility functions
public:
  vtkVector3d GetSegmentColor(std::string segmentID);
  bool GetSegmentColor(std::string segmentID, double* color);
  bool GetSegmentColor(std::string segmentID, double &r, double &g, double &b);

  vtkVector3d GetSegmentOverrideColor(std::string segmentID);
  void SetSegmentOverrideColor(std::string segmentID, double r, double g, double b);
  void SetSegmentOverrideColor(std::string segmentID, vtkVector3d overrideColor);
  void UnsetSegmentOverrideColor(std::string segmentID);

  bool GetSegmentVisibility(std::string segmentID);
  void SetSegmentVisibility(std::string segmentID, bool visible);
  void SetAllSegmentsVisibility(bool visible);

  bool GetSegmentVisibility3D(std::string segmentID);
  void SetSegmentVisibility3D(std::string segmentID, bool visible);
  void SetAllSegmentsVisibility3D(bool visible, bool changeVisibleSegmentsOnly = false);

  bool GetSegmentVisibility2DFill(std::string segmentID);
  void SetSegmentVisibility2DFill(std::string segmentID, bool visible);
  void SetAllSegmentsVisibility2DFill(bool visible, bool changeVisibleSegmentsOnly = false);

  bool GetSegmentVisibility2DOutline(std::string segmentID);
  void SetSegmentVisibility2DOutline(std::string segmentID, bool visible);
  void SetAllSegmentsVisibility2DOutline(bool visible, bool changeVisibleSegmentsOnly = false);

  double GetSegmentOpacity3D(std::string segmentID);
  void SetSegmentOpacity3D(std::string segmentID, double opacity);
  void SetAllSegmentsOpacity3D(double opacity, bool changeVisibleSegmentsOnly = false);

  double GetSegmentOpacity2DFill(std::string segmentID);
  void SetSegmentOpacity2DFill(std::string segmentID, double opacity);
  void SetAllSegmentsOpacity2DFill(double opacity, bool changeVisibleSegmentsOnly = false);

  double GetSegmentOpacity2DOutline(std::string segmentID);
  void SetSegmentOpacity2DOutline(std::string segmentID, double opacity);
  void SetAllSegmentsOpacity2DOutline(double opacity, bool changeVisibleSegmentsOnly = false);

  void SetSegmentOpacity(std::string segmentID, double opacity);
  void SetAllSegmentsOpacity(double opacity, bool changeVisibleSegmentsOnly = false);

  void GetVisibleSegmentIDs(vtkStringArray* segmentIDs);

  void GetVisibleSegmentIDs(std::vector<std::string>& segmentIDs);

protected:
  void GetSegmentIDs(std::vector<std::string>& segmentIDs, bool visibleSegmentsOnly);

  void UpdateSegmentList(bool removeUnusedDisplayProperties = true);

protected:
  vtkMRMLSegmentationDisplayNode();
  virtual ~vtkMRMLSegmentationDisplayNode();
  vtkMRMLSegmentationDisplayNode(const vtkMRMLSegmentationDisplayNode&);
  void operator=(const vtkMRMLSegmentationDisplayNode&);

  friend class vtkMRMLSegmentationNode; // Access to UpdateSegmentList();

protected:
  char* PreferredDisplayRepresentationName2D;

  char* PreferredDisplayRepresentationName3D;

  SegmentDisplayPropertiesMap SegmentationDisplayProperties;

  unsigned int NumberOfGeneratedColors;

  vtkMTimeType SegmentListUpdateTime;
  vtkSegmentation* SegmentListUpdateSource;

  bool Visibility3D;
  bool Visibility2DFill;
  bool Visibility2DOutline;

  double Opacity3D;
  double Opacity2DFill;
  double Opacity2DOutline;
};

#endif