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
|
/*=========================================================================
Program: Visualization Toolkit
Module: vtkImageAnisotropicDiffusion2D.h
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/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 notice for more information.
=========================================================================*/
// .NAME vtkImageAnisotropicDiffusion2D - edge preserving smoothing.
//
// .SECTION Description
// vtkImageAnisotropicDiffusion2D diffuses a 2d image iteratively.
// The neighborhood of the diffusion is determined by the instance
// flags. If "Edges" is on the 4 edge connected voxels
// are included, and if "Corners" is on, the 4 corner connected voxels
// are included. "DiffusionFactor" determines how far a pixel value
// moves toward its neighbors, and is insensitive to the number of
// neighbors chosen. The diffusion is anisotropic because it only occurs
// when a gradient measure is below "GradientThreshold". Two gradient measures
// exist and are toggled by the "GradientMagnitudeThreshold" flag.
// When "GradientMagnitudeThreshold" is on, the magnitude of the gradient,
// computed by central differences, above "DiffusionThreshold"
// a voxel is not modified. The alternative measure examines each
// neighbor independently. The gradient between the voxel and the neighbor
// must be below the "DiffusionThreshold" for diffusion to occur with
// THAT neighbor.
// .SECTION See Also
// vtkImageAnisotropicDiffusion3D
#ifndef vtkImageAnisotropicDiffusion2D_h
#define vtkImageAnisotropicDiffusion2D_h
#include "vtkImagingGeneralModule.h" // For export macro
#include "vtkImageSpatialAlgorithm.h"
class VTKIMAGINGGENERAL_EXPORT vtkImageAnisotropicDiffusion2D : public vtkImageSpatialAlgorithm
{
public:
static vtkImageAnisotropicDiffusion2D *New();
vtkTypeMacro(vtkImageAnisotropicDiffusion2D,vtkImageSpatialAlgorithm);
void PrintSelf(ostream& os, vtkIndent indent);
// Description:
// This method sets the number of interations which also affects the
// input neighborhood needed to compute one output pixel. Each iterations
// requires an extra pixel layer on the neighborhood. This is only relavent
// when you are trying to stream or are requesting a sub extent of the "wholeExtent".
void SetNumberOfIterations(int num);
// Description:
// Get the number of iterations.
vtkGetMacro(NumberOfIterations,int);
// Description:
// Set/Get the difference threshold that stops diffusion.
// when the difference between two pixel is greater than this threshold,
// the pixels are not diffused. This causes diffusion to avoid sharp edges.
// If the GradientMagnitudeThreshold is set, then gradient magnitude is used
// for comparison instead of pixel differences.
vtkSetMacro(DiffusionThreshold,double);
vtkGetMacro(DiffusionThreshold,double);
// Description:
// The diffusion factor specifies how much neighboring pixels effect each other.
// No diffusion occurs with a factor of 0, and a diffusion factor of 1 causes
// the pixel to become the average of all its neighbors.
vtkSetMacro(DiffusionFactor,double);
vtkGetMacro(DiffusionFactor,double);
// Description:
// Choose neighbors to diffuse (6 faces, 12 edges, 8 corners).
vtkSetMacro(Faces,int);
vtkGetMacro(Faces,int);
vtkBooleanMacro(Faces,int);
vtkSetMacro(Edges,int);
vtkGetMacro(Edges,int);
vtkBooleanMacro(Edges,int);
vtkSetMacro(Corners,int);
vtkGetMacro(Corners,int);
vtkBooleanMacro(Corners,int);
// Description:
// Switch between gradient magnitude threshold and pixel gradient threshold.
vtkSetMacro(GradientMagnitudeThreshold,int);
vtkGetMacro(GradientMagnitudeThreshold,int);
vtkBooleanMacro(GradientMagnitudeThreshold,int);
protected:
vtkImageAnisotropicDiffusion2D();
~vtkImageAnisotropicDiffusion2D() {}
int NumberOfIterations;
double DiffusionThreshold;
double DiffusionFactor;
// to determine which neighbors to diffuse
int Faces;
int Edges;
int Corners;
// What threshold to use
int GradientMagnitudeThreshold;
void ThreadedRequestData(vtkInformation *request,
vtkInformationVector **inputVector,
vtkInformationVector *outputVector,
vtkImageData ***inData, vtkImageData **outData,
int extent[6], int id);
void Iterate(vtkImageData *in, vtkImageData *out,
double ar0, double ar1, int *coreExtent, int count);
private:
vtkImageAnisotropicDiffusion2D(const vtkImageAnisotropicDiffusion2D&); // Not implemented.
void operator=(const vtkImageAnisotropicDiffusion2D&); // Not implemented.
};
#endif
|
