itkDiffusionTensor3DNonRigidTransform.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 itkDiffusionTensor3DNonRigidTransform_h
#define itkDiffusionTensor3DNonRigidTransform_h

#include "itkDiffusionTensor3DTransform.h"
#include "itkDiffusionTensor3DFSAffineTransform.h"
#include "itkDiffusionTensor3DPPDAffineTransform.h"
#include <itkTransform.h>

namespace itk
{

template <class TData>
class DiffusionTensor3DNonRigidTransform :
  public DiffusionTensor3DTransform<TData>
{
public:
  typedef TData                                              DataType;
  typedef DiffusionTensor3DNonRigidTransform                 Self;
  typedef DiffusionTensor3DTransform<DataType>               Superclass;
  typedef typename Superclass::TensorDataType                TensorDataType;
  typedef typename Superclass::MatrixTransformType           MatrixTransformType;
  typedef typename Superclass::PointType                     PointType;
  typedef SmartPointer<Self>                                 Pointer;
  typedef SmartPointer<const Self>                           ConstPointer;
  typedef Transform<double, 3, 3>                            TransformType;
  typedef itk::DiffusionTensor3DPPDAffineTransform<DataType> PPDAffineTransformType;
  typedef itk::DiffusionTensor3DFSAffineTransform<DataType>  FSAffineTransformType;
  typedef itk::DiffusionTensor3DAffineTransform<DataType>    AffineTransform;

  itkTypeMacro(DiffusionTensor3DNonRigidTransform, DiffusionTensor3DTransform);

  // SmartPointer
  itkNewMacro( Self );
  // /Set the transform
  itkSetObjectMacro( Transform, TransformType );
  TransformType::Pointer GetTransform() ITK_OVERRIDE;

  // /Evaluate the position of the transformed tensor in the output image
  PointType EvaluateTensorPosition( const PointType & point ) ITK_OVERRIDE;

  // /Evaluate the transformed tensor
  virtual TensorDataType EvaluateTransformedTensor( TensorDataType & tensor, PointType & outputPosition ) ITK_OVERRIDE;

  void SetAffineTransformType(typename AffineTransform::Pointer transform);
protected:
  DiffusionTensor3DNonRigidTransform();
  unsigned long                     m_LatestTime;
  typename TransformType::Pointer   m_Transform;
  typename AffineTransform::Pointer m_Affine;
};

} // end of itk namespace

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

#endif