mitkITKImageImport.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 MITKITKIMAGEIMPORT_H_HEADER_INCLUDED_C1E4861D
#define MITKITKIMAGEIMPORT_H_HEADER_INCLUDED_C1E4861D

#include "itkImageToImageFilterDetail.h"
#include "mitkImageSource.h"
#include <MitkCoreExports.h>

namespace mitk
{
  template <class TInputImage>
  class MITK_EXPORT ITKImageImport : public ImageSource
  {
  public:
    mitkClassMacro(ITKImageImport, ImageSource) itkFactorylessNewMacro(Self) itkCloneMacro(Self)

      typedef TInputImage InputImageType;
    typedef typename InputImageType::Pointer InputImagePointer;
    typedef typename InputImageType::ConstPointer InputImageConstPointer;
    typedef typename InputImageType::RegionType InputImageRegionType;
    typedef typename InputImageType::PixelType InputImagePixelType;

    itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
    itkStaticConstMacro(RegionDimension, unsigned int, mitk::SlicedData::RegionDimension);

    InputImageType *GetInput(void);

    void SetInput(const InputImageType *);
    using itk::ProcessObject::SetInput;

    void SetGeometry(const BaseGeometry *geometry);

  protected:
    ITKImageImport();

    virtual ~ITKImageImport();

    virtual void GenerateOutputInformation() override;

    virtual void GenerateInputRequestedRegion() override;

    virtual void GenerateData() override;

    virtual void SetNthOutput(DataObjectPointerArraySizeType num, itk::DataObject *output) override;

    typedef itk::ImageToImageFilterDetail::ImageRegionCopier<itkGetStaticConstMacro(InputImageDimension),
                                                             itkGetStaticConstMacro(RegionDimension)>
      OutputToInputRegionCopierType;

    BaseGeometry::Pointer m_Geometry;
  };

  template <typename ItkOutputImageType>
  Image::Pointer ImportItkImage(const itk::SmartPointer<ItkOutputImageType> &itkimage,
                                const BaseGeometry *geometry = nullptr,
                                bool update = true);

  template <typename ItkOutputImageType>
  Image::Pointer ImportItkImage(const ItkOutputImageType *itkimage,
                                const BaseGeometry *geometry = nullptr,
                                bool update = true);

  template <typename ItkOutputImageType>
  Image::Pointer GrabItkImageMemory(itk::SmartPointer<ItkOutputImageType> &itkimage,
                                    mitk::Image *mitkImage = nullptr,
                                    const BaseGeometry *geometry = nullptr,
                                    bool update = true);

  template <typename ItkOutputImageType>
  Image::Pointer GrabItkImageMemory(ItkOutputImageType *itkimage,
                                    mitk::Image *mitkImage = nullptr,
                                    const BaseGeometry *geometry = nullptr,
                                    bool update = true);

} // namespace mitk

#ifndef MITK_MANUAL_INSTANTIATION
#include "mitkITKImageImport.txx"
#endif

#endif /* MITKITKIMAGEIMPORT_H_HEADER_INCLUDED_C1E4861D */