itkDiffusionTensor3DInterpolateImageFunctionReimplementation.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 itkDiffusionTensor3DInterpolateImageFunctionReimplementation_h
#define itkDiffusionTensor3DInterpolateImageFunctionReimplementation_h

#include "itkDiffusionTensor3DInterpolateImageFunction.h"
#include <itkImage.h>
#include <itkInterpolateImageFunction.h>
#include "itkSeparateComponentsOfADiffusionTensorImage.h"

namespace itk
{

template <class TData, class TCoordRep = double>
class DiffusionTensor3DInterpolateImageFunctionReimplementation :
  public DiffusionTensor3DInterpolateImageFunction<TData, TCoordRep>
{
public:
  typedef TData                                                          DataType;
  typedef DiffusionTensor3DInterpolateImageFunctionReimplementation      Self;
  typedef DiffusionTensor3DInterpolateImageFunction<DataType, TCoordRep> Superclass;
  typedef typename Superclass::TensorDataType                            TensorDataType;
  typedef typename Superclass::DiffusionImageType                        DiffusionImageType;
  typedef typename Superclass::DiffusionImageTypePointer                 DiffusionImageTypePointer;
  typedef Image<DataType, 3>                                             ImageType;
  typedef typename ImageType::Pointer                                    ImagePointer;
  typedef typename Superclass::PointType                                 PointType;
  typedef SmartPointer<Self>                                             Pointer;
  typedef SmartPointer<const Self>                                       ConstPointer;
  typedef ImageRegionIteratorWithIndex<DiffusionImageType>               IteratorDiffusionImageType;
  typedef ImageRegionIteratorWithIndex<ImageType>                        IteratorImageType;
  typedef InterpolateImageFunction<ImageType, double>                    InterpolateImageFunctionType;
  typedef typename DiffusionImageType::RegionType                        itkRegionType;
  typedef typename DiffusionImageType::SizeType                          SizeType;
  typedef typename Superclass::ContinuousIndexType                       ContinuousIndexType;

  itkTypeMacro(DiffusionTensor3DInterpolateImageFunctionReimplementation, DiffusionTensor3DInterpolateImageFunction);

  // TensorDataType Evaluate( const PointType &point ) ;
  TensorDataType EvaluateAtContinuousIndex( const ContinuousIndexType & index ) const override;

  void SetInputImage( const DiffusionImageType *inputImage ) override;

  itkSetMacro( NumberOfThreads, int );
protected:
  DiffusionTensor3DInterpolateImageFunctionReimplementation();
  virtual void AllocateInterpolator() = 0;

  typename InterpolateImageFunctionType::Pointer m_Interpol[6];
  ImagePointer m_ImageVec[6];
  int          m_NumberOfThreads;
};

} // end namespace itk

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

#endif