ITK  5.0.0
Insight Segmentation and Registration Toolkit
itkGradientNDAnisotropicDiffusionFunction.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright Insight Software Consortium
4  *
5  * Licensed under the Apache License, Version 2.0 (the "License");
6  * you may not use this file except in compliance with the License.
7  * You may obtain a copy of the License at
8  *
9  * http://www.apache.org/licenses/LICENSE-2.0.txt
10  *
11  * Unless required by applicable law or agreed to in writing, software
12  * distributed under the License is distributed on an "AS IS" BASIS,
13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14  * See the License for the specific language governing permissions and
15  * limitations under the License.
16  *
17  *=========================================================================*/
18 #ifndef itkGradientNDAnisotropicDiffusionFunction_h
19 #define itkGradientNDAnisotropicDiffusionFunction_h
20 
24 #include "itkDerivativeOperator.h"
25 
26 namespace itk
27 {
60 template< typename TImage >
61 class ITK_TEMPLATE_EXPORT GradientNDAnisotropicDiffusionFunction:
62  public ScalarAnisotropicDiffusionFunction< TImage >
63 {
64 public:
65  ITK_DISALLOW_COPY_AND_ASSIGN(GradientNDAnisotropicDiffusionFunction);
66 
72 
74  itkNewMacro(Self);
75 
79 
81  using ImageType = typename Superclass::ImageType;
82  using PixelType = typename Superclass::PixelType;
88 
90 
92  static constexpr unsigned int ImageDimension = Superclass::ImageDimension;
93 
95  PixelType ComputeUpdate(const NeighborhoodType & neighborhood,
96  void *globalData,
97  const FloatOffsetType & offset = FloatOffsetType(0.0)
98  ) override;
99 
101  void InitializeIteration() override
102  {
103  m_K = static_cast< PixelType >( this->GetAverageGradientMagnitudeSquared()
104  * this->GetConductanceParameter() * this->GetConductanceParameter() * -2.0f );
105  }
106 
107 protected:
109  ~GradientNDAnisotropicDiffusionFunction() override = default;
110 
113 
115  std::slice x_slice[ImageDimension];
116  std::slice xa_slice[ImageDimension][ImageDimension];
117  std::slice xd_slice[ImageDimension][ImageDimension];
118 
121 
124 
126  NeighborhoodSizeValueType m_Stride[ImageDimension];
127 
128  static double m_MIN_NORM;
129 };
130 } // end namespace itk
131 
132 #ifndef ITK_MANUAL_INSTANTIATION
133 #include "itkGradientNDAnisotropicDiffusionFunction.hxx"
134 #endif
135 
136 #endif
Light weight base class for most itk classes.
unsigned long SizeValueType
Definition: itkIntTypes.h:83
A NeighborhoodOperator for taking an n-th order derivative at a pixel.
Const version of NeighborhoodIterator, defining iteration of a local N-dimensional neighborhood of pi...
typename ConstNeighborhoodIterator< TImage >::RadiusType RadiusType
typename Superclass::FloatOffsetType FloatOffsetType
A templated class holding a n-Dimensional vector.
Definition: itkVector.h:62
DerivativeOperator< PixelType, Self::ImageDimension > m_DerivativeOperator
typename Superclass::NeighborhoodType NeighborhoodType
typename Superclass::TimeStepType TimeStepType