mitkImageRegistrationMethod.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 MITKIMAGEREGISTRATIONMETHOD_H
#define MITKIMAGEREGISTRATIONMETHOD_H

#include "MitkRigidRegistrationExports.h"
#include "itkImageRegistrationMethod.h"
#include "itkSingleValuedNonLinearOptimizer.h"

#include "mitkCommon.h"
#include "mitkImageAccessByItk.h"
#include "mitkImageToImageFilter.h"
#include "mitkRigidRegistrationObserver.h"

#include "itkImageMaskSpatialObject.h"
#include "mitkRigidRegistrationPreset.h"

namespace mitk
{
  class MITKRIGIDREGISTRATION_EXPORT ImageRegistrationMethod : public ImageToImageFilter
  {
  public:
    typedef itk::SingleValuedNonLinearOptimizer OptimizerType;
    typedef itk::ImageMaskSpatialObject<3> MaskType;

    mitkClassMacro(ImageRegistrationMethod, ImageToImageFilter);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      static const int LINEARINTERPOLATOR = 0;
    static const int NEARESTNEIGHBORINTERPOLATOR = 1;

    void SetObserver(RigidRegistrationObserver::Pointer observer);

    void SetInterpolator(int interpolator);

    virtual void GenerateData() override;

    virtual void SetReferenceImage(Image::Pointer fixedImage);

    virtual void SetFixedMask(Image::Pointer fixedMask);

    virtual void SetMovingMask(Image::Pointer movingMask);

    void SetOptimizerScales(itk::Array<double> scales);

    void SetTransform(itk::Object::Pointer transform);

    void SetMetric(itk::Object::Pointer metric);

    void SetOptimizer(itk::Object::Pointer optimizer);

    void SetNumberOfLevels(unsigned int levels);

  protected:
    ImageRegistrationMethod();
    virtual ~ImageRegistrationMethod();

    friend struct ImageRegistrationMethodAccessFunctor;

    RigidRegistrationObserver::Pointer m_Observer;
    int m_Interpolator;
    Image::Pointer m_ReferenceImage;
    Image::Pointer m_FixedMask;
    Image::Pointer m_MovingMask;

    virtual void GenerateOutputInformation() override {}
  private:
    itk::Object::Pointer m_Transform;
    itk::Object::Pointer m_Metric;
    itk::Object::Pointer m_Optimizer;
    itk::Array<double> m_OptimizerScales;

    unsigned int m_MultiResolutionScales;
  };
}

#endif // MITKIMAGEREGISTRATIONMETHOD_H