itkDescoteauxSheetnessFeatureGenerator.h

Go to the documentation of this file.

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

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

#include "itkFeatureGenerator.h"
#include "itkImage.h"
#include "itkImageSpatialObject.h"
#include "itkHessianRecursiveGaussianImageFilter.h"
#include "itkSymmetricSecondRankTensor.h"
#include "itkSymmetricEigenAnalysisImageFilter.h"
#include "itkDescoteauxSheetnessImageFilter.h"
#include "itkRescaleIntensityImageFilter.h"

namespace itk
{

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

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

  itkNewMacro(Self);

  itkTypeMacro(DescoteauxSheetnessFeatureGenerator, 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( SheetnessNormalization, double );
  itkGetMacro( SheetnessNormalization, double );

  itkSetMacro( BloobinessNormalization, double );
  itkGetMacro( BloobinessNormalization, double );

  itkSetMacro( NoiseNormalization, double );
  itkGetMacro( NoiseNormalization, double );

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

protected:
  DescoteauxSheetnessFeatureGenerator();
  ~DescoteauxSheetnessFeatureGenerator() 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 HessianFilterType = HessianRecursiveGaussianImageFilter< InputImageType >;
  using HessianImageType = typename HessianFilterType::OutputImageType;
  using HessianPixelType = typename HessianImageType::PixelType;

  using EigenValueArrayType = FixedArray< double, HessianPixelType::Dimension >;
  using EigenValueImageType = Image< EigenValueArrayType, Dimension >;

  using EigenAnalysisFilterType = SymmetricEigenAnalysisImageFilter< HessianImageType, EigenValueImageType >;

  using SheetnessFilterType = DescoteauxSheetnessImageFilter< EigenValueImageType, OutputImageType >;

  using RescaleFilterType = RescaleIntensityImageFilter< OutputImageType, OutputImageType >;

  typename HessianFilterType::Pointer             m_HessianFilter;
  typename EigenAnalysisFilterType::Pointer       m_EigenAnalysisFilter;
  typename SheetnessFilterType::Pointer           m_SheetnessFilter;
  typename RescaleFilterType::Pointer             m_RescaleFilter;

  double      m_Sigma;
  double      m_SheetnessNormalization;
  double      m_BloobinessNormalization;
  double      m_NoiseNormalization;
  bool        m_DetectBrightSheets;
};

} // end namespace itk

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

#endif