itkCurveFitRegistrationMethod.h

Go to the documentation of this file.

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

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). All rights reserved.

  This software is distributed WITHOUT ANY WARRANTY; without even
  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
  PURPOSE.

  See LICENSE.txt in the top level directory for details.

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

#ifndef itkCurveFitRegistrationMethod_h
#define itkCurveFitRegistrationMethod_h

#include <itkImageToImageFilter.h>

#include <itkImage.h>
#include <itkImageReconstructionMetric.h>
#include <itkMultipleValuedNonLinearOptimizer.h>
#include <itkBSplineCurveFitMetric.h>


namespace itk
{

template < class IntensityType = int >
class ITK_EXPORT CurveFitRegistrationMethod :
  public ImageToImageFilter< Image<IntensityType, 4>, Image<IntensityType, 4> > 
{
  public:
      
  typedef CurveFitRegistrationMethod                    Self;
  typedef ImageToImageFilter< Image<IntensityType, 4>, 
                              Image<IntensityType, 4> > Superclass;
  typedef SmartPointer<Self>                            Pointer;
  typedef SmartPointer<const Self>                      ConstPointer;
      
  itkNewMacro(Self);
      
  itkTypeMacro(CurveFitRegistrationMethod, ImageToImageFilter);
      
  static const unsigned int NDimensions = 4;

  typedef Image<IntensityType, NDimensions>        TemporalVolumeType;
  typedef typename TemporalVolumeType::Pointer     TemporalVolumePointer;
  typedef typename TemporalVolumeType::RegionType  TemporalVolumeRegionType;
  typedef typename TemporalVolumeType::PixelType   TemporalVolumePixelType;
  typedef typename TemporalVolumeType::SizeType    TemporalVolumeSizeType;
  typedef typename TemporalVolumeType::SpacingType TemporalVolumeSpacingType;
  typedef typename TemporalVolumeType::PointType   TemporalVolumePointType;

  typedef MultipleValuedNonLinearOptimizer  OptimizerType;
  typedef typename OptimizerType::Pointer   OptimizerPointer;
      
  typedef BSplineCurveFitMetric< IntensityType > MetricType;
  typedef typename MetricType::Pointer           MetricPointer;
 
  typedef typename MetricType::TransformType TransformType;
  typedef typename TransformType::Pointer    TransformPointer;
     
  typedef typename MetricType::TransformParametersType ParametersType;
      
  void SetInitialTransformParameters( const ParametersType &param );
  itkGetConstReferenceMacro( InitialParameters, ParametersType );

  itkGetConstReferenceMacro( LastParameters, ParametersType );
      
  itkSetObjectMacro( Optimizer,  OptimizerType );
  itkGetObjectMacro( Optimizer,  OptimizerType );
      
  itkSetObjectMacro( Metric, MetricType );
  itkGetObjectMacro( Metric, MetricType );

  itkSetObjectMacro( Transform, TransformType );
  itkGetObjectMacro( Transform, TransformType );
   
  unsigned long GetMTime() const;  

  protected:

  CurveFitRegistrationMethod();
  virtual ~CurveFitRegistrationMethod() {};

  void PrintSelf(std::ostream& os, Indent indent) const;

  virtual void Initialise() throw (ExceptionObject);
       
  void StartOptimization(void);

  void GenerateData();



  private:

  CurveFitRegistrationMethod(const Self&); // purposely not implemented
  void operator=(const Self&);            // purposely not implemented
  
  bool m_FlagInitialised;

  TemporalVolumePointer m_InputTemporalVolume;

  OptimizerPointer      m_Optimizer;
  MetricPointer         m_Metric;

  TransformPointer      m_Transform;

  ParametersType        m_InitialParameters;
  ParametersType        m_LastParameters;

};


} // end namespace itk

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

#endif