itkImageToImageRegistrationMethodTestingHelper.h

Go to the documentation of this file.

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

  Program:   Surface Extraction Program
  Module:    $RCSfile: ExtractSurface.cxx,v $

  Copyright (c) Kitware Inc.
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/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 notice for more information.

=========================================================================*/
#ifndef itkImageToImageRegistrationMethodTestingHelper_h
#define itkImageToImageRegistrationMethodTestingHelper_h

#include "itkImageFileReader.h"
#include "itkImageFileWriter.h"
#include "itkResampleImageFilter.h"
#include "itkTestingComparisonImageFilter.h"

namespace itk
{

template <class TRegistrationMethod>
class ImageToImageRegistrationMethodTestingHelper
{

  typedef TRegistrationMethod                        RegistrationMethodType;
  typedef typename RegistrationMethodType::ImageType ImageType;
  typedef typename RegistrationMethodType::Pointer   RegistrationMethodPointer;

  typedef itk::ImageFileReader<ImageType> ImageReaderType;
  typedef itk::ImageFileWriter<ImageType> ImageWriterType;

  typedef typename ImageReaderType::Pointer ImageReaderPointer;
  typedef typename ImageWriterType::Pointer ImageWriterPointer;

  typedef ResampleImageFilter<ImageType, ImageType> ResampleFilterType;
  typedef typename ResampleFilterType::Pointer      ResampleFilterPointer;

  typedef itk::Testing::ComparisonImageFilter<ImageType, ImageType> DifferenceFilterType;
  typedef typename DifferenceFilterType::Pointer                    DifferenceFilterPointer;
public:

  ImageToImageRegistrationMethodTestingHelper();

  RegistrationMethodType * GetRegistrationMethod();

  void SetFixedImageFileName( const char * filename );

  void SetMovingImageFileName( const char * filename );

  void SetResampledImageFileName( const char * filename );

  void SetBaselineImageFileName( const char * filename );

  void SetDifferenceImageFileName( const char * filename );

  void PrepareRegistration();

  void RunRegistration();

  void ReportResults();

  void ReportImagesMetaData();

  void ResampleOutputImage();

  void PerformRegressionTest();

  void PrintTest();

  int GetTestResult() const;

  void SetNumberOfFailedPixelsTolerance( unsigned long numberOfPixels );

  void SetRadiusTolerance( unsigned long radius );

  void SetIntensityTolerance( double intensity );

protected:
private:

  RegistrationMethodPointer m_RegistrationMethod;
  ImageReaderPointer        m_FixedImageReader;
  ImageReaderPointer        m_MovingImageReader;
  ImageWriterPointer        m_MovingImageWriter;
  ImageReaderPointer        m_BaselineImageReader;
  ImageWriterPointer        m_DifferenceImageWriter;
  DifferenceFilterPointer   m_DifferenceFilter;
  ResampleFilterPointer     m_ResampleFilter;
  unsigned long             m_NumberOfFailedPixelsTolerance;
  unsigned long             m_RadiusTolerance;
  double                    m_IntensityTolerance;

  bool m_ErrorState;
};

} // end namespace itk

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

#endif