00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
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 }
00176
00177
#ifndef ITK_MANUAL_INSTANTIATION
00178
#include "itkSegmentationLevelSetFunction.txx"
00179
#endif
00180
00181
#endif