itkDiffusionTensor3DWrite.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Diffusion Applications
  Module:    $HeadURL$
  Language:  C++
  Date:      $Date$
  Version:   $Revision$

  Copyright (c) Brigham and Women's Hospital (BWH) All Rights Reserved.

  See License.txt or http://www.slicer.org/copyright/copyright.txt for details.

==========================================================================*/
#ifndef itkDiffusionTensor3DWrite_h
#define itkDiffusionTensor3DWrite_h

#include <itkObject.h>
#include <itkMetaDataObject.h>
#include <itkImage.h>
#include <itkMatrix.h>
#include <itkImageFileWriter.h>
#include <itkNrrdImageIO.h>
#include <itkImageIOBase.h>
#include "itkDiffusionTensor3D.h"

namespace itk
{

template <class TData>
class DiffusionTensor3DWrite : public Object
{
public:
  typedef TData                               DataType;
  typedef DiffusionTensor3DWrite              Self;
  typedef DiffusionTensor3D<DataType>         TensorDataType;
  typedef Image<TensorDataType, 3>            DiffusionImageType;
  typedef MetaDataDictionary                  DictionaryType;
  typedef ImageFileWriter<DiffusionImageType> WriterType;
  typedef SmartPointer<Self>                  Pointer;
  typedef SmartPointer<const Self>            ConstPointer;
  typedef std::vector<std::vector<double> >   DoubleVectorType;
  typedef MetaDataObject<DoubleVectorType>    MetaDataDoubleVectorType;
  typedef MetaDataObject<std::string>         MetaDataIntType;

  itkTypeMacro(DiffusionTensor3DWrite, Object);

  itkNewMacro( Self );
  // /Set input tensor image
  itkSetObjectMacro( Input, DiffusionImageType );
  // /Write the image in the given file
  int Update( const char* output );

  // /Set the metadatadictionary of the image, including its measurement frame
  void SetMetaDataDictionary( DictionaryType dic );

  // /Set Number of Threads
  itkSetMacro( NumberOfThreads, unsigned int);
  void SetMeasurementFrame( Matrix<double, 3, 3> measurementFrame );

//  Space:
//  nrrdSpaceUnknown,
//  nrrdSpaceRightAnteriorSuperior,     /*  1: NIFTI-1 (right-handed) */
//  nrrdSpaceLeftAnteriorSuperior,      /*  2: standard Analyze (left-handed) */
//  nrrdSpaceLeftPosteriorSuperior,     /*  3: DICOM 3.0 (right-handed) */
//  nrrdSpaceRightAnteriorSuperiorTime, /*  4: */
//  nrrdSpaceLeftAnteriorSuperiorTime,  /*  5: */
//  nrrdSpaceLeftPosteriorSuperiorTime, /*  6: */
//  nrrdSpaceScannerXYZ,                /*  7: ACR/NEMA 2.0 (pre-DICOM 3.0) */
//  nrrdSpaceScannerXYZTime,            /*  8: */
//  nrrdSpace3DRightHanded,             /*  9: */
//  nrrdSpace3DLeftHanded,              /* 10: */
//  nrrdSpace3DRightHandedTime,         /* 11: */
//  nrrdSpace3DLeftHandedTime,          /* 12: */
//  nrrdSpaceLast
  void SetSpace(int space);

private:
  DiffusionTensor3DWrite();

  typename DiffusionImageType::Pointer m_Input;

  unsigned int   m_NumberOfThreads;
  DictionaryType m_MetaDataDictionary;
};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkDiffusionTensor3DWrite.txx"
#endif

#endif