itkDasTransformImageFilter.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef itkDasTransformImageFilter_h
#define itkDasTransformImageFilter_h
#include <itkVector.h>
#include <itkImage.h>
#include <itkImageToImageFilter.h>
#include <itkLinearInterpolateImageFunction.h>

namespace itk {

template <typename TScalarType, unsigned int NDimensions = 3>
class ITK_EXPORT DasTransformImageFilter:
    public ImageToImageFilter<
                               Image< TScalarType, NDimensions>,
                               Image< TScalarType, NDimensions>
                             >
{

public:  
  typedef DasTransformImageFilter                               Self;
  typedef ImageToImageFilter<
                              Image< TScalarType, NDimensions>,
                              Image< TScalarType, NDimensions>
                            >                                   Superclass;
  typedef SmartPointer<Self>                                    Pointer;
  typedef SmartPointer<const Self>                              ConstPointer;
  typedef Image< TScalarType, NDimensions >                     InputImageType;
  typedef typename InputImageType::IndexType                    InputImageIndexType;
  typedef typename InputImageType::RegionType                   InputImageRegionType;
  typedef Vector< TScalarType, NDimensions >                    VectorPixelType;
  typedef Image< VectorPixelType, NDimensions >                 VectorImageType;
  typedef TScalarType                                           OutputPixelType;
  typedef InputImageType                                        OutputImageType;
  typedef LinearInterpolateImageFunction< InputImageType, 
                                          TScalarType >         LinearInterpolatorType;

  itkNewMacro(Self);

  itkTypeMacro(DasTransformImageFilter, ImageToImageFilter);

  itkStaticConstMacro(Dimension, unsigned int, NDimensions);

  itkSetMacro(FileName, std::string);
  itkGetMacro(FileName, std::string);
  
  itkSetMacro(WriteTransformedMovingImage, bool);
  itkGetMacro(WriteTransformedMovingImage, bool);
  
  itkSetMacro(DefaultValue, OutputPixelType);
  itkGetMacro(DefaultValue, OutputPixelType);
  
  void SetPhiTransformation(VectorImageType* image) { m_PhiTransformation = image; }
  
protected:
  DasTransformImageFilter();
  ~DasTransformImageFilter() {};
  void PrintSelf(std::ostream& os, Indent indent) const;

  // Before we start multi-threaded section.
  virtual void BeforeThreadedGenerateData();

  // The main method to implement in derived classes, note, its threaded.
  virtual void ThreadedGenerateData( const InputImageRegionType &outputRegionForThread, ThreadIdType threadId);

  // After the threaded bit.
  virtual void AfterThreadedGenerateData( void );
  
private:
  
  DasTransformImageFilter(const Self&); 
  void operator=(const Self&); 

  typename VectorImageType::Pointer m_PhiTransformation;

  typename LinearInterpolatorType::Pointer m_Interpolator;
  
  std::string m_FileName;
  
  bool m_WriteTransformedMovingImage;
  
  OutputPixelType m_DefaultValue;
  
}; // end class
    
} // end namespace
    
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkDasTransformImageFilter.txx"
#endif

#endif