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
|
/*=========================================================================
Copyright (c) Kitware, Inc.
All rights reserved.
See Copyright.txt or http://www.kitware.com/VolViewCopyright.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.
=========================================================================*/
/** Generic interface for protocol communication between an ITK filter
and the VolView Plugin Interface */
#ifndef _itkVVCannySegmentationLevelSetModule_h
#define _itkVVCannySegmentationLevelSetModule_h
#include "vvITKFilterModuleBase.h"
#include "itkImportImageFilter.h"
#include "itkCastImageFilter.h"
#include "itkFastMarchingImageFilter.h"
#include "itkCannySegmentationLevelSetImageFilter.h"
#include "itkIntensityWindowingImageFilter.h"
namespace VolView
{
namespace PlugIn
{
template <class TInputPixelType >
class CannySegmentationLevelSetModule : public FilterModuleBase {
public:
// Pixel type of the input buffer
typedef TInputPixelType InputPixelType;
typedef unsigned char OutputPixelType;
typedef float RealPixelType;
itkStaticConstMacro(Dimension, unsigned int, 3 );
typedef itk::Image< InputPixelType, Dimension > InputImageType;
typedef itk::Image< OutputPixelType, Dimension > OutputImageType;
typedef itk::Image< RealPixelType, Dimension > RealImageType;
// Instantiate the ImportImageFilter
// This filter is used for building an ITK image using
// the data passed in a buffer.
typedef itk::ImportImageFilter< InputPixelType, Dimension > ImportFilterType;
typedef typename FilterModuleBase::SizeType SizeType;
typedef typename FilterModuleBase::IndexType IndexType;
typedef typename FilterModuleBase::RegionType RegionType;
// Instantiate the CastImageFilter
// This filter is used for converting the pixel type from the input
// data buffer into the input pixel type of the filter.
typedef itk::CastImageFilter< InputImageType,
RealImageType > CastFilterType;
// Instantiation of the FastMarching filter.
// This filter computes the propagation of the fron starting
// at the seed points. The input of the filter is the speed image.
// The output is a time-crossing map.
typedef itk::FastMarchingImageFilter< RealImageType,
RealImageType > FastMarchingFilterType;
// Type of the CannySegmentationLevelSet filter to use.
typedef itk::CannySegmentationLevelSetImageFilter<
itk::Image< RealPixelType, 3 >,
itk::Image< RealPixelType, 3 >
> CannySegmentationLevelSetFilterType;
// Instantiation of the Intensity windowing filter.
// This filter is used to remove infinite times from the non-visited
// pixels of the time-crossing map. This pixels are set to the stopping value.
typedef itk::IntensityWindowingImageFilter<
RealImageType,
OutputImageType >
IntensityWindowingFilterType;
typedef typename FastMarchingFilterType::NodeType NodeType;
typedef typename FastMarchingFilterType::NodeContainer NodeContainerType;
public:
CannySegmentationLevelSetModule();
~CannySegmentationLevelSetModule();
void ClearSeeds();
void AddSeed( const IndexType & seedPosition );
void SetDistanceFromSeeds( float value );
void SetCurvatureScaling( float value );
void SetPropagationScaling( float value );
void SetAdvectionScaling( float value );
void SetMaximumRMSError( float value );
void SetNumberOfIterations( unsigned int iterations );
void SetThreshold( float value );
void SetVariance( float value );
void ProcessData( const vtkVVProcessDataStruct * pds );
void PostProcessData( const vtkVVProcessDataStruct * pds );
const RealImageType * GetLevelSet();
unsigned int GetElapsedIterations() const;
float GetRMSChange() const;
private:
typename ImportFilterType::Pointer m_ImportFilter;
typename CastFilterType::Pointer m_CastFilter;
typename FastMarchingFilterType::Pointer m_FastMarchingImageFilter;
typename NodeContainerType::Pointer m_NodeContainer;
typename CannySegmentationLevelSetFilterType::Pointer
m_CannySegmentationLevelSetFilter;
typename IntensityWindowingFilterType::Pointer m_IntensityWindowingFilter;
double m_InitialSeedValue;
unsigned long m_CurrentNumberOfSeeds;
bool m_PerformPostprocessing;
};
} // end of namespace PlugIn
} // end of namespace Volview
#endif
|
