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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkShiftScaleLabelMapFilter.txx
Language: C++
Date: $Date$
Version: $Revision$
Copyright (c) Insight Software Consortium. All rights reserved.
See ITKCopyright.txt or http://www.itk.org/HTML/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 notices for more information.
=========================================================================*/
#ifndef __itkShiftScaleLabelMapFilter_txx
#define __itkShiftScaleLabelMapFilter_txx
#include "itkShiftScaleLabelMapFilter.h"
#include "itkProgressReporter.h"
namespace itk {
template <class TImage>
ShiftScaleLabelMapFilter<TImage>
::ShiftScaleLabelMapFilter()
{
m_Shift = 0.0;
m_Scale = 1.0;
m_ChangeBackgroundValue = false;
}
template <class TImage>
void
ShiftScaleLabelMapFilter<TImage>
::GenerateData()
{
// Allocate the output
this->AllocateOutputs();
ImageType * output = this->GetOutput();
// get the label objects
LabelObjectContainerType labelObjects = output->GetLabelObjectContainer();
ProgressReporter progress( this, 0, labelObjects.size() );
// change the background, if requested
if( m_ChangeBackgroundValue )
{
PixelType label = static_cast<PixelType>( m_Scale * output->GetBackgroundValue() + m_Shift );
output->SetBackgroundValue( label );
}
// and put back the objects in the map
output->ClearLabels();
typename LabelObjectContainerType::iterator it = labelObjects.begin();
while( it != labelObjects.end() )
{
LabelObjectType * lo = it->second;
PixelType label = static_cast<PixelType>( m_Scale * lo->GetLabel() + m_Shift );
lo->SetLabel( label );
output->AddLabelObject( lo );
// go to the next label
progress.CompletedPixel();
it++;
}
}
template <class TImage>
void
ShiftScaleLabelMapFilter<TImage>
::PrintSelf(std::ostream& os, Indent indent) const
{
Superclass::PrintSelf(os,indent);
os << indent << "Shift: " << m_Shift << std::endl;
os << indent << "Scale: " << m_Scale << std::endl;
os << indent << "ChangeBackgroundValue: " << m_ChangeBackgroundValue << std::endl;
}
}// end namespace itk
#endif
|
