itkClassifierBase.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkClassifierBase.h,v $
  Language:  C++
  Date:      $Date: 2003/02/23 06:00:37 $
  Version:   $Revision: 1.4 $

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

#include <vector>

#include "itkLightProcessObject.h"
#include "itkMembershipFunctionBase.h"
#include "itkDecisionRuleBase.h"

namespace itk
{

template <class TDataContainer>
class ITK_EXPORT ClassifierBase : public LightProcessObject
{
public:
  typedef ClassifierBase   Self;
  typedef LightProcessObject Superclass;
  typedef SmartPointer<Self>  Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkTypeMacro(ClassifierBase,LightProcessObject);

  itkSetMacro(NumberOfClasses, unsigned int);

  itkGetConstMacro(NumberOfClasses, unsigned int);

  typedef typename TDataContainer::ValueType MeasurementVectorType ;
  
  typedef Statistics::MembershipFunctionBase< MeasurementVectorType > 
    MembershipFunctionType;
  typedef typename MembershipFunctionType::Pointer MembershipFunctionPointer ;

  typedef std::vector< MembershipFunctionPointer > 
    MembershipFunctionPointerVector;

  typedef DecisionRuleBase DecisionRuleType;

  void SetDecisionRule( typename DecisionRuleType::Pointer ptrToDecisionRule )
    {
    m_DecisionRulePointer = ptrToDecisionRule;
    }

  DecisionRuleType * GetDecisionRule(void)
    {
    return m_DecisionRulePointer;
    }
  
  const MembershipFunctionPointerVector GetMembershipFunctions() const
    {
    return m_MembershipFunctions;
    }

  unsigned int GetNumberOfMembershipFunctions() 
    {
    return static_cast<unsigned int>( m_MembershipFunctions.size() );
    }

  unsigned int AddMembershipFunction(MembershipFunctionType * function);

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

  virtual void GenerateData();

private:

  ClassifierBase(const Self&); //purposely not implemented
  void operator=(const Self&); //purposely not implemented

  unsigned int                       m_NumberOfClasses;

  typename DecisionRuleType::Pointer m_DecisionRulePointer;

  MembershipFunctionPointerVector    m_MembershipFunctions ;

}; // class Classifier


} // namespace itk

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



#endif

---