00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef _itkExtensionVelocitiesImageFilter_h
00018 #define _itkExtensionVelocitiesImageFilter_h
00019
00020 #include "itkLevelSetVelocityNeighborhoodExtractor.h"
00021 #include "itkFastMarchingExtensionImageFilter.h"
00022 #include "itkReinitializeLevelSetImageFilter.h"
00023
00024 namespace itk
00025 {
00047 template <
00048 class TLevelSet,
00049 class TAuxValue = float,
00050 unsigned int VAuxDimension = 1
00051 >
00052 class ITK_EXPORT ExtensionVelocitiesImageFilter :
00053 public ReinitializeLevelSetImageFilter<TLevelSet>
00054 {
00055 public:
00056
00058 typedef ExtensionVelocitiesImageFilter Self;
00059 typedef ReinitializeLevelSetImageFilter<TLevelSet> Superclass;
00060 typedef SmartPointer<Self> Pointer;
00061 typedef SmartPointer<const Self> ConstPointer;
00062
00064 itkNewMacro(Self);
00065
00067 itkTypeMacro(ExtensionVelocitiesImageFilter, ReinitializeLevelSetImageFilter);
00068
00070 typedef LevelSetTypeDefault<TLevelSet> LevelSetType;
00071 typedef typename LevelSetType::LevelSetPointer LevelSetPointer;
00072 typedef typename LevelSetType::LevelSetConstPointer LevelSetConstPointer;
00073 typedef typename LevelSetType::PixelType PixelType;
00074 typedef typename LevelSetType::NodeType NodeType;
00075 typedef typename LevelSetType::NodeContainer NodeContainer;
00076 typedef typename LevelSetType::NodeContainerPointer NodeContainerPointer;
00077
00079 itkStaticConstMacro(SetDimension, unsigned int,LevelSetType::SetDimension);
00080
00082 typedef AuxVarTypeDefault<TAuxValue,VAuxDimension,
00083 itkGetStaticConstMacro(SetDimension)> AuxVarType;
00084 typedef typename AuxVarType::AuxValueType AuxValueType;
00085 typedef typename AuxVarType::AuxValueVectorType AuxValueVectorType;
00086 typedef typename AuxVarType::AuxValueContainer AuxValueContainer;
00087 typedef typename AuxVarType::AuxImageType AuxImageType;
00088 typedef typename AuxVarType::AuxImagePointer AuxImagePointer;
00089 typedef typename AuxVarType::AuxImageConstPointer AuxImageConstPointer;
00090
00092 itkStaticConstMacro(AuxDimension, unsigned int,VAuxDimension);
00093
00095 void SetInputVelocityImage(const AuxImageType * ptr, unsigned int idx = 0);
00096 const AuxImageType * GetInputVelocityImage(unsigned int idx = 0);
00097
00099 AuxImageType * GetOutputVelocityImage( unsigned int idx = 0 );
00100
00101 protected:
00102 ExtensionVelocitiesImageFilter();
00103 ~ExtensionVelocitiesImageFilter(){};
00104
00105 virtual void GenerateDataFull();
00106 virtual void GenerateDataNarrowBand();
00107 virtual void AllocateOutput();
00108
00109 virtual void EnlargeOutputRequestedRegion( DataObject * );
00110
00111 private:
00112 ExtensionVelocitiesImageFilter(const Self&);
00113 void operator=(const Self&);
00114
00118 typedef Image<float, itkGetStaticConstMacro(SetDimension) > SpeedImageType;
00119 typedef LevelSetVelocityNeighborhoodExtractor<TLevelSet,TAuxValue,VAuxDimension>
00120 LocatorType;
00121 typedef FastMarchingExtensionImageFilter<TLevelSet,TAuxValue,VAuxDimension,SpeedImageType>
00122 FastMarchingImageFilterType;
00123
00124 typename LocatorType::Pointer m_Locator;
00125 typename FastMarchingImageFilterType::Pointer m_Marcher;
00126
00127 };
00128
00129 }
00130
00131 #ifndef ITK_MANUAL_INSTANTIATION
00132 #include "itkExtensionVelocitiesImageFilter.txx"
00133 #endif
00134
00135 #endif