ITK
4.1.0
Insight Segmentation and Registration Toolkit
|
00001 /*========================================================================= 00002 * 00003 * Copyright Insight Software Consortium 00004 * 00005 * Licensed under the Apache License, Version 2.0 (the "License"); 00006 * you may not use this file except in compliance with the License. 00007 * You may obtain a copy of the License at 00008 * 00009 * http://www.apache.org/licenses/LICENSE-2.0.txt 00010 * 00011 * Unless required by applicable law or agreed to in writing, software 00012 * distributed under the License is distributed on an "AS IS" BASIS, 00013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 00014 * See the License for the specific language governing permissions and 00015 * limitations under the License. 00016 * 00017 *=========================================================================*/ 00018 #ifndef __itkMeanSquareRegistrationFunction_h 00019 #define __itkMeanSquareRegistrationFunction_h 00020 00021 #include "itkPDEDeformableRegistrationFunction.h" 00022 #include "itkPoint.h" 00023 #include "itkLinearInterpolateImageFunction.h" 00024 #include "itkCentralDifferenceImageFunction.h" 00025 00026 namespace itk 00027 { 00052 template< class TFixedImage, class TMovingImage, class TDisplacementField > 00053 class ITK_EXPORT MeanSquareRegistrationFunction: 00054 public PDEDeformableRegistrationFunction< TFixedImage, 00055 TMovingImage, TDisplacementField > 00056 { 00057 public: 00058 00060 typedef MeanSquareRegistrationFunction Self; 00061 typedef PDEDeformableRegistrationFunction< TFixedImage, 00062 TMovingImage, TDisplacementField > Superclass; 00063 typedef SmartPointer< Self > Pointer; 00064 typedef SmartPointer< const Self > ConstPointer; 00065 00067 itkNewMacro(Self); 00068 00070 itkTypeMacro(MeanSquareRegistrationFunction, 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::DisplacementFieldType DisplacementFieldType; 00086 typedef typename DisplacementFieldType::PixelType DisplacementFieldPixelType; 00087 typedef typename Superclass::DisplacementFieldTypePointer 00088 DisplacementFieldTypePointer; 00089 00091 itkStaticConstMacro(ImageDimension, unsigned int, Superclass::ImageDimension); 00092 00094 typedef typename Superclass::PixelType PixelType; 00095 typedef typename Superclass::RadiusType RadiusType; 00096 typedef typename Superclass::NeighborhoodType NeighborhoodType; 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 > DefaultInterpolatorType; 00106 00108 typedef CovariantVector< double, itkGetStaticConstMacro(ImageDimension) > CovariantVectorType; 00109 00111 typedef CentralDifferenceImageFunction< FixedImageType > GradientCalculatorType; 00112 typedef typename GradientCalculatorType::Pointer GradientCalculatorPointer; 00113 00115 void SetMovingImageInterpolator(InterpolatorType *ptr) 00116 { m_MovingImageInterpolator = ptr; } 00117 00119 InterpolatorType * GetMovingImageInterpolator(void) 00120 { return m_MovingImageInterpolator; } 00121 00123 virtual TimeStepType ComputeGlobalTimeStep( void *itkNotUsed(GlobalData) ) const 00124 { return m_TimeStep; } 00125 00128 virtual void * GetGlobalDataPointer() const 00129 { 00130 GlobalDataStruct *global = new GlobalDataStruct(); 00131 00132 return global; 00133 } 00134 00136 virtual void ReleaseGlobalDataPointer(void *GlobalData) const 00137 { delete (GlobalDataStruct *)GlobalData; } 00138 00140 virtual void InitializeIteration(); 00141 00144 virtual PixelType ComputeUpdate( const NeighborhoodType & neighborhood, 00145 void *globalData, 00146 const FloatOffsetType & offset = FloatOffsetType(0.0) ); 00147 00148 protected: 00149 MeanSquareRegistrationFunction(); 00150 ~MeanSquareRegistrationFunction() {} 00151 void PrintSelf(std::ostream & os, Indent indent) const; 00152 00154 typedef ConstNeighborhoodIterator< FixedImageType > FixedImageNeighborhoodIteratorType; 00155 00158 struct GlobalDataStruct { 00159 FixedImageNeighborhoodIteratorType m_FixedImageIterator; 00160 }; 00161 private: 00162 MeanSquareRegistrationFunction(const Self &); //purposely not implemented 00163 void operator=(const Self &); //purposely not implemented 00165 00167 SpacingType m_FixedImageSpacing; 00168 00170 GradientCalculatorPointer m_FixedImageGradientCalculator; 00171 00173 InterpolatorPointer m_MovingImageInterpolator; 00174 00176 TimeStepType m_TimeStep; 00177 00179 double m_DenominatorThreshold; 00180 00182 double m_IntensityDifferenceThreshold; 00183 }; 00184 } // end namespace itk 00185 00186 #ifndef ITK_MANUAL_INSTANTIATION 00187 #include "itkMeanSquareRegistrationFunction.hxx" 00188 #endif 00189 00190 #endif 00191