itkSinglePhaseLevelSetSegmentationModule.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    itkSinglePhaseLevelSetSegmentationModule.h
  Language:  C++
  Date:      $Date$
  Version:   $Revision$

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/
#ifndef itkSinglePhaseLevelSetSegmentationModule_h
#define itkSinglePhaseLevelSetSegmentationModule_h

#include "itkSegmentationModule.h"
#include "itkImageSpatialObject.h"

namespace itk
{

template <unsigned int NDimension>
class ITK_TEMPLATE_EXPORT SinglePhaseLevelSetSegmentationModule : public SegmentationModule<NDimension>
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(SinglePhaseLevelSetSegmentationModule);

  using Self = SinglePhaseLevelSetSegmentationModule;
  using Superclass = SegmentationModule<NDimension>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkNewMacro(Self);

  itkTypeMacro(SinglePhaseLevelSetSegmentationModule, SegmentationModule);

  static constexpr unsigned int Dimension = NDimension;

  using SpatialObjectType = typename Superclass::SpatialObjectType;
  using SpatialObjectPointer = typename Superclass::SpatialObjectPointer;

  using InputPixelType = float;
  using FeaturePixelType = float;
  using OutputPixelType = float;
  using InputImageType = Image< InputPixelType, NDimension >;
  using FeatureImageType = Image< FeaturePixelType, NDimension >;
  using OutputImageType = Image< OutputPixelType, NDimension >;

  using InputSpatialObjectType = ImageSpatialObject< NDimension, InputPixelType >;
  using FeatureSpatialObjectType = ImageSpatialObject< NDimension, FeaturePixelType >;
  using OutputSpatialObjectType = ImageSpatialObject< NDimension, OutputPixelType >;

  itkSetMacro( PropagationScaling, double );
  itkGetMacro( PropagationScaling, double );

  itkSetMacro( CurvatureScaling, double );
  itkGetMacro( CurvatureScaling, double );

  itkSetMacro( AdvectionScaling, double );
  itkGetMacro( AdvectionScaling, double );

  itkSetMacro( MaximumRMSError, double );
  itkGetMacro( MaximumRMSError, double );

  itkSetMacro( MaximumNumberOfIterations, unsigned int );
  itkGetMacro( MaximumNumberOfIterations, unsigned int );

  itkSetMacro( InvertOutputIntensities, bool );
  itkGetMacro( InvertOutputIntensities, bool );
  itkBooleanMacro( InvertOutputIntensities );

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

  void  GenerateData () override;

  void PackOutputImageInOutputSpatialObject( OutputImageType * outputImage );

  const InputImageType * GetInternalInputImage() const;

  const FeatureImageType * GetInternalFeatureImage() const;

private:
  double        m_PropagationScaling;
  double        m_CurvatureScaling;
  double        m_AdvectionScaling;

  unsigned int  m_MaximumNumberOfIterations;
  double        m_MaximumRMSError;

  bool          m_InvertOutputIntensities;

  using ImageConstPointer = typename InputImageType::ConstPointer;
  mutable ImageConstPointer m_ZeroSetInputImage;
};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
# include "itkSinglePhaseLevelSetSegmentationModule.hxx"
#endif

#endif