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
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: itkRobustAutomaticThresholdImageFilter.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.
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 __itkRobustAutomaticThresholdImageFilter_h
#define __itkRobustAutomaticThresholdImageFilter_h
#include "itkImageToImageFilter.h"
#include "itkRobustAutomaticThresholdCalculator.h"
namespace itk {
/** \class RobustAutomaticThresholdImageFilter
* \brief Threshold an image using robust automatic threshold selection (RATS) method.
*
* RobustAutomaticThresholdImageFilter takes two inputs: the image to be thresholded
* and a image of gradient magnitude of that image.
* The threshold is computed as the mean of the pixel values in the input image weighted
* by the pixel values in the gradient image.
* The threshold computed that way should be the mean pixel value where the intensity
* change the most.
*
* http://hdl.handle.net/1926/370
*
* \author Gaetan Lehmann. Biologie du Developpement et de la Reproduction, INRA de Jouy-en-Josas, France.
*
* \ingroup IntensityImageFilters
*/
template<class TInputImage, class TGradientImage=TInputImage, class TOutputImage=TInputImage>
class ITK_EXPORT RobustAutomaticThresholdImageFilter :
public ImageToImageFilter<TInputImage, TOutputImage>
{
public:
/** Standard Self typedef */
typedef RobustAutomaticThresholdImageFilter Self;
typedef ImageToImageFilter<TInputImage,TOutputImage> Superclass;
typedef SmartPointer<Self> Pointer;
typedef SmartPointer<const Self> ConstPointer;
/** Method for creation through the object factory. */
itkNewMacro(Self);
/** Runtime information support. */
itkTypeMacro(RobustAutomaticThresholdImageFilter, ImageToImageFilter);
/** Standard image type within this class. */
typedef TInputImage InputImageType;
typedef TGradientImage GradientImageType;
/** Image pixel value typedef. */
typedef typename TInputImage::PixelType InputPixelType;
typedef typename TOutputImage::PixelType OutputPixelType;
typedef typename TGradientImage::PixelType GradientPixelType;
/** Image related typedefs. */
typedef typename TInputImage::Pointer InputImagePointer;
typedef typename TOutputImage::Pointer OutputImagePointer;
typedef typename TGradientImage::Pointer GradientImagePointer;
typedef typename TInputImage::SizeType InputSizeType;
typedef typename TInputImage::IndexType InputIndexType;
typedef typename TInputImage::RegionType InputImageRegionType;
typedef typename TOutputImage::SizeType OutputSizeType;
typedef typename TOutputImage::IndexType OutputIndexType;
typedef typename TOutputImage::RegionType OutputImageRegionType;
typedef RobustAutomaticThresholdCalculator< TInputImage, TGradientImage >
CalculatorType;
/** Image related typedefs. */
itkStaticConstMacro(InputImageDimension, unsigned int,
TInputImage::ImageDimension );
itkStaticConstMacro(OutputImageDimension, unsigned int,
TOutputImage::ImageDimension );
/** Set the "outside" pixel value. The default value
* NumericTraits<OutputPixelType>::Zero. */
itkSetMacro(OutsideValue,OutputPixelType);
/** Get the "outside" pixel value. */
itkGetConstMacro(OutsideValue,OutputPixelType);
/** Set the "inside" pixel value. The default value
* NumericTraits<OutputPixelType>::max() */
itkSetMacro(InsideValue,OutputPixelType);
/** Get the "inside" pixel value. */
itkGetConstMacro(InsideValue,OutputPixelType);
/** Get the computed threshold. */
itkGetConstMacro(Threshold,InputPixelType);
itkSetMacro(Pow, double);
itkGetConstMacro(Pow, double);
#ifdef ITK_USE_CONCEPT_CHECKING
/** Begin concept checking */
itkConceptMacro(OutputComparableCheck,
(Concept::Comparable<OutputPixelType>));
itkConceptMacro(OutputOStreamWritableCheck,
(Concept::OStreamWritable<OutputPixelType>));
/** End concept checking */
#endif
/** Set the gradient image */
void SetGradientImage(GradientImageType *input)
{
// Process object is not const-correct so the const casting is required.
this->SetNthInput( 1, const_cast<GradientImageType *>(input) );
}
/** Get the gradient image */
GradientImageType * GetGradientImage()
{
return static_cast<GradientImageType*>(const_cast<DataObject *>(this->ProcessObject::GetInput(1)));
}
/** Set the input image */
void SetInput1(TInputImage *input)
{
this->SetInput( input );
}
/** Set the gradient image */
void SetInput2(GradientImageType *input)
{
this->SetGradientImage( input );
}
protected:
RobustAutomaticThresholdImageFilter();
~RobustAutomaticThresholdImageFilter(){};
void PrintSelf(std::ostream& os, Indent indent) const;
void GenerateInputRequestedRegion();
void GenerateData ();
private:
RobustAutomaticThresholdImageFilter(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
double m_Pow;
InputPixelType m_Threshold;
OutputPixelType m_InsideValue;
OutputPixelType m_OutsideValue;
}; // end of class
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkRobustAutomaticThresholdImageFilter.txx"
#endif
#endif
|