1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
/*=========================================================================
*
* Copyright UMC Utrecht and contributors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
/*=========================================================================
*
* Copyright Insight Software Consortium
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkGPUImageToImageFilter_h
#define itkGPUImageToImageFilter_h
#include "itkImageToImageFilter.h"
#include "itkOpenCLKernelManager.h"
namespace itk
{
/** \class GPUImageToImageFilter
*
* \brief class to abstract the behaviour of the GPU filters.
*
* GPUImageToImageFilter is the GPU version of ImageToImageFilter.
* This class can accept both CPU and GPU image as input and output,
* and apply filter accordingly. If GPU is available for use, then
* GPUGenerateData() is called. Otherwise, GenerateData() in the
* parent class (i.e., ImageToImageFilter) will be called.
*
* \note This file was taken from ITK 4.1.0.
* It was modified by Denis P. Shamonin and Marius Staring.
* Division of Image Processing,
* Department of Radiology, Leiden, The Netherlands.
* Added functionality is described in the Insight Journal paper:
* http://hdl.handle.net/10380/3393
*
* \ingroup ITKGPUCommon
*/
template <typename TInputImage,
typename TOutputImage,
typename TParentImageFilter = ImageToImageFilter<TInputImage, TOutputImage>>
class ITK_TEMPLATE_EXPORT ITKOpenCL_EXPORT GPUImageToImageFilter : public TParentImageFilter
{
public:
ITK_DISALLOW_COPY_AND_MOVE(GPUImageToImageFilter);
/** Standard class typedefs. */
using Self = GPUImageToImageFilter;
using Superclass = TParentImageFilter;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Run-time type information (and related methods). */
itkTypeMacro(GPUImageToImageFilter, TParentImageFilter);
/** Superclass typedefs. */
using typename Superclass::DataObjectIdentifierType;
using typename Superclass::OutputImageRegionType;
using typename Superclass::OutputImagePixelType;
/** Some convenient typedefs. */
using InputImageType = TInputImage;
using InputImagePointer = typename InputImageType::Pointer;
using InputImageConstPointer = typename InputImageType::ConstPointer;
using InputImageRegionType = typename InputImageType::RegionType;
using InputImagePixelType = typename InputImageType::PixelType;
using OutputImageType = TOutputImage;
/** ImageDimension constants */
itkStaticConstMacro(InputImageDimension, unsigned int, TInputImage::ImageDimension);
itkStaticConstMacro(OutputImageDimension, unsigned int, TOutputImage::ImageDimension);
// macro to set if GPU is used
itkSetMacro(GPUEnabled, bool);
itkGetConstMacro(GPUEnabled, bool);
itkBooleanMacro(GPUEnabled);
void
GraftOutput(DataObject * graft) override;
void
GraftOutput(const DataObjectIdentifierType & key, DataObject * graft) override;
void
SetNumberOfWorkUnits(ThreadIdType _arg) override;
protected:
GPUImageToImageFilter();
~GPUImageToImageFilter() override = default;
void
PrintSelf(std::ostream & os, Indent indent) const override;
void
GenerateData() override;
virtual void
GPUGenerateData()
{}
// GPU kernel manager
typename OpenCLKernelManager::Pointer m_GPUKernelManager{};
private:
bool m_GPUEnabled{};
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkGPUImageToImageFilter.hxx"
#endif
#endif
|
