itkBayesianClassifierInitializationImageFilter.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkBayesianClassifierInitializationImageFilter.h,v $
  Language:  C++
  Date:      $Date: 2009-05-02 05:43:54 $
  Version:   $Revision: 1.8 $

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

#include "itkVectorImage.h"
#include "itkVectorContainer.h"
#include "itkImageToImageFilter.h"
#include "itkImageRegionIterator.h"
#include "itkImageRegionConstIterator.h"
#ifdef ITK_USE_REVIEW_STATISTICS
#include "itkMembershipFunctionBase.h"
#else
#include "itkDensityFunction.h"
#endif

namespace itk
{
  
template< class TInputImage, class TProbabilityPrecisionType=float >
class ITK_EXPORT BayesianClassifierInitializationImageFilter :
    public
ImageToImageFilter<TInputImage, VectorImage< TProbabilityPrecisionType, 
                  ::itk::GetImageDimension< TInputImage >::ImageDimension > >
{
public:
  typedef BayesianClassifierInitializationImageFilter  Self;
  typedef TInputImage                                  InputImageType;
  typedef TProbabilityPrecisionType                    ProbabilityPrecisionType;

  itkStaticConstMacro( Dimension, unsigned int, 
                     ::itk::GetImageDimension< InputImageType >::ImageDimension );

  typedef VectorImage< ProbabilityPrecisionType, 
          itkGetStaticConstMacro(Dimension) >                   OutputImageType;
  typedef ImageToImageFilter< InputImageType, OutputImageType > Superclass;
  typedef SmartPointer<Self>                                    Pointer;
  typedef SmartPointer<const Self>                              ConstPointer;

  itkNewMacro(Self);

  itkTypeMacro(BayesianClassifierInitializationImageFilter, ImageToImageFilter);

  typedef ImageRegionConstIterator< InputImageType > InputImageIteratorType;

  typedef typename InputImageType::PixelType  InputPixelType;
  typedef typename OutputImageType::PixelType OutputPixelType;

  typedef VectorImage< ProbabilityPrecisionType, 
          itkGetStaticConstMacro(Dimension) >             MembershipImageType;
  typedef typename MembershipImageType::PixelType         MembershipPixelType;
  typedef typename MembershipImageType::Pointer           MembershipImagePointer;
  typedef ImageRegionIterator< MembershipImageType >      MembershipImageIteratorType;

  typedef Vector< InputPixelType, 1 >                     MeasurementVectorType;

#ifdef ITK_USE_REVIEW_STATISTICS
  typedef Statistics::MembershipFunctionBase< MeasurementVectorType > MembershipFunctionType;
#else
  typedef Statistics::DensityFunction< MeasurementVectorType >
                                                        MembershipFunctionType;
#endif

  typedef typename MembershipFunctionType::Pointer      MembershipFunctionPointer;

  typedef VectorContainer< unsigned int, 
       MembershipFunctionPointer >  MembershipFunctionContainerType;
  typedef typename MembershipFunctionContainerType::Pointer
                                    MembershipFunctionContainerPointer;

  virtual void SetMembershipFunctions( MembershipFunctionContainerType 
                                                * densityFunctionContainer );
  itkGetObjectMacro(  MembershipFunctionContainer, MembershipFunctionContainerType );

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

  virtual void GenerateOutputInformation(); 

#ifdef ITK_USE_CONCEPT_CHECKING

  itkConceptMacro(InputMultiplyOperatorCheck,
    (Concept::MultiplyOperator<InputPixelType>));
  itkConceptMacro(DoubleConvertibleToProbabilityCheck,
    (Concept::Convertible<double, TProbabilityPrecisionType>));
  itkConceptMacro(InputHasNumericTraitsCheck,
    (Concept::HasNumericTraits<InputPixelType>));
  itkConceptMacro(ProbabilityHasNumericTraitsCheck,
    (Concept::HasNumericTraits<TProbabilityPrecisionType>));
  itkConceptMacro(DoublePlusInputCheck,
    (Concept::AdditiveOperators<double, InputPixelType>));

#endif

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

  virtual void InitializeMembershipFunctions();

  virtual void GenerateData();

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

  bool                      m_UserSuppliesMembershipFunctions;
  unsigned int              m_NumberOfClasses;

  typename MembershipFunctionContainerType::Pointer m_MembershipFunctionContainer;
};

} // end namespace itk

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

#endif