ITK  4.3.0
Insight Segmentation and Registration Toolkit
itkImageToImageMetric.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright Insight Software Consortium
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 #ifndef __itkImageToImageMetric_h
19 #define __itkImageToImageMetric_h
20 
21 #include "itkBSplineBaseTransform.h"
22 #include "itkBSplineInterpolateImageFunction.h"
23 #include "itkSingleValuedCostFunction.h"
24 #include "itkGradientRecursiveGaussianImageFilter.h"
25 #include "itkSpatialObject.h"
26 #include "itkCentralDifferenceImageFunction.h"
27 
28 namespace itk
29 {
50 template< class TFixedImage, class TMovingImage >
51 class ITK_EXPORT ImageToImageMetric:
52  public SingleValuedCostFunction
53 {
54 public:
56  typedef ImageToImageMetric Self;
57  typedef SingleValuedCostFunction Superclass;
58  typedef SmartPointer< Self > Pointer;
59  typedef SmartPointer< const Self > ConstPointer;
60 
62  typedef typename Superclass::ParametersValueType CoordinateRepresentationType;
63 
65  itkTypeMacro(ImageToImageMetric, SingleValuedCostFunction);
66 
68  typedef TMovingImage MovingImageType;
69  typedef typename TMovingImage::PixelType MovingImagePixelType;
70  typedef typename MovingImageType::ConstPointer MovingImageConstPointer;
71 
73  typedef TFixedImage FixedImageType;
74  typedef typename TFixedImage::PixelType FixedImagePixelType;
75  typedef typename FixedImageType::ConstPointer FixedImageConstPointer;
76  typedef typename FixedImageType::RegionType FixedImageRegionType;
77 
79  itkStaticConstMacro(MovingImageDimension,
80  unsigned int,
81  TMovingImage::ImageDimension);
82  itkStaticConstMacro(FixedImageDimension,
83  unsigned int,
84  TFixedImage::ImageDimension);
86 
88  typedef Transform< CoordinateRepresentationType,
89  itkGetStaticConstMacro(MovingImageDimension),
90  itkGetStaticConstMacro(FixedImageDimension) >
91  TransformType;
92 
93  typedef typename TransformType::Pointer TransformPointer;
94  typedef typename TransformType::InputPointType InputPointType;
95  typedef typename TransformType::OutputPointType OutputPointType;
96  typedef typename TransformType::ParametersType TransformParametersType;
97  typedef typename TransformType::JacobianType TransformJacobianType;
98 
100  typedef typename FixedImageType::IndexType FixedImageIndexType;
101  typedef typename FixedImageIndexType::IndexValueType FixedImageIndexValueType;
102  typedef typename MovingImageType::IndexType MovingImageIndexType;
103  typedef typename TransformType::InputPointType FixedImagePointType;
104  typedef typename TransformType::OutputPointType MovingImagePointType;
105 
106  typedef std::vector< FixedImageIndexType > FixedImageIndexContainer;
107 
109  typedef InterpolateImageFunction< MovingImageType, CoordinateRepresentationType > InterpolatorType;
110 
112  typedef typename NumericTraits< MovingImagePixelType >::RealType RealType;
113  typedef CovariantVector< RealType, itkGetStaticConstMacro(MovingImageDimension) > GradientPixelType;
114  typedef Image< GradientPixelType, itkGetStaticConstMacro(MovingImageDimension) > GradientImageType;
115  typedef SmartPointer< GradientImageType > GradientImagePointer;
116  typedef GradientRecursiveGaussianImageFilter< MovingImageType, GradientImageType > GradientImageFilterType;
117  typedef typename GradientImageFilterType::Pointer GradientImageFilterPointer;
119 
120  typedef typename InterpolatorType::Pointer InterpolatorPointer;
121 
124  typedef SpatialObject< itkGetStaticConstMacro(FixedImageDimension) > FixedImageMaskType;
125  typedef typename FixedImageMaskType::Pointer FixedImageMaskPointer;
126  typedef typename FixedImageMaskType::ConstPointer FixedImageMaskConstPointer;
127 
130  typedef SpatialObject< itkGetStaticConstMacro(MovingImageDimension) > MovingImageMaskType;
131  typedef typename MovingImageMaskType::Pointer MovingImageMaskPointer;
132  typedef typename MovingImageMaskType::ConstPointer MovingImageMaskConstPointer;
133 
135  typedef typename Superclass::MeasureType MeasureType;
136 
138  typedef typename Superclass::DerivativeType DerivativeType;
139 
141  typedef typename Superclass::ParametersType ParametersType;
142 
144  itkSetConstObjectMacro( FixedImage, FixedImageType );
145 
147  itkGetConstObjectMacro( FixedImage, FixedImageType );
148 
150  itkSetConstObjectMacro( MovingImage, MovingImageType );
151 
153  itkGetConstObjectMacro( MovingImage, MovingImageType );
154 
156  itkSetObjectMacro( Transform, TransformType );
157 
159  itkGetConstObjectMacro(Transform, TransformType);
160 
162  itkSetObjectMacro(Interpolator, InterpolatorType);
163 
165  itkGetConstObjectMacro(Interpolator, InterpolatorType);
166 
168  SizeValueType GetNumberOfMovingImageSamples(void)
169  {
170  return this->GetNumberOfPixelsCounted();
171  }
172 
173  itkGetConstReferenceMacro(NumberOfPixelsCounted, SizeValueType);
174 
176  void SetFixedImageRegion(const FixedImageRegionType reg);
177 
179  itkGetConstReferenceMacro(FixedImageRegion, FixedImageRegionType);
180 
182  itkSetObjectMacro(MovingImageMask, MovingImageMaskType);
183  itkSetConstObjectMacro(MovingImageMask, MovingImageMaskType);
184  itkGetConstObjectMacro(MovingImageMask, MovingImageMaskType);
186 
188  itkSetObjectMacro(FixedImageMask, FixedImageMaskType);
189  itkSetConstObjectMacro(FixedImageMask, FixedImageMaskType);
190  itkGetConstObjectMacro(FixedImageMask, FixedImageMaskType);
192 
195  void SetFixedImageIndexes(const FixedImageIndexContainer & indexes);
196 
197  void SetUseFixedImageIndexes(bool useIndex);
198 
199  itkGetConstReferenceMacro(UseFixedImageIndexes, bool);
200 
202  void SetNumberOfThreads(ThreadIdType numberOfThreads);
203  itkGetConstReferenceMacro(NumberOfThreads, ThreadIdType);
205 
207  itkSetMacro(ComputeGradient, bool);
208  itkGetConstReferenceMacro(ComputeGradient, bool);
209  itkBooleanMacro(ComputeGradient);
211 
213  virtual void ComputeGradient(void);
214 
216  itkGetConstObjectMacro(GradientImage, GradientImageType);
217 
219  void SetTransformParameters(const ParametersType & parameters) const;
220 
222  unsigned int GetNumberOfParameters(void) const
223  {
224  return m_Transform->GetNumberOfParameters();
225  }
226 
229  virtual void Initialize(void)
230  throw ( ExceptionObject );
231 
233  virtual void MultiThreadingInitialize(void) throw ( ExceptionObject );
234 
237  virtual void SetNumberOfFixedImageSamples(SizeValueType numSamples);
238  itkGetConstReferenceMacro(NumberOfFixedImageSamples, SizeValueType);
240 
243  void SetNumberOfSpatialSamples(SizeValueType num)
244  {
245  this->SetNumberOfFixedImageSamples(num);
246  }
247 
248  SizeValueType GetNumberOfSpatialSamples(void)
249  {
250  return this->GetNumberOfFixedImageSamples();
251  }
252 
255  void SetFixedImageSamplesIntensityThreshold(const FixedImagePixelType & thresh);
256 
257  itkGetConstReferenceMacro(FixedImageSamplesIntensityThreshold, FixedImagePixelType);
258 
259  void SetUseFixedImageSamplesIntensityThreshold(bool useThresh);
260 
261  itkGetConstReferenceMacro(UseFixedImageSamplesIntensityThreshold, bool);
262 
266  void SetUseAllPixels(bool useAllPixels);
267 
268  void UseAllPixelsOn(void)
269  {
270  this->SetUseAllPixels(true);
271  }
272 
273  void UseAllPixelsOff(void)
274  {
275  this->SetUseAllPixels(false);
276  }
277 
278  itkGetConstReferenceMacro(UseAllPixels, bool);
279 
284  void SetUseSequentialSampling(bool sequentialSampling);
285 
286  itkGetConstReferenceMacro(UseSequentialSampling, bool);
287 
297  void ReinitializeSeed();
298  void ReinitializeSeed(int seed);
300 
317  itkSetMacro(UseCachingOfBSplineWeights, bool);
318  itkGetConstReferenceMacro(UseCachingOfBSplineWeights, bool);
319  itkBooleanMacro(UseCachingOfBSplineWeights);
321 
322  typedef MultiThreader MultiThreaderType;
324  itkGetConstObjectMacro(Threader, MultiThreaderType);
325  const TransformPointer * GetThreaderTransform()
326  {
327  return m_ThreaderTransform;
328  }
330 
331 protected:
332  ImageToImageMetric();
333  virtual ~ImageToImageMetric();
334 
335  void PrintSelf(std::ostream & os, Indent indent) const;
336 
342  class FixedImageSamplePoint
343  {
344 public:
345  FixedImageSamplePoint()
346  {
347  point.Fill(0.0);
348  value = 0;
349  valueIndex = 0;
350  }
351 
352  ~FixedImageSamplePoint() {}
353 
354 public:
355  FixedImagePointType point;
356  double value;
357  unsigned int valueIndex;
358  };
359 
360  bool m_UseFixedImageIndexes;
361  FixedImageIndexContainer m_FixedImageIndexes;
362 
363  bool m_UseFixedImageSamplesIntensityThreshold;
364  FixedImagePixelType m_FixedImageSamplesIntensityThreshold;
365 
367  typedef std::vector< FixedImageSamplePoint > FixedImageSampleContainer;
368 
370  virtual void SampleFixedImageRegion(FixedImageSampleContainer & samples) const;
371 
372  virtual void SampleFixedImageIndexes(FixedImageSampleContainer & samples) const;
373 
375  virtual void SampleFullFixedImageRegion(FixedImageSampleContainer &
376  samples) const;
377 
379  FixedImageSampleContainer m_FixedImageSamples;
380 
381  SizeValueType m_NumberOfParameters;
382 
383  SizeValueType m_NumberOfFixedImageSamples;
384  //m_NumberOfPixelsCounted must be mutable because the const
385  //thread consolidation functions merge each threads valus
386  //onto this accumulator variable.
387  mutable SizeValueType m_NumberOfPixelsCounted;
388 
389  FixedImageConstPointer m_FixedImage;
390  MovingImageConstPointer m_MovingImage;
391 
393  TransformPointer m_Transform;
394 
397  TransformPointer *m_ThreaderTransform;
398 
399  InterpolatorPointer m_Interpolator;
400 
401  bool m_ComputeGradient;
402  GradientImagePointer m_GradientImage;
403 
404  FixedImageMaskConstPointer m_FixedImageMask;
405  MovingImageMaskConstPointer m_MovingImageMask;
406 
407  ThreadIdType m_NumberOfThreads;
408 
409  bool m_UseAllPixels;
410  bool m_UseSequentialSampling;
411 
412  bool m_ReseedIterator;
413 
414  int m_RandomSeed;
415 
423  bool m_TransformIsBSpline;
424 
427  SizeValueType m_NumBSplineWeights;
428 
429  itkStaticConstMacro(DeformationSplineOrder, unsigned int, 3);
430 
431  typedef BSplineBaseTransform< CoordinateRepresentationType,
432  FixedImageType ::ImageDimension,
433  itkGetStaticConstMacro(DeformationSplineOrder) > BSplineTransformType;
434 
435  typedef typename BSplineTransformType::WeightsType BSplineTransformWeightsType;
436  typedef typename BSplineTransformWeightsType::ValueType WeightsValueType;
437  typedef Array2D< WeightsValueType > BSplineTransformWeightsArrayType;
438 
439  typedef typename BSplineTransformType::ParameterIndexArrayType BSplineTransformIndexArrayType;
440  typedef typename BSplineTransformIndexArrayType::ValueType IndexValueType;
441  typedef Array2D< IndexValueType > BSplineTransformIndicesArrayType;
442 
443  typedef std::vector< MovingImagePointType > MovingImagePointArrayType;
444  typedef std::vector< bool > BooleanArrayType;
445  typedef FixedArray< SizeValueType, FixedImageType ::ImageDimension > BSplineParametersOffsetType;
451  typedef BSplineInterpolateImageFunction< MovingImageType,
452  CoordinateRepresentationType >
453  BSplineInterpolatorType;
454 
456  typedef CentralDifferenceImageFunction< MovingImageType,
457  CoordinateRepresentationType >
458  DerivativeFunctionType;
459  typedef CovariantVector< double, itkGetStaticConstMacro(MovingImageDimension) >
460  ImageDerivativesType;
461 
462  typename BSplineTransformType::Pointer m_BSplineTransform;
463 
464  BSplineTransformWeightsArrayType m_BSplineTransformWeightsArray;
465  BSplineTransformIndicesArrayType m_BSplineTransformIndicesArray;
466  MovingImagePointArrayType m_BSplinePreTransformPointsArray;
467  BooleanArrayType m_WithinBSplineSupportRegionArray;
468 
469  BSplineParametersOffsetType m_BSplineParametersOffset;
470 
471  // Variables needed for optionally caching values when using a BSpline
472  // transform.
473  bool m_UseCachingOfBSplineWeights;
474  mutable BSplineTransformWeightsType m_BSplineTransformWeights;
475  mutable BSplineTransformIndexArrayType m_BSplineTransformIndices;
476 
477  mutable BSplineTransformWeightsType *m_ThreaderBSplineTransformWeights;
478  mutable BSplineTransformIndexArrayType *m_ThreaderBSplineTransformIndices;
479 
480  virtual void PreComputeTransformValues(void);
481 
484  virtual void TransformPoint(unsigned int sampleNumber,
485  MovingImagePointType & mappedPoint,
486  bool & sampleWithinSupportRegion,
487  double & movingImageValue,
488  ThreadIdType threadID) const;
489 
490  virtual void TransformPointWithDerivatives(unsigned int sampleNumber,
491  MovingImagePointType & mappedPoint,
492  bool & sampleWithinSupportRegion,
493  double & movingImageValue,
494  ImageDerivativesType & gradient,
495  ThreadIdType threadID) const;
496 
498  bool m_InterpolatorIsBSpline;
499 
501  typename BSplineInterpolatorType::Pointer m_BSplineInterpolator;
502 
504  typename DerivativeFunctionType::Pointer m_DerivativeCalculator;
505 
507  virtual void ComputeImageDerivatives(const MovingImagePointType & mappedPoint,
508  ImageDerivativesType & gradient,
509  ThreadIdType threadID) const;
510 
515  struct MultiThreaderParameterType {
516  ImageToImageMetric *metric;
517  };
518 
519  MultiThreaderType::Pointer m_Threader;
520  MultiThreaderParameterType m_ThreaderParameter;
521  mutable unsigned int * m_ThreaderNumberOfMovingImageSamples;
522  bool m_WithinThreadPreProcess;
523  bool m_WithinThreadPostProcess;
524 
525  void GetValueMultiThreadedPreProcessInitiate(void) const;
526 
527  void GetValueMultiThreadedInitiate(void) const;
528 
529  void GetValueMultiThreadedPostProcessInitiate(void) const;
530 
531  static ITK_THREAD_RETURN_TYPE GetValueMultiThreadedPreProcess(void *arg);
532 
533  static ITK_THREAD_RETURN_TYPE GetValueMultiThreaded(void *arg);
534 
535  static ITK_THREAD_RETURN_TYPE GetValueMultiThreadedPostProcess(void *arg);
536 
537  virtual inline void GetValueThread(ThreadIdType threadID) const;
538 
539  virtual inline void GetValueThreadPreProcess(
540  ThreadIdType itkNotUsed(threadID),
541  bool itkNotUsed(withinSampleThread) ) const
542  {}
543  virtual inline bool GetValueThreadProcessSample(
544  ThreadIdType itkNotUsed(threadID),
545  SizeValueType itkNotUsed(fixedImageSample),
546  const MovingImagePointType & itkNotUsed(mappedPoint),
547  double itkNotUsed(movingImageValue) ) const
548  { return false; }
549  virtual inline void GetValueThreadPostProcess(
550  ThreadIdType itkNotUsed(threadID),
551  bool itkNotUsed(withinSampleThread) ) const
552  {}
553 
554  void GetValueAndDerivativeMultiThreadedPreProcessInitiate(void) const;
555 
556  void GetValueAndDerivativeMultiThreadedInitiate(void) const;
557 
558  void GetValueAndDerivativeMultiThreadedPostProcessInitiate(void) const;
559 
560  static ITK_THREAD_RETURN_TYPE GetValueAndDerivativeMultiThreadedPreProcess(void *arg);
561 
562  static ITK_THREAD_RETURN_TYPE GetValueAndDerivativeMultiThreaded(void *arg);
563 
564  static ITK_THREAD_RETURN_TYPE GetValueAndDerivativeMultiThreadedPostProcess(void *arg);
565 
566  virtual inline void GetValueAndDerivativeThread(ThreadIdType threadID) const;
567 
568  virtual inline void GetValueAndDerivativeThreadPreProcess(
569  ThreadIdType itkNotUsed(threadID),
570  bool itkNotUsed(withinSampleThread) ) const
571  {}
572  virtual inline bool GetValueAndDerivativeThreadProcessSample(
573  ThreadIdType itkNotUsed(threadID),
574  SizeValueType itkNotUsed(fixedImageSample),
575  const MovingImagePointType & itkNotUsed(mappedPoint),
576  double itkNotUsed(movingImageValue),
577  const ImageDerivativesType & itkNotUsed(movingImageGradientValue) ) const
578  { return false; }
579  virtual inline void GetValueAndDerivativeThreadPostProcess(
580  ThreadIdType itkNotUsed(threadID),
581  bool itkNotUsed(withinSampleThread) ) const
582  {}
583 
587  virtual void SynchronizeTransforms() const;
588 
589 private:
590  ImageToImageMetric(const Self &); //purposely not implemented
591  void operator=(const Self &); //purposely not implemented
592 
593  FixedImageRegionType m_FixedImageRegion;
594 };
595 } // end namespace itk
596 
597 #ifndef ITK_MANUAL_INSTANTIATION
598 #include "itkImageToImageMetric.hxx"
599 #endif
600 
601 #endif
602 
Tarballs of the nightly generated Doxygen documentation are available for the html, xml, and tag file.
Generated on Sun Dec 9 2012 01:09:36 for ITK by   doxygen 1.8.2