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
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkTernaryFunctorImageFilter.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 __itkTernaryFunctorImageFilter_txx
#define __itkTernaryFunctorImageFilter_txx
#include "itkTernaryFunctorImageFilter.h"
#include "itkImageRegionIterator.h"
#include "itkImageRegionConstIterator.h"
#include "itkProgressReporter.h"
namespace itk
{
/**
* Constructor
*/
template < class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
TernaryFunctorImageFilter<TInputImage1,TInputImage2,TInputImage3,TOutputImage,TFunction>
::TernaryFunctorImageFilter()
{
this->InPlaceOff();
}
/**
* Connect one of the operands for pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
void
TernaryFunctorImageFilter<TInputImage1,TInputImage2,TInputImage3,TOutputImage,TFunction>
::SetInput1( const TInputImage1 *image1 )
{
// The ProcessObject is not const-correct so the const_cast is required here
this->SetNthInput( 0, const_cast<TInputImage1 *>( image1 ) );
}
/**
* Connect one of the operands for pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
void
TernaryFunctorImageFilter<TInputImage1,TInputImage2,TInputImage3,TOutputImage,TFunction>
::SetInput2( const TInputImage2 *image2 )
{
// The ProcessObject is not const-correct so the const_cast is required here
this->SetNthInput( 1, const_cast<TInputImage2 *>( image2 ) );
}
/**
* Connect one of the operands for pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
void
TernaryFunctorImageFilter<TInputImage1,TInputImage2,TInputImage3,TOutputImage,TFunction>
::SetInput3( const TInputImage3 *image3 )
{
// The ProcessObject is not const-correct so the const_cast is required here
this->SetNthInput( 2, const_cast<TInputImage3 *>( image3 ) );
}
/**
* BeforeThreadedGenerateData function. Validate inputs
*/
template <class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
void
TernaryFunctorImageFilter<TInputImage1,TInputImage2,TInputImage3,TOutputImage,TFunction>
::BeforeThreadedGenerateData()
{
Input1ImagePointer inputPtr1
= dynamic_cast<const TInputImage1*>((ProcessObject::GetInput(0)));
Input2ImagePointer inputPtr2
= dynamic_cast<const TInputImage2*>((ProcessObject::GetInput(1)));
Input3ImagePointer inputPtr3
= dynamic_cast<const TInputImage3*>((ProcessObject::GetInput(2)));
if (inputPtr1.IsNull() || inputPtr2.IsNull() || inputPtr3.IsNull())
{
itkExceptionMacro(<< "At least one input is missing."
<< " Input1 is " << inputPtr1.GetPointer() << ", "
<< " Input2 is " << inputPtr2.GetPointer() << ", "
<< " Input3 is " << inputPtr3.GetPointer());
}
}
/**
* ThreadedGenerateData function. Performs the pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TInputImage3, class TOutputImage, class TFunction >
void
TernaryFunctorImageFilter<TInputImage1, TInputImage2, TInputImage3, TOutputImage, TFunction>
::ThreadedGenerateData( const OutputImageRegionType &outputRegionForThread,
int threadId)
{
// We use dynamic_cast since inputs are stored as DataObjects. The
// ImageToImageFilter::GetInput(int) always returns a pointer to a
// TInputImage1 so it cannot be used for the second or third input.
Input1ImagePointer inputPtr1
= dynamic_cast<const TInputImage1*>((ProcessObject::GetInput(0)));
Input2ImagePointer inputPtr2
= dynamic_cast<const TInputImage2*>((ProcessObject::GetInput(1)));
Input3ImagePointer inputPtr3
= dynamic_cast<const TInputImage3*>((ProcessObject::GetInput(2)));
OutputImagePointer outputPtr = this->GetOutput(0);
ImageRegionConstIterator<TInputImage1> inputIt1(inputPtr1, outputRegionForThread);
ImageRegionConstIterator<TInputImage2> inputIt2(inputPtr2, outputRegionForThread);
ImageRegionConstIterator<TInputImage3> inputIt3(inputPtr3, outputRegionForThread);
ImageRegionIterator<TOutputImage> outputIt(outputPtr, outputRegionForThread);
ProgressReporter progress(this, threadId, outputRegionForThread.GetNumberOfPixels());
inputIt1.GoToBegin();
inputIt2.GoToBegin();
inputIt3.GoToBegin();
outputIt.GoToBegin();
while( !inputIt1.IsAtEnd() )
{
outputIt.Set( m_Functor(inputIt1.Get(), inputIt2.Get(), inputIt3.Get() ) );
++inputIt1;
++inputIt2;
++inputIt3;
++outputIt;
progress.CompletedPixel(); // potential exception thrown here
}
}
} // end namespace itk
#endif
|
