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 150 151 152 153 154 155 156 157 158
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkTernaryFunctorImageFilter.txx,v $
Language: C++
Date: $Date: 2006-06-06 15:40:48 $
Version: $Revision: 1.38 $
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
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
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
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
|