mitkMAPAlgorithmHelper.h

Go to the documentation of this file.

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

The Medical Imaging Interaction Toolkit (MITK)

Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.

Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.

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


#ifndef mitkMAPAlgorithmHelper_h
#define mitkMAPAlgorithmHelper_h


//MatchPoint
#include "mapRegistrationAlgorithmBase.h"
#include "mapRegistrationBase.h"

//MITK
#include <mitkImage.h>
#include <mitkPointSet.h>

//MITK
#include "MitkMatchPointRegistrationExports.h"
#include "mitkMAPRegistrationWrapper.h"

namespace mitk
{
  class MITKMATCHPOINTREGISTRATION_EXPORT MAPAlgorithmHelper
  {
  public:

    MAPAlgorithmHelper(map::algorithm::RegistrationAlgorithmBase* algorithm);

    void SetData(const mitk::BaseData* moving, const mitk::BaseData* target);

    void SetAllowImageCasting(bool allowCasting);
    bool GetAllowImageCasting() const;

    static bool HasImageAlgorithmInterface(const map::algorithm::RegistrationAlgorithmBase* algorithm);
    static bool HasPointSetAlgorithmInterface(const map::algorithm::RegistrationAlgorithmBase* algorithm);


    struct CheckError
    {
      enum Type
      {
        none = 0,
        onlyByCasting = 1,
        wrongDimension = 2,
        unsupportedDataType = 3
      };
    };

    bool CheckData(const mitk::BaseData* moving, const mitk::BaseData* target,
                   CheckError::Type& error) const;

    map::core::RegistrationBase::Pointer GetRegistration() const;

    mitk::MAPRegistrationWrapper::Pointer GetMITKRegistrationWrapper() const;

    ~MAPAlgorithmHelper() {}

  private:

    MAPAlgorithmHelper& operator = (const MAPAlgorithmHelper&);
    MAPAlgorithmHelper(const MAPAlgorithmHelper&);

    template<typename TInImageType, typename TOutImageType>
    typename TOutImageType::Pointer CastImage(const TInImageType* input) const;

    template<typename TPixelType1, unsigned int VImageDimension1,
             typename TPixelType2, unsigned int VImageDimension2>
    void DoSetImages(const itk::Image<TPixelType1, VImageDimension1>* moving,
                     const itk::Image<TPixelType2, VImageDimension2>* target);

    template<typename TPixelType1, unsigned int VImageDimension1,
             typename TPixelType2, unsigned int VImageDimension2>
    void DoCheckImages(const itk::Image<TPixelType1, VImageDimension1>* moving,
                       const itk::Image<TPixelType2, VImageDimension2>* target) const;

    map::algorithm::RegistrationAlgorithmBase::Pointer m_AlgorithmBase;

    bool m_AllowImageCasting;

    mutable CheckError::Type m_Error;
  };

  mitk::MAPRegistrationWrapper::Pointer MITKMATCHPOINTREGISTRATION_EXPORT GenerateIdentityRegistration3D();

}

#endif