itkSatoVesselnessSigmoidFeatureGenerator.h

Go to the documentation of this file.

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

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

#include "itkSatoVesselnessFeatureGenerator.h"
#include "itkSigmoidImageFilter.h"

namespace itk
{

template <unsigned int NDimension>
class ITK_TEMPLATE_EXPORT SatoVesselnessSigmoidFeatureGenerator : public SatoVesselnessFeatureGenerator<NDimension>
{
public:
  ITK_DISALLOW_COPY_AND_ASSIGN(SatoVesselnessSigmoidFeatureGenerator);

  using Self = SatoVesselnessSigmoidFeatureGenerator;
  using Superclass = SatoVesselnessFeatureGenerator<NDimension>;
  using Pointer = SmartPointer<Self>;
  using ConstPointer = SmartPointer<const Self>;

  itkNewMacro(Self);

  itkTypeMacro(SatoVesselnessSigmoidFeatureGenerator, SatoVesselnessFeatureGenerator);

  static constexpr unsigned int Dimension = NDimension;

  itkSetMacro( SigmoidAlpha, double );
  itkGetMacro( SigmoidAlpha, double );

  itkSetMacro( SigmoidBeta, double );
  itkGetMacro( SigmoidBeta, double );

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

  void  GenerateData () override;

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

  using OutputPixelType = InternalPixelType;
  using OutputImageType = InternalImageType;

  using OutputImageSpatialObjectType = ImageSpatialObject< NDimension, OutputPixelType >;

  using SigmoidFilterType = SigmoidImageFilter< InternalImageType, InternalImageType >;

  typename SigmoidFilterType::Pointer                 m_SigmoidFilter;

  double      m_SigmoidAlpha;
  double      m_SigmoidBeta;
};

} // end namespace itk

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

#endif