vtkMRMLVolumePropertyNode.h

Go to the documentation of this file.

#ifndef __vtkMRMLVolumePropertyNode_h
#define __vtkMRMLVolumePropertyNode_h

// VolumeRendering includes
#include "vtkSlicerVolumeRenderingModuleMRMLExport.h"

// MRML includes
#include "vtkMRMLStorableNode.h"

// VTK includes
class vtkColorTransferFunction;
class vtkIntArray;
class vtkPiecewiseFunction;
class vtkVolumeProperty;

// STD includes
#include <string>
#include <vector>

#define COUNT_CROPPING_REGION_PLANES 6

class VTK_SLICER_VOLUMERENDERING_MODULE_MRML_EXPORT vtkMRMLVolumePropertyNode
  : public vtkMRMLStorableNode
{
public:
  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------

  static vtkMRMLVolumePropertyNode *New();
  vtkTypeMacro(vtkMRMLVolumePropertyNode,vtkMRMLStorableNode);
  void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE;

  vtkGetObjectMacro(VolumeProperty,vtkVolumeProperty);

  void SetScalarOpacity(vtkPiecewiseFunction* newScalarOpacity, int component = 0);\
  vtkPiecewiseFunction* GetScalarOpacity(int component = 0);

  void SetGradientOpacity(vtkPiecewiseFunction* newGradientOpacity, int component = 0);
  vtkPiecewiseFunction* GetGradientOpacity(int component = 0);

  void SetColor(vtkColorTransferFunction* newColorFunction, int component = 0);
  vtkColorTransferFunction* GetColor(int component = 0);

  static std::string GetPiecewiseFunctionString(vtkPiecewiseFunction* function);

  static std::string GetColorTransferFunctionString(vtkColorTransferFunction* function);

  static void GetPiecewiseFunctionFromString(const std::string& str,
                                             vtkPiecewiseFunction* result);

  static void GetColorTransferFunctionFromString(const std::string& str,
                                                 vtkColorTransferFunction* result);

  static inline void GetPiecewiseFunctionFromString(const char *str,
                                                    vtkPiecewiseFunction* result);

  static inline void GetColorTransferFunctionFromString(const char *str,
                                                        vtkColorTransferFunction* result);

  static double NextHigher(double value);
  static double HigherAndUnique(double value, double & previousValue);

  //--------------------------------------------------------------------------
  //--------------------------------------------------------------------------
  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;

  void CopyParameterSet(vtkMRMLNode *node);

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

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

  virtual vtkMRMLStorageNode* CreateDefaultStorageNode() VTK_OVERRIDE;

  virtual bool GetModifiedSinceRead() VTK_OVERRIDE;

protected:
  vtkMRMLVolumePropertyNode(void);
  ~vtkMRMLVolumePropertyNode(void);

  static int NodesFromString(const std::string& dataString, double* &data, int nodeSize);
  static int DataFromString(const std::string& dataString, double* &data);
  static std::string DataToString(double* data, int size);

  vtkIntArray* ObservedEvents;

  vtkVolumeProperty* VolumeProperty;

private:
  vtkMRMLVolumePropertyNode(const vtkMRMLVolumePropertyNode&);//Not implemented
  void operator=(const vtkMRMLVolumePropertyNode&);

};

//---------------------------------------------------------------------------
void vtkMRMLVolumePropertyNode
::GetPiecewiseFunctionFromString(const char *str,
                                 vtkPiecewiseFunction* result)
{
  vtkMRMLVolumePropertyNode::GetPiecewiseFunctionFromString(
    std::string(str), result);
}

//---------------------------------------------------------------------------
void vtkMRMLVolumePropertyNode
::GetColorTransferFunctionFromString(const char *str,
                                     vtkColorTransferFunction* result)
{
  vtkMRMLVolumePropertyNode::GetColorTransferFunctionFromString(
    std::string (str), result);
}

#endif