itkChangeLabelImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkChangeLabelImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2005/07/27 19:30:11 $
  Version:   $Revision: 1.1 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even 
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef __itkChangeLabelImageFilter_h
#define __itkChangeLabelImageFilter_h

#include "itkUnaryFunctorImageFilter.h"
#include "itkConceptChecking.h"
#include "itkSimpleDataObjectDecorator.h"

namespace itk
{
  
#include <map>

namespace Functor {  
  
template< class TInput, class TOutput>
class ChangeLabel
{
public:
  ChangeLabel() {};
  ~ChangeLabel() {};

  typedef std::map<TInput, TOutput> ChangeMapType;

  TInput GetChange( const TInput & original )
  { 
    return m_ChangeMap[original]; 
  }

  void SetChange( const TInput & original, const TOutput & result )
  { 
    m_ChangeMap[original] = result; 
  }
  
  void SetChangeMap( ChangeMapType & changeMap )
  { 
    m_ChangeMap = changeMap; 
  }

  inline TOutput operator()( const TInput & A )
  {
    if ( m_ChangeMap.find(A) != m_ChangeMap.end() )
      {
      return m_ChangeMap[A];
      }
    return A;
  }

private:

  ChangeMapType m_ChangeMap;

};
}

template <class TInputImage, class TOutputImage>
class ITK_EXPORT ChangeLabelImageFilter :
    public
UnaryFunctorImageFilter<TInputImage,TOutputImage, 
                        Functor::ChangeLabel< 
  typename TInputImage::PixelType, 
  typename TOutputImage::PixelType> >
{
public:
  typedef ChangeLabelImageFilter  Self;
  typedef UnaryFunctorImageFilter<TInputImage,TOutputImage, 
    Functor::ChangeLabel< 
    typename TInputImage::PixelType, 
    typename TOutputImage::PixelType>   
  >  Superclass;
  typedef SmartPointer<Self>   Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ChangeLabelImageFilter, UnaryFunctorImageFilter);

  typedef typename TInputImage::PixelType  InputPixelType;
  typedef typename TOutputImage::PixelType OutputPixelType;

  itkConceptMacro(PixelTypeComparable, (Concept::Comparable<InputPixelType>));

  typedef std::map<InputPixelType, OutputPixelType> ChangeMapType;

  void SetChange( const InputPixelType & original, const OutputPixelType & result );
  
  void SetChangeMap( const ChangeMapType & changeMap );
  
protected:
  ChangeLabelImageFilter();
  virtual ~ChangeLabelImageFilter() {}
  void PrintSelf(std::ostream& os, Indent indent) const;

private:
  ChangeLabelImageFilter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkChangeLabelImageFilter.txx"
#endif

#endif

---