itkCoherenceEnhancingDiffusionFilter.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 itkCoherenceEnhancingDiffusionFilter_h
#define itkCoherenceEnhancingDiffusionFilter_h

#include "itkAnisotropicDiffusionLBRImageFilter.h"


namespace itk
{
template< typename TImage, typename TScalar=typename TImage::PixelType >
class CoherenceEnhancingDiffusionFilter:
  public AnisotropicDiffusionLBRImageFilter< TImage, TScalar >
{
public:
  typedef CoherenceEnhancingDiffusionFilter                     Self;
  typedef AnisotropicDiffusionLBRImageFilter< TImage, TScalar > Superclass;
  typedef SmartPointer< Self >                                  Pointer;
  typedef SmartPointer< const Self >                            ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(CoherenceEnhancingDiffusionFilter, Superclass);

  static const unsigned int Dimension = Superclass::Dimension;

  typedef typename Superclass::EigenValuesArrayType EigenValuesArrayType;
  virtual EigenValuesArrayType EigenValuesTransform(const EigenValuesArrayType &) const;

  typedef typename Superclass::ScalarType 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));}

  CoherenceEnhancingDiffusionFilter();
};

} // end namespace itk

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

#endif