niftkQImageToMitkImageFilter.h

Go to the documentation of this file.

/*=============================================================================

  NifTK: A software platform for medical image computing.

  Copyright (c) University College London (UCL). 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 in the top level directory for details.

=============================================================================*/

#ifndef niftkQImageToMitkImageFilter_h
#define niftkQImageToMitkImageFilter_h

#include <mitkCommon.h>
#include <mitkImageSource.h>
#include <itkMacro.h>
#include <itkImage.h>
#include <itkRGBPixel.h>
#include <itkRGBAPixel.h>
#include <QImage>

#include "niftkIGIDataSourcesExports.h"

namespace niftk
{

class NIFTKIGIDATASOURCES_EXPORT QImageToMitkImageFilter : public mitk::ImageSource
{
  public:
    typedef itk::RGBPixel< unsigned char > UCRGBPixelType;
    typedef itk::RGBPixel< unsigned short > USRGBPixelType;
    typedef itk::RGBPixel< float > FloatRGBPixelType;
    typedef itk::RGBPixel< double > DoubleRGBPixelType;

    typedef itk::RGBAPixel< unsigned char > UCRGBAPixelType;
    typedef itk::RGBAPixel< unsigned short > USRGBAPixelType;
    typedef itk::RGBAPixel< float > FloatRGBAPixelType;
    typedef itk::RGBAPixel< double > DoubleRGBAPixelType;

    mitkClassMacro(QImageToMitkImageFilter, mitk::ImageSource)
    itkNewMacro(QImageToMitkImageFilter)

    void SetQImage(const QImage* image);
    itkGetMacro(QImage, const QImage*);

    OutputImageType* GetOutput();

    void SetGeometryImage ( mitk::Image::Pointer GeomImage);

protected:

    QImageToMitkImageFilter(); // purposely hidden
    virtual ~QImageToMitkImageFilter();

    virtual DataObjectPointer MakeOutput(unsigned int idx);
    virtual void GenerateData() override;

protected:

    const QImage*        m_QImage;
    mitk::Image::Pointer m_Image;
    mitk::Image::Pointer m_GeomImage;

private:

    template <typename TPixel, unsigned int VImageDimension>
    static mitk::Image::Pointer ConvertQImageToMitkImage(const QImage* input, const mitk::Image::Pointer GeomImage);
    template <typename TPixel, unsigned int VImageDimension>
    static mitk::Image::Pointer Convert8BitQImageToMitkImage(const QImage* input, const mitk::Image::Pointer GeomImage);
};

} // end namespace

#endif