00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017 #ifndef __itkCurvatureNDAnisotropicDiffusionFunction_h_
00018 #define __itkCurvatureNDAnisotropicDiffusionFunction_h_
00019
00020 #include "itkScalarAnisotropicDiffusionFunction.h"
00021 #include "itkNeighborhoodAlgorithm.h"
00022 #include "itkNeighborhoodInnerProduct.h"
00023 #include "itkDerivativeOperator.h"
00024
00025 namespace itk {
00026
00070 template <class TImage>
00071 class ITK_EXPORT CurvatureNDAnisotropicDiffusionFunction :
00072 public ScalarAnisotropicDiffusionFunction<TImage>
00073 {
00074 public:
00076 typedef CurvatureNDAnisotropicDiffusionFunction Self;
00077 typedef ScalarAnisotropicDiffusionFunction<TImage> Superclass;
00078 typedef SmartPointer<Self> Pointer;
00079 typedef SmartPointer<const Self> ConstPointer;
00080
00082 itkNewMacro(Self);
00083
00085 itkTypeMacro( CurvatureNDAnisotropicDiffusionFunction,
00086 ScalarAnisotropicDiffusionFunction );
00087
00089 typedef typename Superclass::ImageType ImageType;
00090 typedef typename Superclass::PixelType PixelType;
00091 typedef typename Superclass::TimeStepType TimeStepType;
00092 typedef typename Superclass::RadiusType RadiusType;
00093 typedef typename Superclass::NeighborhoodType NeighborhoodType;
00094 typedef typename Superclass::FloatOffsetType FloatOffsetType;
00095
00097 itkStaticConstMacro(ImageDimension, unsigned int,Superclass::ImageDimension);
00098
00100 virtual PixelType ComputeUpdate(const NeighborhoodType &neighborhood,
00101 void *globalData,
00102 const FloatOffsetType& offset = FloatOffsetType(0.0)
00103 );
00104
00106 virtual void InitializeIteration()
00107 {
00108 m_K = static_cast<PixelType>(this->GetAverageGradientMagnitudeSquared() *
00109 this->GetConductanceParameter() *
00110 this->GetConductanceParameter() * -2.0f);
00111 }
00112
00113 protected:
00114 CurvatureNDAnisotropicDiffusionFunction();
00115 ~CurvatureNDAnisotropicDiffusionFunction() {}
00116 void PrintSelf(std::ostream& os, Indent indent) const
00117 {
00118 Superclass::PrintSelf(os,indent);
00119 }
00120
00121 private:
00122 CurvatureNDAnisotropicDiffusionFunction(const Self&);
00123 void operator=(const Self&);
00124
00126 NeighborhoodInnerProduct<ImageType> m_InnerProduct;
00127
00129 std::slice x_slice[ImageDimension];
00130 std::slice xa_slice[ImageDimension][ImageDimension];
00131 std::slice xd_slice[ImageDimension][ImageDimension];
00132
00134 DerivativeOperator<PixelType, itkGetStaticConstMacro(ImageDimension)> dx_op;
00135
00137 PixelType m_K;
00138
00140 static double m_MIN_NORM;
00141
00142 unsigned long m_Center;
00143 unsigned long m_Stride[ImageDimension];
00144
00145 };
00146
00147
00148
00149 }
00150
00151 #ifndef ITK_MANUAL_INSTANTIATION
00152 #include "itkCurvatureNDAnisotropicDiffusionFunction.txx"
00153 #endif
00154
00155 #endif