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
|
/*=========================================================================
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 _itkVVWatershedModule_h
#define _itkVVWatershedModule_h
#include "vtkVVPluginAPI.h"
#include <string.h>
#include <stdlib.h>
#include "vvITKFilterModuleBase.h"
#include "itkImage.h"
#include "itkImportImageFilter.h"
#include "itkGradientMagnitudeRecursiveGaussianImageFilter.h"
#include "itkWatershedImageFilter.h"
#include "itkImageRegionConstIterator.h"
#include <vector>
namespace VolView
{
namespace PlugIn
{
template <class TInputPixelType >
class WatershedModule : public FilterModuleBase {
public:
// Pixel type of the input buffer
typedef TInputPixelType InputPixelType;
typedef float RealPixelType;
typedef unsigned char OutputPixelType;
itkStaticConstMacro( Dimension, unsigned int, 3 );
typedef itk::Image< InputPixelType, Dimension > InputImageType;
typedef itk::Image< RealPixelType, Dimension > RealImageType;
typedef itk::Image< OutputPixelType, Dimension > OutputImageType;
// 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 FilterModuleBase::RegionType RegionType;
typedef FilterModuleBase::IndexType IndexType;
typedef FilterModuleBase::SizeType SizeType;
// Instantiate the GradientMagnitude filter.
// This filter is the first stage in
// the generation of the watershed landscape.
typedef itk::GradientMagnitudeRecursiveGaussianImageFilter<
InputImageType,
RealImageType > GradientMagnitudeFilterType;
// Instantiation of the Watershed 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::WatershedImageFilter< RealImageType > WatershedFilterType;
// This structure will hold the seed points used to recolect basins
// at the end.
typedef std::vector< IndexType > SeedsContainerType;
public:
WatershedModule();
~WatershedModule();
void ClearSeeds();
void AddSeed( const IndexType & seedPosition );
void SetSigma( float value );
void SetThreshold( float value );
void SetWaterLevel( float value );
void ProcessData( const vtkVVProcessDataStruct * pds );
void PostProcessData( const vtkVVProcessDataStruct * pds );
/** This methods allows to disable postprocessing. Useful when
this module is used for initializing other level set methods */
void SetPerformPostProcessing( bool value );
private:
typename ImportFilterType::Pointer m_ImportFilter;
typename GradientMagnitudeFilterType::Pointer m_GradientMagnitudeFilter;
typename WatershedFilterType::Pointer m_WatershedFilter;
SeedsContainerType m_Seeds;
bool m_PerformPostprocessing;
};
} // end of namespace PlugIn
} // end of namespace Volview
#endif
|
