itkKLMRegionGrowImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright NumFOCUS
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         https://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/
#ifndef itkKLMRegionGrowImageFilter_h
#define itkKLMRegionGrowImageFilter_h
 
#include "itkImage.h"
#include "itkRegionGrowImageFilter.h"
#include "itkKLMSegmentationBorder.h"
#include "itkImageRegionIterator.h"
#include "itkConceptChecking.h"
#include <algorithm>
#include <functional>
 
namespace itk
{
 
template <typename TInputImage, typename TOutputImage>
class ITK_TEMPLATE_EXPORT KLMRegionGrowImageFilter : public RegionGrowImageFilter<TInputImage, TOutputImage>
{
public:
  ITK_DISALLOW_COPY_AND_MOVE(KLMRegionGrowImageFilter);

  using Self = KLMRegionGrowImageFilter;
  using Superclass = RegionGrowImageFilter<TInputImage, TOutputImage>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkNewMacro(Self);

  itkOverrideGetNameOfClassMacro(KLMRegionGrowImageFilter);

  using InputImageType = TInputImage;
  using InputImagePointer = typename TInputImage::Pointer;
  using InputImageConstPointer = typename TInputImage::ConstPointer;

  using InputImagePixelType = typename TInputImage::PixelType;

  using InputImageVectorType = typename TInputImage::PixelType::VectorType;

  static constexpr unsigned int InputImageVectorDimension = InputImagePixelType::Dimension;

  using InputImageIndexType = typename TInputImage::IndexType;

  using InputImageIterator = ImageRegionIterator<TInputImage>;
  using InputImageConstIterator = ImageRegionConstIterator<TInputImage>;

  using InputRegionType = typename TInputImage::RegionType;

  using typename Superclass::GridSizeType;

  using OutputImageType = TOutputImage;
  using OutputImagePointer = typename TOutputImage::Pointer;

  static constexpr unsigned int InputImageDimension = TInputImage::ImageDimension;

  static constexpr unsigned int OutputImageDimension = TOutputImage::ImageDimension;

  using OutputImagePixelType = typename TOutputImage::PixelType;

  using OutputImageVectorType = typename TOutputImage::PixelType::VectorType;

  static constexpr unsigned int OutputImageVectorDimension = OutputImagePixelType::Dimension;

  using OutputImageIndexType = typename TOutputImage::IndexType;

  using OutputImageIterator = ImageRegionIterator<TOutputImage>;

  using RegionLabelType = typename KLMSegmentationRegion::RegionLabelType;

  static constexpr RegionLabelType LabelImageDimension = InputImageDimension;

  using LabelImageType = Image<RegionLabelType, Self::LabelImageDimension>;

  using LabelImagePointer = typename LabelImageType::Pointer;

  using LabelImagePixelType = typename LabelImageType::PixelType;

  using LabelImageIndexType = typename LabelImageType::IndexType;

  using LabelImageIterator = ImageRegionIterator<LabelImageType>;

  using MeanRegionIntensityType = vnl_vector<double>;

  using BorderType = KLMSegmentationBorder;

  using KLMSegmentationBorderArrayPtr = KLMDynamicBorderArray<BorderType>;

  itkSetMacro(MaximumLambda, double);
  itkGetConstReferenceMacro(MaximumLambda, double);

  itkSetMacro(NumberOfRegions, unsigned int);
  itkGetConstReferenceMacro(NumberOfRegions, unsigned int);

  LabelImagePointer
  GetLabelledImage();

  void
  PrintAlgorithmRegionStats();

  void
  PrintAlgorithmBorderStats();
 
  itkConceptMacro(InputHasNumericTraitsCheck, (Concept::HasNumericTraits<typename InputImagePixelType::ValueType>));
  itkConceptMacro(SameDimension, (Concept::SameDimension<Self::InputImageDimension, Self::OutputImageDimension>));
#if defined(THIS_CONCEPT_FAILS_ON_GCC)
  itkConceptMacro(SameVectorDimension,
                  (Concept::SameDimension<Self::InputImageVectorDimension, Self::OutputImageVectorDimension>));
#endif // end THIS_CONCEPT_FAILS_ON_GCC
 
protected:
  KLMRegionGrowImageFilter();
  ~KLMRegionGrowImageFilter() override = default;
  void
  PrintSelf(std::ostream & os, Indent indent) const override;

  void
  GenerateData() override;

  void
  GenerateInputRequestedRegion() override;

  void
  EnlargeOutputRequestedRegion(DataObject *) override;

  void
  ApplyRegionGrowImageFilter() override;

  void
  MergeRegions() override;

  virtual void
  GenerateOutputImage();

  void
  ApplyKLM();

  void
  InitializeKLM();

  LabelImagePointer
  GenerateLabelledImage(LabelImageType * labelImagePtr);

  virtual void
  InitializeRegionParameters(InputRegionType region);

  virtual void
  ResolveRegions();
 
private:
  using InputImageSizeType = typename TInputImage::SizeType;
  using KLMSegmentationRegionPtr = typename KLMSegmentationRegion::Pointer;
  using KLMSegmentationBorderPtr = typename KLMSegmentationBorder::Pointer;
 
  double       m_MaximumLambda{ 1000 };
  unsigned int m_NumberOfRegions{ 0 };

 
  double       m_InternalLambda{ 0 };
  unsigned int m_InitialNumberOfRegions{ 0 };
  double       m_TotalBorderLength{ 0.0 };
 
  std::vector<KLMSegmentationRegionPtr>      m_RegionsPointer{};
  std::vector<KLMSegmentationBorderPtr>      m_BordersPointer{};
  std::vector<KLMSegmentationBorderArrayPtr> m_BordersDynamicPointer{};
  KLMSegmentationBorderArrayPtr *            m_BorderCandidate{ nullptr };
 
  MeanRegionIntensityType m_InitialRegionMean{};
  double                  m_InitialRegionArea{ 0 };
}; // class KLMRegionGrowImageFilter
} // namespace itk
 
#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkKLMRegionGrowImageFilter.hxx"
#endif
 
#endif