#ifndef __itkMultiResolutionLogDomainDeformableRegistrationTensor_h #define __itkMultiResolutionLogDomainDeformableRegistrationTensor_h #include "itkExponentialDisplacementFieldImageFilter2.h" #include "itkImage.h" #include "itkImageToImageFilter.h" #include "itkLogDomainDeformableRegistrationFilter.h" #include "itkLogDomainDemonsRegistrationTensorFilter.h" #include "itkMultiResolutionPyramidTensorImageFilter.h" #include "itkVectorResampleImageFilter.h" #include "itkTensor.h" #include namespace itk { /** * \class MultiResolutionLogDomainDeformableRegistrationTensor * \brief Framework for performing multi-resolution log-domain * deformable registration. * * MultiResolutionLogDomainDeformableRegistrationTensor provides a generic * framework to peform multi-resolution deformable registration of tensor * images. It is heavily based on * MultiResolutionLogDomainDeformableRegistration, but modifies this to handle * tensor images and their registration. In fact, * MultiResolutionLogDomainDeformableRegistrationTensor could be simply * modified so that this class is not necessary. * * At each resolution level a LogDomainDeformableRegistrationFilter is used * to register two images by computing the velocity field whose exponential * will map a moving tensor image onto a fixed tensor image. * * See A Sweet and X Pennec. "Log-Domain Diffeomorphic Registration of * Diffusion Tensor Images". Workshop on Biomedical Image Registration 2010. * * Velocity and deformation fields are represented as images whose pixel type * is some vector type with at least N elements, where N is the dimension of * the fixed image. The vector type must support element access via operator * []. It is assumed that the vector elements behave like floating point * scalars. * * The internal LogDomainDeformableRegistrationFilter can be set using * SetRegistrationFilter. By default a LogDomainDemonsRegistrationTensorFilter * is used. * * The input fixed and moving images are set via methods SetFixedImage * and SetMovingImage respectively. An initial velocity field maybe set via * SetInitialVelocityField if is matches the characteristics of the coarsest * pyramid level. If no such assumption can be made (e.g. the velocity field * has the same characteristics as the input images), an initial velocity * field can still be set via SetArbitraryInitialVelocityField or * SetInput. The filter will then take care of mathching the coarsest level * characteristics. If no initial field is set a zero field is used as the * initial condition. * * MultiResolutionPyramidImageFilters are used to downsample the fixed * and moving images. A VectorExpandImageFilter is used to upsample * the velocity field as we move from a coarse to fine solution. * * This class is templated over the fixed image type, the moving image type, * and the velocity/deformation Field type. * * \warning This class assumes that the fixed, moving and * field image types all have the same number of dimensions. * * \author Florence Dru, INRIA and Tom Vercauteren, MKT * * \sa LogDomainDeformableRegistrationFilter * \sa LogDomainDemonsRegistrationTensorFilter * \sa MultiResolutionPyramidTensorImageFilter * \sa VectorExpandImageFilter * * The current implementation of this class does not support streaming. * * \ingroup DeformableImageRegistration * * \author Andrew Sweet */ template class ITK_EXPORT MultiResolutionLogDomainDeformableRegistrationTensor : public ImageToImageFilter { public: /** Standard class typedefs */ typedef MultiResolutionLogDomainDeformableRegistrationTensor Self; typedef ImageToImageFilter Superclass; typedef SmartPointer Pointer; typedef SmartPointer ConstPointer; /** Method for creation through the object factory. */ itkNewMacro(Self); /** Run-time type information (and related methods). */ itkTypeMacro( MultiResolutionLogDomainDeformableRegistrationTensor, ImageToImageFilter ); /** Fixed image type. */ typedef TFixedImage FixedImageType; typedef typename FixedImageType::Pointer FixedImagePointer; typedef typename FixedImageType::ConstPointer FixedImageConstPointer; /** Moving image type. */ typedef TMovingImage MovingImageType; typedef typename MovingImageType::Pointer MovingImagePointer; typedef typename MovingImageType::ConstPointer MovingImageConstPointer; /** Velocity field type. */ typedef TField VelocityFieldType; typedef typename VelocityFieldType::Pointer VelocityFieldPointer; /** Displacement field image type. */ typedef TField DisplacementFieldType; typedef typename DisplacementFieldType::Pointer DisplacementFieldPointer; /** ImageDimension. */ itkStaticConstMacro(ImageDimension, unsigned int, FixedImageType::ImageDimension); /** Internal float image type. */ typedef typename TFixedImage::PixelType TensorType; typedef Image FloatImageType; /** The internal registration type. */ typedef LogDomainDeformableRegistrationFilter RegistrationType; typedef typename RegistrationType::Pointer RegistrationPointer; /** The default registration type. */ typedef LogDomainDemonsRegistrationTensorFilter DefaultRegistrationType; typedef typename DefaultRegistrationType::Pointer DefaultRegistrationPointer; /** The fixed multi-resolution image pyramid type. */ typedef MultiResolutionPyramidTensorImageFilter FixedImagePyramidType; typedef typename FixedImagePyramidType::Pointer FixedImagePyramidPointer; /** The moving multi-resolution image pyramid type. */ typedef MultiResolutionPyramidTensorImageFilter MovingImagePyramidType; typedef typename MovingImagePyramidType::Pointer MovingImagePyramidPointer; /** The schedule type for pyramid construction. */ typedef typename FixedImagePyramidType::ScheduleType ScheduleType; /** The velocity field expander type. */ typedef VectorResampleImageFilter FieldExpanderType; typedef typename FieldExpanderType::Pointer FieldExpanderPointer; /** Set the fixed image. */ virtual void SetFixedImage( const FixedImageType * ptr ); /** Get the fixed image. */ const FixedImageType * GetFixedImage(void) const; /** Set the moving image. */ virtual void SetMovingImage( const MovingImageType * ptr ); /** Get the moving image. */ const MovingImageType * GetMovingImage(void) const; /** Set initial velocity field to be used as is (no smoothing, no * subsampling at the coarsest level of the pyramid. */ virtual void SetInitialVelocityField( VelocityFieldType * ptr ) { this->m_InitialVelocityField=ptr; } /** Set initial velocity field. No assumption is made on the * input. It will therefore be smoothed and resampled to match the * images characteristics at the coarsest level of the pyramid. */ virtual void SetArbitraryInitialVelocityField( VelocityFieldType * ptr ) { this->SetInput( ptr ); } /** Get output velocity field. */ VelocityFieldType * GetVelocityField() { return this->GetOutput(); } /** Get output deformation field. */ DisplacementFieldPointer GetDisplacementField(); /** Get output inverse deformation field. */ DisplacementFieldPointer GetInverseDisplacementField(); /** Get the number of valid inputs. For * MultiResolutionLogDomainDeformableRegistrationTensor, this checks whether the * fixed and moving images have been set. While * MultiResolutionLogDomainDeformableRegistrationTensor can take a third input * as an initial velocity field, this input is not a required input. */ virtual std::vector >::size_type GetNumberOfValidRequiredInputs() const; /** Set the internal registrator. */ itkSetObjectMacro( RegistrationFilter, RegistrationType ); /** Get the internal registrator. */ itkGetObjectMacro( RegistrationFilter, RegistrationType ); /** Set the fixed image pyramid. */ itkSetObjectMacro( FixedImagePyramid, FixedImagePyramidType ); /** Get the fixed image pyramid. */ itkGetObjectMacro( FixedImagePyramid, FixedImagePyramidType ); /** Set the moving image pyramid. */ itkSetObjectMacro( MovingImagePyramid, MovingImagePyramidType ); /** Get the moving image pyramid. */ itkGetObjectMacro( MovingImagePyramid, MovingImagePyramidType ); /** Set number of multi-resolution levels. */ virtual void SetNumberOfLevels( unsigned int num ); /** Get number of multi-resolution levels. */ itkGetConstReferenceMacro( NumberOfLevels, unsigned int ); /** Get the current resolution level being processed. */ itkGetConstReferenceMacro( CurrentLevel, unsigned int ); /** Set number of iterations per multi-resolution levels. */ itkSetVectorMacro( NumberOfIterations, unsigned int, m_NumberOfLevels ); /** Set the moving image pyramid. */ itkSetObjectMacro( FieldExpander, FieldExpanderType ); /** Get the moving image pyramid. */ itkGetObjectMacro( FieldExpander, FieldExpanderType ); /** Get number of iterations per multi-resolution levels. */ virtual const unsigned int * GetNumberOfIterations() const { return &(m_NumberOfIterations[0]); } /** Stop the registration after the current iteration. */ virtual void StopRegistration(); /** Set a multi-resolution schedule. The input schedule must have only * ImageDimension number of columns and NumberOfLevels number of rows. For * each dimension, the shrink factor must be non-increasing with respect to * subsequent levels. This function will clamp shrink factors to satisify * this condition. All shrink factors less than one will also be clamped * to the value of 1. */ void SetSchedule( const ScheduleType& schedule ); protected: MultiResolutionLogDomainDeformableRegistrationTensor(); ~MultiResolutionLogDomainDeformableRegistrationTensor() {} void PrintSelf(std::ostream& os, Indent indent) const; /** Exponential type */ typedef ExponentialDisplacementFieldImageFilter FieldExponentiatorType; typedef typename FieldExponentiatorType::Pointer FieldExponentiatorPointer; itkSetObjectMacro( Exponentiator, FieldExponentiatorType ); itkGetObjectMacro( Exponentiator, FieldExponentiatorType ); /** Generate output data by performing the registration * at each resolution level. */ virtual void GenerateData(); /** The current implementation of this class does not support * streaming. As such it requires the largest possible region * for the moving, fixed and input velocity field. */ virtual void GenerateInputRequestedRegion(); /** By default, the output velocity field has the same * spacing, origin and LargestPossibleRegion as the input/initial * velocity field. If the initial velocity field is not set, the output * information is copied from the fixed image. */ virtual void GenerateOutputInformation(); /** The current implementation of this class does not supprot * streaming. As such it produces the output for the largest * possible region. */ virtual void EnlargeOutputRequestedRegion( DataObject *ptr ); /** This method returns true to indicate that the registration should * terminate at the current resolution level. */ virtual bool Halt(); private: MultiResolutionLogDomainDeformableRegistrationTensor(const Self&); //purposely not implemented void operator=(const Self&); //purposely not implemented RegistrationPointer m_RegistrationFilter; FixedImagePyramidPointer m_FixedImagePyramid; MovingImagePyramidPointer m_MovingImagePyramid; FieldExpanderPointer m_FieldExpander; VelocityFieldPointer m_InitialVelocityField; unsigned int m_NumberOfLevels; unsigned int m_CurrentLevel; std::vector m_NumberOfIterations; /** Flag to indicate user stop registration request. */ bool m_StopRegistrationFlag; FieldExponentiatorPointer m_Exponentiator; }; } // end namespace itk #ifndef ITK_MANUAL_INSTANTIATION #include "itkMultiResolutionLogDomainDeformableRegistrationTensor.txx" #endif #endif