mitkDemonsRegistration.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 MITKDEMONSREGISTRATION_H
#define MITKDEMONSREGISTRATION_H

#include "MitkDeformableRegistrationExports.h"
#include "itkDemonsRegistrationFilter.h"

#include "mitkImageAccessByItk.h"
#include "mitkRegistrationBase.h"

namespace mitk
{
  class MITKDEFORMABLEREGISTRATION_EXPORT DemonsRegistration : public RegistrationBase
  {
  public:
    mitkClassMacro(DemonsRegistration, RegistrationBase);

    itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      
      void SetNumberOfIterations(int iterations);

    void SetStandardDeviation(float deviation);

    void SetSaveDeformationField(bool saveField);

    void SetDeformationFieldFileName(const char *fieldName);

    void SetSaveResult(bool saveResult);

    void SetResultFileName(const char *resultName);

    itk::Image<itk::Vector<float, 3>, 3>::Pointer GetDeformationField();

    virtual void GenerateData() override
    {
      if (this->GetInput())
      {
        AccessByItk(this->GetInput(), GenerateData2);
      }
    }

  protected:
    DemonsRegistration();

    virtual ~DemonsRegistration();

    template <typename TPixel, unsigned int VImageDimension>
    void GenerateData2(const itk::Image<TPixel, VImageDimension> *itkImage1);

    int m_Iterations;
    float m_StandardDeviation;
    const char *m_FieldName;
    const char *m_ResultName;
    bool m_SaveField;
    bool m_SaveResult;
    itk::Image<class itk::Vector<float, 3>, 3>::Pointer m_DeformationField;
  };
}

#endif // MITKDEMONSREGISTRATION_H