RTK  2.6.0
Reconstruction Toolkit
rtkCudaConjugateGradientImageFilter.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 rtkCudaConjugateGradientImageFilter_h
20 #define rtkCudaConjugateGradientImageFilter_h
21 
22 #include "rtkConfiguration.h"
23 // Conditional definition of the class to pass ITKHeaderTest
24 #ifdef RTK_USE_CUDA
25 
27 # include <itkCudaImageToImageFilter.h>
28 # include "RTKExport.h"
29 
30 namespace rtk
31 {
32 
43 template <class TImage>
44 class ITK_TEMPLATE_EXPORT CudaConjugateGradientImageFilter
45  : public itk::CudaImageToImageFilter<TImage, TImage, ConjugateGradientImageFilter<TImage>>
46 {
47 public:
48  ITK_DISALLOW_COPY_AND_MOVE(CudaConjugateGradientImageFilter);
49 
51  using Self = rtk::CudaConjugateGradientImageFilter<TImage>;
55 
57  itkNewMacro(Self);
58 
60 # ifdef itkOverrideGetNameOfClassMacro
61  itkOverrideGetNameOfClassMacro(CudaConjugateGradientImageFilter);
62 # else
63  itkTypeMacro(CudaConjugateGradientImageFilter, ConjugateGradientImageFilter);
64 # endif
65 
66 
67 protected:
68  CudaConjugateGradientImageFilter();
69  ~CudaConjugateGradientImageFilter() {}
70 
71  virtual void
72  GPUGenerateData();
73 
74 }; // end of class
75 
76 } // end namespace rtk
77 
78 # ifndef ITK_MANUAL_INSTANTIATION
79 # include "rtkCudaConjugateGradientImageFilter.hxx"
80 # endif
81 
82 
83 #endif // end conditional definition of the class
84 
85 #endif
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass
Solves AX = B by conjugate gradient.