itkFluidMultiResolutionMethod.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 itkFluidMultiResolutionMethod_h
#define itkFluidMultiResolutionMethod_h

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

namespace itk
{

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

  itkNewMacro(Self);

  itkTypeMacro(FluidMultiResolutionMethod, MultiResolutionDeformableImageRegistrationMethod);

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

  typedef FluidDeformableTransform<TInputImageType, TScalarType, NDimensions, float >         FluidDeformableTransformType;
  typedef typename FluidDeformableTransformType::Pointer                             FluidDeformableTransformPointer;

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

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

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

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

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

  FluidDeformableTransformPointer m_Transform;

};

} // end namespace itk

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

#endif