forked from InsightSoftwareConsortium/ITK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
itkCenteredTransformInitializer.h
158 lines (127 loc) · 5.78 KB
/
itkCenteredTransformInitializer.h
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
158
/*=========================================================================
*
* 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 itkCenteredTransformInitializer_h
#define itkCenteredTransformInitializer_h
#include "itkObject.h"
#include "itkObjectFactory.h"
#include "itkImageMomentsCalculator.h"
#include <iostream>
namespace itk
{
/** \class CenteredTransformInitializer
* \brief CenteredTransformInitializer is a helper class intended to
* initialize the center of rotation and the translation of Transforms having
* the center of rotation among their parameters.
*
* This class is connected to the fixed image, moving image and transform
* involved in the registration. Two modes of operation are possible:
*
* - Geometrical,
* - Center of mass
*
* In the first mode, the geometrical center of the moving image is passed as
* initial center of rotation to the transform and the vector from the center
* of the fixed image to the center of the moving image is passed as the
* initial translation. This mode basically assumes that the anatomical objects
* to be registered are centered in their respective images. Hence the best
* initial guess for the registration is the one that superimposes those two
* centers.
*
* In the second mode, the moments of gray level values are computed
* for both images. The center of mass of the moving image is then
* used as center of rotation. The vector between the two centers of
* mass is passes as the initial translation to the transform. This
* second approach assumes that the moments of the anatomical objects
* are similar for both images and hence the best initial guess for
* registration is to superimpose both mass centers. Note that this
* assumption will probably not hold in multi-modality registration.
*
* \ingroup ITKRegistrationCommon
* \ingroup ITKTransform
*/
template< typename TTransform,
typename TFixedImage,
typename TMovingImage >
class ITK_TEMPLATE_EXPORT CenteredTransformInitializer:public Object
{
public:
ITK_DISALLOW_COPY_AND_ASSIGN(CenteredTransformInitializer);
/** Standard class type aliases. */
using Self = CenteredTransformInitializer;
using Superclass = Object;
using Pointer = SmartPointer< Self >;
using ConstPointer = SmartPointer< const Self >;
/** New macro for creation of through a Smart Pointer. */
itkNewMacro(Self);
/** Run-time type information (and related methods). */
itkTypeMacro(CenteredTransformInitializer, Object);
/** Type of the transform to initialize */
using TransformType = TTransform;
using TransformPointer = typename TransformType::Pointer;
/** Dimension of parameters. */
static constexpr unsigned int InputSpaceDimension = TransformType::InputSpaceDimension;
static constexpr unsigned int OutputSpaceDimension = TransformType::OutputSpaceDimension;
/** Image Types to use in the initialization of the transform */
using FixedImageType = TFixedImage;
using MovingImageType = TMovingImage;
using FixedImagePointer = typename FixedImageType::ConstPointer;
using MovingImagePointer = typename MovingImageType::ConstPointer;
/** Moment calculators */
using FixedImageCalculatorType = ImageMomentsCalculator< FixedImageType >;
using MovingImageCalculatorType = ImageMomentsCalculator< MovingImageType >;
using FixedImageCalculatorPointer = typename FixedImageCalculatorType::Pointer;
using MovingImageCalculatorPointer = typename MovingImageCalculatorType::Pointer;
/** Offset type. */
using OffsetType = typename TransformType::OffsetType;
/** Point type. */
using InputPointType = typename TransformType::InputPointType;
/** Vector type. */
using OutputVectorType = typename TransformType::OutputVectorType;
/** Set the transform to be initialized */
itkSetObjectMacro(Transform, TransformType);
/** Set the fixed image used in the registration process */
itkSetConstObjectMacro(FixedImage, FixedImageType);
/** Set the moving image used in the registration process */
itkSetConstObjectMacro(MovingImage, MovingImageType);
/** Initialize the transform using data from the images */
virtual void InitializeTransform();
/** Select between using the geometrical center of the images or
using the center of mass given by the image intensities. */
void GeometryOn() { m_UseMoments = false; }
void MomentsOn() { m_UseMoments = true; }
/** Get() access to the moments calculators */
itkGetModifiableObjectMacro(FixedCalculator, FixedImageCalculatorType);
itkGetModifiableObjectMacro(MovingCalculator, MovingImageCalculatorType);
protected:
CenteredTransformInitializer();
~CenteredTransformInitializer() override = default;
void PrintSelf(std::ostream & os, Indent indent) const override;
itkGetModifiableObjectMacro(Transform, TransformType);
private:
TransformPointer m_Transform;
FixedImagePointer m_FixedImage;
MovingImagePointer m_MovingImage;
bool m_UseMoments;
FixedImageCalculatorPointer m_FixedCalculator;
MovingImageCalculatorPointer m_MovingCalculator;
}; //class CenteredTransformInitializer
} // namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkCenteredTransformInitializer.hxx"
#endif
#endif /* itkCenteredTransformInitializer_h */