itkLevelSetCurvatureFunction.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkLevelSetCurvatureFunction.h,v $
  Language:  C++
  Date:      $Date: 2002/09/11 19:57:13 $
  Version:   $Revision: 1.22 $

  Copyright (c) 2002 Insight 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 _itkLevelSetCurvatureFunction_h
#define _itkLevelSetCurvatureFunction_h

#include "itkImageFunction.h"

#include "vnl/vnl_vector_fixed.h"
#include "vnl/vnl_matrix_fixed.h"

namespace itk
{

template <class TInputImage, class TCoordRep = float >
class ITK_EXPORT LevelSetCurvatureFunction :
  public ImageFunction<TInputImage,double,TCoordRep>
{
public:
  typedef LevelSetCurvatureFunction Self;
  typedef ImageFunction<TInputImage,double,TCoordRep> Superclass;
  typedef SmartPointer<Self> Pointer;
  typedef SmartPointer<const Self>  ConstPointer;
  
  itkTypeMacro(LevelSetCurvatureFunction, ImageFunction);

  itkNewMacro(Self);

  typedef TInputImage InputImageType;

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

  typedef typename Superclass::IndexType IndexType;

  typedef typename Superclass::ContinuousIndexType ContinuousIndexType;

  typedef typename Superclass::PointType PointType;

  virtual void SetInputImage( const InputImageType * ptr );

  void SetEpsilonMagnitude( double value )
    { m_EpsilonMagnitude = value; }

  double GetEpsilonMagnitude() const
    { return m_EpsilonMagnitude; }

  virtual double EvaluateAtIndex( const IndexType& index ) const;

  virtual double Evaluate( const PointType& point ) const
    { 
      IndexType index;
      this->ConvertPointToNearestIndex( point, index );
      return this->EvaluateAtIndex( index ); 
    }

  virtual double EvaluateAtContinuousIndex( 
    const ContinuousIndexType& cindex ) const
    { 
      IndexType index;
      this->ConvertContinuousIndexToNearestIndex( cindex, index );
      return this->EvaluateAtIndex( index ) ; 
    }
  
  double GetCurvature() const
    { return m_Curvature; }

  double GetMagnitude() const
    { return m_Magnitude; }

protected:
  LevelSetCurvatureFunction(){}
  ~LevelSetCurvatureFunction(){}
  void PrintSelf(std::ostream& os, Indent indent) const;

private:
  LevelSetCurvatureFunction( const Self& ); //purposely not implemented
  void operator=( const Self& ); //purposely not implemented

  signed long                                 m_ImageSize[ImageDimension];
  bool                                        m_ImageSizeOK;

  mutable double                              m_Curvature;
  mutable double                              m_Magnitude;
  double                                      m_EpsilonMagnitude;
  mutable bool                                m_BorderPixel;

  mutable vnl_vector_fixed<double,itkGetStaticConstMacro(ImageDimension)>     
                                              m_FirstDerivative;
  mutable vnl_matrix_fixed<double,itkGetStaticConstMacro(ImageDimension),
                           itkGetStaticConstMacro(ImageDimension)>
                                              m_SecondDerivative;

  mutable IndexType                           m_NeighIndex;
  mutable IndexType                           m_RightIndex;
  mutable IndexType                           m_LeftIndex;
  mutable double                              m_CenterValue;
  mutable double                              m_DiffValue;

  mutable vnl_matrix_fixed<unsigned int,itkGetStaticConstMacro(ImageDimension),
                           itkGetStaticConstMacro(ImageDimension)>
                                                            m_Variable;

  void CalculateDerivatives( const IndexType& index ) const;
  void CalculateCurvature() const;
  void CalculateCurvature2D() const;
  void CalculateCurvature3D() const;

};

} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkLevelSetCurvatureFunction.txx"
#endif

#endif

---