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
|
/*=========================================================================
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 _itkVVWatershedRGBModule_h
#define _itkVVWatershedRGBModule_h
#include "vtkVVPluginAPI.h"
#include <string.h>
#include <stdlib.h>
#include "vvITKFilterModuleBase.h"
#include "itkImage.h"
#include "itkImportImageFilter.h"
#include "itkWatershedImageFilter.h"
#include "itkImageRegionConstIterator.h"
#include "itkRGBPixel.h"
#include "itkScalarToRGBPixelFunctor.h"
#include "itkUnaryFunctorImageFilter.h"
#include "itkCastImageFilter.h"
#include <vector>
namespace VolView
{
namespace PlugIn
{
template <class TInputPixelType >
class WatershedRGBModule : 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 Cast filter.
// This filter is the first stage in
// the generation of the watershed landscape.
typedef itk::CastImageFilter< InputImageType,
RealImageType > CastFilterType;
// 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;
// Type of the output labeled image
typedef typename WatershedFilterType::OutputImageType LabeledImageType;
// Type declaration for RGB image that will hold the color
// encoding of the water basins
typedef itk::RGBPixel< unsigned char > RGBPixelType;
typedef itk::Image< RGBPixelType, Dimension > RGBImageType;
typedef itk::Functor::ScalarToRGBPixelFunctor<unsigned long>
ColorMapFunctorType;
typedef itk::UnaryFunctorImageFilter< LabeledImageType,
RGBImageType,
ColorMapFunctorType
> ColorMapFilterType;
public:
WatershedRGBModule();
~WatershedRGBModule();
void SetWaterLevel( float value );
void ProcessData( const vtkVVProcessDataStruct * pds );
void CopyOutputData( const vtkVVProcessDataStruct * pds );
private:
typename ImportFilterType::Pointer m_ImportFilter;
typename CastFilterType::Pointer m_CastFilter;
typename WatershedFilterType::Pointer m_WatershedFilter;
typename ColorMapFilterType::Pointer m_ColorEncoder;
};
} // end of namespace PlugIn
} // end of namespace Volview
#endif
|
