itkVelocityFieldMultiResolutionMethod.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 itkVelocityFieldMultiResolutionMethod_h
#define itkVelocityFieldMultiResolutionMethod_h

#include "itkMultiResolutionDeformableImageRegistrationMethod.h"
#include <itkUCLBSplineTransform.h>
#include <itkVelocityFieldDeformableTransform.h>

namespace itk
{

template <typename TInputImageType, class TScalarType, unsigned int NDimensions, class TPyramidFilter >
class VelocityFieldMultiResolutionMethod : 
public MultiResolutionDeformableImageRegistrationMethod<TInputImageType, TScalarType, NDimensions, float, TPyramidFilter> 
{
public:
  typedef VelocityFieldMultiResolutionMethod                                         Self;
  typedef MultiResolutionDeformableImageRegistrationMethod<TInputImageType, TScalarType, NDimensions, float> Superclass;
  typedef SmartPointer<Self>                                                 Pointer;
  typedef SmartPointer<const Self>                                           ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(VelocityFieldMultiResolutionMethod, MultiResolutionDeformableImageRegistrationMethod);

  typedef TInputImageType                                                    InputImageType;
  typedef typename InputImageType::SpacingType                               InputImageSpacingType;

  typedef VelocityFieldDeformableTransform<TInputImageType, TScalarType, NDimensions, float >         VelocityFieldDeformableTransformType;
  typedef typename VelocityFieldDeformableTransformType::Pointer                             VelocityFieldDeformableTransformPointer;

  typedef VelocityFieldGradientDescentOptimizer<TInputImageType, TInputImageType, double, float> OptimizerType;
  typedef OptimizerType*                                                     OptimizerPointer;

  itkSetObjectMacro( Transform, VelocityFieldDeformableTransformType );
  itkGetObjectMacro( Transform, VelocityFieldDeformableTransformType );
  
  itkSetMacro(MaxStepSize, double);
  itkGetMacro(MaxStepSize, double);

  itkSetMacro(MinDeformationSize, double);
  itkGetMacro(MinDeformationSize, double);
  
  virtual void StartRegistration();
  
protected:
  VelocityFieldMultiResolutionMethod();
  virtual ~VelocityFieldMultiResolutionMethod() {};

  virtual void BeforeSingleResolutionRegistration();
  
  double m_MaxStepSize;
  
  double m_MinDeformationSize;
  
  typename VelocityFieldDeformableTransformType::DeformableParameterPointerType m_InitialDeformableTransformParametersOfNextLevel; 

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

  VelocityFieldDeformableTransformPointer m_Transform;

};

} // end namespace itk

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

#endif