itkExtractChannelFromRgbaImageFilter.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 __itkExtractChannelFromRgbaImageFilter_h__
#define __itkExtractChannelFromRgbaImageFilter_h__

#include <itkImageToImageFilter.h>
#include <itkImage.h>
#include <itkRGBAPixel.h>
#include <itkVectorImage.h>

namespace itk{

template< class ReferenceImageType, class OutputImageType >
class ExtractChannelFromRgbaImageFilter : public ImageToImageFilter< itk::Image<itk::RGBAPixel<unsigned char>,3>, OutputImageType >
{

public:

  enum Channel {
    RED,
    GREEN,
    BLUE,
    ALPHA
  };

  typedef ExtractChannelFromRgbaImageFilter Self;
  typedef ProcessObject Superclass;
  typedef SmartPointer< Self > Pointer;
  typedef SmartPointer< const Self > ConstPointer;

  typedef itk::Image<itk::RGBAPixel<unsigned char>,3> InputImageType;
  typedef typename OutputImageType::PixelType OutPixelType;
  typedef typename InputImageType::PixelType InPixelType;

  itkFactorylessNewMacro(Self)
  itkCloneMacro(Self)
  itkTypeMacro( ExtractChannelFromRgbaImageFilter, ImageSource );

  void GenerateData();

  itkSetMacro(Channel, Channel)
  itkSetMacro(ReferenceImage, typename ReferenceImageType::Pointer)

protected:

  ExtractChannelFromRgbaImageFilter();
  virtual ~ExtractChannelFromRgbaImageFilter();
  Channel m_Channel;
  typename ReferenceImageType::Pointer m_ReferenceImage;
};

}

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkExtractChannelFromRgbaImageFilter.cpp"
#endif

#endif // __itkExtractChannelFromRgbaImageFilter_h__