itkImageGaussianModelEstimator.h

Go to the documentation of this file.

/*=========================================================================

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkImageGaussianModelEstimator.h,v $
  Language:  C++
  Date:      $Date: 2009-01-24 20:02:54 $
  Version:   $Revision: 1.12 $

  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 __itkImageGaussianModelEstimator_h
#define __itkImageGaussianModelEstimator_h

#include <math.h>
#include <float.h>

#include "vnl/vnl_vector.h"
#include "vnl/vnl_matrix.h"
#include "vnl/vnl_matrix_fixed.h"
#include "vnl/vnl_math.h"
#include "vnl/algo/vnl_matrix_inverse.h"


#include "itkImageRegionIterator.h"
#include "itkExceptionObject.h"

#include "itkImageModelEstimatorBase.h"

namespace itk
{

template <class TInputImage, 
          class TMembershipFunction,
          class TTrainingImage>
class ITK_EXPORT ImageGaussianModelEstimator: 
    public ImageModelEstimatorBase<TInputImage, TMembershipFunction>
{
public:
  typedef ImageGaussianModelEstimator                              Self;
  typedef ImageModelEstimatorBase<TInputImage,TMembershipFunction> Superclass;
  typedef SmartPointer<Self>                                       Pointer;
  typedef SmartPointer<const Self>                                 ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(ImageGaussianModelEstimator, ImageModelEstimatorBase);

  typedef typename TInputImage::Pointer   InputImagePointer;

  typedef typename TTrainingImage::Pointer TrainingImagePointer;

  typedef typename TInputImage::PixelType InputImagePixelType;

  typedef typename TTrainingImage::PixelType TrainingImagePixelType;

  typedef ImageRegionIterator< TInputImage >    InputImageIterator;
  typedef ImageRegionIterator< TTrainingImage > TrainingImageIterator;

  typedef typename TMembershipFunction::Pointer MembershipFunctionPointer;

  itkSetMacro(TrainingImage,TrainingImagePointer);

  itkGetMacro(TrainingImage,TrainingImagePointer);

protected: 
  ImageGaussianModelEstimator();
  ~ImageGaussianModelEstimator();
  virtual void PrintSelf(std::ostream& os, Indent indent) const;

  void GenerateData();

private:
  ImageGaussianModelEstimator(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  typedef vnl_matrix<double> MatrixType; 
  typedef vnl_vector<double> VectorType;

  typedef typename TInputImage::SizeType InputImageSizeType;

  itkStaticConstMacro(VectorDimension, unsigned int,
                      InputImagePixelType::Dimension);
  typedef vnl_matrix_fixed<double,1,itkGetStaticConstMacro(VectorDimension)> ColumnVectorType;

  MatrixType            m_NumberOfSamples;
  MatrixType            m_Means;
  MatrixType           *m_Covariance;  

  TrainingImagePointer  m_TrainingImage;

  virtual void EstimateModels();

  void EstimateGaussianModelParameters();

}; // class ImageGaussianModelEstimator

} // namespace itk

#ifndef ITK_MANUAL_INSTANTIATION
#include "itkImageGaussianModelEstimator.txx"
#endif

#endif