itkDiscreteHessianGaussianImageFunction.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkDiscreteHessianGaussianImageFunction.h,v $
  Language:  C++
  Date:      $Date: 2009-04-23 03:43:41 $
  Version:   $Revision: 1.8 $

  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 __itkDiscreteHessianGaussianImageFunction_h
#define __itkDiscreteHessianGaussianImageFunction_h

#include "itkNeighborhoodOperatorImageFunction.h"
#include "itkImageFunction.h"
#include "itkGaussianOperator.h"
#include "itkGaussianDerivativeOperator.h"
#include "itkSymmetricSecondRankTensor.h"

namespace itk
{

template <class TInputImage,class TOutput=double>
class ITK_EXPORT DiscreteHessianGaussianImageFunction :
  public ImageFunction< TInputImage,
    SymmetricSecondRankTensor<TOutput,::itk::GetImageDimension<TInputImage>::ImageDimension>,
    TOutput >
{
public:

  typedef DiscreteHessianGaussianImageFunction   Self;

  typedef ImageFunction<TInputImage,
    SymmetricSecondRankTensor<TOutput,::itk::GetImageDimension<TInputImage>::ImageDimension>,
    TOutput > Superclass;

  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro( DiscreteHessianGaussianImageFunction, ImageFunction );

  typedef typename Superclass::InputImageType       InputImageType;
  typedef typename Superclass::InputPixelType       InputPixelType;
  typedef typename Superclass::IndexType            IndexType;
  typedef typename Superclass::ContinuousIndexType  ContinuousIndexType;
  typedef typename Superclass::PointType            PointType;

  itkStaticConstMacro(ImageDimension2, unsigned int,
                      InputImageType::ImageDimension);

  typedef SymmetricSecondRankTensor<TOutput,
    ::itk::GetImageDimension<TInputImage>::ImageDimension>  TensorType;
  typedef typename Superclass::OutputType                   OutputType;

  typedef FixedArray<double,itkGetStaticConstMacro(ImageDimension2)>  VarianceArrayType;

  typedef itk::GaussianDerivativeOperator<TOutput,
    itkGetStaticConstMacro(ImageDimension2)>            GaussianDerivativeOperatorType;

  typedef FixedArray<GaussianDerivativeOperatorType,
    3*itkGetStaticConstMacro(ImageDimension2)>          GaussianDerivativeOperatorArrayType;

  typedef Neighborhood<TOutput,itkGetStaticConstMacro(ImageDimension2)>  KernelType;

  typedef FixedArray<KernelType,itkGetStaticConstMacro(ImageDimension2)*
    (itkGetStaticConstMacro(ImageDimension2)+1)/2>      KernelArrayType;

  typedef NeighborhoodOperatorImageFunction
    <InputImageType, TOutput>                           OperatorImageFunctionType;
  typedef typename OperatorImageFunctionType::Pointer   OperatorImageFunctionPointer;

  enum InterpolationModeType { NearestNeighbourInterpolation, LinearInterpolation };

public:

  virtual OutputType Evaluate(const PointType& point) const;

  virtual OutputType EvaluateAtIndex( const IndexType & index ) const;

  virtual OutputType EvaluateAtContinuousIndex(
    const ContinuousIndexType & index ) const;

  itkSetMacro( Variance, VarianceArrayType );
  itkGetConstMacro( Variance, const VarianceArrayType );
  itkSetVectorMacro( Variance, double, VarianceArrayType::Length );

  virtual void SetVariance( double variance )
    {
    m_Variance.Fill( variance );
    this->Modified();
    }

  void SetSigma( const double sigma )
    {
    SetVariance( sigma * sigma );
    }

  itkSetClampMacro( MaximumError, double, 0.00001, 0.99999 );
  itkGetConstMacro( MaximumError, double );

  itkSetMacro( NormalizeAcrossScale, bool );
  itkGetConstMacro( NormalizeAcrossScale, bool );
  itkBooleanMacro( NormalizeAcrossScale );

  itkSetMacro( UseImageSpacing, bool );
  itkGetConstMacro( UseImageSpacing, bool );
  itkBooleanMacro( UseImageSpacing );

  itkSetMacro( MaximumKernelWidth, unsigned int );
  itkGetConstMacro( MaximumKernelWidth, unsigned int );

  itkSetMacro( InterpolationMode, InterpolationModeType );
  itkGetConstMacro( InterpolationMode, InterpolationModeType );

  virtual void SetInputImage( const InputImageType * ptr );

  virtual void Initialize( ) { RecomputeGaussianKernel(); }

protected:

  DiscreteHessianGaussianImageFunction();
  DiscreteHessianGaussianImageFunction( const Self& ){};

  ~DiscreteHessianGaussianImageFunction(){};

  void operator=( const Self& ){};
  void PrintSelf(std::ostream& os, Indent indent) const;

  void RecomputeGaussianKernel();

private:

  VarianceArrayType m_Variance;

  double m_MaximumError;

  unsigned int m_MaximumKernelWidth;

  mutable GaussianDerivativeOperatorArrayType m_OperatorArray;

  KernelArrayType m_KernelArray;

  OperatorImageFunctionPointer m_OperatorImageFunction;

  bool m_NormalizeAcrossScale;

  bool m_UseImageSpacing;

  InterpolationModeType m_InterpolationMode;

};

} // namespace itk

// Define instantiation macro for this template.
#define ITK_TEMPLATE_DiscreteHessianGaussianImageFunction(_, EXPORT, x, y) namespace itk { \
  _(2(class EXPORT DiscreteHessianGaussianImageFunction< ITK_TEMPLATE_2 x >)) \
  namespace Templates { typedef DiscreteHessianGaussianImageFunction< ITK_TEMPLATE_2 x > \
                                                  DiscreteHessianGaussianImageFunction##y; } \
  }

#if ITK_TEMPLATE_EXPLICIT
# include "Templates/itkDiscreteHessianGaussianImageFunction+-.h"
#endif

#if ITK_TEMPLATE_TXX
# include "itkDiscreteHessianGaussianImageFunction.txx"
#endif

#endif