RTK  2.5.0
Reconstruction Toolkit
rtkCudaFFTProjectionsConvolutionImageFilter.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 rtkCudaFFTProjectionsConvolutionImageFilter_h
20 #define rtkCudaFFTProjectionsConvolutionImageFilter_h
21 
22 #include "rtkConfiguration.h"
23 // Conditional definition of the class to pass ITKHeaderTest
24 #ifdef RTK_USE_CUDA
25 
26 # include <itkCudaImage.h>
27 # include <itkCudaImageToImageFilter.h>
28 
29 namespace rtk
30 {
31 
48 template <class TParentImageFilter>
49 class ITK_TEMPLATE_EXPORT CudaFFTProjectionsConvolutionImageFilter
50  : public itk::CudaImageToImageFilter<itk::CudaImage<float, 3>, itk::CudaImage<float, 3>, TParentImageFilter>
51 {
52 public:
53  ITK_DISALLOW_COPY_AND_MOVE(CudaFFTProjectionsConvolutionImageFilter);
54 
56  using Self = CudaFFTProjectionsConvolutionImageFilter;
57  using Superclass = TParentImageFilter;
60 
62  using RegionType = typename TParentImageFilter::RegionType;
63  using FFTInputImagePointer = typename TParentImageFilter::FFTInputImagePointer;
64  using CudaImageType = itk::CudaImage<float, 3>;
65  using CudaFFTOutputImageType = itk::CudaImage<std::complex<float>, 3>;
66  using CudaFFTOutputImagePointer = CudaFFTOutputImageType::Pointer;
67 
69 # ifdef itkOverrideGetNameOfClassMacro
70  itkOverrideGetNameOfClassMacro(CudaFFTProjectionsConvolutionImageFilter);
71 # else
72  itkTypeMacro(CudaFFTProjectionsConvolutionImageFilter, TParentImageFilter);
73 # endif
74 
75 
76 protected:
77  CudaFFTProjectionsConvolutionImageFilter();
78  ~CudaFFTProjectionsConvolutionImageFilter() {}
79 
80  virtual void
81  GPUGenerateData();
82 
83  virtual FFTInputImagePointer
84  PadInputImageRegion(const RegionType & inputRegion);
85 
86 private:
87  CudaFFTOutputImagePointer m_KernelFFTCUDA;
88 }; // end of class
89 
90 } // end namespace rtk
91 
92 # ifndef ITK_MANUAL_INSTANTIATION
93 # include "rtkCudaFFTProjectionsConvolutionImageFilter.hxx"
94 # endif
95 
96 #endif // end conditional definition of the class
97 
98 #endif
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass