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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkScalarConnectedComponentImageFilter.h
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.
Portions of this code are covered under the VTK copyright.
See VTKCopyright.txt or http://www.kitware.com/VTKCopyright.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 __itkScalarConnectedComponentImageFilter_h
#define __itkScalarConnectedComponentImageFilter_h
#include "vnl/vnl_math.h"
#include "itkNumericTraits.h"
#include "itkConnectedComponentFunctorImageFilter.h"
namespace itk
{
/** \class ScalarConnectedComponentImageFilter
*
* \brief A connected components filter that labels the
* objects in an arbitrary image. Two pixels are
* similar if they are within threshold of each other.
* Uses ConnectedComponentFunctorImageFilter.
*/
namespace Functor {
template<class TInput>
class SimilarPixelsFunctor
{
public:
SimilarPixelsFunctor()
{
m_Threshold = NumericTraits<TInput>::Zero;
}
~SimilarPixelsFunctor()
{
}
bool operator!=( const SimilarPixelsFunctor & other ) const
{
if( m_Threshold != other.m_Threshold )
{
return true;
}
return false;
}
bool operator==( const SimilarPixelsFunctor & other ) const
{
return !(*this != other);
}
void SetDistanceThreshold(const TInput &thresh)
{
m_Threshold = thresh;
}
TInput GetDistanceThreshold()
{
return (m_Threshold);
}
bool operator()(const TInput &a, const TInput &b) const
{
typedef typename NumericTraits<TInput>::RealType InputRealType;
TInput absDifference = static_cast<TInput>( vnl_math_abs(
static_cast<InputRealType>(a)
- static_cast<InputRealType>(b) ) );
if( absDifference <= m_Threshold )
{
return true;
}
else
{
return false;
}
}
protected:
TInput m_Threshold;
};
} // end namespace Functor
template <class TInputImage, class TOutputImage, class TMaskImage=TInputImage>
class ITK_EXPORT ScalarConnectedComponentImageFilter :
public ConnectedComponentFunctorImageFilter<TInputImage,TOutputImage,
Functor::SimilarPixelsFunctor<typename TInputImage::ValueType>,
TMaskImage>
{
public:
/** Standard class typedefs. */
typedef ScalarConnectedComponentImageFilter Self;
typedef ConnectedComponentFunctorImageFilter<
TInputImage,TOutputImage,
Functor::SimilarPixelsFunctor<typename TInputImage::ValueType>,
TMaskImage> Superclass;
typedef SmartPointer<Self> Pointer;
typedef SmartPointer<const Self> ConstPointer;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Run-time type information (and related methods). */
itkTypeMacro(ScalarConnectedComponentImageFilter,ConnectedComponentFunctorImageFilter);
typedef typename TInputImage::PixelType InputPixelType;
virtual void SetDistanceThreshold(const InputPixelType& thresh)
{this->GetFunctor().SetDistanceThreshold(thresh);}
virtual InputPixelType GetDistanceThreshold()
{return (this->GetFunctor().GetDistanceThreshold());}
#ifdef ITK_USE_CONCEPT_CHECKING
/** Begin concept checking */
itkConceptMacro(InputEqualityComparableCheck,
(Concept::EqualityComparable<InputPixelType>));
itkConceptMacro(OutputEqualityComparableCheck,
(Concept::EqualityComparable<typename TOutputImage::PixelType>));
itkConceptMacro(MaskEqualityComparableCheck,
(Concept::EqualityComparable<typename TMaskImage::PixelType>));
itkConceptMacro(OutputIncrementDecrementOperatorsCheck,
(Concept::IncrementDecrementOperators<typename TOutputImage::PixelType>));
/** End concept checking */
#endif
protected:
ScalarConnectedComponentImageFilter() {};
virtual ~ScalarConnectedComponentImageFilter() {};
private:
ScalarConnectedComponentImageFilter(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
};
} // end namespace itk
#endif
|