itkDWISingleVoxelGenerator.h

Go to the documentation of this file.

#ifndef __itkDWISingleVoxelGenerator_h
#define __itkDWISingleVoxelGenerator_h

#include "utl.h"
#include "itkDWIGeneratorBase.h"

namespace itk
{

template <class TOutputImage, class TScalarImage=Image<float,3> >
class ITK_EXPORT DWISingleVoxelGenerator : public DWIGeneratorBase<TOutputImage, TScalarImage>
{
public:
  typedef DWISingleVoxelGenerator                Self;
  typedef DWIGeneratorBase<TOutputImage, TScalarImage>   Superclass;
  typedef SmartPointer<Self>          Pointer;
  
  itkNewMacro(Self);

  itkTypeMacro(DWISingleVoxelGenerator, DWIGeneratorBase);

  typedef typename Superclass::OutputImageType                  OutputImageType;
  typedef typename Superclass::OutputImagePointer               OutputImagePointer;
  typedef typename Superclass::OutputImageSizeType              OutputImageSizeType;
  typedef typename Superclass::OutputImageSpacingType           OutputImageSpacingType;
  typedef typename Superclass::OutputImageIndexType             OutputImageIndexType;
  typedef typename Superclass::OutputImagePointType             OutputImagePointType;
  typedef typename Superclass::OutputImageDirectionType         OutputImageDirectionType;
  typedef typename Superclass::OutputImageRegionType            OutputImageRegionType;
  typedef typename Superclass::OutputImagePixelType             OutputImagePixelType;
  typedef typename Superclass::OutputImageInternalPixelType     OutputImageInternalPixelType;

  itkStaticConstMacro(OutputImageDimension, unsigned int, OutputImageType::ImageDimension);

  typedef typename Superclass::ScalarImageType           ScalarImageType;
  typedef typename Superclass::ScalarImagePointer        ScalarImagePointer;

  typedef typename Superclass::PrecisionType         PrecisionType;
  typedef typename Superclass::MatrixType            MatrixType;
  typedef typename Superclass::MatrixPointer         MatrixPointer;
  typedef typename Superclass::STDVectorType         STDVectorType;
  typedef typename Superclass::STDVectorPointer      STDVectorPointer;
    
  typedef typename Superclass::DiffusionParameterValuesType    DiffusionParameterValuesType;
  typedef typename Superclass::DiffusionParameterContainerType DiffusionParameterContainerType;
  
  typedef enum 
    {
    FIXED=0, 
    UNIFORM 
    } RandomType;
  
  itkSetMacro( StoredOrientationMatrix, MatrixPointer);
  itkGetMacro( StoredOrientationMatrix, MatrixPointer );

  void SetUniformFromElectrostaticEnergy(const int num)
    {
    m_StoredOrientationMatrix = utl::ReadGradElectricRepulsion<double>(num,DIRECTION_NODUPLICATE,CARTESIAN_TO_CARTESIAN, DIRECTION_NOFLIP, DIRECTION_NOFLIP, DIRECTION_NOFLIP, true);
    this->Modified();
    }
  void SetUniformFromTessellation(const int num)
    {
    m_StoredOrientationMatrix = utl::ReadGrad<double>(num,DIRECTION_NODUPLICATE,CARTESIAN_TO_CARTESIAN,DIRECTION_NOFLIP, DIRECTION_NOFLIP, DIRECTION_NOFLIP, true);
    this->Modified();
    }


  
  itkSetMacro( RandomType, RandomType );
  itkGetMacro( RandomType, RandomType );

  itkSetMacro( DiffusionParameterValues, DiffusionParameterValuesType );
  itkGetMacro( DiffusionParameterValues, DiffusionParameterValuesType );
  
protected:
  DWISingleVoxelGenerator();
  ~DWISingleVoxelGenerator(){};

  typename LightObject::Pointer InternalClone() const;
  void PrintSelf(std::ostream& os, Indent indent) const;
  
  void GenerateData();
  
  void Initialization();
  
  DiffusionParameterValuesType m_DiffusionParameterValues;

  RandomType m_RandomType;

  MatrixPointer m_StoredOrientationMatrix;

private:
  DWISingleVoxelGenerator(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
  
  
};

}

#if !defined(ITK_MANUAL_INSTANTIATION) && !defined(__itkDWISingleVoxelGenerator_hxx)
#include "itkDWISingleVoxelGenerator.hxx"
#endif


#endif