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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkBinaryFunctorImageFilter.txx,v $
Language: C++
Date: $Date: 2007-08-30 15:01:09 $
Version: $Revision: 1.39 $
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 _itkBinaryFunctorImageFilter_txx
#define _itkBinaryFunctorImageFilter_txx
#include "itkBinaryFunctorImageFilter.h"
#include "itkImageRegionIterator.h"
#include "itkImageRegionConstIterator.h"
#include "itkProgressReporter.h"
namespace itk
{
/**
* Constructor
*/
template <class TInputImage1, class TInputImage2,
class TOutputImage, class TFunction >
BinaryFunctorImageFilter<TInputImage1,TInputImage2,TOutputImage,TFunction>
::BinaryFunctorImageFilter()
{
this->SetNumberOfRequiredInputs( 2 );
this->InPlaceOff();
}
/**
* Connect one of the operands for pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TOutputImage, class TFunction >
void
BinaryFunctorImageFilter<TInputImage1,TInputImage2,TOutputImage,TFunction>
::SetInput1( const TInputImage1 * image1 )
{
// Process object is not const-correct so the const casting is required.
this->SetNthInput(0, const_cast<TInputImage1 *>( image1 ));
}
/**
* Connect one of the operands for pixel-wise addition
*/
template <class TInputImage1, class TInputImage2,
class TOutputImage, class TFunction >
void
BinaryFunctorImageFilter<TInputImage1,TInputImage2,TOutputImage,TFunction>
::SetInput2( const TInputImage2 * image2 )
{
// Process object is not const-correct so the const casting is required.
this->SetNthInput(1, const_cast<TInputImage2 *>( image2 ));
}
/**
* ThreadedGenerateData Performs the pixel-wise addition
*/
template <class TInputImage1, class TInputImage2, class TOutputImage, class TFunction >
void
BinaryFunctorImageFilter<TInputImage1, TInputImage2, 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 input.
Input1ImagePointer inputPtr1
= dynamic_cast<const TInputImage1*>(ProcessObject::GetInput(0));
Input2ImagePointer inputPtr2
= dynamic_cast<const TInputImage2*>(ProcessObject::GetInput(1));
OutputImagePointer outputPtr = this->GetOutput(0);
ImageRegionConstIterator<TInputImage1> inputIt1(inputPtr1, outputRegionForThread);
ImageRegionConstIterator<TInputImage2> inputIt2(inputPtr2, outputRegionForThread);
ImageRegionIterator<TOutputImage> outputIt(outputPtr, outputRegionForThread);
ProgressReporter progress(this, threadId, outputRegionForThread.GetNumberOfPixels());
inputIt1.GoToBegin();
inputIt2.GoToBegin();
outputIt.GoToBegin();
while( !inputIt1.IsAtEnd() )
{
outputIt.Set( m_Functor( inputIt1.Get(), inputIt2.Get() ) );
++inputIt2;
++inputIt1;
++outputIt;
progress.CompletedPixel(); // potential exception thrown here
}
}
} // end namespace itk
#endif
|