00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkMultiResolutionPDEDeformableRegistration_h
00018 #define __itkMultiResolutionPDEDeformableRegistration_h
00019
00020 #include "itkImage.h"
00021 #include "itkImageToImageFilter.h"
00022 #include "itkPDEDeformableRegistrationFilter.h"
00023 #include "itkDemonsRegistrationFilter.h"
00024 #include "itkMultiResolutionPyramidImageFilter.h"
00025 #include "itkVectorExpandImageFilter.h"
00026
00027 #include <vector>
00028
00029 namespace itk
00030 {
00075 template <class TFixedImage, class TMovingImage, class TDeformationField>
00076 class MultiResolutionPDEDeformableRegistration :
00077 public ImageToImageFilter <TDeformationField, TDeformationField>
00078 {
00079 public:
00081 typedef MultiResolutionPDEDeformableRegistration Self;
00082 typedef ImageToImageFilter<TDeformationField, TDeformationField>
00083 Superclass;
00084 typedef SmartPointer<Self> Pointer;
00085 typedef SmartPointer<const Self> ConstPointer;
00086
00088 itkNewMacro(Self);
00089
00091 itkTypeMacro( MultiResolutionPDEDeformableRegistration,
00092 ImageToImageFilter );
00093
00095 typedef TFixedImage FixedImageType;
00096 typedef typename FixedImageType::Pointer FixedImagePointer;
00097 typedef typename FixedImageType::ConstPointer FixedImageConstPointer;
00098
00100 typedef TMovingImage MovingImageType;
00101 typedef typename MovingImageType::Pointer MovingImagePointer;
00102 typedef typename MovingImageType::ConstPointer MovingImageConstPointer;
00103
00105 typedef TDeformationField DeformationFieldType;
00106 typedef typename DeformationFieldType::Pointer DeformationFieldPointer;
00107
00109 itkStaticConstMacro(ImageDimension, unsigned int,
00110 FixedImageType::ImageDimension);
00111
00113 typedef Image<float,itkGetStaticConstMacro(ImageDimension)> FloatImageType;
00114
00116 typedef PDEDeformableRegistrationFilter<
00117 FloatImageType, FloatImageType, DeformationFieldType > RegistrationType;
00118 typedef typename RegistrationType::Pointer RegistrationPointer;
00119
00121 typedef DemonsRegistrationFilter<
00122 FloatImageType, FloatImageType, DeformationFieldType > DefaultRegistrationType;
00123
00125 typedef MultiResolutionPyramidImageFilter<
00126 FixedImageType, FloatImageType > FixedImagePyramidType;
00127 typedef typename FixedImagePyramidType::Pointer FixedImagePyramidPointer;
00128
00130 typedef MultiResolutionPyramidImageFilter<
00131 MovingImageType, FloatImageType > MovingImagePyramidType;
00132 typedef typename MovingImagePyramidType::Pointer MovingImagePyramidPointer;
00133
00135 typedef VectorExpandImageFilter<
00136 DeformationFieldType, DeformationFieldType > FieldExpanderType;
00137 typedef typename FieldExpanderType::Pointer FieldExpanderPointer;
00138
00140 virtual void SetFixedImage( const FixedImageType * ptr );
00141
00143 const FixedImageType * GetFixedImage(void);
00144
00146 virtual void SetMovingImage( const MovingImageType * ptr );
00147
00149 const MovingImageType * GetMovingImage(void);
00150
00152 virtual void SetInitialDeformationField( DeformationFieldType * ptr )
00153 {
00154 itkExceptionMacro( << "This feature not implemented yet" );
00155
00156 }
00157
00159 const DeformationFieldType * GetDeformationField(void)
00160 { return this->GetOutput(); }
00161
00163 itkSetObjectMacro( RegistrationFilter, RegistrationType );
00164
00166 itkGetObjectMacro( RegistrationFilter, RegistrationType );
00167
00169 itkSetObjectMacro( FixedImagePyramid, FixedImagePyramidType );
00170
00172 itkGetObjectMacro( FixedImagePyramid, FixedImagePyramidType );
00173
00175 itkSetObjectMacro( MovingImagePyramid, MovingImagePyramidType );
00176
00178 itkGetObjectMacro( MovingImagePyramid, MovingImagePyramidType );
00179
00181 virtual void SetNumberOfLevels( unsigned int num );
00182
00184 itkGetMacro( NumberOfLevels, unsigned int );
00185
00187 itkGetMacro( CurrentLevel, unsigned int );
00188
00190 itkSetVectorMacro( NumberOfIterations, unsigned int, m_NumberOfLevels );
00191
00193 virtual const unsigned int * GetNumberOfIterations() const
00194 { return &(m_NumberOfIterations[0]); }
00195
00196 protected:
00197 MultiResolutionPDEDeformableRegistration();
00198 ~MultiResolutionPDEDeformableRegistration() {}
00199 void PrintSelf(std::ostream& os, Indent indent) const;
00200
00203 virtual void GenerateData();
00204
00208 virtual void GenerateInputRequestedRegion();
00209
00216 virtual void GenerateOutputInformation();
00217
00221 virtual void EnlargeOutputRequestedRegion( DataObject *ptr );
00222
00223 private:
00224 MultiResolutionPDEDeformableRegistration(const Self&);
00225 void operator=(const Self&);
00226
00227 RegistrationPointer m_RegistrationFilter;
00228 FixedImagePyramidPointer m_FixedImagePyramid;
00229 MovingImagePyramidPointer m_MovingImagePyramid;
00230 FieldExpanderPointer m_FieldExpander;
00231
00232 unsigned int m_NumberOfLevels;
00233 unsigned int m_CurrentLevel;
00234 std::vector<unsigned int> m_NumberOfIterations;
00235
00236 };
00237
00238
00239 }
00240
00241 #ifndef ITK_MANUAL_INSTANTIATION
00242 #include "itkMultiResolutionPDEDeformableRegistration.txx"
00243 #endif
00244
00245
00246 #endif