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
|
/*=========================================================================
*
* 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 itkMinMaxCurvatureFlowFunction_h
#define itkMinMaxCurvatureFlowFunction_h
#include "itkCurvatureFlowFunction.h"
#include "itkNeighborhoodOperator.h"
namespace itk
{
/**
* \class MinMaxCurvatureFlowFunction
*
* This class encapsulate the finite difference equation which drives a
* min/max curvature flow denoising algorithm.
*
* This class uses a zero flux Neumann boundary condition when computing
* derivatives near the data boundary.
*
* This class operates as part of the finite difference solver hierarchy.
*
* \sa MinMaxCurvatureFlowImageFilter
* \sa ZeroFluxNeumannBoundaryCondition
* \ingroup FiniteDifferenceFunctions
* \ingroup ITKCurvatureFlow
*/
template <typename TImage>
class ITK_TEMPLATE_EXPORT MinMaxCurvatureFlowFunction : public CurvatureFlowFunction<TImage>
{
public:
ITK_DISALLOW_COPY_AND_MOVE(MinMaxCurvatureFlowFunction);
/** Standard class type aliases. */
using Self = MinMaxCurvatureFlowFunction;
using Superclass = CurvatureFlowFunction<TImage>;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(MinMaxCurvatureFlowFunction);
/** Inherit some parameters from the superclass type. */
using typename Superclass::ImageType;
using typename Superclass::PixelType;
using typename Superclass::RadiusType;
using typename Superclass::NeighborhoodType;
using typename Superclass::FloatOffsetType;
/** Extract superclass dimension. */
static constexpr unsigned int ImageDimension = Superclass::ImageDimension;
/** Typedef support for the stencil radius. */
using RadiusValueType = typename RadiusType::SizeValueType;
/** Set/Get the stencil radius. */
void
SetStencilRadius(const RadiusValueType value);
const RadiusValueType &
GetRadiusValueType() const
{
return m_StencilRadius;
}
/** Convenience function for symmetry with SetStencilRadius. */
const RadiusValueType &
GetStencilRadius() const
{
return GetRadiusValueType();
}
/** Compute the solution update for each pixel that does not lie on a the data set boundary. */
PixelType
ComputeUpdate(const NeighborhoodType & it,
void * globalData,
const FloatOffsetType & offset = FloatOffsetType(0.0)) override;
protected:
MinMaxCurvatureFlowFunction();
~MinMaxCurvatureFlowFunction() override = default;
using StencilOperatorType = Neighborhood<PixelType, Self::ImageDimension>;
StencilOperatorType m_StencilOperator{};
/** Initialize the stencil operator to be an N-Dimensional sphere
* of radius m_StencilRadius. */
void
InitializeStencilOperator();
private:
RadiusValueType m_StencilRadius{};
// To control overloaded versions of ComputeThreshold
struct DispatchBase
{};
template <int VDimension>
struct Dispatch : public DispatchBase
{};
/** Compute the threshold by averaging the intensity in direction perpendicular to the image gradient. */
PixelType
ComputeThreshold(const Dispatch<2> &, const NeighborhoodType & it) const;
PixelType
ComputeThreshold(const Dispatch<3> &, const NeighborhoodType & it) const;
PixelType
ComputeThreshold(const DispatchBase &, const NeighborhoodType & it) const;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkMinMaxCurvatureFlowFunction.hxx"
#endif
#endif
|
