itkListSample.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkListSample.h,v $
  Language:  C++
  Date:      $Date: 2007-02-16 14:53:28 $
  Version:   $Revision: 1.29 $

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

#include "itkObjectFactory.h"
#include "itkMacro.h"
#include "itkListSampleBase.h"
#include "itkFixedArray.h"
#include "itkSmartPointer.h"

#include <vector>

namespace itk{ 
namespace Statistics{

template< class TMeasurementVector >
class ITK_EXPORT ListSample : public ListSampleBase< TMeasurementVector >
{
public:
  typedef ListSample  Self;
  typedef ListSampleBase< TMeasurementVector > Superclass;
  typedef SmartPointer< Self > Pointer;
  typedef SmartPointer<const Self> ConstPointer;

  itkTypeMacro(ListSample, ListSampleBase);

  itkNewMacro(Self) ;

  typedef typename Superclass::MeasurementVectorType MeasurementVectorType;
  typedef typename Superclass::MeasurementVectorSizeType MeasurementVectorSizeType;
  typedef typename Superclass::MeasurementType MeasurementType;
  typedef typename Superclass::FrequencyType FrequencyType ;
  typedef typename Superclass::TotalFrequencyType TotalFrequencyType ;
  typedef typename Superclass::InstanceIdentifier InstanceIdentifier;
  typedef typename Superclass::SearchResultVectorType SearchResultVectorType ;

  typedef MeasurementVectorType ValueType ;


  typedef std::vector< MeasurementVectorType > InternalDataContainerType ;

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

  void Clear() 
  { m_InternalContainer.clear() ; }

  void PushBack( MeasurementVectorType mv )
  { m_InternalContainer.push_back( mv ) ; }

  unsigned int Size() const
  { return static_cast<unsigned int>( m_InternalContainer.size() ); }

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

  void SetMeasurement(const InstanceIdentifier &id, 
                      const unsigned int &dim,
                      const MeasurementType &value) ;

  void SetMeasurementVector(const InstanceIdentifier &id, 
                            const MeasurementVectorType &mv) ;

  FrequencyType GetFrequency(const InstanceIdentifier &id) const ;

  TotalFrequencyType GetTotalFrequency() const
  { return static_cast<TotalFrequencyType>( m_InternalContainer.size() ); }

  class ConstIterator;
 
  class Iterator
  {

    friend class ConstIterator;

  public:
    
    Iterator(){}
    
    Iterator(typename InternalDataContainerType::iterator iter, 
             InstanceIdentifier iid)
      :m_Iter(iter), m_InstanceIdentifier(iid)
    {}
    
    FrequencyType GetFrequency() const
    { return 1 ;}

    const MeasurementVectorType & GetMeasurementVector() const
    { return (MeasurementVectorType&) *m_Iter ;} 

    InstanceIdentifier GetInstanceIdentifier() const
    { return m_InstanceIdentifier ;}

    Iterator& operator++()
    { 
      ++m_Iter; 
      ++m_InstanceIdentifier; 
      return *this;
    }
    
    Iterator& operator--()
    { 
      --m_Iter ; 
      --m_InstanceIdentifier;
      return *this ;
    }

    bool operator!=(const Iterator &it)
    { 
      return (m_Iter != it.m_Iter);
    }
    
    bool operator==(const Iterator &it)
    { 
      return (m_Iter == it.m_Iter);
    }
    
    Iterator& operator =(const Iterator & iter)
    { 
      m_Iter = iter.m_Iter; 
      m_InstanceIdentifier = iter.m_InstanceIdentifier ;
      return *this ;
    }

    Iterator(const Iterator &iter)
    {
      m_Iter = iter.m_Iter; 
      m_InstanceIdentifier = iter.m_InstanceIdentifier ;
    }
    
  private:
    typename InternalDataContainerType::iterator m_Iter ;
    InstanceIdentifier m_InstanceIdentifier ;
  } ;

 
  class ConstIterator
  {
  public:
    
    ConstIterator(){}
    
    ConstIterator(typename InternalDataContainerType::const_iterator iter, 
             InstanceIdentifier iid)
      :m_Iter(iter), m_InstanceIdentifier(iid)
    {}
    
    FrequencyType GetFrequency() const
    { return 1 ;}

    const MeasurementVectorType & GetMeasurementVector() const
    { return static_cast<const MeasurementVectorType&>(*m_Iter) ;} 

    InstanceIdentifier GetInstanceIdentifier() const
    { return m_InstanceIdentifier ;}

    ConstIterator& operator++()
    { 
      ++m_Iter; 
      ++m_InstanceIdentifier; 
      return *this;
    }
    
    ConstIterator& operator--()
    { 
      --m_Iter; 
      --m_InstanceIdentifier; 
      return *this ;
    }

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

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


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

    ConstIterator(const Iterator &iter)
    {
      m_Iter = iter.m_Iter; 
      m_InstanceIdentifier = iter.m_InstanceIdentifier ;
    }
    
  private:
    typename InternalDataContainerType::const_iterator m_Iter ;
    InstanceIdentifier m_InstanceIdentifier ;
  } ;

  Iterator Begin()
  { 
    Iterator iter(m_InternalContainer.begin(), 0);
    return iter; 
  }

  Iterator End()        
  {
    Iterator iter(m_InternalContainer.end(), m_InternalContainer.size()); 
    return iter; 
  }

  ConstIterator Begin() const
  { 
    ConstIterator iter(m_InternalContainer.begin(), 0);
    return iter; 
  }

  ConstIterator End() const
  {
    ConstIterator iter(m_InternalContainer.end(), m_InternalContainer.size()); 
    return iter; 
  }

  virtual MeasurementVectorSizeType GetMeasurementVectorSize() const
    {
    if ( !this->Superclass::GetMeasurementVectorSize() && this->Size())
      { // determined from the length of the first vector in the sample
        // at run time
      return MeasurementVectorTraits::GetLength(this->GetMeasurementVector(0));
      }
    return this->Superclass::GetMeasurementVectorSize();
    }
 
protected:
  ListSample() ;
  virtual ~ListSample() {}
  void PrintSelf(std::ostream& os, Indent indent) const; 
  

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

  InternalDataContainerType m_InternalContainer ;

};

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

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

#endif