RTK  2.0.1
Reconstruction Toolkit
rtkCudaWarpImageFilter.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 rtkCudaWarpImageFilter_h
20 #define rtkCudaWarpImageFilter_h
21 
22 #include "rtkConfiguration.h"
23 //Conditional definition of the class to pass ITKHeaderTest
24 #ifdef RTK_USE_CUDA
25 
26 #include "RTKExport.h"
27 
28 #include <itkCudaImage.h>
29 #include <itkWarpImageFilter.h>
30 #include <itkCudaInPlaceImageFilter.h>
31 
32 namespace rtk
33 {
34 
46 class RTK_EXPORT CudaWarpImageFilter :
47  public itk::CudaInPlaceImageFilter< itk::CudaImage<float,3>, itk::CudaImage<float,3>,
48  itk::WarpImageFilter< itk::CudaImage<float,3>, itk::CudaImage<float,3>, itk::CudaImage<itk::CovariantVector<float, 3>, 3> > >
49 {
50 public:
51  ITK_DISALLOW_COPY_AND_ASSIGN(CudaWarpImageFilter);
52 
54  using ImageType = itk::CudaImage<float,3>;
55  using DisplacementVectorType = itk::CovariantVector<float, 3>;
56  using DVFType = itk::CudaImage<DisplacementVectorType, 3>;
57  using WarpImageFilterType = itk::WarpImageFilter< ImageType, ImageType, DVFType>;
58  using Self = CudaWarpImageFilter;
59  using Superclass = itk::CudaInPlaceImageFilter<ImageType, ImageType,
60  WarpImageFilterType>;
61  using Pointer = itk::SmartPointer<Self>;
62  using ConstPointer = itk::SmartPointer<const Self>;
63 
64  using OutputImageRegionType = ImageType::RegionType;
65 
67  itkNewMacro(Self);
68 
70  itkTypeMacro(CudaWarpImageFilter, Superclass);
71 
72 protected:
73  CudaWarpImageFilter();
74  virtual ~CudaWarpImageFilter() {};
75 
76  virtual void GPUGenerateData();
77 
78 };
79 
80 } // end namespace rtk
81 
82 #endif //end conditional definition of the class
83 
84 #endif