RTK  2.0.1
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  * 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 
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 CudaFFTProjectionsConvolutionImageFilter:
50  public itk::CudaImageToImageFilter< itk::CudaImage<float,3>,
51  itk::CudaImage<float,3>,
52  TParentImageFilter >
53 {
54 public:
55  ITK_DISALLOW_COPY_AND_ASSIGN(CudaFFTProjectionsConvolutionImageFilter);
56 
58  using Self = CudaFFTProjectionsConvolutionImageFilter;
59  using Superclass = TParentImageFilter;
60  using Pointer = itk::SmartPointer<Self>;
61  using ConstPointer = itk::SmartPointer<const Self>;
62 
64  using RegionType = typename TParentImageFilter::RegionType;
65  using FFTInputImagePointer = typename TParentImageFilter::FFTInputImagePointer;
66  using CudaImageType = itk::CudaImage<float,3>;
67  using CudaFFTOutputImageType = itk::CudaImage< std::complex<float>, 3 >;
68  using CudaFFTOutputImagePointer = CudaFFTOutputImageType::Pointer;
69 
71  itkTypeMacro(CudaFFTProjectionsConvolutionImageFilter, TParentImageFilter);
72 
73 protected:
74  CudaFFTProjectionsConvolutionImageFilter();
75  ~CudaFFTProjectionsConvolutionImageFilter(){}
76 
77  virtual void GPUGenerateData();
78 
79  virtual FFTInputImagePointer PadInputImageRegion(const RegionType &inputRegion);
80 
81 private:
82  CudaFFTOutputImagePointer m_KernelFFTCUDA;
83 }; // end of class
84 
85 } // end namespace rtk
86 
87 #ifndef ITK_MANUAL_INSTANTIATION
88 #include "rtkCudaFFTProjectionsConvolutionImageFilter.hxx"
89 #endif
90 
91 #endif //end conditional definition of the class
92 
93 #endif
ImageBaseType::RegionType RegionType