itkSegmentationLevelSetFunction.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkSegmentationLevelSetFunction.h,v $
  Language:  C++
  Date:      $Date: 2009-03-03 15:09:09 $
  Version:   $Revision: 1.19 $

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

#include "itkLevelSetFunction.h"
#include "itkLinearInterpolateImageFunction.h"
#include "itkVectorLinearInterpolateImageFunction.h"
#include "itkVectorCastImageFilter.h"

namespace itk {

template <class TImageType, class TFeatureImageType = TImageType>
class ITK_EXPORT SegmentationLevelSetFunction
  : public LevelSetFunction<TImageType>
{
public:
  typedef SegmentationLevelSetFunction Self;
  typedef LevelSetFunction<TImageType> Superclass;
  typedef SmartPointer<Self>           Pointer;
  typedef SmartPointer<const Self>     ConstPointer;

  itkTypeMacro( SegmentationLevelSetFunction, LevelSetFunction );

  typedef typename Superclass::ImageType        ImageType;
  typedef typename Superclass::RadiusType       RadiusType;
  typedef typename Superclass::PixelRealType    PixelRealType;
  typedef TFeatureImageType                     FeatureImageType;
  typedef typename Superclass::FloatOffsetType  FloatOffsetType;
  typedef typename Superclass::ScalarValueType  ScalarValueType;
  typedef typename Superclass::NeighborhoodType NeighborhoodType;
  typedef typename FeatureImageType::PixelType  FeatureScalarType;
  typedef typename ImageType::IndexType         IndexType;
  typedef typename Superclass::VectorType       VectorType;
  typedef typename Superclass::GlobalDataStruct GlobalDataStruct;

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

  typedef Image<VectorType, itkGetStaticConstMacro(ImageDimension)> VectorImageType;


  typedef LinearInterpolateImageFunction<ImageType>  InterpolatorType;

  typedef VectorLinearInterpolateImageFunction<VectorImageType> VectorInterpolatorType;

  typedef typename InterpolatorType::ContinuousIndexType ContinuousIndexType;

  virtual const FeatureImageType *GetFeatureImage() const
  { return m_FeatureImage.GetPointer(); }
  virtual void SetFeatureImage(const FeatureImageType *f)
  {    m_FeatureImage = f;  }

  virtual ImageType *GetSpeedImage() 
    { return m_SpeedImage.GetPointer(); }
  void SetSpeedImage( ImageType *s );

  virtual VectorImageType *GetAdvectionImage() const
  { return m_AdvectionImage.GetPointer(); } 
  void SetAdvectionImage( VectorImageType *s );

  
  virtual void Initialize(const RadiusType &r);

  virtual void CalculateSpeedImage() {}

  virtual void CalculateAdvectionImage() {}

  virtual void AllocateSpeedImage();

  virtual void AllocateAdvectionImage();

  virtual void ReverseExpansionDirection();

protected:
  typename FeatureImageType::ConstPointer m_FeatureImage;

  typename ImageType::Pointer        m_SpeedImage;

  typename VectorImageType::Pointer  m_AdvectionImage;

  Functor::VectorCast< ITK_TYPENAME VectorInterpolatorType::OutputType,
                       VectorType > m_VectorCast;

  virtual ScalarValueType PropagationSpeed(const NeighborhoodType &,
                                           const FloatOffsetType &, GlobalDataStruct *gd) const;

  virtual VectorType AdvectionField(const NeighborhoodType &,
                                    const FloatOffsetType &, GlobalDataStruct *gd) const;

  
  virtual ~SegmentationLevelSetFunction() {}
  SegmentationLevelSetFunction()
    {
    m_SpeedImage = ImageType::New();
    m_AdvectionImage = VectorImageType::New();
    m_Interpolator = InterpolatorType::New();
    m_VectorInterpolator = VectorInterpolatorType::New();
    }

  typename InterpolatorType::Pointer       m_Interpolator;
  typename VectorInterpolatorType::Pointer m_VectorInterpolator;

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

} // end namespace

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

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

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


#endif