00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef _itkNCCRegistrationFunction_h_
00018 #define _itkNCCRegistrationFunction_h_
00019
00020 #include "itkPDEDeformableRegistrationFunction.h"
00021 #include "itkPoint.h"
00022 #include "itkCovariantVector.h"
00023 #include "itkInterpolateImageFunction.h"
00024 #include "itkLinearInterpolateImageFunction.h"
00025 #include "itkCentralDifferenceImageFunction.h"
00026
00027 namespace itk {
00028
00052 template<class TFixedImage, class TMovingImage, class TDeformationField>
00053 class ITK_EXPORT NCCRegistrationFunction :
00054 public PDEDeformableRegistrationFunction< TFixedImage,
00055 TMovingImage, TDeformationField>
00056 {
00057 public:
00058
00060 typedef NCCRegistrationFunction Self;
00061 typedef PDEDeformableRegistrationFunction< TFixedImage,
00062 TMovingImage, TDeformationField > Superclass;
00063 typedef SmartPointer<Self> Pointer;
00064 typedef SmartPointer<const Self> ConstPointer;
00065
00067 itkNewMacro(Self);
00068
00070 itkTypeMacro( NCCRegistrationFunction,
00071 PDEDeformableRegistrationFunction );
00072
00074 typedef typename Superclass::MovingImageType MovingImageType;
00075 typedef typename Superclass::MovingImagePointer MovingImagePointer;
00076
00078 typedef typename Superclass::FixedImageType FixedImageType;
00079 typedef typename Superclass::FixedImagePointer FixedImagePointer;
00080 typedef typename FixedImageType::IndexType IndexType;
00081 typedef typename FixedImageType::SizeType SizeType;
00082 typedef typename FixedImageType::SpacingType SpacingType;
00083
00085 typedef typename Superclass::DeformationFieldType DeformationFieldType;
00086 typedef typename Superclass::DeformationFieldTypePointer
00087 DeformationFieldTypePointer;
00088
00090 itkStaticConstMacro(ImageDimension, unsigned int,Superclass::ImageDimension);
00091
00093 typedef typename Superclass::PixelType PixelType;
00094 typedef typename Superclass::RadiusType RadiusType;
00095 typedef typename Superclass::NeighborhoodType NeighborhoodType;
00096
00097 typedef typename Superclass::FloatOffsetType FloatOffsetType;
00098 typedef typename Superclass::TimeStepType TimeStepType;
00099
00101 typedef double CoordRepType;
00102 typedef InterpolateImageFunction<MovingImageType,CoordRepType> InterpolatorType;
00103 typedef typename InterpolatorType::Pointer InterpolatorPointer;
00104 typedef typename InterpolatorType::PointType PointType;
00105 typedef LinearInterpolateImageFunction<MovingImageType,CoordRepType>
00106 DefaultInterpolatorType;
00107
00109 typedef CovariantVector<double,itkGetStaticConstMacro(ImageDimension)> CovariantVectorType;
00110
00112 typedef CentralDifferenceImageFunction<FixedImageType> GradientCalculatorType;
00113 typedef typename GradientCalculatorType::Pointer GradientCalculatorPointer;
00114
00116 void SetMovingImageInterpolator( InterpolatorType * ptr )
00117 { m_MovingImageInterpolator = ptr; }
00118
00120 InterpolatorType * GetMovingImageInterpolator(void)
00121 { return m_MovingImageInterpolator; }
00122
00124 virtual TimeStepType ComputeGlobalTimeStep(void * itkNotUsed(GlobalData)) const
00125 { return m_TimeStep; }
00126
00129 virtual void *GetGlobalDataPointer() const
00130 {
00131 GlobalDataStruct *global = new GlobalDataStruct();
00132 return global;
00133 }
00135
00137 virtual void ReleaseGlobalDataPointer( void *GlobalData ) const
00138 { delete (GlobalDataStruct *) GlobalData; }
00139
00141 virtual void InitializeIteration();
00142
00145 virtual PixelType ComputeUpdate(const NeighborhoodType &neighborhood,
00146 void *globalData,
00147 const FloatOffsetType &offset = FloatOffsetType(0.0));
00148
00149 protected:
00150 NCCRegistrationFunction();
00151 ~NCCRegistrationFunction() {}
00152 void PrintSelf(std::ostream& os, Indent indent) const;
00153
00155 typedef ConstNeighborhoodIterator<FixedImageType> FixedImageNeighborhoodIteratorType;
00156
00159 struct GlobalDataStruct
00160 {
00161 FixedImageNeighborhoodIteratorType m_FixedImageIterator;
00162 };
00163
00164
00165 private:
00166 NCCRegistrationFunction(const Self&);
00167 void operator=(const Self&);
00168
00170 SpacingType m_FixedImageSpacing;
00171 PointType m_FixedImageOrigin;
00172
00174 GradientCalculatorPointer m_FixedImageGradientCalculator;
00175
00176
00178 InterpolatorPointer m_MovingImageInterpolator;
00179
00181 TimeStepType m_TimeStep;
00182
00184 double m_DenominatorThreshold;
00185
00187 double m_IntensityDifferenceThreshold;
00188
00189 mutable double m_MetricTotal;
00190
00191 };
00192
00193
00194 }
00195
00196 #ifndef ITK_MANUAL_INSTANTIATION
00197 #include "itkNCCRegistrationFunction.txx"
00198 #endif
00199
00200 #endif
00201