Numerics/Statistics/itkListSample.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkListSample.h,v $
  Language:  C++
  Date:      $Date: 2009-03-04 15:23:51 $
  Version:   $Revision: 1.30 $

  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