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 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219
|
/*=========================================================================
Program: Insight Segmentation & Registration Toolkit
Module: $RCSfile: itkHistogramMatchingImageFilter.h,v $
Language: C++
Date: $Date: 2007-07-31 23:09:16 $
Version: $Revision: 1.10 $
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 __itkHistogramMatchingImageFilter_h
#define __itkHistogramMatchingImageFilter_h
#include "itkImageToImageFilter.h"
#include "itkHistogram.h"
#include "vnl/vnl_matrix.h"
namespace itk
{
/** \class HistogramMatchingImageFilter
* \brief Normalize the grayscale values between two image by histogram
* matching.
*
* HistogramMatchingImageFilter normalizes the grayscale values of a source
* image based on the grayscale values of a reference image.
* This filter uses a histogram matching technique where the histograms of the
* two images are matched only at a specified number of quantile values.
*
* This filter was orginally designed to normalize MR images of the same
* MR protocol and same body part. The algorithm works best if background
* pixels are excluded from both the source and reference histograms.
* A simple background exclusion method is to exclude all pixels whose
* grayscale values are smaller than the mean grayscale value.
* ThresholdAtMeanIntensityOn() switches on this simple background
* exclusion method.
*
* The source image can be set via either SetInput() or SetSourceImage().
* The reference image can be set via SetReferenceImage().
*
* SetNumberOfHistogramLevels() sets the number of bins used when
* creating histograms of the source and reference images.
* SetNumberOfMatchPoints() governs the number of quantile values to be
* matched.
*
* This filter assumes that both the source and reference are of the same
* type and that the input and output image type have the same number of
* dimension and have scalar pixel types.
*
* \ingroup IntensityImageFilters Multithreaded
*
*/
/* THistogramMeasurement -- The precision level for which to do HistogramMeasurmenets */
template <class TInputImage, class TOutputImage, class THistogramMeasurement=ITK_TYPENAME TInputImage::PixelType>
class ITK_EXPORT HistogramMatchingImageFilter:
public ImageToImageFilter<TInputImage,TOutputImage>
{
public:
/** Standard class typedefs. */
typedef HistogramMatchingImageFilter Self;
typedef ImageToImageFilter<TInputImage,TOutputImage> 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(HistogramMatchingImageFilter, ImageToImageFilter);
/** ImageDimension enumeration. */
itkStaticConstMacro(ImageDimension, unsigned int,
TInputImage::ImageDimension);
itkStaticConstMacro(OutputImageDimension, unsigned int,
TOutputImage::ImageDimension);
/** Typedef to describe the output image region type. */
typedef typename TOutputImage::RegionType OutputImageRegionType;
/** Inherited typedefs. */
typedef typename Superclass::InputImageType InputImageType;
typedef typename Superclass::InputImagePointer InputImagePointer;
typedef typename Superclass::InputImageConstPointer InputImageConstPointer;
typedef typename Superclass::OutputImageType OutputImageType;
typedef typename Superclass::OutputImagePointer OutputImagePointer;
/** Pixel related typedefs. */
typedef typename InputImageType::PixelType InputPixelType;
typedef typename OutputImageType::PixelType OutputPixelType;
/** Histogram related typedefs. */
typedef Statistics::Histogram<THistogramMeasurement, 1> HistogramType;
typedef typename HistogramType::Pointer HistogramPointer;
/** Set/Get the source image. */
void SetSourceImage( const InputImageType * source )
{ this->SetInput( source ); }
const InputImageType * GetSourceImage(void)
{ return this->GetInput(); }
/** Set/Get the reference image. */
void SetReferenceImage( const InputImageType * reference );
const InputImageType * GetReferenceImage(void);
/** Set/Get the number of histogram levels used. */
itkSetMacro( NumberOfHistogramLevels, unsigned long );
itkGetMacro( NumberOfHistogramLevels, unsigned long );
/** Set/Get the number of match points used. */
itkSetMacro( NumberOfMatchPoints, unsigned long );
itkGetMacro( NumberOfMatchPoints, unsigned long );
/** Set/Get the threshold at mean intensity flag.
* If true, only source (reference) pixels which are greater
* than the mean source (reference) intensity is used in
* the histogram matching. If false, all pixels are
* used. */
itkSetMacro( ThresholdAtMeanIntensity, bool );
itkGetMacro( ThresholdAtMeanIntensity, bool );
itkBooleanMacro( ThresholdAtMeanIntensity );
/** This filter requires all of the input to be in the buffer. */
virtual void GenerateInputRequestedRegion();
/** Methods to get the histograms of the source, reference, and
* output. Objects are only valid after Update() has been called
* on this filter. */
itkGetObjectMacro(SourceHistogram, HistogramType);
itkGetObjectMacro(ReferenceHistogram, HistogramType);
itkGetObjectMacro(OutputHistogram, HistogramType);
#ifdef ITK_USE_CONCEPT_CHECKING
/** Begin concept checking */
itkConceptMacro(IntConvertibleToInputCheck,
(Concept::Convertible<int, InputPixelType>));
itkConceptMacro(SameDimensionCheck,
(Concept::SameDimension<ImageDimension, OutputImageDimension>));
itkConceptMacro(DoubleConvertibleToInputCheck,
(Concept::Convertible<double, InputPixelType>));
itkConceptMacro(DoubleConvertibleToOutputCheck,
(Concept::Convertible<double, OutputPixelType>));
itkConceptMacro(InputConvertibleToDoubleCheck,
(Concept::Convertible<InputPixelType, double>));
itkConceptMacro(OutputConvertibleToDoubleCheck,
(Concept::Convertible<OutputPixelType, double>));
itkConceptMacro(SameTypeCheck,
(Concept::SameType<InputPixelType, OutputPixelType>));
/** End concept checking */
#endif
protected:
HistogramMatchingImageFilter();
~HistogramMatchingImageFilter() {};
void PrintSelf(std::ostream& os, Indent indent) const;
void BeforeThreadedGenerateData();
void AfterThreadedGenerateData();
void ThreadedGenerateData(const OutputImageRegionType& outputRegionForThread,
int threadId );
/** Compute min, max and mean of an image. */
void ComputeMinMaxMean( const InputImageType * image,
THistogramMeasurement& minValue, THistogramMeasurement& maxValue, THistogramMeasurement& meanValue );
/** Construct a histogram from an image. */
void ConstructHistogram( const InputImageType * image,
HistogramType * histogram, const THistogramMeasurement minValue,
const THistogramMeasurement maxValue );
private:
HistogramMatchingImageFilter(const Self&); //purposely not implemented
void operator=(const Self&); //purposely not implemented
unsigned long m_NumberOfHistogramLevels;
unsigned long m_NumberOfMatchPoints;
bool m_ThresholdAtMeanIntensity;
InputPixelType m_SourceIntensityThreshold;
InputPixelType m_ReferenceIntensityThreshold;
OutputPixelType m_OutputIntensityThreshold;
THistogramMeasurement m_SourceMinValue;
THistogramMeasurement m_SourceMaxValue;
THistogramMeasurement m_SourceMeanValue;
THistogramMeasurement m_ReferenceMinValue;
THistogramMeasurement m_ReferenceMaxValue;
THistogramMeasurement m_ReferenceMeanValue;
THistogramMeasurement m_OutputMinValue;
THistogramMeasurement m_OutputMaxValue;
THistogramMeasurement m_OutputMeanValue;
HistogramPointer m_SourceHistogram;
HistogramPointer m_ReferenceHistogram;
HistogramPointer m_OutputHistogram;
typedef vnl_matrix<double> TableType;
TableType m_QuantileTable;
typedef vnl_vector<double> GradientArrayType;
GradientArrayType m_Gradients;
double m_LowerGradient;
double m_UpperGradient;
};
} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkHistogramMatchingImageFilter.txx"
#endif
#endif
|