Review/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-05-22 12:55:02 $
  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 __itkListSample_h
#define __itkListSample_h

#include "itkObjectFactory.h"
#include "itkFixedArray.h"
#include "itkSample.h"

#include <vector>

namespace itk {
namespace Statistics {

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

  itkTypeMacro(ListSample, Sample);

  itkNewMacro(Self);

  itkSuperclassTraitMacro( MeasurementVectorType )
  itkSuperclassTraitMacro( MeasurementVectorSizeType )
  itkSuperclassTraitMacro( MeasurementType )
  itkSuperclassTraitMacro( AbsoluteFrequencyType )
  itkSuperclassTraitMacro( TotalAbsoluteFrequencyType )
  itkSuperclassTraitMacro( InstanceIdentifier )

  typedef MeasurementVectorType ValueType;


  typedef std::vector< MeasurementVectorType > InternalDataContainerType;

  void Resize( InstanceIdentifier newsize );

  void Clear();

  void PushBack( const MeasurementVectorType & mv );

  InstanceIdentifier Size() const;

  const MeasurementVectorType & GetMeasurementVector(InstanceIdentifier id) const;

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

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

  AbsoluteFrequencyType GetFrequency( InstanceIdentifier id ) const;

  TotalAbsoluteFrequencyType GetTotalFrequency() const;

  virtual void Graft( const DataObject *thatObject );

  class ConstIterator
    {
    friend class ListSample;

    public:

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

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

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


    AbsoluteFrequencyType 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;
      }

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

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

#if !(defined(_MSC_VER) && (_MSC_VER <= 1200))
  protected:
#endif
    // This method should only be available to the ListSample class
    ConstIterator(
      typename InternalDataContainerType::const_iterator iter,
      InstanceIdentifier iid)
      {
      m_Iter = iter;
      m_InstanceIdentifier = iid;
      }

    // This method is purposely not implemented
    ConstIterator();

  private:
    typedef typename InternalDataContainerType::const_iterator InternalIterator;
    InternalIterator     m_Iter;
    InstanceIdentifier   m_InstanceIdentifier;

    };

  class Iterator : public ConstIterator
  {

  friend class ListSample;

  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( typename InternalDataContainerType::const_iterator iter, InstanceIdentifier iid);
    Iterator(const ConstIterator & it);
    ConstIterator& operator=(const ConstIterator& it);
    Iterator(
      typename InternalDataContainerType::iterator iter,
      InstanceIdentifier iid):ConstIterator( iter, iid )
      {
      }

  private:
  };


  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;
    }

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