itkConnectedAdaptiveThresholdImageFilter.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
#ifndef itkConnectedAdaptiveThresholdImageFilter_h
#define itkConnectedAdaptiveThresholdImageFilter_h
 
#include <itkConnectedThresholdImageFilter.h>
#include <itkImage.h>
 
namespace itk
{
  template <class TInputImage, class TOutputImage>
  class ITK_EXPORT ConnectedAdaptiveThresholdImageFilter
    : public ConnectedThresholdImageFilter<TInputImage, TOutputImage>
  {
  public:
    typedef ConnectedAdaptiveThresholdImageFilter Self;
    typedef ConnectedThresholdImageFilter<TInputImage, TOutputImage> Superclass;
    typedef SmartPointer<Self> Pointer;
    typedef SmartPointer<const Self> ConstPointer;
 
    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);
 
      itkTypeMacro(ConnectedAdaptiveThresholdImageFilter, ConnectedThresholdImageFilter);
 
    typedef TInputImage InputImageType;
    typedef TOutputImage OutputImageType;
    typedef typename OutputImageType::Pointer OutputImagePointer;
    typedef typename InputImageType::IndexType IndexType;
    typedef typename InputImageType::PixelType PixelType;
 
    void SetGrowingDirectionIsUpwards(bool upwards) { m_GrowingDirectionIsUpwards = upwards; }
 
    void SetFineDetectionMode(bool fine)
    {
      m_FineDetectionMode = fine;
      m_DiscardLastPreview = false;
    }
 
    int GetSeedpointValue(void) { return m_SeedpointValue; }
 
    int GetLeakagePoint(void) { return m_DetectedLeakagePoint; }
 
    IndexType CorrectSeedPointPosition(unsigned int sizeOfVolume, int lowerTh, int upperTh);
 
    void CropMask(unsigned int croppingSize);
 
    unsigned int AdjustIteratorMask();
 
    void SetParameterForFineSegmentation(TOutputImage *iteratorMaskForFineSegmentation,
                                         unsigned int adjLowerTh,
                                         unsigned int adjUpperTh,
                                         itk::Index<3> seedPoint,
                                         bool discardLeafSegmentation);
 
    TOutputImage *GetResultImage();
 
  protected:
    ConnectedAdaptiveThresholdImageFilter();
    ~ConnectedAdaptiveThresholdImageFilter() override{};
 
    void GenerateData() override;
 
  private:
    OutputImagePointer m_OutoutImageMaskFineSegmentation;
    bool m_GrowingDirectionIsUpwards;
    PixelType m_SeedpointValue;
    PixelType m_DetectedLeakagePoint;
    PixelType m_InitValue;
    unsigned int m_AdjLowerTh;
    unsigned int m_AdjUpperTh;
    itk::Index<3> m_SeedPointIndex;
 
    /* Flag for switching between raw segmentation and fine segmentation (Bronchial tree segmentation) */
    bool m_FineDetectionMode;
 
    bool m_DiscardLastPreview;
    bool m_SegmentationCancelled;
  };
 
} // end namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#include <itkConnectedAdaptiveThresholdImageFilter.tpp>
#endif
 
#endif