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 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkLabelMapToBinaryImageFilter.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 __itkLabelMapToBinaryImageFilter_txx
#define __itkLabelMapToBinaryImageFilter_txx
#include "itkLabelMapToBinaryImageFilter.h"
#include "itkNumericTraits.h"
#include "itkProgressReporter.h"
#include "itkImageRegionConstIterator.h"
#include "itkImageRegionIterator.h"
namespace itk {
template <class TInputImage, class TOutputImage>
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::LabelMapToBinaryImageFilter()
{
this->m_BackgroundValue = NumericTraits<OutputImagePixelType>::NonpositiveMin();
this->m_ForegroundValue = NumericTraits<OutputImagePixelType>::max();
}
template <class TInputImage, class TOutputImage>
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::GenerateInputRequestedRegion()
{
// call the superclass' implementation of this method
Superclass::GenerateInputRequestedRegion();
// We need all the input.
InputImagePointer input = const_cast<InputImageType *>(this->GetInput());
if( input )
{
input->SetRequestedRegion( input->GetLargestPossibleRegion() );
}
}
template <class TInputImage, class TOutputImage>
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::EnlargeOutputRequestedRegion(DataObject *)
{
this->GetOutput()->SetRequestedRegion( this->GetOutput()->GetLargestPossibleRegion() );
}
template<class TInputImage, class TOutputImage>
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::BeforeThreadedGenerateData()
{
unsigned long numberOfThreads = this->GetNumberOfThreads();
if( itk::MultiThreader::GetGlobalMaximumNumberOfThreads() != 0 )
{
numberOfThreads = vnl_math_min(
this->GetNumberOfThreads(), itk::MultiThreader::GetGlobalMaximumNumberOfThreads() );
}
// number of threads can be constrained by the region size, so call the
// SplitRequestedRegion to get the real number of threads which will be used
typename TOutputImage::RegionType splitRegion; // dummy region - just to call the following method
numberOfThreads = this->SplitRequestedRegion(0, numberOfThreads, splitRegion);
m_Barrier = Barrier::New();
m_Barrier->Initialize( numberOfThreads );
this->Superclass::BeforeThreadedGenerateData();
}
template<class TInputImage, class TOutputImage>
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::ThreadedGenerateData( const OutputImageRegionType& outputRegionForThread, int threadId )
{
OutputImageType * output = this->GetOutput();
// fill the output with background value - they will be overridden with the
// foreground value later, if there is some objects
if( this->GetNumberOfInputs() == 2 )
{
// fill the background with the background values from the background image
ImageRegionConstIterator< OutputImageType > bgIt( this->GetBackgroundImage(), outputRegionForThread );
ImageRegionIterator< OutputImageType > oIt( output, outputRegionForThread );
bgIt.GoToBegin();
oIt.GoToBegin();
while( !oIt.IsAtEnd() )
{
const OutputImagePixelType & bg = bgIt.Get();
if( bg != this->m_ForegroundValue )
{
oIt.Set( bg );
}
else
{
oIt.Set( this->m_BackgroundValue );
}
++oIt;
++bgIt;
}
}
else
{
// fill the background with the background value
ImageRegionIterator< OutputImageType > oIt( output, outputRegionForThread );
oIt.GoToBegin();
while( !oIt.IsAtEnd() )
{
oIt.Set( this->m_BackgroundValue );
++oIt;
}
}
// wait for the other threads to complete that part
this->m_Barrier->Wait();
// and delegate to the superclass implementation to use the thread support for the label objects
this->Superclass::ThreadedGenerateData( outputRegionForThread, threadId );
}
template<class TInputImage, class TOutputImage>
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::ThreadedProcessLabelObject( LabelObjectType * labelObject )
{
OutputImageType * output = this->GetOutput();
typedef typename LabelObjectType::LineContainerType LineContainerType;
typename LineContainerType::const_iterator lit;
LineContainerType & lineContainer = labelObject->GetLineContainer();
for( lit = lineContainer.begin(); lit != lineContainer.end(); lit++ )
{
IndexType idx = lit->GetIndex();
unsigned long length = lit->GetLength();
for( unsigned int i=0; i<length; i++)
{
output->SetPixel( idx, this->m_ForegroundValue );
idx[0]++;
}
}
}
template< class TInputImage, class TOutputImage >
void
LabelMapToBinaryImageFilter<TInputImage, TOutputImage>
::PrintSelf(std::ostream& os, Indent indent) const
{
this->Superclass::PrintSelf(os,indent);
os << indent << "ForegroundValue: "
<< static_cast<typename NumericTraits<OutputImagePixelType>::PrintType>(this->m_ForegroundValue) << std::endl;
os << indent << "BackgroundValue: "
<< static_cast<typename NumericTraits<OutputImagePixelType>::PrintType>(this->m_BackgroundValue) << std::endl;
os << indent << "Barrier object: " << this->m_Barrier.GetPointer() << std::endl;
}
} // end namespace itk
#endif
|