Review/Statistics/itkMembershipSample.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkMembershipSample.h,v $
  Language:  C++
  Date:      $Date: 2009-05-22 12:55:07 $
  Version:   $Revision: 1.2 $

  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"

namespace itk {
namespace Statistics {

template< class TSample >
class ITK_EXPORT MembershipSample : public DataObject
{
public:
  typedef MembershipSample            Self;
  typedef DataObject                  Superclass;
  typedef SmartPointer< Self >        Pointer;
  typedef SmartPointer< const Self >  ConstPointer;

  itkTypeMacro(MembershipSample, DataObject);
  itkNewMacro(Self);

  typedef TSample                                    SampleType;
  typedef typename SampleType::MeasurementVectorType MeasurementVectorType;
  typedef typename SampleType::MeasurementType       MeasurementType;
  typedef typename SampleType::InstanceIdentifier    InstanceIdentifier;
  typedef typename SampleType::ConstPointer          SampleConstPointer;

  typedef typename SampleType::AbsoluteFrequencyType         AbsoluteFrequencyType;
  typedef typename SampleType::TotalAbsoluteFrequencyType    TotalAbsoluteFrequencyType;
  

  typedef unsigned long              ClassLabelType;  
  typedef std::vector< ClassLabelType > UniqueClassLabelsType;

  typedef itk::hash_map< InstanceIdentifier, ClassLabelType > ClassLabelHolderType;

  typedef Subsample< SampleType >                   ClassSampleType;
  typedef typename ClassSampleType::Pointer         ClassSamplePointer;
  typedef typename ClassSampleType::ConstPointer    ClassSampleConstPointer;

  itkSetConstObjectMacro( Sample, SampleType );
  itkGetConstObjectMacro( Sample, SampleType );

  void SetNumberOfClasses( unsigned int numberOfClasses );

  itkGetConstMacro( NumberOfClasses, unsigned int );

  void AddInstance(const ClassLabelType &classLabel, const InstanceIdentifier &id);

  unsigned int GetClassLabel(const InstanceIdentifier &id) const;

  const ClassSampleType* GetClassSample(const  ClassLabelType &classLabel) const;

  const ClassLabelHolderType GetClassLabelHolder() const;

  const MeasurementVectorType & GetMeasurementVector(const InstanceIdentifier &id) const;

  MeasurementType GetMeasurement(const InstanceIdentifier &id, 
                                  const unsigned int &dimension);

  AbsoluteFrequencyType GetFrequency(const InstanceIdentifier &id) const;

  TotalAbsoluteFrequencyType GetTotalFrequency() const;

  virtual void Graft( const DataObject *thatObject );

//  void PrintSelf(std::ostream& os, Indent indent) const;  

  class ConstIterator
    {
    friend class MembershipSample;
    public:

    ConstIterator( const Self * sample )
      {
      *this = sample->Begin();
      }

    ConstIterator(const ConstIterator& iter)
      {
      m_Sample = iter.m_Sample;
      m_MembershipSample = iter.m_MembershipSample;
      m_InstanceIdentifier = iter.m_InstanceIdentifier;
      }

    ConstIterator& operator=(const ConstIterator& iter)
      {
      m_Sample = iter.m_Sample;
      m_MembershipSample = iter.m_MembershipSample;
      m_InstanceIdentifier = iter.m_InstanceIdentifier;
      return *this;
      }

    bool operator!=(const ConstIterator& it) 
      {
      return (m_InstanceIdentifier != it.m_InstanceIdentifier);
      }
    
    bool operator==(const ConstIterator& it) 
      { 
      return (m_InstanceIdentifier == it.m_InstanceIdentifier);
      }

    ConstIterator& operator++() 
      { 
      ++m_InstanceIdentifier;
      return *this;
      }
 
    AbsoluteFrequencyType GetFrequency() const
      {
      return  m_Sample->GetFrequency(m_InstanceIdentifier);
      }
    
    const MeasurementVectorType & GetMeasurementVector() const
      {
      return m_Sample->GetMeasurementVector(m_InstanceIdentifier);
      } 
    
    InstanceIdentifier GetInstanceIdentifier() const   
      {
      return m_InstanceIdentifier; 
      }

    unsigned int   GetClassLabel() const
      {
      return m_MembershipSample->GetClassLabel( m_InstanceIdentifier );
      }
 
#if !(defined(_MSC_VER) && (_MSC_VER <= 1200))
    protected:
#endif
    // Purposely not implemented
    ConstIterator();

    // Only to be called from the MembershipSample
    ConstIterator(
             const Self* memberSample, InstanceIdentifier iid)
      :m_Sample(memberSample->GetSample()), m_MembershipSample( memberSample), m_InstanceIdentifier(iid)
      {

      }


    //typename SampleType::ConstIterator m_Iter; 
    const TSample*                 m_Sample;
    const MembershipSample*        m_MembershipSample;
    InstanceIdentifier             m_InstanceIdentifier;
    };

  class Iterator: public ConstIterator
    {
    friend class MembershipSample;

    public:
    
    Iterator(Self * sample):ConstIterator( sample )
      {
      }

    Iterator(const Iterator &iter):ConstIterator( iter )
      {
      }

    Iterator& operator =(const Iterator & iter)
      {
      this->ConstIterator::operator=( iter );
      return *this;
      }

#if !(defined(_MSC_VER) && (_MSC_VER <= 1200))
    protected:
#endif
    // To ensure const-correctness these method must not be in the public API.
    // The are purposly not implemented, since they should never be called.
    Iterator();
    Iterator(const Self * sample);
    Iterator(const ConstIterator & it);
    ConstIterator& operator=(const ConstIterator& it);

    // Only to be called from the MembershipSample
    Iterator(Self * memberSample,
             InstanceIdentifier iid)
      :ConstIterator( memberSample, iid )
      {} 
     
    private:
    };

  Iterator Begin()
    { 
    Iterator iter(this, 0);
    return iter; 
    }

  Iterator  End()
    {
    Iterator iter(this, m_Sample->Size()); 
    return iter; 
    }

  ConstIterator Begin() const
    { 
    ConstIterator iter(this,  0);
    return iter; 
    }
  
  ConstIterator  End()  const
    {
    ConstIterator iter(this, m_Sample->Size()); 
    return iter; 
    }
 
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

  int GetInternalClassLabel(const ClassLabelType classLabel ) const;


  UniqueClassLabelsType                 m_UniqueClassLabels;
  ClassLabelHolderType                  m_ClassLabelHolder;
  std::vector< ClassSamplePointer >     m_ClassSamples;
  SampleConstPointer                    m_Sample;
  unsigned int                          m_NumberOfClasses;

}; // end of class


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


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

#endif