RTK  2.5.0
Reconstruction Toolkit
rtkCudaPolynomialGainCorrectionImageFilter.h
Go to the documentation of this file.
1 /*=========================================================================
2  *
3  * Copyright RTK 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  * 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 
19 #ifndef rtkCudaPolynomialGainCorrectionImageFilter_h
20 #define rtkCudaPolynomialGainCorrectionImageFilter_h
21 
22 #include "rtkConfiguration.h"
23 // Conditional definition of the class to pass ITKHeaderTest
24 #ifdef RTK_USE_CUDA
25 
27 # include "RTKExport.h"
28 
29 # include <itkCudaImage.h>
30 # include <itkCudaImageToImageFilter.h>
31 
32 # include "rtkConfiguration.h"
33 
34 namespace rtk
35 {
36 
48 class RTK_EXPORT CudaPolynomialGainCorrectionImageFilter
49  : public itk::CudaImageToImageFilter<
50  itk::CudaImage<unsigned short, 3>,
51  itk::CudaImage<float, 3>,
52  PolynomialGainCorrectionImageFilter<itk::CudaImage<unsigned short, 3>, itk::CudaImage<float, 3>>>
53 {
54 public:
55  ITK_DISALLOW_COPY_AND_MOVE(CudaPolynomialGainCorrectionImageFilter);
56 
58  using ImageType = itk::CudaImage<float, 3>;
59  using CPUPolyGainFilterType =
60  PolynomialGainCorrectionImageFilter<itk::CudaImage<unsigned short, 3>, itk::CudaImage<float, 3>>;
61 
63  using Self = CudaPolynomialGainCorrectionImageFilter;
64  using Superclass = itk::CudaImageToImageFilter<ImageType, ImageType, CPUPolyGainFilterType>;
67 
69  itkNewMacro(Self);
70 
72 # ifdef itkOverrideGetNameOfClassMacro
73  itkOverrideGetNameOfClassMacro(CudaPolynomialGainCorrectionImageFilter);
74 # else
75  itkTypeMacro(CudaPolynomialGainCorrectionImageFilter, itk::CudaImageToImageFilter);
76 # endif
77 
78 
79 protected:
81  CudaPolynomialGainCorrectionImageFilter();
82 
84  virtual ~CudaPolynomialGainCorrectionImageFilter();
85 
86  virtual void
87  GPUGenerateData();
88 };
89 
90 } // namespace rtk
91 
92 #endif // end conditional definition of the class
93 
94 #endif // rtkCudaPolynomialGainCorrectionImageFilter_h
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass