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 159 160 161 162 163 164 165 166
|
/*=========================================================================
*
* Copyright NumFOCUS
*
* 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
*
* https://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_MOVE(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);
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(CenteredTransformInitializer);
/** 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() = default;
~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{ false };
const FixedImageCalculatorPointer m_FixedCalculator{ FixedImageCalculatorType::New() };
const MovingImageCalculatorPointer m_MovingCalculator{ MovingImageCalculatorType::New() };
}; // class CenteredTransformInitializer
} // namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkCenteredTransformInitializer.hxx"
#endif
#endif /* itkCenteredTransformInitializer_h */
|