itkGroupwiseRegistrationMethod.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 itkGroupwiseRegistrationMethod_h
#define itkGroupwiseRegistrationMethod_h

#include <itkProcessObject.h>
#include <itkImage.h>
#include <itkMeanVoxelwiseIntensityOfMultipleImages.h>
#include <itkImageRegistrationFilter.h>

namespace itk
{

template <typename TImageType, 
          unsigned int Dimension, 
          class TScalarType, 
          typename TDeformationScalar >
class ITK_EXPORT GroupwiseRegistrationMethod : public ImageSource< TImageType >
{
public:
  typedef GroupwiseRegistrationMethod  Self;
  typedef ProcessObject                Superclass;
  typedef SmartPointer<Self>           Pointer;
  typedef SmartPointer<const Self>     ConstPointer;

  itkNewMacro(Self);
  
  itkTypeMacro(GroupwiseRegistrationMethod, ProcessObject);

  itkStaticConstMacro(ImageDimension, unsigned int, TImageType::ImageDimension);

  // Some convenient typedefs.
  typedef TImageType                       ImageType;
  typedef typename ImageType::Pointer      ImagePointer;
  typedef typename ImageType::ConstPointer ImageConstPointer;
  typedef typename ImageType::RegionType   ImageRegionType; 
  typedef typename ImageType::PixelType    ImagePixelType; 
  typedef typename ImageType::SpacingType  ImageSpacingType;
  typedef typename ImageType::IndexType    ImageIndexType;
  typedef typename ImageType::PointType    ImagePointType;

  typedef typename ImageRegionType::SizeType  ImageSizeType;

  virtual void SetInput( const ImageType *image);
  virtual void SetInput( unsigned int, const TImageType * image);

  const ImageType * GetInput(void);
  const ImageType * GetInput(unsigned int idx);

  typedef itk::ImageRegistrationFilter<ImageType, ImageType, 
                                       Dimension, TScalarType, TDeformationScalar> ImageRegistrationFilterType;
  
  typedef typename ImageRegistrationFilterType::Pointer ImageRegistrationFilterPointerType;
  
  typedef typename itk::MeanVoxelwiseIntensityOfMultipleImages<ImageType, 
                                                               ImageType> MeanVoxelwiseIntensityOfMultipleImagesType;

  itkSetObjectMacro( SumImagesFilter, MeanVoxelwiseIntensityOfMultipleImagesType );
  itkGetObjectMacro( SumImagesFilter, MeanVoxelwiseIntensityOfMultipleImagesType );

  itkSetMacro( NumberOfIterations, unsigned int );
  itkGetMacro( NumberOfIterations, unsigned int );

  void SetRegistrationFilters( std::vector< ImageRegistrationFilterPointerType > &regnFilters ) {
    m_RegistrationFilters = regnFilters;
    this->Modified();
  }

  virtual void Initialise() throw (ExceptionObject);
    
protected:
  GroupwiseRegistrationMethod();
  virtual ~GroupwiseRegistrationMethod() {};
  void PrintSelf(std::ostream& os, Indent indent) const;

  void ComputeInitialSumOfInputImages();

  virtual void GenerateInputRequestedRegion();
  void GenerateOutputInformation();
  
  void StartOptimization(void);

  void  GenerateData ();


private:
  GroupwiseRegistrationMethod(const Self&); // purposely not implemented
  void operator=(const Self&);            // purposely not implemented

  bool m_FlagInitialSumComputed;

  bool m_FlagInitialised;

  unsigned int m_NumberOfIterations;

  ImageRegionType  m_OutRegion;
  ImageSizeType    m_OutSize;
  ImageSpacingType m_OutSpacing;
  ImagePointType   m_OutOrigin;

  typename MeanVoxelwiseIntensityOfMultipleImagesType::Pointer m_SumImagesFilter;

  std::vector< ImageRegistrationFilterPointerType > m_RegistrationFilters;

};


} // end namespace itk

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

#endif