itkCannyEdgesDistanceAdvectionFieldFeatureGenerator.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    itkCannyEdgesDistanceAdvectionFieldFeatureGenerator.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 itkCannyEdgesDistanceAdvectionFieldFeatureGenerator_h
#define itkCannyEdgesDistanceAdvectionFieldFeatureGenerator_h

#include "itkFeatureGenerator.h"
#include "itkImage.h"
#include "itkImageSpatialObject.h"
#include "itkCastImageFilter.h"
#include "itkCannyEdgeDetectionRecursiveGaussianImageFilter.h"
#include "itkSignedMaurerDistanceMapImageFilter.h"
#include "itkGradientImageFilter.h"
#include "itkMultiplyImageFilter.h"

namespace itk
{

template <unsigned int NDimension>
class ITK_TEMPLATE_EXPORT CannyEdgesDistanceAdvectionFieldFeatureGenerator : public FeatureGenerator<NDimension>
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(CannyEdgesDistanceAdvectionFieldFeatureGenerator);

  using Self = CannyEdgesDistanceAdvectionFieldFeatureGenerator;
  using Superclass = FeatureGenerator<NDimension>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  using InternalPixelType = float;

  itkNewMacro(Self);

  itkTypeMacro(CannyEdgesDistanceAdvectionFieldFeatureGenerator, FeatureGenerator);

  static constexpr unsigned int Dimension = NDimension;

  using InputPixelType = signed short;
  using InputImageType = Image< InputPixelType, Dimension >;
  using InputImageSpatialObjectType = ImageSpatialObject< NDimension, InputPixelType >;
  using InputImageSpatialObjectPointer = typename InputImageSpatialObjectType::Pointer;
  using SpatialObjectType = typename Superclass::SpatialObjectType;

  using ProcessObject::SetInput;
  void SetInput( const SpatialObjectType * input );
  const SpatialObjectType * GetInput() const;

  const SpatialObjectType * GetFeature() const;

  itkSetMacro( Sigma, double );
  itkGetMacro( Sigma, double );
  itkSetMacro( UpperThreshold, double );
  itkGetMacro( UpperThreshold, double );
  itkSetMacro( LowerThreshold, double );
  itkGetMacro( LowerThreshold, double );

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

  void  GenerateData () override;

private:
  using InternalImageType = Image< InternalPixelType, Dimension >;

  using CastFilterType = CastImageFilter<
    InputImageType, InternalImageType >;
  using CastFilterPointer = typename CastFilterType::Pointer;
  using CannyEdgeFilterType = CannyEdgeDetectionRecursiveGaussianImageFilter<
    InternalImageType, InternalImageType >;
  using CannyEdgeFilterPointer = typename CannyEdgeFilterType::Pointer;

  using DistanceMapFilterType = SignedMaurerDistanceMapImageFilter<
    InternalImageType, InternalImageType >;
  using DistanceMapFilterPointer = typename DistanceMapFilterType::Pointer;

  using GradientFilterType = GradientImageFilter< InternalImageType,
          InternalPixelType, InternalPixelType >;
  using GradientFilterPointer = typename GradientFilterType::Pointer;
  using CovariantVectorImageType = typename GradientFilterType::OutputImageType;

  using OutputPixelType = typename CovariantVectorImageType::PixelType;
  using OutputImageSpatialObjectType = ImageSpatialObject< NDimension, OutputPixelType >;
  using OutputImageType = Image< OutputPixelType, Dimension >;

  using MultiplyFilterType = MultiplyImageFilter<
    CovariantVectorImageType, InternalImageType,
                     CovariantVectorImageType >;
  using MultiplyFilterPointer = typename MultiplyFilterType::Pointer;

  CastFilterPointer                                   m_CastFilter;
  DistanceMapFilterPointer                            m_DistanceMapFilter;
  CannyEdgeFilterPointer                              m_CannyFilter;
  GradientFilterPointer                               m_GradientFilter;
  MultiplyFilterPointer                               m_MultiplyFilter;

  double                                              m_UpperThreshold;
  double                                              m_LowerThreshold;
  double                                              m_Sigma;

};

} // end namespace itk

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

#endif