forked from InsightSoftwareConsortium/ITK
-
Notifications
You must be signed in to change notification settings - Fork 0
/
itkShapePriorSegmentationLevelSetFunction.h
158 lines (129 loc) · 5.99 KB
/
itkShapePriorSegmentationLevelSetFunction.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
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkShapePriorSegmentationLevelSetFunction.h
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/Copyright.htm for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notices for more information.
=========================================================================*/
#ifndef __itkShapePriorSegmentationLevelSetFunction_h_
#define __itkShapePriorSegmentationLevelSetFunction_h_
#include "itkSegmentationLevelSetFunction.h"
#include "itkShapeSignedDistanceFunction.h"
namespace itk {
/** \class ShapePriorSegmentationLevelSetFunction
*
* \brief This function is used in ShapePriorSegmentationLevelSetFilter to
* segment structures in an image based on user supplied edge potential map and
* shape model.
*
* This class extends the basic LevelSetFunction with a shape prior term
* as developed in [1].
*
* \f$ \zeta( \phi^{*} - \phi) \f$
*
* where \f$ \phi^{*} \f$ is the signed distance function from a target shape
* and \f$ \zeta \f$ is a scalar constant.
*
* The target shape signed distance function is supplied through a
* ShapeSignedDistanceFunction object.
*
* \sa LevelSetFunction
* \sa SegmentationLevelSetImageFunction
* \sa ShapeSignedDistanceFunction
*
* \par REFERENCES
* \par
* [1] Leventon, M.E. et al. "Statistical Shape Influence in Geodesic Active Contours", CVPR 2000.
*
* \ingroup FiniteDifferenceFunctions
*/
template <class TImageType, class TFeatureImageType = TImageType>
class ITK_EXPORT ShapePriorSegmentationLevelSetFunction
: public SegmentationLevelSetFunction<TImageType, TFeatureImageType>
{
public:
/** Standard class typedefs. */
typedef ShapePriorSegmentationLevelSetFunction Self;
typedef SegmentationLevelSetFunction<TImageType,TFeatureImageType> Superclass;
typedef SmartPointer<Self> Pointer;
typedef SmartPointer<const Self> ConstPointer;
typedef TFeatureImageType FeatureImageType;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Run-time type information (and related methods) */
itkTypeMacro( ShapePriorSegmentationLevelSetFunction, SegmentationLevelSetFunction );
/** Extract some parameters from the superclass. */
typedef typename Superclass::ImageType ImageType;
typedef typename Superclass::NeighborhoodType NeighborhoodType;
typedef typename Superclass::ScalarValueType ScalarValueType;
typedef typename Superclass::FeatureScalarType FeatureScalarType;
typedef typename Superclass::RadiusType RadiusType;
typedef typename Superclass::FloatOffsetType FloatOffsetType;
typedef typename Superclass::VectorImageType VectorImageType;
typedef typename Superclass::PixelType PixelType;
typedef typename Superclass::TimeStepType TimeStepType;
typedef typename Superclass::IndexType IndexType;
typedef typename Superclass::ContinuousIndexType ContinuousIndexType;
/** Extract some parameters from the superclass. */
itkStaticConstMacro(ImageDimension, unsigned int,
Superclass::ImageDimension);
/** ShapeFunction typedef support. */
typedef ShapeSignedDistanceFunction<double,
itkGetStaticConstMacro(ImageDimension)> ShapeFunctionType;
typedef typename ShapeFunctionType::ConstPointer ShapeFunctionPointer;
/** Zeta. The ShapePriorWeight scales the shape prior term values. */
void SetShapePriorWeight( const ScalarValueType p )
{ m_ShapePriorWeight = p; }
ScalarValueType GetShapePriorWeight() const
{ return m_ShapePriorWeight; }
/** The ShapeFunction encapsulates the signed distance to the shape used to
* influence the evolution of the level set. */
void SetShapeFunction( const ShapeFunctionType * ptr )
{ m_ShapeFunction = ptr; }
const ShapeFunctionType * GetShapeFunction() const
{ return m_ShapeFunction; }
/** Compute the equation value with the additional shape prior term. */
virtual PixelType ComputeUpdate(const NeighborhoodType &neighborhood,
void *globalData,
const FloatOffsetType& = FloatOffsetType(0.0));
/** Compute global time step from the global data structure. */
virtual TimeStepType ComputeGlobalTimeStep( void *globalData ) const;
/** A global data type used to store values needed to compute the time step. */
typedef typename Superclass::GlobalDataStruct GlobalDataStruct;
struct ShapePriorGlobalDataStruct : public GlobalDataStruct
{
ScalarValueType m_MaxShapePriorChange;
};
/** Returns a pointer to a global data structure for computing time step. */
virtual void *GetGlobalDataPointer() const
{
ShapePriorGlobalDataStruct *ans = new ShapePriorGlobalDataStruct();
ans->m_MaxAdvectionChange = NumericTraits<ScalarValueType>::Zero;
ans->m_MaxPropagationChange = NumericTraits<ScalarValueType>::Zero;
ans->m_MaxCurvatureChange = NumericTraits<ScalarValueType>::Zero;
ans->m_MaxShapePriorChange = NumericTraits<ScalarValueType>::Zero;
return ans;
}
/** Release the global data structure. */
virtual void ReleaseGlobalDataPointer(void *GlobalData) const
{ delete (ShapePriorGlobalDataStruct *) GlobalData; }
protected:
ShapePriorSegmentationLevelSetFunction();
virtual ~ShapePriorSegmentationLevelSetFunction() {}
ShapePriorSegmentationLevelSetFunction(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
void PrintSelf(std::ostream& os, Indent indent) const;
private:
ShapeFunctionPointer m_ShapeFunction;
ScalarValueType m_ShapePriorWeight;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkShapePriorSegmentationLevelSetFunction.txx"
#endif
#endif