File: itkGPUCompositeTransformBase.h

package info (click to toggle)
elastix 5.2.0-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 42,480 kB
  • sloc: cpp: 68,403; lisp: 4,118; python: 1,013; xml: 182; sh: 177; makefile: 33
file content (157 lines) | stat: -rw-r--r-- 5,624 bytes parent folder | download
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
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
/*=========================================================================
 *
 *  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.
 *
 *=========================================================================*/
#ifndef itkGPUCompositeTransformBase_h
#define itkGPUCompositeTransformBase_h

#include "itkGPUTransformBase.h"
#include "itkTransform.h"

namespace itk
{
/** \class GPUCompositeTransformBaseBase
 * \brief Base class for all GPU composite transforms.
 *
 * \author Denis P. Shamonin and Marius Staring. Division of Image Processing,
 * Department of Radiology, Leiden, The Netherlands
 *
 * \note This work was funded by the Netherlands Organisation for
 * Scientific Research (NWO NRG-2010.02 and NWO 639.021.124).
 *
 * \ingroup GPUCommon
 */
template <typename TScalarType = float, unsigned int NDimensions = 3>
class ITK_TEMPLATE_EXPORT GPUCompositeTransformBase : public GPUTransformBase
{
public:
  /** Standard class typedefs. */
  using Self = GPUCompositeTransformBase;
  using GPUSuperclass = GPUTransformBase;

  /** Run-time type information (and related methods). */
  itkTypeMacro(GPUCompositeTransformBase, GPUSuperclass);

  /** Sub transform types. */
  using ScalarType = TScalarType;
  using TransformType = Transform<TScalarType, NDimensions, NDimensions>;
  using TransformTypePointer = typename TransformType::Pointer;
  using TransformTypeConstPointer = typename TransformType::ConstPointer;

  /** Dimension of the domain space. */
  itkStaticConstMacro(InputSpaceDimension, unsigned int, NDimensions);
  itkStaticConstMacro(OutputSpaceDimension, unsigned int, NDimensions);

  /** Get number of transforms in composite transform. */
  virtual SizeValueType
  GetNumberOfTransforms() const = 0;

  /** Get the Nth transform. */
  virtual const TransformTypePointer
  GetNthTransform(SizeValueType n) const = 0;

  /** Returns true if the derived composite transform has identity transform,
   * false otherwise. */
  virtual bool
  HasIdentityTransform() const;

  /** Returns true if the derived composite transform has matrix offset transform,
   * false otherwise. */
  virtual bool
  HasMatrixOffsetTransform() const;

  /** Returns true if the derived composite transform has translation transform,
   * false otherwise. */
  virtual bool
  HasTranslationTransform() const;

  /** Returns true if the derived composite transform has BSpline transform,
   * false otherwise. */
  virtual bool
  HasBSplineTransform() const;

  /** Returns true if the transform at \a index is identity transform,
   * false otherwise. */
  virtual bool
  IsIdentityTransform(const std::size_t index) const;

  /** Returns true if the transform at \a index is matrix offset transform,
   * false otherwise. */
  virtual bool
  IsMatrixOffsetTransform(const std::size_t index) const;

  /** Returns true if the transform at \a index is translation transform,
   * false otherwise. */
  virtual bool
  IsTranslationTransform(const std::size_t index) const;

  /** Returns true if the transform at \a index is BSpline transform,
   * false otherwise. */
  virtual bool
  IsBSplineTransform(const std::size_t index) const;

protected:
  GPUCompositeTransformBase() = default;
  ~GPUCompositeTransformBase() override = default;

  /** Returns OpenCL \a source code for the transform.
   * Returns true if source code was combined, false otherwise. */
  bool
  GetSourceCode(std::string & source) const override;

  /** Returns data manager that stores all settings for the transform \a index.
   * Used by combination transforms. */
  GPUDataManager::Pointer
  GetParametersDataManager(const std::size_t index) const override;

private:
  GPUCompositeTransformBase(const Self & other) = delete;
  const Self &
  operator=(const Self &) = delete;

  /** \internal
   * Returns true if the transform at \a index is identity transform,
   * false otherwise. If \a loadSource is true, the OpenCL \a source code is loaded. */
  bool
  IsIdentityTransform(const std::size_t index, const bool loadSource, std::string & source) const;

  /** \internal
   * Returns true if the transform at \a index is matrix offset transform,
   * false otherwise. If \a loadSource is true, the OpenCL \a source code is loaded. */

  bool
  IsMatrixOffsetTransform(const std::size_t index, const bool loadSource, std::string & source) const;

  /** \internal
   * Returns true if the transform at \a index is translation transform,
   * false otherwise. If \a loadSource is true, the OpenCL \a source code is loaded. */
  bool
  IsTranslationTransform(const std::size_t index, const bool loadSource, std::string & source) const;

  /** \internal
   * Returns true if the transform at \a index is BSpline transform,
   * false otherwise. If \a loadSource is true, the OpenCL \a source code is loaded. */
  bool
  IsBSplineTransform(const std::size_t index, const bool loadSource, std::string & source) const;
};

} // end namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#  include "itkGPUCompositeTransformBase.hxx"
#endif

#endif /* itkGPUCompositeTransformBase_h */