itkSegmentationLevelSetFunction.h Source File

00001 /*=========================================================================
00002 
00003   Program:   Insight Segmentation & Registration Toolkit
00004   Module:    $RCSfile: itkSegmentationLevelSetFunction.h,v $
00005   Language:  C++
00006   Date:      $Date: 2003/09/10 14:29:25 $
00007   Version:   $Revision: 1.12 $
00008 
00009   Copyright (c) Insight Software Consortium. All rights reserved.
00010   See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
00011 
00012      This software is distributed WITHOUT ANY WARRANTY; without even 
00013      the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
00014      PURPOSE.  See the above copyright notices for more information.
00015 
00016 =========================================================================*/
00017 #ifndef __itkSegmentationLevelSetFunction_h_
00018 #define __itkSegmentationLevelSetFunction_h_
00019 
00020 #include "itkLevelSetFunction.h"
00021 #include "itkLinearInterpolateImageFunction.h"
00022 #include "itkVectorLinearInterpolateImageFunction.h"
00023 #include "itkVectorCastImageFilter.h"
00024 
00025 namespace itk {
00026 
00044 template <class TImageType, class TFeatureImageType = TImageType>
00045 class ITK_EXPORT SegmentationLevelSetFunction
00046   : public LevelSetFunction<TImageType>
00047 {
00048 public:
00050   typedef SegmentationLevelSetFunction Self;
00051   typedef LevelSetFunction<TImageType> Superclass;
00052   typedef SmartPointer<Self> Pointer;
00053   typedef SmartPointer<const Self> ConstPointer;
00054 
00056   itkTypeMacro( SegmentationLevelSetFunction, LevelSetFunction );
00057 
00059   typedef typename Superclass::ImageType ImageType;
00060   typedef typename Superclass::RadiusType RadiusType;
00061   typedef TFeatureImageType FeatureImageType;
00062   typedef typename Superclass::FloatOffsetType FloatOffsetType;
00063   typedef typename Superclass::ScalarValueType ScalarValueType;
00064   typedef typename Superclass::NeighborhoodType NeighborhoodType;
00065   typedef typename FeatureImageType::PixelType FeatureScalarType;
00066   typedef typename ImageType::IndexType IndexType;
00067   typedef typename Superclass::VectorType VectorType;
00068   typedef typename Superclass::GlobalDataStruct GlobalDataStruct;
00069 
00071   typedef Image<VectorType, 
00072                 ::itk::GetImageDimension<TImageType>::ImageDimension > VectorImageType;
00073 
00074 
00076   typedef LinearInterpolateImageFunction<ImageType>  InterpolatorType;
00077 
00079   typedef VectorLinearInterpolateImageFunction<VectorImageType> VectorInterpolatorType;
00080   
00082   typedef typename InterpolatorType::ContinuousIndexType ContinuousIndexType;
00083   
00085   itkStaticConstMacro(ImageDimension, unsigned int,Superclass::ImageDimension);
00086 
00088   virtual const FeatureImageType *GetFeatureImage() const
00089   { return m_FeatureImage.GetPointer(); }
00090   virtual void SetFeatureImage(const FeatureImageType *f)
00091   {    m_FeatureImage = f;  }
00092   
00094   virtual ImageType *GetSpeedImage() 
00095   { return m_SpeedImage.GetPointer(); }
00096 
00098   virtual VectorImageType *GetAdvectionImage() const
00099   { return m_AdvectionImage.GetPointer(); } 
00100 
00104   virtual void Initialize(const RadiusType &r);
00105 
00110   virtual void CalculateSpeedImage() {}
00111 
00116   virtual void CalculateAdvectionImage() {}
00117 
00120   virtual void AllocateSpeedImage();
00121 
00124   virtual void AllocateAdvectionImage();
00125 
00136   virtual void ReverseExpansionDirection();
00137   
00138 protected:
00140   typename FeatureImageType::ConstPointer m_FeatureImage;
00141 
00143   typename ImageType::Pointer        m_SpeedImage;
00144 
00146   typename VectorImageType::Pointer  m_AdvectionImage;
00147 
00149   Functor::VectorCast< ITK_TYPENAME VectorInterpolatorType::OutputType,
00150                        VectorType > m_VectorCast;
00151 
00153   virtual ScalarValueType PropagationSpeed(const NeighborhoodType &,
00154                                            const FloatOffsetType &, GlobalDataStruct *gd) const;
00155 
00157   virtual VectorType AdvectionField(const NeighborhoodType &,
00158                                     const FloatOffsetType &, GlobalDataStruct *gd) const;
00159   
00160   
00161   virtual ~SegmentationLevelSetFunction() {}
00162   SegmentationLevelSetFunction()
00163   {
00164     m_SpeedImage = ImageType::New();
00165     m_AdvectionImage = VectorImageType::New();
00166     m_Interpolator = InterpolatorType::New();
00167     m_VectorInterpolator = VectorInterpolatorType::New();
00168   }
00169 
00170   typename InterpolatorType::Pointer m_Interpolator;
00171   typename VectorInterpolatorType::Pointer m_VectorInterpolator;
00172   
00173 };
00174 
00175 } // end namespace
00176 
00177 #ifndef ITK_MANUAL_INSTANTIATION
00178 #include "itkSegmentationLevelSetFunction.txx"
00179 #endif
00180 
00181 #endif