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
|
/*=========================================================================
*
* 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 itkLabelToRGBImageFilter_h
#define itkLabelToRGBImageFilter_h
#include "itkUnaryFunctorImageFilter.h"
#include "itkLabelToRGBFunctor.h"
namespace itk
{
/**
* \class LabelToRGBImageFilter
* \brief Apply a colormap to a label image
*
* Apply a colormap to a label image. The set of colors
* is a good selection of distinct colors. The user can choose to use a background
* value. In that case, a gray pixel with the same intensity than the background
* label is produced.
*
* This code was contributed in the Insight Journal paper:
* "The watershed transform in ITK - discussion and new developments"
* by Beare R., Lehmann G.
* https://www.insight-journal.org/browse/publication/92
*
* \author Gaetan Lehmann. Biologie du Developpement et de la Reproduction, INRA de Jouy-en-Josas, France.
* \author Richard Beare. Department of Medicine, Monash University, Melbourne, Australia.
*
* \sa LabelOverlayImageFilter
* \sa LabelMapToRGBImageFilter, LabelToRGBFunctor, ScalarToRGBPixelFunctor
* \ingroup MultiThreaded
*
* \ingroup ITKImageFusion
*/
template <typename TLabelImage, typename TOutputImage>
class ITK_TEMPLATE_EXPORT LabelToRGBImageFilter
: public UnaryFunctorImageFilter<
TLabelImage,
TOutputImage,
Functor::LabelToRGBFunctor<typename TLabelImage::PixelType, typename TOutputImage::PixelType>>
{
public:
ITK_DISALLOW_COPY_AND_MOVE(LabelToRGBImageFilter);
/** Standard class type aliases. */
using Self = LabelToRGBImageFilter;
using Pointer = SmartPointer<Self>;
using ConstPointer = SmartPointer<const Self>;
using Superclass = UnaryFunctorImageFilter<
TLabelImage,
TOutputImage,
Functor::LabelToRGBFunctor<typename TLabelImage::PixelType, typename TOutputImage::PixelType>>;
using OutputImageType = TOutputImage;
using LabelImageType = TLabelImage;
using OutputPixelType = typename TOutputImage::PixelType;
using LabelPixelType = typename TLabelImage::PixelType;
using OutputPixelValueType = typename NumericTraits<OutputPixelType>::ValueType;
/** \see LightObject::GetNameOfClass() */
itkOverrideGetNameOfClassMacro(LabelToRGBImageFilter);
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Set/Get the background value */
itkSetMacro(BackgroundValue, LabelPixelType);
itkGetConstReferenceMacro(BackgroundValue, LabelPixelType);
/** Set/Get the background color in the output image */
itkSetMacro(BackgroundColor, OutputPixelType);
itkGetConstReferenceMacro(BackgroundColor, OutputPixelType);
/** Empty the color LUT container */
void
ResetColors();
/** Get number of colors in the LUT container */
unsigned int
GetNumberOfColors() const;
/** Type of the color component */
using ComponentType = typename OutputPixelType::ComponentType;
/** Add color to the LUT container */
void
AddColor(ComponentType r, ComponentType g, ComponentType b);
protected:
LabelToRGBImageFilter();
~LabelToRGBImageFilter() override = default;
/** Process to execute before entering the multithreaded section */
void
BeforeThreadedGenerateData() override;
/** Print internal ivars */
void
PrintSelf(std::ostream & os, Indent indent) const override;
void
GenerateOutputInformation() override;
private:
OutputPixelType m_BackgroundColor{};
LabelPixelType m_BackgroundValue{};
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
# include "itkLabelToRGBImageFilter.hxx"
#endif
#endif
|
