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