RTK  2.5.0
Reconstruction Toolkit
rtkCudaWarpBackProjectionImageFilter.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 rtkCudaWarpBackProjectionImageFilter_h
20 #define rtkCudaWarpBackProjectionImageFilter_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 <itkCudaInPlaceImageFilter.h>
31 
32 namespace rtk
33 {
34 
47 class RTK_EXPORT CudaWarpBackProjectionImageFilter
48  : public itk::CudaInPlaceImageFilter<itk::CudaImage<float, 3>,
49  itk::CudaImage<float, 3>,
50  BackProjectionImageFilter<itk::CudaImage<float, 3>, itk::CudaImage<float, 3>>>
51 {
52 public:
53  ITK_DISALLOW_COPY_AND_MOVE(CudaWarpBackProjectionImageFilter);
54 
56  using ImageType = itk::CudaImage<float, 3>;
57  using DVFType = itk::CudaImage<itk::CovariantVector<float, 3>, 3>;
58  using BackProjectionImageFilterType = BackProjectionImageFilter<ImageType, ImageType>;
59  using Self = CudaWarpBackProjectionImageFilter;
60  using Superclass = itk::CudaInPlaceImageFilter<ImageType, ImageType, BackProjectionImageFilterType>;
63 
64  using OutputImageRegionType = ImageType::RegionType;
65  using ProjectionImageType = itk::CudaImage<float, 2>;
66  using ProjectionImagePointer = ProjectionImageType::Pointer;
67 
69  itkNewMacro(Self);
70 
72 # ifdef itkOverrideGetNameOfClassMacro
73  itkOverrideGetNameOfClassMacro(CudaWarpBackProjectionImageFilter);
74 # else
75  itkTypeMacro(CudaWarpBackProjectionImageFilter, Superclass);
76 # endif
77 
78 
80  void
81  SetInputProjectionStack(const InputImageType * ProjectionStack);
82  InputImageType::Pointer
83  GetInputProjectionStack();
85 
87  void
88  SetInputVolume(const InputImageType * Volume);
89  InputImageType::Pointer
90  GetInputVolume();
92 
94  void
95  SetDisplacementField(const DVFType * MVF);
96  DVFType::Pointer
97  GetDisplacementField();
99 
100 protected:
101  CudaWarpBackProjectionImageFilter();
102  ~CudaWarpBackProjectionImageFilter(){};
103 
104  virtual void
105  GenerateInputRequestedRegion();
106 
107  virtual void
108  GPUGenerateData();
109 };
110 
111 } // end namespace rtk
112 
113 #endif // end conditional definition of the class
114 
115 #endif
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass