mitkMaskedAlgorithmHelper.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 mitkMaskedAlgorithmHelper_h
#define mitkMaskedAlgorithmHelper_h

#include "itkSpatialObject.h"

//MatchPoint
#include "mapRegistrationAlgorithmBase.h"

//MITK
#include <mitkImage.h>

//MITK
#include "MitkMatchPointRegistrationExports.h"

namespace mitk
{
  class MITKMATCHPOINTREGISTRATION_EXPORT MaskedAlgorithmHelper
  {
  public:

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

    bool SetMasks(const mitk::Image* movingMask, const mitk::Image* targetMask);

    bool CheckSupport(const mitk::Image* movingMask, const mitk::Image* targetMask) const;

    static bool HasMaskedRegistrationAlgorithmInterface(const map::algorithm::RegistrationAlgorithmBase* algorithm);

    ~MaskedAlgorithmHelper() {}

  private:
    using MaskPixelType = unsigned char;

    MaskedAlgorithmHelper& operator = (const MaskedAlgorithmHelper&);
    MaskedAlgorithmHelper(const MaskedAlgorithmHelper&);

    template<unsigned int VImageDimension1, unsigned int VImageDimension2>
    bool DoSetMasks(const mitk::Image* movingMask, const mitk::Image* targetMask);

    template<typename TPixelType, unsigned int VImageDimension>
    void DoConvertMask(const itk::Image<TPixelType, VImageDimension>* mask);
    template<unsigned int VImageDimension>
    void DoConvertMask(const itk::Image<MaskPixelType, VImageDimension>* mask);

    template<unsigned int VImageDimension>
    typename itk::SpatialObject<VImageDimension>::Pointer ConvertMaskSO(const itk::Image<MaskPixelType, VImageDimension>* mask) const;

    itk::DataObject::Pointer m_convertResult;

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

}

#endif