vtkTeemNRRDWriter.h

Go to the documentation of this file.

 
#ifndef __vtkTeemNRRDWriter_h
#define __vtkTeemNRRDWriter_h
 
#include "vtkWriter.h"
 
#include "vtkDoubleArray.h"
#include "vtkMatrix4x4.h"
#include "vtkSmartPointer.h"
#include "teem/nrrd.h"
 
#include "vtkTeemConfigure.h"
 
class vtkImageData;
class AttributeMapType;
class AxisInfoMapType;

class VTK_Teem_EXPORT vtkTeemNRRDWriter : public vtkWriter
{
public:
  vtkTypeMacro(vtkTeemNRRDWriter, vtkWriter);
  void PrintSelf(ostream& os, vtkIndent indent) override;
 
  static vtkTeemNRRDWriter* New();

  vtkImageData* GetInput();
  vtkImageData* GetInput(int port);

  vtkSetStringMacro(FileName);
  vtkGetStringMacro(FileName);
 
  vtkSetObjectMacro(DiffusionGradients, vtkDoubleArray);
  vtkGetObjectMacro(DiffusionGradients, vtkDoubleArray);
 
  vtkSetObjectMacro(BValues, vtkDoubleArray);
  vtkGetObjectMacro(BValues, vtkDoubleArray);
 
  vtkSetObjectMacro(IJKToRASMatrix, vtkMatrix4x4);
  vtkGetObjectMacro(IJKToRASMatrix, vtkMatrix4x4);
 
  vtkSetObjectMacro(MeasurementFrameMatrix, vtkMatrix4x4);
  vtkGetObjectMacro(MeasurementFrameMatrix, vtkMatrix4x4);
 
  vtkSetMacro(UseCompression, int);
  vtkGetMacro(UseCompression, int);
  vtkBooleanMacro(UseCompression, int);
 
  vtkSetClampMacro(CompressionLevel, int, 0, 9);
  vtkGetMacro(CompressionLevel, int);
 
  vtkSetClampMacro(FileType, int, VTK_ASCII, VTK_BINARY);
  vtkGetMacro(FileType, int);
  void SetFileTypeToASCII() { this->SetFileType(VTK_ASCII); };
  void SetFileTypeToBinary() { this->SetFileType(VTK_BINARY); };

  vtkBooleanMacro(WriteError, int);
  vtkSetMacro(WriteError, int);
  vtkGetMacro(WriteError, int);

  void SetAttribute(const std::string& name, const std::string& value);

  void SetAxisLabel(unsigned int axis, const char* label);

  void SetAxisUnit(unsigned int axis, const char* unit);

  void SetVectorAxisKind(int kind);

  vtkSetMacro(Space, int);
  vtkGetMacro(Space, int);

  void vtkSetSpaceToRAS() { this->SetSpace(nrrdSpaceRightAnteriorSuperior); };
  void vtkSetSpaceToRAST() { this->SetSpace(nrrdSpaceRightAnteriorSuperiorTime); };

  void vtkSetSpaceToLPS() { this->SetSpace(nrrdSpaceLeftPosteriorSuperior); };
  void vtkSetSpaceToLPST() { this->SetSpace(nrrdSpaceLeftPosteriorSuperiorTime); };

  vtkSetMacro(ForceRangeAxis, bool);
  vtkGetMacro(ForceRangeAxis, bool);
  vtkBooleanMacro(ForceRangeAxis, bool);

  void* MakeNRRD();
 
protected:
  vtkTeemNRRDWriter();
  ~vtkTeemNRRDWriter() override;
 
  int FillInputPortInformation(int port, vtkInformation* info) override;

  void WriteData() override;

  int WriteError;
 
  char* FileName;
 
  vtkDoubleArray* BValues;
  vtkDoubleArray* DiffusionGradients;
 
  vtkMatrix4x4* IJKToRASMatrix;
  vtkMatrix4x4* MeasurementFrameMatrix;
 
  int UseCompression;
  int CompressionLevel;
  int FileType;
 
  AttributeMapType* Attributes;
  AxisInfoMapType* AxisLabels;
  AxisInfoMapType* AxisUnits;
  int VectorAxisKind;
  int Space;
 
  bool ForceRangeAxis;
 
private:
  vtkTeemNRRDWriter(const vtkTeemNRRDWriter&) = delete;
  void operator=(const vtkTeemNRRDWriter&) = delete;
  void vtkImageDataInfoToNrrdInfo(vtkImageData* in, int& nrrdKind, size_t& numComp, int& vtkType, void** buffer);
  int VTKToNrrdPixelType(const int vtkPixelType);
  int DiffusionWeightedData;
};
 
#endif