mitkAlgorithmHelper.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 mitkAlgorithmHelper_h
#define mitkAlgorithmHelper_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 MITKAlgorithmHelper
  {
  public:

    MITKAlgorithmHelper(map::algorithm::RegistrationAlgorithmBase* algorithm = NULL);

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

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

    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;

    ~MITKAlgorithmHelper() {}

  private:

    MITKAlgorithmHelper& operator = (const MITKAlgorithmHelper&);
    MITKAlgorithmHelper(const MITKAlgorithmHelper&);

    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