mitkTransformFactory.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center,
Division of Medical and Biological Informatics.
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 or http://www.mitk.org for details.

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

#ifndef MITKTRANSFORMFACTORY_H
#define MITKTRANSFORMFACTORY_H

#include "MitkRigidRegistrationExports.h"
#include "itkImage.h"
#include "itkTransform.h"
#include "mitkTransformParameters.h"

namespace mitk
{
  template <class TPixelType, unsigned int VImageDimension>
  class MITKRIGIDREGISTRATION_EXPORT TransformFactory : public itk::Object
  {
  public:
    mitkClassMacroItkParent(TransformFactory, itk::Object);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      typedef typename itk::Image<TPixelType, VImageDimension> FixedImageType;
    typedef typename itk::Image<TPixelType, VImageDimension> MovingImageType;
    typedef typename itk::Image<TPixelType, 2> FixedImage2DType;
    typedef typename itk::Image<TPixelType, 2> MovingImage2DType;
    typedef typename itk::Image<TPixelType, 3> FixedImage3DType;
    typedef typename itk::Image<TPixelType, 3> MovingImage3DType;

    typedef typename itk::Transform<double, VImageDimension, VImageDimension> TransformType;

    typedef typename TransformType::Pointer TransformPointer;

    TransformPointer GetTransform();

    void SetFixedImage(FixedImageType *fixed);

    void SetMovingImage(MovingImageType *moving);

    void SetTransformParameters(TransformParameters::Pointer transformParameters)
    {
      m_TransformParameters = transformParameters;
    }

    TransformParameters::Pointer GetTransformParameters() { return m_TransformParameters; }
  protected:
    TransformFactory();
    ~TransformFactory(){};

    TransformParameters::Pointer m_TransformParameters;
    typename FixedImageType::Pointer m_FixedImage;
    typename MovingImageType::Pointer m_MovingImage;
    typename FixedImage2DType::Pointer m_FixedImage2D;
    typename MovingImage2DType::Pointer m_MovingImage2D;
    typename FixedImage3DType::Pointer m_FixedImage3D;
    typename MovingImage3DType::Pointer m_MovingImage3D;
  };

} // namespace mitk

#include "mitkTransformFactory.txx"

#endif // MITKTRANSFORMFACTORY_H