vtkMRMLDiffusionTensorVolumeDisplayNode.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: vtkMRMLDiffusionTensorVolumeDisplayNode.h,v $
  Date:      $Date: 2006/03/19 17:12:29 $
  Version:   $Revision: 1.3 $
 
=========================================================================auto=*/
 
#ifndef __vtkMRMLDiffusionTensorVolumeDisplayNode_h
#define __vtkMRMLDiffusionTensorVolumeDisplayNode_h
 
#include "vtkMRMLGlyphableVolumeDisplayNode.h"
#include "vtkMRMLDiffusionTensorDisplayPropertiesNode.h"
class vtkMRMLGlyphableVolumeSliceDisplayNode;
 
class vtkAlgorithmOutput;
class vtkDiffusionTensorMathematics;
class vtkDiffusionTensorGlyph;
class vtkImageCast;
class vtkImageData;
class vtkImageExtractComponents;
class vtkImageShiftScale;
class vtkImageMathematics;
class vtkMatrix4x4;

class VTK_MRML_EXPORT vtkMRMLDiffusionTensorVolumeDisplayNode : public vtkMRMLGlyphableVolumeDisplayNode
{
public:
  static vtkMRMLDiffusionTensorVolumeDisplayNode* New();
  vtkTypeMacro(vtkMRMLDiffusionTensorVolumeDisplayNode, vtkMRMLGlyphableVolumeDisplayNode);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  vtkMRMLNode* CreateNodeInstance() override;

  void ReadXMLAttributes(const char** atts) override;

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

  void Copy(vtkMRMLNode* node) override;

  const char* GetNodeTagName() override { return "DiffusionTensorVolumeDisplay"; }
 
  // virtual vtkPolyData* ExecuteGlyphPipeLineAndGetPolyData( vtkImageData* );

  void UpdateReferences() override;

  void UpdateScene(vtkMRMLScene* scene) override;

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

  void ProcessMRMLEvents(vtkObject* /*caller*/, unsigned long /*event*/, void* /*callData*/) override;
 
  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------
 
  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  vtkGetMacro(ScalarInvariant, int);

  vtkSetMacro(ScalarInvariant, int);

  void SetScalarInvariantToTrace() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::Trace); };
 
  // Description:
  void SetScalarInvariantToRelativeAnisotropy() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::RelativeAnisotropy); };

  void SetScalarInvariantToFractionalAnisotropy() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::FractionalAnisotropy); };

  void SetScalarInvariantToLinearMeasure() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::LinearMeasure); };

  void SetScalarInvariantToPlanarMeasure() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::PlanarMeasure); };

  void SetScalarInvariantToSphericalMeasure() { this->SetScalarInvariant(vtkMRMLDiffusionTensorDisplayPropertiesNode::SphericalMeasure); };

  virtual const char* GetScalarInvariantAsString() { return vtkMRMLDiffusionTensorDisplayPropertiesNode::GetScalarEnumAsString(this->ScalarInvariant); };

  vtkAlgorithmOutput* GetInputImageDataConnection() override;

  vtkAlgorithmOutput* GetBackgroundImageStencilDataConnection() override;
 
  void UpdateImageDataPipeline() override;

  void SetTensorRotationMatrix(vtkMatrix4x4*);
 
  vtkGetObjectMacro(DTIMathematics, vtkDiffusionTensorMathematics);
  vtkGetObjectMacro(DTIMathematicsAlpha, vtkDiffusionTensorMathematics);
  vtkGetObjectMacro(ShiftScale, vtkImageShiftScale);

  std::vector<vtkMRMLGlyphableVolumeSliceDisplayNode*> GetSliceGlyphDisplayNodes(vtkMRMLVolumeNode* node) override;

  void AddSliceGlyphDisplayNodes(vtkMRMLVolumeNode* node) override;

  void GetDisplayScalarRange(double range[2]) override;
 
  static int GetNumberOfScalarInvariants();
  static int GetNthScalarInvariant(int i);
 
protected:
  vtkMRMLDiffusionTensorVolumeDisplayNode();
  ~vtkMRMLDiffusionTensorVolumeDisplayNode() override;
  vtkMRMLDiffusionTensorVolumeDisplayNode(const vtkMRMLDiffusionTensorVolumeDisplayNode&);
  void operator=(const vtkMRMLDiffusionTensorVolumeDisplayNode&);

  void SetInputToImageDataPipeline(vtkAlgorithmOutput* imageDataConnection) override;
 
  vtkAlgorithmOutput* GetScalarImageDataConnection() override;
 
  static std::vector<int> GetSupportedColorModes();
 
  vtkDiffusionTensorGlyph* DiffusionTensorGlyphFilter;

  vtkDiffusionTensorMathematics* DTIMathematics;
  vtkDiffusionTensorMathematics* DTIMathematicsAlpha;
 
  vtkImageShiftScale* ShiftScale;
 
  vtkImageMathematics* ImageMath;
 
  vtkImageCast* ImageCast;

  int ScalarInvariant;
};
 
#endif