freRigidRegistrationProcessor.h

Go to the documentation of this file.

/*=========================================================================
 
  Program:   F.R.E.E. - flexible registration evaluation engine
  Version:   v.1.0.0
  Date:      $Date: 2006/09/01 12:00:00 $
  Module:    $RCSfile: freRigidRegistrationProcessor.h,v $
  Language:  C++
 
 
 
  Copyright (c) 2007 Ralf o Floca (Department of Medical Informatics,
  Institute for Medical Biometry and Informatics, University of Heidelberg,
  Germany). All rights reserved.
  See FREECopyright.txt or http://www.mi.med.uni-hd.de/free/copyright.htm
  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 __freRigidRegistrationProcessor_h
#define __freRigidRegistrationProcessor_h

#include "freRigidRegistrationProcessorBase.h"

#include "itkMultiResolutionImageRegistrationMethod.h"
#include "itkMultiResolutionPyramidImageFilter.h"

#include "itkImageRegistrationMethod.h"

#include "itkTransform.h"
#include "itkInterpolateImageFunction.h"
#include "itkImageToImageMetric.h"
#include "itkSingleValuedNonLinearOptimizer.h"

namespace FREE
{

template <class TRegistrationMethod, class TTransformBase, class TInterpolateBase,
class TMetricBase, class TOptimizerBase, unsigned int VImageDimension>
class SingleRigidRegistrationProcessorBase : public RigidRegistrationProcessorBase<TRegistrationMethod, TTransformBase, TInterpolateBase, TMetricBase, TOptimizerBase, VImageDimension>
{
public:
    typedef SingleRigidRegistrationProcessorBase<TRegistrationMethod, TTransformBase, TInterpolateBase, TMetricBase, TOptimizerBase, VImageDimension> Self;
    typedef RigidRegistrationProcessorBase<TRegistrationMethod, TTransformBase, TInterpolateBase, TMetricBase, TOptimizerBase, VImageDimension> Superclass;
    typedef itk::SmartPointer<Self>        Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    typedef typename Superclass::RegistrationType RegistrationType;
    typedef typename Superclass::RegistrationPointer RegistrationPointer;

    typedef typename Superclass::TransformType TransformType;
    typedef typename Superclass::TransformPointer TransformPointer;

    typedef typename Superclass::InterpolatorType InterpolatorType;
    typedef typename Superclass::InterpolatorPointer InterpolatorPointer;

    typedef typename Superclass::MetricType MetricType;
    typedef typename Superclass::MetricPointer MetricPointer;

    typedef typename Superclass::OptimizerType OptimizerType;
    typedef typename Superclass::OptimizerPointer OptimizerPointer;

    typedef typename Superclass::ImageType ImageType;
    typedef typename Superclass::RegionType RegionType;
    typedef typename Superclass::ImagePointer ImagePointer;
    typedef typename Superclass::PointType PointType;
    typedef typename Superclass::TransformFieldType TransformFieldType;
    typedef typename Superclass::TransformFieldPointer TransformFieldPointer;

    itkTypeMacro( SingleRigidRegistrationProcessorBase, RigidRegistrationProcessorBase );
    itkNewMacro(Self);

protected:
    virtual void ComputeRegistration();

    SingleRigidRegistrationProcessorBase();

    virtual ~SingleRigidRegistrationProcessorBase();
};

template <unsigned int VImageDimension>
class RigidRegistrationProcessor : public SingleRigidRegistrationProcessorBase < itk::ImageRegistrationMethod < typename ImageTypes<VImageDimension>::InternalImageType , typename ImageTypes<VImageDimension>::InternalImageType >,
            itk::Transform< ScalarType, VImageDimension, VImageDimension >,
            itk::InterpolateImageFunction< typename ImageTypes<VImageDimension>::InternalImageType, ScalarType >,
            itk::ImageToImageMetric<  typename ImageTypes<VImageDimension>::InternalImageType, typename ImageTypes<VImageDimension>::InternalImageType >,
            itk::SingleValuedNonLinearOptimizer, VImageDimension >
{
public:
    typedef RigidRegistrationProcessor< VImageDimension > Self;
    typedef SingleRigidRegistrationProcessorBase < itk::ImageRegistrationMethod < typename ImageTypes<VImageDimension>::InternalImageType , typename ImageTypes<VImageDimension>::InternalImageType >,
    itk::Transform< ScalarType, VImageDimension, VImageDimension >,
    itk::InterpolateImageFunction< typename ImageTypes<VImageDimension>::InternalImageType, ScalarType >,
    itk::ImageToImageMetric<  typename ImageTypes<VImageDimension>::InternalImageType, typename ImageTypes<VImageDimension>::InternalImageType >,
    itk::SingleValuedNonLinearOptimizer, VImageDimension > Superclass;

    typedef itk::SmartPointer<Self>        Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    typedef typename Superclass::RegistrationType RegistrationType;
    typedef typename Superclass::RegistrationPointer RegistrationPointer;

    typedef typename Superclass::TransformType TransformType;
    typedef typename Superclass::TransformPointer TransformPointer;

    typedef typename Superclass::InterpolatorType InterpolatorType;
    typedef typename Superclass::InterpolatorPointer InterpolatorPointer;

    typedef typename Superclass::MetricType MetricType;
    typedef typename Superclass::MetricPointer MetricPointer;

    typedef typename Superclass::OptimizerType OptimizerType;
    typedef typename Superclass::OptimizerPointer OptimizerPointer;

    typedef typename Superclass::ImageType ImageType;
    typedef typename Superclass::RegionType RegionType;
    typedef typename Superclass::ImagePointer ImagePointer;
    typedef typename Superclass::PointType PointType;
    typedef typename Superclass::TransformFieldType TransformFieldType;
    typedef typename Superclass::TransformFieldPointer TransformFieldPointer;

    itkTypeMacro( RigidRegistrationProcessor, SingleRigidRegistrationProcessorBase );
    itkNewMacro(Self);

protected:
    RigidRegistrationProcessor();

    virtual ~RigidRegistrationProcessor();
};

template <unsigned int VImageDimension>
class MultiRigidRegistrationProcessor   : public RigidRegistrationProcessorBase< itk::MultiResolutionImageRegistrationMethod < typename ImageTypes<VImageDimension>::InternalImageType , typename ImageTypes<VImageDimension>::InternalImageType >,
            itk::Transform< ScalarType, VImageDimension, VImageDimension >,
            itk::InterpolateImageFunction< typename ImageTypes<VImageDimension>::InternalImageType, ScalarType >,
            itk::ImageToImageMetric<  typename ImageTypes<VImageDimension>::InternalImageType, typename ImageTypes<VImageDimension>::InternalImageType >,
            itk::SingleValuedNonLinearOptimizer, VImageDimension >
{
public:
    typedef MultiRigidRegistrationProcessor<VImageDimension> Self;
    typedef RigidRegistrationProcessorBase<itk::MultiResolutionImageRegistrationMethod < typename ImageTypes<VImageDimension>::InternalImageType , typename ImageTypes<VImageDimension>::InternalImageType >,
    itk::Transform< ScalarType, VImageDimension, VImageDimension >,
    itk::InterpolateImageFunction< typename ImageTypes<VImageDimension>::InternalImageType, ScalarType >,
    itk::ImageToImageMetric<  typename ImageTypes<VImageDimension>::InternalImageType, typename ImageTypes<VImageDimension>::InternalImageType >,
    itk::SingleValuedNonLinearOptimizer, VImageDimension > Superclass;

    typedef itk::SmartPointer<Self>        Pointer;
    typedef itk::SmartPointer<const Self>  ConstPointer;

    typedef typename Superclass::RegistrationType RegistrationType;
    typedef typename Superclass::RegistrationPointer RegistrationPointer;

    typedef typename Superclass::TransformType TransformType;
    typedef typename Superclass::TransformPointer TransformPointer;

    typedef typename Superclass::InterpolatorType InterpolatorType;
    typedef typename Superclass::InterpolatorPointer InterpolatorPointer;

    typedef typename Superclass::MetricType MetricType;
    typedef typename Superclass::MetricPointer MetricPointer;

    typedef typename Superclass::OptimizerType OptimizerType;
    typedef typename Superclass::OptimizerPointer OptimizerPointer;

    typedef typename Superclass::ImageType ImageType;
    typedef typename Superclass::RegionType RegionType;
    typedef typename Superclass::ImagePointer ImagePointer;
    typedef typename Superclass::PointType PointType;
    typedef typename Superclass::TransformFieldType TransformFieldType;
    typedef typename Superclass::TransformFieldPointer TransformFieldPointer;

    itkTypeMacro( MultiRigidRegistrationProcessor, RigidRegistrationProcessorBase );

    itkNewMacro(Self);

    typedef itk::MultiResolutionPyramidImageFilter< ImageType, ImageType > PyramidType;
    typedef typename PyramidType::Pointer PyramidPointer;

    IterationObserver::Pointer m_LevelObserver;
    typename NotificationEvent<Self>::Pointer m_LevelNotificationEvent;

    PyramidPointer smpFixedImagePyramid;
    PyramidPointer smpMovingImagePyramid;

    virtual const int GetResolutionLevelCount () const;

protected:
    MultiRigidRegistrationProcessor();

    virtual ~MultiRigidRegistrationProcessor();

    virtual void ComputeRegistration();

    void OnNewLevel(void* pSender, long threadID);

    //Indicator of the first level is still in processing. Needed because the OnLevel event
    //is also triggered in the first level.
    bool m_bFirstResLevel;
};

typedef RigidRegistrationProcessor<2> RigidRegistrationProcessor2D;
typedef RigidRegistrationProcessor<3> RigidRegistrationProcessor3D;

typedef MultiRigidRegistrationProcessor<2> MultiRigidRegistrationProcessor2D;
typedef MultiRigidRegistrationProcessor<3> MultiRigidRegistrationProcessor3D;

}//End of Namespace free

#ifndef ITK_MANUAL_INSTANTIATION
#include "freRigidRegistrationProcessor.txx"
#endif

#endif

---