RTK  2.5.0
Reconstruction Toolkit
rtkCudaWarpForwardProjectionImageFilter.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 rtkCudaWarpForwardProjectionImageFilter_h
20 #define rtkCudaWarpForwardProjectionImageFilter_h
21 
22 #include "rtkConfiguration.h"
23 // Conditional definition of the class to pass ITKHeaderTest
24 #ifdef RTK_USE_CUDA
25 
27 # include "itkCudaInPlaceImageFilter.h"
28 # include "itkCudaUtil.h"
29 # include "RTKExport.h"
30 
44 namespace rtk
45 {
46 
47 class RTK_EXPORT CudaWarpForwardProjectionImageFilter
48  : public itk::CudaInPlaceImageFilter<itk::CudaImage<float, 3>,
49  itk::CudaImage<float, 3>,
50  ForwardProjectionImageFilter<itk::CudaImage<float, 3>, itk::CudaImage<float, 3>>>
51 {
52 public:
53  ITK_DISALLOW_COPY_AND_MOVE(CudaWarpForwardProjectionImageFilter);
54 
56  using InputImageType = itk::CudaImage<float, 3>;
57  using DisplacementVectorType = itk::CovariantVector<float, 3>;
58  using Self = CudaWarpForwardProjectionImageFilter;
59  using Superclass = ForwardProjectionImageFilter<InputImageType, InputImageType>;
60  using GPUSuperclass = itk::CudaInPlaceImageFilter<InputImageType, InputImageType, Superclass>;
63  using DVFType = itk::CudaImage<DisplacementVectorType, 3>;
64 
66  itkNewMacro(Self);
67 
69 # ifdef itkOverrideGetNameOfClassMacro
70  itkOverrideGetNameOfClassMacro(CudaWarpForwardProjectionImageFilter);
71 # else
72  itkTypeMacro(CudaWarpForwardProjectionImageFilter, ForwardProjectionImageFilter);
73 # endif
74 
75 
77  void
78  SetInputProjectionStack(const InputImageType * ProjectionStack);
79  InputImageType::Pointer
80  GetInputProjectionStack();
82 
84  void
85  SetInputVolume(const InputImageType * Volume);
86  InputImageType::Pointer
87  GetInputVolume();
89 
91  void
92  SetDisplacementField(const DVFType * DVF);
93  DVFType::Pointer
94  GetDisplacementField();
96 
98  itkGetConstMacro(StepSize, double);
99  itkSetMacro(StepSize, double);
101 
102 protected:
103  CudaWarpForwardProjectionImageFilter();
104  ~CudaWarpForwardProjectionImageFilter(){};
105 
106  virtual void
107  GenerateInputRequestedRegion();
108 
109  virtual void
110  GPUGenerateData();
111 
112 private:
113  double m_StepSize;
114 }; // end of class
115 
116 } // end namespace rtk
117 
118 #endif // end conditional definition of the class
119 
120 #endif
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass
#define itkSetMacro(name, type)