00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkVectorGradientNDAnisotropicDiffusionFunction_h_
00018 #define __itkVectorGradientNDAnisotropicDiffusionFunction_h_
00019
00020 #include "itkVectorAnisotropicDiffusionFunction.h"
00021 #include "itkNeighborhoodAlgorithm.h"
00022 #include "itkVectorNeighborhoodInnerProduct.h"
00023 #include "itkDerivativeOperator.h"
00024
00025 namespace itk {
00026
00041 template <class TImage>
00042 class ITK_EXPORT VectorGradientNDAnisotropicDiffusionFunction :
00043 public VectorAnisotropicDiffusionFunction<TImage>
00044 {
00045 public:
00047 typedef VectorGradientNDAnisotropicDiffusionFunction Self;
00048 typedef VectorAnisotropicDiffusionFunction<TImage> Superclass;
00049 typedef SmartPointer<Self> Pointer;
00050 typedef SmartPointer<const Self> ConstPointer;
00051
00053 itkNewMacro(Self);
00054
00056 itkTypeMacro( VectorGradientNDAnisotropicDiffusionFunction,
00057 ScalarAnisotropicDiffusionFunction );
00058
00060 typedef typename Superclass::ImageType ImageType;
00061 typedef typename Superclass::PixelType PixelType;
00062 typedef typename Superclass::TimeStepType TimeStepType;
00063 typedef typename Superclass::RadiusType RadiusType;
00064 typedef typename Superclass::NeighborhoodType NeighborhoodType;
00065 typedef typename Superclass::FloatOffsetType FloatOffsetType;
00066
00068 itkStaticConstMacro(ImageDimension, unsigned int,
00069 Superclass::ImageDimension );
00070 itkStaticConstMacro(VectorDimension, unsigned int,
00071 Superclass::VectorDimension );
00072
00074 typedef typename PixelType::ValueType ScalarValueType;
00075
00077 virtual PixelType ComputeUpdate(const NeighborhoodType &neighborhood,
00078 void * globalData,
00079 const FloatOffsetType& offset = FloatOffsetType(0.0)
00080 );
00081
00083 virtual void InitializeIteration()
00084 {
00085 m_K = this->GetAverageGradientMagnitudeSquared() * this->GetConductanceParameter() *
00086 this->GetConductanceParameter() * -2.0f;
00087 }
00088
00089 protected:
00090 VectorGradientNDAnisotropicDiffusionFunction();
00091 ~VectorGradientNDAnisotropicDiffusionFunction() {}
00092
00093 private:
00094 VectorGradientNDAnisotropicDiffusionFunction(const Self&);
00095 void operator=(const Self&);
00096
00098 VectorNeighborhoodInnerProduct<ImageType> m_InnerProduct;
00099
00101 std::slice x_slice[ImageDimension];
00102 std::slice xa_slice[ImageDimension][ImageDimension];
00103 std::slice xd_slice[ImageDimension][ImageDimension];
00104
00106 DerivativeOperator<ScalarValueType,
00107 itkGetStaticConstMacro(ImageDimension)> dx_op;
00108
00110 ScalarValueType m_K;
00111
00112 static double m_MIN_NORM;
00113
00114 unsigned long int m_Stride[ImageDimension];
00115 unsigned long int m_Center;
00116
00117 };
00118
00119
00120
00121 }
00122
00123 #ifndef ITK_MANUAL_INSTANTIATION
00124 #include "itkVectorGradientNDAnisotropicDiffusionFunction.txx"
00125 #endif
00126
00127 #endif