RTK  2.6.0
Reconstruction Toolkit
rtkCudaLastDimensionTVDenoisingImageFilter.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 rtkCudaLastDimensionTVDenoisingImageFilter_h
20 #define rtkCudaLastDimensionTVDenoisingImageFilter_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 "RTKExport.h"
29 
30 namespace rtk
31 {
32 
44 class RTK_EXPORT CudaLastDimensionTVDenoisingImageFilter
45  : public itk::CudaInPlaceImageFilter<
46  itk::CudaImage<float, 4>,
47  itk::CudaImage<float, 4>,
48  TotalVariationDenoisingBPDQImageFilter<itk::CudaImage<float, 4>,
49  itk::CudaImage<itk::CovariantVector<float, 1>, 4>>>
50 {
51 public:
52  ITK_DISALLOW_COPY_AND_MOVE(CudaLastDimensionTVDenoisingImageFilter);
53 
55  using Self = rtk::CudaLastDimensionTVDenoisingImageFilter;
56  using OutputImageType = itk::CudaImage<float, 4>;
57  using GradientType = itk::CudaImage<itk::CovariantVector<float, 1>, 4>;
61 
63  itkNewMacro(Self);
64 
66 # ifdef itkOverrideGetNameOfClassMacro
67  itkOverrideGetNameOfClassMacro(CudaLastDimensionTVDenoisingImageFilter);
68 # else
69  itkTypeMacro(CudaLastDimensionTVDenoisingImageFilter, TotalVariationDenoisingBPDQImageFilter);
70 # endif
71 
72 
73 protected:
74  CudaLastDimensionTVDenoisingImageFilter();
75  ~CudaLastDimensionTVDenoisingImageFilter() {}
76 
77  virtual void
78  GPUGenerateData();
79 
80 }; // end of class
81 
82 } // end namespace rtk
83 
84 #endif // end conditional definition of the class
85 
86 #endif
BinaryGeneratorImageFilter< TInputImage1, TInputImage2, TOutputImage > Superclass
Applies a total variation denoising, only alm_SingularValueThresholdFilterong the dimensions specifie...