vtkMRMLFiberBundleDisplayNode.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: vtkMRMLFiberBundleDisplayNode.h,v $
  Date:      $Date: 2006/03/19 17:12:28 $
  Version:   $Revision: 1.6 $

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

#ifndef __vtkMRMLFiberBundleDisplayNode_h
#define __vtkMRMLFiberBundleDisplayNode_h

// MRML includes
#include <vtkMRMLModelDisplayNode.h>

// Tractography includes
#include "vtkSlicerTractographyDisplayModuleMRMLExport.h"

class vtkMRMLDiffusionTensorDisplayPropertiesNode;

class VTK_SLICER_TRACTOGRAPHYDISPLAY_MODULE_MRML_EXPORT vtkMRMLFiberBundleDisplayNode : public vtkMRMLModelDisplayNode
{
 public:
  vtkTypeMacro(vtkMRMLFiberBundleDisplayNode, vtkMRMLModelDisplayNode);
  void PrintSelf(ostream& os, vtkIndent indent);

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  virtual void ReadXMLAttributes ( const char** atts );

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


  virtual void Copy ( vtkMRMLNode *node );

  virtual const char* GetNodeTagName() = 0;

  virtual void UpdateReferences();

  virtual void UpdateScene(vtkMRMLScene *scene);

  virtual void UpdateReferenceID(const char *oldID, const char *newID);

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------


  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  enum
  {
    colorModeSolid = 0,
    colorModeScalar = 1,
    colorModeFunctionOfScalar = 2,
    colorModeUseCellScalars = 3,
    colorModeScalarData = 4,
    colorModePointFiberOrientation = 5,
    colorModeMeanFiberOrientation = 6,
  };

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  vtkGetMacro ( ColorMode, int );
  vtkSetMacro ( ColorMode, int );

  void SetColorModeToSolid ( ) {
    this->SetColorMode ( this->colorModeSolid );
  };

  void SetColorModeToScalar ( ) {
    this->SetColorMode ( this->colorModeScalar );
  };

  void SetColorModeToFunctionOfScalar ( ) {
    this->SetColorMode ( this->colorModeFunctionOfScalar );
  };

  void SetColorModeToUseCellScalars ( ) {
    this->SetColorMode ( this->colorModeUseCellScalars );
  };

  void SetColorModeToScalarData ( ) {
    this->SetColorMode ( this->colorModeScalarData );
  };

  void SetColorModeToMeanFiberOrientation ( ) {
    this->SetColorMode ( this->colorModeMeanFiberOrientation );
  };

  void SetColorModeToPointFiberOrientation ( ) {
    this->SetColorMode ( this->colorModePointFiberOrientation );
  };

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------


  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------



  vtkMRMLDiffusionTensorDisplayPropertiesNode* GetDiffusionTensorDisplayPropertiesNode ( );

  void SetAndObserveDiffusionTensorDisplayPropertiesNodeID ( const char *ID );

  vtkGetStringMacro(DiffusionTensorDisplayPropertiesNodeID);


  static int GetNumberOfScalarInvariants();
  static int GetNthScalarInvariant(int i);

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

  vtkMRMLDiffusionTensorDisplayPropertiesNode *DiffusionTensorDisplayPropertiesNode;
  char *DiffusionTensorDisplayPropertiesNodeID;

  virtual void SetDiffusionTensorDisplayPropertiesNodeID(const char* id);

  static std::vector<int> GetSupportedColorModes();

  int ColorMode;

  //double ScalarRange[2];
  //
};

#endif