RTK  2.7.0
Reconstruction Toolkit
rtkTotalVariationDenoiseSequenceImageFilter.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 rtkTotalVariationDenoiseSequenceImageFilter_h
20 #define rtkTotalVariationDenoiseSequenceImageFilter_h
21 
22 #include "rtkConstantImageSource.h"
23 
24 #include <itkExtractImageFilter.h>
25 #include <itkPasteImageFilter.h>
26 #include <itkCastImageFilter.h>
27 
28 #ifdef RTK_USE_CUDA
30 #else
32 #endif
33 
34 namespace rtk
35 {
78 template <typename TImageSequence>
80  : public itk::ImageToImageFilter<TImageSequence, TImageSequence>
81 {
82 public:
83  ITK_DISALLOW_COPY_AND_MOVE(TotalVariationDenoiseSequenceImageFilter);
84 
89 
91  itkNewMacro(Self);
92 
94  itkOverrideGetNameOfClassMacro(TotalVariationDenoiseSequenceImageFilter);
95 
97  itkGetMacro(Gamma, double);
98  itkSetMacro(Gamma, double);
100 
101  itkGetMacro(NumberOfIterations, int);
102  itkSetMacro(NumberOfIterations, int);
103 
104  void
105  SetDimensionsProcessed(bool * arg);
106 
109  using ImageType = typename TImageSequence::template RebindImageType<typename TImageSequence::PixelType,
110  TImageSequence::ImageDimension - 1>;
111 #ifdef RTK_USE_CUDA
112  using TVDenoisingFilterType = typename std::conditional_t<std::is_same_v<TImageSequence, CPUImageSequenceType>,
115 #else
117 #endif
122 
123 protected:
125  ~TotalVariationDenoiseSequenceImageFilter() override = default;
126 
128  void
129  GenerateData() override;
130 
131  void
132  GenerateOutputInformation() override;
133  void
134  GenerateInputRequestedRegion() override;
135 
137  typename TVDenoisingFilterType::Pointer m_TVDenoisingFilter;
142 
144  typename TImageSequence::RegionType m_ExtractAndPasteRegion;
145 
147  double m_Gamma{ 1. };
148  int m_NumberOfIterations{ 1 };
149  bool m_DimensionsProcessed[ImageType::ImageDimension];
150 };
151 } // namespace rtk
152 
153 
154 #ifndef ITK_MANUAL_INSTANTIATION
155 # include "rtkTotalVariationDenoiseSequenceImageFilter.hxx"
156 #endif
157 
158 #endif
Implements the TotalVariationDenoisingBPDQImageFilter on GPU.
Generate an n-dimensional image with constant pixel values.
Applies 3D total variation denoising to a 3D + time sequence of images.
Applies a total variation denoising, only alm_SingularValueThresholdFilterong the dimensions specifie...
typename std::conditional_t< std::is_same_v< TImageSequence, CPUImageSequenceType >, TotalVariationDenoisingBPDQImageFilter< ImageType >, CudaTotalVariationDenoisingBPDQImageFilter > TVDenoisingFilterType
#define itkSetMacro(name, type)
typename itk::Image< typename TImageSequence::PixelType, TImageSequence::ImageDimension > CPUImageSequenceType
typename TImageSequence::template RebindImageType< typename TImageSequence::PixelType, TImageSequence::ImageDimension - 1 > ImageType