itkMultiInputImageToImageMetricBase.h

Go to the documentation of this file.

/*======================================================================

  This file is part of the elastix software.

  Copyright (c) University Medical Center Utrecht. All rights reserved.
  See src/CopyrightElastix.txt or http://elastix.isi.uu.nl/legal.php for
  details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE. See the above copyright notices for more information.

======================================================================*/
#ifndef __itkMultiInputImageToImageMetricBase_h
#define __itkMultiInputImageToImageMetricBase_h

#include "itkAdvancedImageToImageMetric.h"
#include <vector>

#define itkSetNumberOfMacro( name ) \
  virtual void SetNumberOf##name##s( const unsigned int _arg ) \
  { \
    if( this->m_NumberOf##name##s != _arg ) \
    { \
      this->m_##name##Vector.resize( _arg ); \
      this->m_NumberOf##name##s = _arg; \
      this->Modified(); \
    } \
  } // comments for allowing ; after calling the macro

namespace itk
{

template< class TFixedImage, class TMovingImage >
class MultiInputImageToImageMetricBase :
  public AdvancedImageToImageMetric< TFixedImage, TMovingImage >
{
public:

  typedef MultiInputImageToImageMetricBase                        Self;
  typedef AdvancedImageToImageMetric< TFixedImage, TMovingImage > Superclass;
  typedef SmartPointer< Self >                                    Pointer;
  typedef SmartPointer< const Self >                              ConstPointer;

  itkTypeMacro( MultiInputImageToImageMetricBase, AdvancedImageToImageMetric );

  itkStaticConstMacro( MovingImageDimension, unsigned int, TMovingImage::ImageDimension );
  itkStaticConstMacro( FixedImageDimension, unsigned int, TFixedImage::ImageDimension );

  typedef typename Superclass::CoordinateRepresentationType CoordinateRepresentationType;
  typedef typename Superclass::MovingImageType              MovingImageType;
  typedef typename Superclass::MovingImagePixelType         MovingImagePixelType;
  typedef typename Superclass::MovingImagePointer           MovingImagePointer;
  typedef typename Superclass::MovingImageConstPointer      MovingImageConstPointer;
  typedef typename Superclass::FixedImageType               FixedImageType;
  typedef typename Superclass::FixedImagePointer            FixedImagePointer;
  typedef typename Superclass::FixedImageConstPointer       FixedImageConstPointer;
  typedef typename Superclass::FixedImageRegionType         FixedImageRegionType;
  typedef typename Superclass::TransformType                TransformType;
  typedef typename Superclass::TransformPointer             TransformPointer;
  typedef typename Superclass::InputPointType               InputPointType;
  typedef typename Superclass::OutputPointType              OutputPointType;
  typedef typename Superclass::TransformParametersType      TransformParametersType;
  typedef typename Superclass::TransformJacobianType        TransformJacobianType;
  typedef typename Superclass::InterpolatorType             InterpolatorType;
  typedef typename Superclass::InterpolatorPointer          InterpolatorPointer;
  typedef typename Superclass::RealType                     RealType;
  typedef typename Superclass::GradientPixelType            GradientPixelType;
  typedef typename Superclass::GradientImageType            GradientImageType;
  typedef typename Superclass::GradientImagePointer         GradientImagePointer;
  typedef typename Superclass::GradientImageFilterType      GradientImageFilterType;
  typedef typename Superclass::GradientImageFilterPointer   GradientImageFilterPointer;
  typedef typename Superclass::FixedImageMaskType           FixedImageMaskType;
  typedef typename Superclass::FixedImageMaskPointer        FixedImageMaskPointer;
  typedef typename Superclass::MovingImageMaskType          MovingImageMaskType;
  typedef typename Superclass::MovingImageMaskPointer       MovingImageMaskPointer;
  typedef typename Superclass::MeasureType                  MeasureType;
  typedef typename Superclass::DerivativeType               DerivativeType;
  typedef typename Superclass::ParametersType               ParametersType;

  typedef InterpolateImageFunction<
    FixedImageType, CoordinateRepresentationType >        FixedImageInterpolatorType;
  typedef typename FixedImageInterpolatorType::Pointer FixedImageInterpolatorPointer;

  typedef std::vector< FixedImageConstPointer >        FixedImageVectorType;
  typedef std::vector< FixedImageMaskPointer >         FixedImageMaskVectorType;
  typedef std::vector< FixedImageRegionType >          FixedImageRegionVectorType;
  typedef std::vector< MovingImageConstPointer >       MovingImageVectorType;
  typedef std::vector< MovingImageMaskPointer >        MovingImageMaskVectorType;
  typedef std::vector< InterpolatorPointer >           InterpolatorVectorType;
  typedef std::vector< FixedImageInterpolatorPointer > FixedImageInterpolatorVectorType;

  virtual void SetFixedImage( const FixedImageType * _arg, unsigned int pos );

  virtual void SetFixedImage( const FixedImageType * _arg )
  {
    this->SetFixedImage( _arg, 0 );
  }


  virtual const FixedImageType * GetFixedImage( unsigned int pos ) const;

  virtual const FixedImageType * GetFixedImage( void ) const
  {
    return this->GetFixedImage( 0 );
  }


  itkSetNumberOfMacro( FixedImage );

  itkGetConstMacro( NumberOfFixedImages, unsigned int );

  virtual void SetFixedImageMask( FixedImageMaskType * _arg, unsigned int pos );

  virtual void SetFixedImageMask( FixedImageMaskType * _arg )
  {
    this->SetFixedImageMask( _arg, 0 );
  }


  virtual FixedImageMaskType * GetFixedImageMask( unsigned int pos ) const;

  virtual FixedImageMaskType * GetFixedImageMask( void ) const
  {
    return this->GetFixedImageMask( 0 );
  }


  itkSetNumberOfMacro( FixedImageMask );

  itkGetConstMacro( NumberOfFixedImageMasks, unsigned int );

  virtual void SetFixedImageRegion( const FixedImageRegionType _arg, unsigned int pos );

  virtual void SetFixedImageRegion( const FixedImageRegionType _arg )
  {
    this->SetFixedImageRegion( _arg, 0 );
  }


  virtual const FixedImageRegionType & GetFixedImageRegion( unsigned int pos ) const;

  virtual const FixedImageRegionType & GetFixedImageRegion( void ) const
  {
    return this->GetFixedImageRegion( 0 );
  }


  itkSetNumberOfMacro( FixedImageRegion );

  itkGetConstMacro( NumberOfFixedImageRegions, unsigned int );

  virtual void SetMovingImage( const MovingImageType * _arg, unsigned int pos );

  virtual void SetMovingImage( const MovingImageType * _arg )
  {
    this->SetMovingImage( _arg, 0 );
  }


  virtual const MovingImageType * GetMovingImage( unsigned int pos ) const;

  virtual const MovingImageType * GetMovingImage( void ) const
  {
    return this->GetMovingImage( 0 );
  }


  itkSetNumberOfMacro( MovingImage );

  itkGetConstMacro( NumberOfMovingImages, unsigned int );

  virtual void SetMovingImageMask( MovingImageMaskType * _arg, unsigned int pos );

  virtual void SetMovingImageMask( MovingImageMaskType * _arg )
  {
    this->SetMovingImageMask( _arg, 0 );
  }


  virtual MovingImageMaskType * GetMovingImageMask( unsigned int pos ) const;

  virtual MovingImageMaskType * GetMovingImageMask( void ) const
  {
    return this->GetMovingImageMask( 0 );
  }


  itkSetNumberOfMacro( MovingImageMask );

  itkGetConstMacro( NumberOfMovingImageMasks, unsigned int );

  virtual void SetInterpolator( InterpolatorType * _arg, unsigned int pos );

  virtual void SetInterpolator( InterpolatorType * _arg )
  {
    return this->SetInterpolator( _arg, 0 );
  }


  virtual InterpolatorType * GetInterpolator( unsigned int pos ) const;

  virtual InterpolatorType * GetInterpolator( void ) const
  {
    return this->GetInterpolator( 0 );
  }


  itkSetNumberOfMacro( Interpolator );

  itkGetConstMacro( NumberOfInterpolators, unsigned int );

  itkGetConstMacro( InterpolatorsAreBSpline, bool );

  virtual void SetFixedImageInterpolator( FixedImageInterpolatorType * _arg, unsigned int pos );

  virtual void SetFixedImageInterpolator( FixedImageInterpolatorType * _arg )
  {
    return this->SetFixedImageInterpolator( _arg, 0 );
  }


  virtual FixedImageInterpolatorType * GetFixedImageInterpolator( unsigned int pos ) const;

  virtual FixedImageInterpolatorType * GetFixedImageInterpolator( void ) const
  {
    return this->GetFixedImageInterpolator( 0 );
  }


  itkSetNumberOfMacro( FixedImageInterpolator );

  itkGetConstMacro( NumberOfFixedImageInterpolators, unsigned int );

  virtual void Initialize( void ) throw ( ExceptionObject );

protected:

  MultiInputImageToImageMetricBase();

  virtual ~MultiInputImageToImageMetricBase() {}

  typedef typename Superclass::MovingImagePointType           MovingImagePointType;
  typedef typename Superclass::MovingImageIndexType           MovingImageIndexType;
  typedef typename Superclass::MovingImageDerivativeType      MovingImageDerivativeType;
  typedef typename Superclass::MovingImageContinuousIndexType MovingImageContinuousIndexType;

  typedef typename Superclass::BSplineInterpolatorType BSplineInterpolatorType;
  typedef typename BSplineInterpolatorType::Pointer    BSplineInterpolatorPointer;
  typedef std::vector< BSplineInterpolatorPointer >    BSplineInterpolatorVectorType;

  virtual void InitializeImageSampler( void ) throw ( ExceptionObject );

  virtual void CheckForBSplineInterpolators( void );

  virtual bool EvaluateMovingImageValueAndDerivative(
    const MovingImagePointType & mappedPoint,
    RealType & movingImageValue,
    MovingImageDerivativeType * gradient ) const;

  virtual bool IsInsideMovingMask(
    const MovingImagePointType & mappedPoint ) const;

  FixedImageVectorType             m_FixedImageVector;
  FixedImageMaskVectorType         m_FixedImageMaskVector;
  FixedImageRegionVectorType       m_FixedImageRegionVector;
  MovingImageVectorType            m_MovingImageVector;
  MovingImageMaskVectorType        m_MovingImageMaskVector;
  InterpolatorVectorType           m_InterpolatorVector;
  FixedImageInterpolatorVectorType m_FixedImageInterpolatorVector;

  bool                          m_InterpolatorsAreBSpline;
  BSplineInterpolatorVectorType m_BSplineInterpolatorVector;

private:

  MultiInputImageToImageMetricBase( const Self & ); // purposely not implemented
  void operator=( const Self & );                   // purposely not implemented

  FixedImageRegionType m_DummyFixedImageRegion;

  unsigned int m_NumberOfFixedImages;
  unsigned int m_NumberOfFixedImageMasks;
  unsigned int m_NumberOfFixedImageRegions;
  unsigned int m_NumberOfMovingImages;
  unsigned int m_NumberOfMovingImageMasks;
  unsigned int m_NumberOfInterpolators;
  unsigned int m_NumberOfFixedImageInterpolators;

};

} // end namespace itk

#undef itkSetNumberOfMacro

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkMultiInputImageToImageMetricBase.hxx"
#endif

#endif // end #ifndef __itkMultiInputImageToImageMetricBase_h