itkUCLLabelVotingImageFilter.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 itkUCLLabelVotingImageFilter_h
#define itkUCLLabelVotingImageFilter_h

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

namespace itk
{

template <typename TInputImage, typename TOutputImage>
class ITK_EXPORT UCLLabelVotingImageFilter :
    public ImageToImageFilter< TInputImage, TOutputImage >
{
public:
  typedef UCLLabelVotingImageFilter Self;
  typedef ImageToImageFilter< TInputImage, TOutputImage > Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(UCLLabelVotingImageFilter, ImageToImageFilter);
  
  typedef typename TOutputImage::PixelType OutputPixelType;
  typedef typename TInputImage::PixelType  InputPixelType;
  
  itkStaticConstMacro(InputImageDimension, unsigned int,
                      TInputImage::ImageDimension );
  itkStaticConstMacro(ImageDimension, unsigned int,
                      TOutputImage::ImageDimension);
  
  typedef TInputImage                           InputImageType;
  typedef TOutputImage                          OutputImageType;
  typedef typename InputImageType::ConstPointer InputImagePointer;
  typedef typename OutputImageType::Pointer     OutputImagePointer;
  
  typedef typename Superclass::OutputImageRegionType OutputImageRegionType;
  
  virtual void SetLabelForUndecidedPixels( const OutputPixelType l )
  {
    this->m_LabelForUndecidedPixels = l;
    this->m_HasLabelForUndecidedPixels = true;
    this->Modified();
  }
  
  virtual OutputPixelType GetLabelForUndecidedPixels() const
  {
    return this->m_LabelForUndecidedPixels;
  }
  
  virtual void UnsetLabelForUndecidedPixels()
  {
    if ( this->m_HasLabelForUndecidedPixels )
    {
      this->m_HasLabelForUndecidedPixels = false;
      this->Modified();
    }
  }
    
protected:   
  UCLLabelVotingImageFilter() { this->m_HasLabelForUndecidedPixels = false; srand(time(NULL)); }
  virtual ~UCLLabelVotingImageFilter() {}  
  virtual InputPixelType ComputeMaximumInputValue(); 
  virtual void BeforeThreadedGenerateData(); 
  virtual void ThreadedGenerateData(const OutputImageRegionType &outputRegionForThread, ThreadIdType threadId);
  
protected:
  OutputPixelType m_LabelForUndecidedPixels;
  bool m_HasLabelForUndecidedPixels;
  InputPixelType m_TotalLabelCount;
  
private:
  UCLLabelVotingImageFilter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
  
}; 
    
  
}

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


#endif