ITK  4.3.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< class 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 );
96 
98  virtual void InitializeIteration()
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];
114  std::slice xa_slice[ImageDimension][ImageDimension];
115  std::slice xd_slice[ImageDimension][ImageDimension];
116 
119 
122 
124  NeighborhoodSizeValueType m_Stride[ImageDimension];
125 
126  static double m_MIN_NORM;
127 
128 private:
129  GPUGradientNDAnisotropicDiffusionFunction(const Self &); //purposely not
130  // implemented
131  void operator=(const Self &); //purposely not
132 
133  // implemented
134 };
135 } // end namespace itk
136 
137 #ifndef ITK_MANUAL_INSTANTIATION
138 #include "itkGPUGradientNDAnisotropicDiffusionFunction.hxx"
139 #endif
140 
141 #endif
142