RTK  2.0.1
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  * 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 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>, itk::CudaImage<float,3>,
49  ForwardProjectionImageFilter< itk::CudaImage<float,3>, itk::CudaImage<float,3> > >
50 {
51 public:
52  ITK_DISALLOW_COPY_AND_ASSIGN(CudaWarpForwardProjectionImageFilter);
53 
55  using InputImageType = itk::CudaImage<float, 3>;
56  using DisplacementVectorType = itk::CovariantVector<float, 3>;
57  using Self = CudaWarpForwardProjectionImageFilter;
58  using Superclass = ForwardProjectionImageFilter<InputImageType, InputImageType>;
59  using GPUSuperclass = itk::CudaInPlaceImageFilter<InputImageType, InputImageType, Superclass >;
60  using Pointer = itk::SmartPointer<Self>;
61  using ConstPointer = itk::SmartPointer<const Self>;
62  using DVFType = itk::CudaImage<DisplacementVectorType, 3>;
63 
65  itkNewMacro(Self);
66 
68  itkTypeMacro(CudaWarpForwardProjectionImageFilter, ForwardProjectionImageFilter);
69 
71  void SetInputProjectionStack(const InputImageType* ProjectionStack);
72  InputImageType::Pointer GetInputProjectionStack();
74 
76  void SetInputVolume(const InputImageType* Volume);
77  InputImageType::Pointer GetInputVolume();
79 
81  void SetDisplacementField(const DVFType* DVF);
82  DVFType::Pointer GetDisplacementField();
84 
86  itkGetConstMacro(StepSize, double);
87  itkSetMacro(StepSize, double);
89 
90 protected:
91  CudaWarpForwardProjectionImageFilter();
92  ~CudaWarpForwardProjectionImageFilter() {};
93 
94  virtual void GenerateInputRequestedRegion();
95 
96  virtual void GPUGenerateData();
97 
98 private:
99  double m_StepSize;
100 }; // end of class
101 
102 } // end namespace rtk
103 
104 #endif //end conditional definition of the class
105 
106 #endif
#define itkSetMacro(name, type)