itkMIDASMorphologicalSegmentorLargestConnectedComponentImageFilter.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 itkMIDASMorphologicalSegmentorLargestConnectedComponentImageFilter_h
#define itkMIDASMorphologicalSegmentorLargestConnectedComponentImageFilter_h

#include <vector>
#include <stack>
#include <itkImage.h>
#include <itkImageRegionConstIterator.h>
#include <itkImageRegionIterator.h>
#include <itkImageToImageFilter.h>

namespace itk {
  template <class TInputImageType, class TOutputImageType>
  class MIDASMorphologicalSegmentorLargestConnectedComponentImageFilter : public ImageToImageFilter<TInputImageType, TOutputImageType> {

  public:
    typedef TInputImageType InputImageType;
    typedef TOutputImageType OutputImageType;
    typedef typename InputImageType::IndexType IndexType;
    typedef typename OutputImageType::PixelType OutputImagePixelType;
    typedef typename InputImageType::PixelType InputImagePixelType;
    typedef typename InputImageType::RegionType InputImageRegionType;
    typedef MIDASMorphologicalSegmentorLargestConnectedComponentImageFilter Self;
    typedef ImageToImageFilter<InputImageType, OutputImageType> SuperClass;
    typedef SmartPointer<Self> Pointer;
    typedef SmartPointer<const Self> ConstPointer;
    itkNewMacro(Self);

    itkSetMacro(InputBackgroundValue, InputImagePixelType);
    itkGetConstMacro(InputBackgroundValue, InputImagePixelType);

    itkSetMacro(OutputBackgroundValue, OutputImagePixelType);
    itkGetConstMacro(OutputBackgroundValue, OutputImagePixelType);

    itkSetMacro(OutputForegroundValue, OutputImagePixelType);
    itkGetConstMacro(OutputForegroundValue, OutputImagePixelType);

    itkGetConstMacro(Capacity, unsigned int);
    void SetCapacity(unsigned capacity)
    {
      m_Capacity = capacity;
    }

    itkGetConstMacro(NumberOfConnectedComponents, unsigned int);

  protected:
    MIDASMorphologicalSegmentorLargestConnectedComponentImageFilter(void);
    virtual ~MIDASMorphologicalSegmentorLargestConnectedComponentImageFilter(void) {}

    void _ProcessRegion(std::vector<unsigned int> &r_regionIndices, const unsigned int &startIndex);

    void _SetComponentPixels(const std::vector<unsigned int> &regionIndices);

    virtual void BeforeThreadedGenerateData();

    virtual void ThreadedGenerateData(const InputImageRegionType &outputRegionForThread, ThreadIdType ThreadID);

    virtual void AfterThreadedGenerateData();

  private:
    OutputImagePixelType           m_OutputBackgroundValue;
    OutputImagePixelType           m_OutputForegroundValue;
    InputImagePixelType            m_InputBackgroundValue;
    unsigned int                   m_Capacity;
    std::vector<unsigned long int> m_NumberOfLabelledPixelsPerThread;
    unsigned int                   m_NumberOfConnectedComponents;
  };

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkMIDASMorphologicalSegmentorLargestConnectedComponentImageFilter.txx"
#endif
}

#endif /* ITKMIDASMORPHOLOGICALSEGMENTORLARGESTCONNECTEDCOMPONENTIMAGEFILTER_H_ */