itkImageModelEstimatorBase.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkImageModelEstimatorBase.h,v $
  Language:  C++
  Date:      $Date: 2004/11/04 20:40:32 $
  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 _itkImageModelEstimatorBase_h
#define _itkImageModelEstimatorBase_h

#include "itkLightProcessObject.h"

namespace itk
{

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

  itkTypeMacro(ImageModelEstimatorBase,LightProcessObject);

  itkSetMacro(NumberOfModels, unsigned int);

  itkGetConstReferenceMacro(NumberOfModels, unsigned int);

  typedef typename TMembershipFunction::Pointer MembershipFunctionPointer ;

  typedef std::vector< MembershipFunctionPointer > 
  MembershipFunctionPointerVector;

  typedef          TInputImage          InputImageType;
  typedef typename TInputImage::Pointer InputImagePointer;

  //typedef typename TTrainingImage::Pointer TrainingImagePointer;

  itkSetObjectMacro(InputImage,InputImageType);

  itkGetObjectMacro(InputImage,InputImageType);

  void SetMembershipFunctions(MembershipFunctionPointerVector 
                              membershipFunctions)
  {
    m_MembershipFunctions = membershipFunctions;
  }
  
  const MembershipFunctionPointerVector GetMembershipFunctions() const
  {
    return m_MembershipFunctions;
  }

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

  void DeleteAllMembershipFunctions() 
  {
    m_MembershipFunctions.resize(0);
  }

  unsigned int AddMembershipFunction(MembershipFunctionPointer function);

  void Update() ;

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

  virtual void GenerateData();

private:

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

  unsigned int                    m_NumberOfModels;

  MembershipFunctionPointerVector m_MembershipFunctions;

  InputImagePointer               m_InputImage;

  virtual void EstimateModels() = 0;
}; // class ImageModelEstimator


} // namespace itk

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



#endif

---