00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef _itkLevelSetVelocityNeighborhoodExtractor_h
00018 #define _itkLevelSetVelocityNeighborhoodExtractor_h
00019
00020 #include "itkLevelSetNeighborhoodExtractor.h"
00021 #include "itkLevelSet.h"
00022 #include "itkIndex.h"
00023
00024 namespace itk
00025 {
00026
00054 template <
00055 class TLevelSet,
00056 class TAuxValue,
00057 unsigned int VAuxDimension = 1
00058 >
00059 class ITK_EXPORT LevelSetVelocityNeighborhoodExtractor :
00060 public LevelSetNeighborhoodExtractor<TLevelSet>
00061 {
00062 public:
00064 typedef LevelSetVelocityNeighborhoodExtractor Self;
00065 typedef LevelSetNeighborhoodExtractor<TLevelSet> Superclass;
00066 typedef SmartPointer<Self> Pointer;
00067 typedef SmartPointer<const Self> ConstPointer;
00068
00070 itkNewMacro(Self);
00071
00073 itkTypeMacro(LevelSetVelocityNeighborhoodExtractor,
00074 LevelSetNeighborhoodExtractor);
00075
00077 typedef LevelSetTypeDefault<TLevelSet> LevelSetType;
00078
00080 itkStaticConstMacro(SetDimension, unsigned int,
00081 LevelSetType::SetDimension);
00082
00084 typedef Index<itkGetStaticConstMacro(SetDimension)> Index;
00085
00087 typedef AuxVarTypeDefault<TAuxValue,VAuxDimension,itkGetStaticConstMacro(SetDimension)> AuxVarType;
00088 typedef typename AuxVarType::AuxValueType AuxValueType;
00089 typedef typename AuxVarType::AuxValueVectorType AuxValueVectorType;
00090 typedef typename AuxVarType::AuxValueContainer AuxValueContainer;
00091 typedef typename AuxVarType::AuxImageType AuxImageType;
00092 typedef typename AuxVarType::AuxImagePointer AuxImagePointer;
00093 typedef typename AuxVarType::AuxImageConstPointer AuxImageConstPointer;
00094
00096 void SetAuxImage( const AuxImageType * ptr, unsigned int idx = 0 )
00097 {
00098 if ( idx < VAuxDimension && m_AuxImage[idx] != ptr )
00099 {
00100 m_AuxImage[idx] = ptr;
00101 }
00102 this->Modified();
00103 }
00104
00106 AuxImageConstPointer GetAuxImage( unsigned int idx = 0 )
00107 {
00108 if ( idx >= VAuxDimension )
00109 {
00110 return NULL;
00111 }
00112 else
00113 {
00114 return m_AuxImage[idx];
00115 }
00116 }
00117
00120 itkGetObjectMacro( AuxInsideValues, AuxValueContainer );
00121
00124 itkGetObjectMacro( AuxOutsideValues, AuxValueContainer );
00125
00126 protected:
00127 LevelSetVelocityNeighborhoodExtractor();
00128 ~LevelSetVelocityNeighborhoodExtractor(){};
00129 void PrintSelf( std::ostream& os, Indent indent ) const;
00130
00131 virtual void Initialize();
00132 virtual double CalculateDistance( Index & index );
00133
00134 private:
00135 LevelSetVelocityNeighborhoodExtractor(const Self&);
00136 void operator=(const Self&);
00137
00138 typename AuxValueContainer::Pointer m_AuxInsideValues;
00139 typename AuxValueContainer::Pointer m_AuxOutsideValues;
00140 AuxImageConstPointer m_AuxImage[VAuxDimension];
00141
00142 };
00143
00144 }
00145
00146 #ifndef ITK_MANUAL_INSTANTIATION
00147 #include "itkLevelSetVelocityNeighborhoodExtractor.txx"
00148 #endif
00149
00150 #endif
00151
00152