itkChangeLabelImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkChangeLabelImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2009-03-30 11:32:14 $
  Version:   $Revision: 1.7 $

  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;

  bool operator!=( const ChangeLabel & other ) const
    {
    if (m_ChangeMap != other.m_ChangeMap)
      {
      return true;
      }
    return false;
    }
  bool operator==( const ChangeLabel & other ) const
    {
    return !(*this != other);
    }
  TOutput GetChange( const TInput & original )
    {  
    return m_ChangeMap[original]; 
    }

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

  void ClearChangeMap( )
    { 
    m_ChangeMap.clear(); 
    }

  inline TOutput operator()( const TInput & A ) const
    {
    const typename ChangeMapType::const_iterator it = m_ChangeMap.find(A);
    if ( it != m_ChangeMap.end() )
      {
      return it->second;
      }
    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;

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

  void SetChange( const InputPixelType & original, const OutputPixelType & result );

  void SetChangeMap( const ChangeMapType & changeMap );

  void ClearChangeMap( );

#ifdef ITK_USE_CONCEPT_CHECKING

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

#endif

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