itkCoherenceEnhancingDiffusionImageFilter.h

Go to the documentation of this file.

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  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
 *
 *         http://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.
 *
 *=========================================================================*/
//
//  Created by Jean-Marie Mirebeau on 06/03/2014.
//
//

#ifndef itkCoherenceEnhancingDiffusionImageFilter_h
#define itkCoherenceEnhancingDiffusionImageFilter_h

#include "itkAnisotropicDiffusionLBRImageFilter.h"


namespace itk
{
template< typename TImage, typename TScalar = typename NumericTraits< typename TImage::PixelType >::RealType >
class CoherenceEnhancingDiffusionImageFilter:
  public AnisotropicDiffusionLBRImageFilter< TImage, TScalar >
{
public:
  using Self = CoherenceEnhancingDiffusionImageFilter;
  using Superclass = AnisotropicDiffusionLBRImageFilter< TImage, TScalar >;
  using Pointer = SmartPointer< Self >;
  using ConstPointer = SmartPointer< const Self >;

  itkNewMacro(Self);

  itkTypeMacro(CoherenceEnhancingDiffusionImageFilter, Superclass);

  static const unsigned int Dimension = Superclass::Dimension;

  using EigenValuesArrayType = typename Superclass::EigenValuesArrayType;
  EigenValuesArrayType EigenValuesTransform(const EigenValuesArrayType &) const override;

  using ScalarType = typename Superclass::ScalarType;
  itkSetMacro(Exponent, ScalarType);
  itkSetMacro(Lambda, ScalarType);
  itkSetMacro(Alpha, ScalarType);

  itkGetMacro(Exponent, ScalarType);
  itkGetMacro(Lambda, ScalarType);
  itkGetMacro(Alpha, ScalarType);

  enum EnhancementType {CED, cCED, EED, cEED, Isotropic};
  itkSetEnumMacro(Enhancement, EnhancementType);
  itkGetEnumMacro(Enhancement, EnhancementType);

protected:
  ScalarType      m_Lambda;
  ScalarType      m_Exponent;
  ScalarType      m_Alpha;
  EnhancementType m_Enhancement;

  ScalarType g_CED(ScalarType s) const {return s<=0 ? m_Alpha : m_Alpha + (1-m_Alpha)*exp(-pow(m_Lambda/s,m_Exponent));}
  ScalarType g_EED(ScalarType s) const {return s<=0 ? 1 : 1 - (1-m_Alpha)*exp(-pow(m_Lambda/s,m_Exponent));}

  CoherenceEnhancingDiffusionImageFilter();
};

} // end namespace itk

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

#endif