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