mitkImageToItk.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 IMAGETOITK_H_HEADER_INCLUDED_C1C2FCD2
#define IMAGETOITK_H_HEADER_INCLUDED_C1C2FCD2

#include "mitkImage.h"
#include "mitkImageDataItem.h"
#include "mitkImageWriteAccessor.h"

#include <itkImage.h>
#include <itkImageSource.h>

namespace mitk
{
  template <class TOutputImage>
  class ImageToItk : public itk::ImageSource<TOutputImage>
  {
  protected:
    mitk::Image::Pointer m_MitkImage;
    mitk::ImageDataItem::Pointer m_ImageDataItem;

  public:
    typedef ImageToItk Self;
    typedef itk::ImageSource<TOutputImage> Superclass;
    typedef itk::SmartPointer<Self> Pointer;
    typedef itk::SmartPointer<const Self> ConstPointer;

    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

      typedef typename Superclass::OutputImageRegionType OutputImageRegionType;

    typedef mitk::Image InputImageType;
    typedef InputImageType::Pointer InputImagePointer;
    typedef InputImageType::ConstPointer InputImageConstPointer;
    typedef SlicedData::RegionType InputImageRegionType;
    typedef typename TOutputImage::SizeType SizeType;
    typedef typename TOutputImage::IndexType IndexType;
    typedef typename TOutputImage::RegionType RegionType;
    typedef typename TOutputImage::PixelType PixelType;
    typedef typename TOutputImage::InternalPixelType InternalPixelType;
    typedef typename TOutputImage::PixelContainer PixelContainer;

    virtual void SetInput(mitk::Image *input);
    virtual void SetInput(const mitk::Image *input);
    // virtual void SetInput(unsigned int index, mitk::Image * image);
    // virtual void SetInput(unsigned int index, const mitk::Image * image);

    void UpdateOutputInformation() override;

    itkGetMacro(Channel, int);
    itkSetMacro(Channel, int);

    itkSetMacro(CopyMemFlag, bool);
    itkGetMacro(CopyMemFlag, bool);
    itkBooleanMacro(CopyMemFlag);

    itkSetMacro(Options, int);
    itkGetMacro(Options, int);

  protected:
    using itk::ProcessObject::SetInput;
    mitk::Image *GetInput(void);
    const mitk::Image *GetInput() const;

    ImageToItk() : m_CopyMemFlag(false), m_Channel(0), m_Options(mitk::ImageAccessorBase::DefaultBehavior) {}
    ~ImageToItk() override {}
    void PrintSelf(std::ostream &os, itk::Indent indent) const override;

    void GenerateData() override;

    void GenerateOutputInformation() override;

  private:
    bool m_CopyMemFlag;
    int m_Channel;
    int m_Options;

    bool m_ConstInput;

    // ImageToItk(const Self&); //purposely not implemented
    void operator=(const Self &); // purposely not implemented

    void CheckInput(const mitk::Image *image) const;
  };

  template <typename TPixel, unsigned int VDimension>
  typename ImageTypeTrait<TPixel, VDimension>::ImageType::Pointer ImageToItkImage(mitk::Image *mitkImage)
  {
    typedef typename ImageTypeTrait<TPixel, VDimension>::ImageType ImageType;
    typedef mitk::ImageToItk<ImageType> ImageToItkType;
    itk::SmartPointer<ImageToItkType> imagetoitk = ImageToItkType::New();
    imagetoitk->SetInput(mitkImage);
    imagetoitk->Update();
    return imagetoitk->GetOutput();
  }

  template <typename TPixel, unsigned int VDimension>
  typename ImageTypeTrait<TPixel, VDimension>::ImageType::ConstPointer ImageToItkImage(const mitk::Image *mitkImage)
  {
    typedef typename ImageTypeTrait<TPixel, VDimension>::ImageType ImageType;
    typedef mitk::ImageToItk<ImageType> ImageToItkType;
    itk::SmartPointer<ImageToItkType> imagetoitk = ImageToItkType::New();
    imagetoitk->SetInput(mitkImage);
    imagetoitk->Update();
    return imagetoitk->GetOutput();
  }

} // end namespace mitk

#ifndef ITK_MANUAL_INSTANTIATION
#include "mitkImageToItk.txx"
#endif

#endif // IMAGETOITK_H_HEADER_INCLUDED_C1C2FCD2