itkIntelligentBinaryClosingFilter.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 __itkIntelligentBinaryClosingFilter_h
#define __itkIntelligentBinaryClosingFilter_h

#if defined(_MSC_VER)
#pragma warning(disable : 4786)
#endif

#include <itkBinaryBallStructuringElement.h>
#include <itkBinaryDilateImageFilter.h>
#include <itkBinaryErodeImageFilter.h>
#include <itkConnectedComponentImageFilter.h>
#include <itkConnectedComponentImageFilter.h>
#include <itkGrayscaleDilateImageFilter.h>
#include <itkImageRegionIterator.h>
#include <itkImageToImageFilter.h>
#include <itkRelabelComponentImageFilter.h>
#include <itkSubtractImageFilter.h>

namespace itk
{
  template <class TInputImage, class TOutputImage>
  class ITK_EXPORT IntelligentBinaryClosingFilter : public ImageToImageFilter<TInputImage, TOutputImage>
  {
  public:
    typedef IntelligentBinaryClosingFilter Self;

    typedef TInputImage InputImageType;

    itkStaticConstMacro(ImageDimension, unsigned int, TInputImage::ImageDimension);

    typedef TOutputImage OutputImageType;

    typedef ImageToImageFilter<InputImageType, OutputImageType> Superclass;

    typedef SmartPointer<Self> Pointer;

    itkTypeMacro(IntelligentBinaryClosingFilter, ImageToImageFilter);

    itkFactorylessNewMacro(Self);
    itkCloneMacro(Self);

      void GenerateData() override;

    void SetInput(const InputImageType *input) override
    {
      // processObject is not const-correct so a const_cast is needed here
      this->ProcessObject::SetNthInput(0, const_cast<InputImageType *>(input));
      m_DilateImageFilter->SetInput(input);
    }

    using Superclass::SetInput;
    void SetInput(unsigned int i, const TInputImage *image) override
    {
      if (i != 0)
      {
        itkExceptionMacro(<< "Filter has only one input.");
      }
      else
      {
        this->SetInput(image);
      }
    }

    itkGetMacro(ClosingRadius, float);
    itkSetMacro(ClosingRadius, float);

    itkGetMacro(SurfaceRatio, float);
    itkSetMacro(SurfaceRatio, float);

  protected:
    IntelligentBinaryClosingFilter();
    ~IntelligentBinaryClosingFilter() override {}
    void PrintSelf(std::ostream &os, Indent indent) const override;

  private:
    typedef typename InputImageType::PixelType InputPixelType;
    typedef BinaryBallStructuringElement<InputPixelType, ImageDimension> StructuringElementType;
    typedef BinaryErodeImageFilter<InputImageType, InputImageType, StructuringElementType> BinaryErodeImageFilterType;
    typedef BinaryDilateImageFilter<InputImageType, InputImageType, StructuringElementType> BinaryDilateImageFilterType;
    typedef SubtractImageFilter<InputImageType, InputImageType, InputImageType> SubtractImageFilterType;
    typedef ConnectedComponentImageFilter<InputImageType, OutputImageType> ConnectedComponentImageFilterType;
    typedef RelabelComponentImageFilter<OutputImageType, OutputImageType> RelabelComponentImageFilterType;
    typedef GrayscaleDilateImageFilter<OutputImageType, OutputImageType, StructuringElementType>
      DilateComponentImageFilterType;
    typedef ImageRegionIterator<InputImageType> InputIteratorType;
    typedef ImageRegionConstIterator<InputImageType> ConstInputIteratorType;
    typedef ImageRegionIterator<OutputImageType> OutputIteratorType;

    typename BinaryErodeImageFilterType::Pointer m_ErodeImageFilter;
    typename BinaryDilateImageFilterType::Pointer m_DilateImageFilter;
    typename SubtractImageFilterType::Pointer m_SubtractImageFilter;
    typename ConnectedComponentImageFilterType::Pointer m_ConnectedComponentImageFilter;
    typename RelabelComponentImageFilterType::Pointer m_RelabelComponentImageFilter;
    typename DilateComponentImageFilterType::Pointer m_BorderDetectionDilateFilter;
    // typename BinaryDilateImageFilterType::Pointer       m_BorderAdjacencyDilateFilter;
    float m_ClosingRadius;
    float m_SurfaceRatio;
  };

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkIntelligentBinaryClosingFilter.txx"

#endif

#endif // IntelligentBinaryClosingFilter_h