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 mitkImageToItk_h
#define mitkImageToItk_h
 
#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