ITK  5.4.0
Insight Toolkit
itkGPUGradientNDAnisotropicDiffusionFunction.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright NumFOCUS
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  * https://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 itkGPUGradientNDAnisotropicDiffusionFunction_h
19 #define itkGPUGradientNDAnisotropicDiffusionFunction_h
20 
24 #include "itkDerivativeOperator.h"
25 
26 namespace itk
27 {
57 itkGPUKernelClassMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel);
58 
59 template <typename TImage>
62 {
63 public:
64  ITK_DISALLOW_COPY_AND_MOVE(GPUGradientNDAnisotropicDiffusionFunction);
65 
71 
73  itkNewMacro(Self);
74 
76  itkOverrideGetNameOfClassMacro(GPUGradientNDAnisotropicDiffusionFunction);
77 
79  using typename Superclass::ImageType;
80  using typename Superclass::PixelType;
81  using typename Superclass::PixelRealType;
82  using typename Superclass::TimeStepType;
83  using typename Superclass::RadiusType;
84  using typename Superclass::NeighborhoodType;
85  using typename Superclass::FloatOffsetType;
86 
88 
90  static constexpr unsigned int ImageDimension = Superclass::ImageDimension;
91 
93  itkGetOpenCLSourceFromKernelMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel);
94 
96  void
97  GPUComputeUpdate(const typename TImage::Pointer output, typename TImage::Pointer buffer, void * globalData) override;
98 
100  void
102  {
103  m_K = static_cast<PixelType>(this->GetAverageGradientMagnitudeSquared() * this->GetConductanceParameter() *
104  this->GetConductanceParameter() * -2.0f);
105  }
106 
107 protected:
109  ~GPUGradientNDAnisotropicDiffusionFunction() 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 
123  PixelType m_K{};
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 "itkGPUGradientNDAnisotropicDiffusionFunction.hxx"
134 #endif
135 
136 #endif
Pointer
SmartPointer< Self > Pointer
Definition: itkAddImageFilter.h:93
itk::FiniteDifferenceFunction< TImage >::RadiusType
typename ConstNeighborhoodIterator< TImage >::RadiusType RadiusType
Definition: itkFiniteDifferenceFunction.h:97
itkNeighborhoodAlgorithm.h
itkGPUScalarAnisotropicDiffusionFunction.h
itk::Vector
A templated class holding a n-Dimensional vector.
Definition: itkVector.h:62
itk::FiniteDifferenceFunction< TImage >::PixelRealType
double PixelRealType
Definition: itkFiniteDifferenceFunction.h:84
itk::GPUGradientNDAnisotropicDiffusionFunction::InitializeIteration
void InitializeIteration() override
Definition: itkGPUGradientNDAnisotropicDiffusionFunction.h:101
itk::SmartPointer< Self >
itk::FiniteDifferenceFunction< TImage >::ImageType
TImage ImageType
Definition: itkFiniteDifferenceFunction.h:82
itk::GPUGradientNDAnisotropicDiffusionFunction::m_MIN_NORM
static double m_MIN_NORM
Definition: itkGPUGradientNDAnisotropicDiffusionFunction.h:128
itk::LightObject
Light weight base class for most itk classes.
Definition: itkLightObject.h:55
itk::GPUFiniteDifferenceFunction< TImage >
itk::FiniteDifferenceFunction< TImage >::PixelType
typename ImageType::PixelType PixelType
Definition: itkFiniteDifferenceFunction.h:83
itk::GPUScalarAnisotropicDiffusionFunction
Definition: itkGPUScalarAnisotropicDiffusionFunction.h:38
itkNeighborhoodInnerProduct.h
itk::itkGPUKernelClassMacro
itkGPUKernelClassMacro(GPUImageOpsKernel)
itk
The "itk" namespace contains all Insight Segmentation and Registration Toolkit (ITK) classes....
Definition: itkAnnulusOperator.h:24
itk::DerivativeOperator< PixelType, Self::ImageDimension >
itk::ConstNeighborhoodIterator
Const version of NeighborhoodIterator, defining iteration of a local N-dimensional neighborhood of pi...
Definition: itkConstNeighborhoodIterator.h:51
itk::GPUGradientNDAnisotropicDiffusionFunction
Definition: itkGPUGradientNDAnisotropicDiffusionFunction.h:60
itk::NeighborhoodInnerProduct< ImageType >
itkDerivativeOperator.h
itk::SizeValueType
unsigned long SizeValueType
Definition: itkIntTypes.h:83
itk::GPUGradientNDAnisotropicDiffusionFunction::NeighborhoodSizeValueType
SizeValueType NeighborhoodSizeValueType
Definition: itkGPUGradientNDAnisotropicDiffusionFunction.h:87
itk::FiniteDifferenceFunction< TImage >::TimeStepType
double TimeStepType
Definition: itkFiniteDifferenceFunction.h:90