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 167 168 169 170
|
/*=========================================================================
*
* 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 itkLabeledPointSetToPointSetMetricv4_h
#define itkLabeledPointSetToPointSetMetricv4_h
#include "itkPointSetToPointSetMetricv4.h"
#include <vector>
namespace itk
{
/** \class LabeledPointSetToPointSetMetricv4
* \brief Computes the distance metric and gradient between two
* labeled point sets.
*
* This class is generic in that it takes an unlabeled point set metric,
* such as one of the following options
*
* \li \c EuclideanDistancePointSetToPointSetMetricv4 (default)
* \li \c ExpectationBasedPointSetToPointSetMetricv4
* \li \c JensenHavrdaCharvatTsallisPointSetToPointSetMetricv4
*
* where it is assumed that the point set \c PixelType associated with
* each point is a label (i.e. \c LabelEnum). This class is used to find
* the total metric and total derivative based on the matching between the
* separate fixed and moving labeled point sets. This class first determines
* the common label set between the fixed and moving point sets. For each
* common label, the specified unlabeled point set metric is cloned and the
* fixed and moving point subset associated with that common label is specified
* as the fixed and moving point set input. Functionality in the base class
* (\c PointSetToPointSetMetric) is used to accumulated the value and gradient.
*
* \author Nick Tustison
*
* \ingroup ITKMetricsv4
*/
template <typename TFixedPointSet,
typename TMovingPointSet = TFixedPointSet,
class TInternalComputationValueType = double>
class ITK_TEMPLATE_EXPORT LabeledPointSetToPointSetMetricv4
: public PointSetToPointSetMetricv4<TFixedPointSet, TMovingPointSet, TInternalComputationValueType>
{
public:
ITK_DISALLOW_COPY_AND_MOVE(LabeledPointSetToPointSetMetricv4);
/** Standard class type aliases. */
using Self = LabeledPointSetToPointSetMetricv4;
using Superclass = PointSetToPointSetMetricv4<TFixedPointSet, TMovingPointSet, TInternalComputationValueType>;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(LabeledPointSetToPointSetMetricv4);
using FixedPointSetType = TFixedPointSet;
using FixedPointSetPointer = typename FixedPointSetType::Pointer;
using MovingPointSetType = TMovingPointSet;
using MovingPointSetPointer = typename MovingPointSetType::Pointer;
/** Types transferred from the base class */
using typename Superclass::MeasureType;
using typename Superclass::DerivativeType;
using typename Superclass::LocalDerivativeType;
using typename Superclass::PointType;
using typename Superclass::PointIdentifier;
using LabelType = typename Superclass::PixelType;
using LabelSetType = std::vector<LabelType>;
using PointSetMetricType = Superclass;
using PointSetMetricPointer = typename PointSetMetricType::Pointer;
/**
* Initialize the metric by making sure that all the components
* are present and plugged together correctly.
*/
void
Initialize() override;
/**
* Calculates the local metric value for a single point. The label type
* is used to segregate the computation.
*/
MeasureType
GetLocalNeighborhoodValue(const PointType &, const LabelType &) const override;
/**
* Calculates the local value and derivative for a single point. The label type
* is used to segregate the computation.
*/
void
GetLocalNeighborhoodValueAndDerivative(const PointType &,
MeasureType &,
LocalDerivativeType &,
const LabelType &) const override;
/**
* Set/get the specific unlabeled point set metric type. Default is
* the \c EuclideanDistancePointSetToPointSetMetricv4.
*/
itkSetObjectMacro(PointSetMetric, PointSetMetricType);
itkGetModifiableObjectMacro(PointSetMetric, PointSetMetricType);
/**
* Ensure label type is an integer type
*/
itkConceptMacro(LabelTypeIsInteger, (Concept::IsInteger<LabelType>));
protected:
LabeledPointSetToPointSetMetricv4();
~LabeledPointSetToPointSetMetricv4() override = default;
/** PrintSelf function */
void
PrintSelf(std::ostream & os, Indent indent) const override;
private:
/**
* Private function to find the common label set for the moving
* and fixed point sets.
*/
void
DetermineCommonPointSetLabels();
/**
* Private function to create a fixed point set from the input fixed point
* set with a specific label
*/
FixedPointSetPointer
GetLabeledFixedPointSet(const LabelType) const;
/**
* Private function to create a moving point set from the input moving point
* set with a specific label
*/
MovingPointSetPointer
GetLabeledMovingPointSet(const LabelType) const;
PointSetMetricPointer m_PointSetMetric{};
std::vector<PointSetMetricPointer> m_PointSetMetricClones{};
LabelSetType m_FixedPointSetLabels{};
LabelSetType m_MovingPointSetLabels{};
LabelSetType m_CommonPointSetLabels{};
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkLabeledPointSetToPointSetMetricv4.hxx"
#endif
#endif
|