ITK  4.9.0
Insight Segmentation and Registration Toolkit
itkGPUGradientNDAnisotropicDiffusionFunction.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 itkGPUGradientNDAnisotropicDiffusionFunction_h
19 #define itkGPUGradientNDAnisotropicDiffusionFunction_h
20 
24 #include "itkDerivativeOperator.h"
25 
26 namespace itk
27 {
56 itkGPUKernelClassMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel);
57 
58 template< typename TImage >
61 {
62 public:
68 
70  itkNewMacro(Self);
71 
75 
77  typedef typename Superclass::ImageType ImageType;
78  typedef typename Superclass::PixelType PixelType;
84 
86 
88  itkStaticConstMacro(ImageDimension, unsigned int, Superclass::ImageDimension);
89 
91  itkGetOpenCLSourceFromKernelMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel);
92 
94  virtual void GPUComputeUpdate( const typename TImage::Pointer output, typename TImage::Pointer buffer,
95  void *globalData ) ITK_OVERRIDE;
96 
98  virtual void InitializeIteration() ITK_OVERRIDE
99  {
100  m_K = static_cast< PixelType >( this->GetAverageGradientMagnitudeSquared()
101  * this->GetConductanceParameter() * this->GetConductanceParameter() * -2.0f );
102  }
103 
104 protected:
107  }
108 
111 
113  std::slice x_slice[ImageDimension];
116 
119 
122 
125 
126  static double m_MIN_NORM;
127 
128 private:
129  GPUGradientNDAnisotropicDiffusionFunction(const Self &) ITK_DELETE_FUNCTION;
130  void operator=(const Self &) ITK_DELETE_FUNCTION;
131 };
132 } // end namespace itk
133 
134 #ifndef ITK_MANUAL_INSTANTIATION
135 #include "itkGPUGradientNDAnisotropicDiffusionFunction.hxx"
136 #endif
137 
138 #endif
DerivativeOperator< PixelType, itkGetStaticConstMacro(ImageDimension) > dx_op
itkGetOpenCLSourceFromKernelMacro(GPUGradientNDAnisotropicDiffusionFunctionKernel)
Light weight base class for most itk classes.
virtual void GPUComputeUpdate(const typename TImage::Pointer output, typename TImage::Pointer buffer, void *globalData) override
itkGPUKernelClassMacro(GPUImageOpsKernel)
unsigned long SizeValueType
Definition: itkIntTypes.h:143