itkMembershipSample.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkMembershipSample.h,v $
  Language:  C++
  Date:      $Date: 2003/09/10 14:29:47 $
  Version:   $Revision: 1.25 $

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

#include "itk_hash_map.h"
#include "itkSample.h"
#include "itkSubsample.h"

#include "itkExceptionObject.h"

namespace itk{ 
namespace Statistics{

template< class TSample >
class ITK_EXPORT MembershipSample : 
    public Sample< typename TSample::MeasurementVectorType >
{
public:
  typedef MembershipSample Self;
  typedef Sample< typename TSample::MeasurementVectorType > Superclass ;
  typedef SmartPointer< Self > Pointer ;

  itkTypeMacro(MembershipSample, Sample);
  itkNewMacro(Self) ;
  
  typedef typename TSample::MeasurementVectorType MeasurementVectorType;
  typedef typename TSample::MeasurementType MeasurementType;
  typedef typename TSample::InstanceIdentifier InstanceIdentifier;
  typedef typename TSample::FrequencyType FrequencyType ;
  //    typedef typename TSample::SizeType SizeType ;
  //    typedef typename TSample::SizeValueType SizeValueType ;
  
  itkStaticConstMacro(MeasurementVectorSize, unsigned int,
                      TSample::MeasurementVectorSize);
  
  typedef std::vector< unsigned int > UniqueClassLabelsType ;

  typedef itk::hash_map< InstanceIdentifier, unsigned int> ClassLabelHolderType ;
  
  typedef Subsample< TSample > ClassSampleType ;
  
  void SetSample(TSample* sample) ;

  TSample* GetSample() ;
  
  void SetNumberOfClasses(unsigned int numberOfClasses) ;
  
  unsigned int GetNumberOfClasses() const ;

  void AddInstance(const unsigned int &classLabel, const InstanceIdentifier &id) ;
  unsigned int GetClassLabel(const InstanceIdentifier &id) const ;

  int GetInternalClassLabel(const unsigned int classLabel ) const ;

  unsigned int GetClassSampleSize(const unsigned int &classLabel) const ;

  ClassSampleType* GetClassSample(const unsigned int &classLabel) ;
  
  ClassLabelHolderType* GetClassLabels()
  { return &m_ClassLabelHolder ; }

  unsigned int Size(void) const ;
  
  MeasurementVectorType GetMeasurementVector(const InstanceIdentifier &id) ;
  
  MeasurementType GetMeasurement(const InstanceIdentifier &id, 
                                  const unsigned int &dimension) ;

  FrequencyType GetFrequency(const InstanceIdentifier &id) const ;
  
  FrequencyType GetTotalFrequency() const ;

  void Resize(unsigned int n) 
  {
    m_ClassLabelHolder.resize(n) ;
  }

  class Iterator;
  friend class Iterator;
  
  Iterator Begin()
  { 
    Iterator iter(0, this) ;
    return iter; 
  }
  
  Iterator  End()        
  {
    Iterator iter(this->Size(), this) ; 
    return iter; 
  }
  
  class Iterator
  {
  public:
    Iterator(InstanceIdentifier id, Self* membershipSample)
      :m_Id(id), m_MembershipSample(membershipSample),
       m_Sample(membershipSample->GetSample())
    {}
    
    FrequencyType GetFrequency() const
    { return  m_Sample->GetFrequency(m_Id) ; }
    
    MeasurementVectorType GetMeasurementVector()
    { return m_Sample->GetMeasurementVector(m_Id) ; } 
    
    InstanceIdentifier GetInstanceIdentifier() const
    { return m_Id ; }

    void SetClassLabel(unsigned int classLabel)
    { m_MembershipSample->AddInstance(classLabel, m_Id) ; }

    unsigned int GetClassLabel() const
    { return m_MembershipSample->GetClassLabel(m_Id) ; }

    Iterator& operator++() 
    { 
      ++m_Id ;
      return *this ;
    }
    
    bool operator!=(const Iterator& it) 
    { 
      if (m_Id != it.m_Id || 
          m_MembershipSample != it.m_MembershipSample ||
          m_Sample != it.m_Sample)
        {
        return true ;
        }
      else
        {
        return false ;
        }
    }

    bool operator==(const Iterator& it) 
    { 
      if (m_Id == it.m_Id && 
          m_MembershipSample == it.m_MembershipSample &&
          m_Sample == it.m_Sample)
        {
        return true ;
        }
      else
        {
        return false ;
        }
    }
    
    Iterator& operator=(const Iterator& it)
    {
      m_Id = it.m_Id;
      m_MembershipSample = it.m_MembershipSample ;
      m_Sample = it.m_Sample ;
      return *this ;
    }

    Iterator(const Iterator& it)
    {
      m_Id = it.m_Id;
      m_MembershipSample = it.m_MembershipSample ;
      m_Sample = it.m_Sample ;
    }
    
  private:
    // identifier for the instance
    InstanceIdentifier m_Id ;  
    // Pointer to MemebershipSample object
    Self* m_MembershipSample ;
    TSample* m_Sample ;
  } ;

protected:
  MembershipSample() ;
  virtual ~MembershipSample() {}
  void PrintSelf(std::ostream& os, Indent indent) const;  
  
private:
  MembershipSample(const Self&) ; //purposely not implemented
  void operator=(const Self&) ; //purposely not implemented

  TSample*                        m_Sample ;
  unsigned int                    m_CurrentClassLabel ;
  UniqueClassLabelsType           m_UniqueClassLabels ;
  ClassLabelHolderType            m_ClassLabelHolder ;
  unsigned int                    m_NumberOfClasses ;
  std::vector< unsigned int >     m_ClassSampleSizes ;
  std::vector< typename ClassSampleType::Pointer > m_ClassSamples ;
} ; // end of class


} // end of namespace Statistics 
} // end of namespace itk


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

#endif

---