mitkITKImageImport.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 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();

    ~ITKImageImport() override;

    void GenerateOutputInformation() override;

    void GenerateInputRequestedRegion() override;

    void GenerateData() override;

    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 */