ITK  6.0.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
25
26namespace itk
27{
57itkGPUKernelClassMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel);
58
59template <typename TImage>
62{
63public:
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
107protected:
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 "itkGPUGradientNDAnisotropicDiffusionFunction.hxx"
134#endif
135
136#endif
Const version of NeighborhoodIterator, defining iteration of a local N-dimensional neighborhood of pi...
typename ConstNeighborhoodIterator< TImage >::RadiusType RadiusType
void GPUComputeUpdate(const typename TImage::Pointer output, typename TImage::Pointer buffer, void *globalData) override
itkGetOpenCLSourceFromKernelMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel)
~GPUGradientNDAnisotropicDiffusionFunction() override=default
Light weight base class for most itk classes.
A templated class holding a n-Dimensional vector.
Definition: itkVector.h:63
SmartPointer< Self > Pointer
The "itk" namespace contains all Insight Segmentation and Registration Toolkit (ITK) classes....
itkGPUKernelClassMacro(GPUImageOpsKernel)
unsigned long SizeValueType
Definition: itkIntTypes.h:86