itkGibbsPriorFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkGibbsPriorFilter.h,v $
  Language:  C++
  Date:      $Date: 2002/09/11 19:39:01 $
  Version:   $Revision: 1.22 $

  Copyright (c) 2002 Insight 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 __itkGibbsPriorFilter_h
#define __itkGibbsPriorFilter_h

//#include "itkObject.h"
#include "vnl/vnl_vector.h"
#include "vnl/vnl_matrix.h"

#include "itkImageToImageFilter.h"
//#include "itkSupervisedClassifier.h"
#include "itkMRFImageFilter.h"
#include "itkImageRegionIteratorWithIndex.h"


namespace itk
{

template <class TInputImage, class TClassifiedImage>
class ITK_EXPORT GibbsPriorFilter : public MRFImageFilter<TInputImage, 
  TClassifiedImage>
{
public:
  typedef GibbsPriorFilter  Self;
  typedef MRFImageFilter<TInputImage, TClassifiedImage> Superclass;
  typedef SmartPointer<Self>  Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkNewMacro(Self);
  
  itkTypeMacro(GibbsPriorFilter,MRFImageFilter);

  typedef typename TInputImage::Pointer              InputImageType;  

  typedef typename TInputImage::PixelType            InputPixelType;

  typedef typename TClassifiedImage::Pointer         TrainingImageType;

  typedef typename TClassifiedImage::PixelType       TrainingPixelType;

  typedef typename TClassifiedImage::Pointer         LabelledImageType;
      
  typedef typename TClassifiedImage::PixelType       LabelledPixelType;

  typedef typename TClassifiedImage::IndexType       LabelledImageIndexType;

  typedef typename Superclass::ClassifierType ClassifierType;
//  typedef Classifier<TInputImage,TClassifiedImage> ClassifierType;

  typename ClassifierType::Pointer m_ClassifierPtr;
  typedef typename TInputImage::PixelType      InputImagePixelType;
  typedef typename TClassifiedImage::PixelType TrainingImagePixelType;
  typedef typename TClassifiedImage::PixelType LabelledImagePixelType;

  typedef ImageRegionIteratorWithIndex< TInputImage > InputImageIterator;
  typedef ImageRegionIteratorWithIndex<TClassifiedImage> LabelledImageIterator;
  typedef typename TInputImage::PixelType    InputImageVectorType;

  void SetTrainingImage(TrainingImageType image);

  void SetLabelledImage(LabelledImageType LabelledImage);

  LabelledImageType GetLabelledImage()
    { return m_LabelledImage; }

  void SetClassifier( typename ClassifierType::Pointer ptrToClassifier );

  itkSetMacro(NumberOfClasses, unsigned int);
  itkGetMacro(NumberOfClasses, unsigned int);

  itkSetMacro(MaximumNumberOfIterations, unsigned int);
  itkGetMacro(MaximumNumberOfIterations, unsigned int);

  itkSetMacro(ErrorTolerance, double);
  itkGetMacro(ErrorTolerance, double);

  itkSetMacro(ClusterSize, unsigned int);

  itkSetMacro(ObjectLabel, unsigned int);

  itkStaticConstMacro(ImageDimension, unsigned int,
                      TInputImage::ImageDimension);

  void SetStartPoint (int x, int y, int z); 
  void SetBoundaryGradient(int a);
  void Advance();

protected:
  GibbsPriorFilter();
  ~GibbsPriorFilter() {};
  void PrintSelf(std::ostream& os, Indent indent) const;

  virtual void MinimizeFunctional();
  virtual void GenerateData();
  void ApplyGibbsLabeller();
  virtual void ApplyMRFImageFilter();
  void Allocate();

  InputImageType         m_InputImage;
  TrainingImageType      m_TrainingImage;
  LabelledImageType      m_LabelledImage;

  float m_BoundaryWeight; 
  float m_GibbsPriorWeight; 
  int m_StartRadius;

  int m_StartPoint[3]; 
  int m_StartModelSize; 
  int m_GibbsNeighborsThreshold; 
  int m_BoundaryGradient;
  int m_RecursiveNum;

  unsigned int          m_NumberOfClasses;
  unsigned int          m_MaximumNumberOfIterations;
  unsigned int          *m_LabelStatus;
  
  double                m_ErrorTolerance;
  double                *m_ClassProbability; //Class liklihood

  int                   m_ErrorCounter;
  int                   m_ImageWidth;
  int                   m_ImageHeight;
  int                   m_ImageDepth;
  int                   m_ClusterSize;
  int                   m_ObjectLabel;

  unsigned short    *m_Region;
  unsigned short    *m_RegionCount;

  void  GenerateInputRequestedRegion();
  void  EnlargeOutputRequestedRegion( DataObject * );
  void  GenerateOutputInformation();

  void  GibbsTotalEnergy(int i);
  int   GreyScalarBoundary(LabelledImageIndexType Index3D); 
  float GibbsEnergy(int i, int k, int k1);
  int Sim(int a, int b);
  int LabelRegion(int i, int l, int change);
  void  RegionEraser();

private:
  GibbsPriorFilter(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented
  
  typedef typename TInputImage::SizeType InputImageSizeType;
};

} // end namespace itk
#ifndef ITK_MANUAL_INSTANTIATION
#include "itkGibbsPriorFilter.txx"
#endif
#endif

---