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
|
/*=========================================================================
*
* Copyright Insight Software Consortium
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0.txt
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
*=========================================================================*/
#ifndef itkLabelMapToBinaryImageFilter_hxx
#define itkLabelMapToBinaryImageFilter_hxx
#include "itkLabelMapToBinaryImageFilter.h"
#include "itkNumericTraits.h"
#include "itkProgressReporter.h"
#include "itkImageRegionIterator.h"
namespace itk
{
template< typename TInputImage, typename TOutputImage >
LabelMapToBinaryImageFilter< TInputImage, TOutputImage >
::LabelMapToBinaryImageFilter()
{
this->m_BackgroundValue = NumericTraits< OutputImagePixelType >::NonpositiveMin();
this->m_ForegroundValue = NumericTraits< OutputImagePixelType >::max();
}
template< typename TInputImage, typename 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< typename TInputImage, typename TOutputImage >
void
LabelMapToBinaryImageFilter< TInputImage, TOutputImage >
::EnlargeOutputRequestedRegion(DataObject *)
{
this->GetOutput()->SetRequestedRegion( this->GetOutput()->GetLargestPossibleRegion() );
}
template< typename TInputImage, typename TOutputImage >
void
LabelMapToBinaryImageFilter< TInputImage, TOutputImage >
::BeforeThreadedGenerateData()
{
ThreadIdType numberOfThreads = this->GetNumberOfThreads();
if ( itk::MultiThreader::GetGlobalMaximumNumberOfThreads() != 0 )
{
numberOfThreads = std::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< typename TInputImage, typename TOutputImage >
void
LabelMapToBinaryImageFilter< TInputImage, TOutputImage >
::ThreadedGenerateData(const OutputImageRegionType & outputRegionForThread, ThreadIdType 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->GetNumberOfIndexedInputs() == 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< typename TInputImage, typename TOutputImage >
void
LabelMapToBinaryImageFilter< TInputImage, TOutputImage >
::ThreadedProcessLabelObject(LabelObjectType *labelObject)
{
OutputImageType *output = this->GetOutput();
typename LabelObjectType::ConstIndexIterator it( labelObject );
while( ! it.IsAtEnd() )
{
output->SetPixel( it.GetIndex(), this->m_ForegroundValue );
++it;
}
}
template< typename TInputImage, typename 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
|