itkImageToListSampleAdaptor.h

Go to the documentation of this file.

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

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

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

#include <typeinfo>

#include "itkImage.h"
#include "itkPixelTraits.h"
#include "itkListSample.h"
#include "itkSmartPointer.h"
#include "itkImageRegionIterator.h"
#include "itkFixedArray.h"
#include "itkMacro.h"
#include "itkMeasurementVectorTraits.h"

namespace itk { 
namespace Statistics {

template < class TImage >
class ITK_EXPORT ImageToListSampleAdaptor : 
    public ListSample< typename MeasurementVectorPixelTraits< typename TImage::PixelType >::MeasurementVectorType >
{
public:
  typedef ImageToListSampleAdaptor               Self;

  typedef ListSample< typename MeasurementVectorPixelTraits<
     typename TImage::PixelType >::MeasurementVectorType >
                                                 Superclass;

  typedef SmartPointer< Self >                   Pointer;
  typedef SmartPointer<const Self>               ConstPointer;
  
  itkTypeMacro(ImageToListSampleAdaptor, ListSample);

  itkNewMacro(Self);

  typedef TImage                                          ImageType;
  typedef typename ImageType::Pointer                     ImagePointer;
  typedef typename ImageType::ConstPointer                ImageConstPointer;
  typedef typename ImageType::IndexType                   IndexType;
  typedef typename ImageType::PixelType                   PixelType;
  typedef typename ImageType::PixelContainerConstPointer  PixelContainerConstPointer;

  typedef ImageRegionIterator< ImageType >               ImageIteratorType; 
  typedef ImageRegionConstIterator< ImageType >          ImageConstIteratorType; 
  typedef PixelTraits< typename TImage::PixelType >      PixelTraitsType;

  typedef typename MeasurementVectorPixelTraits<
         PixelType >::MeasurementVectorType                    MeasurementVectorType; 
  typedef typename MeasurementVectorTraitsTypes< 
             MeasurementVectorType >::ValueType                MeasurementType;
  itkSuperclassTraitMacro( AbsoluteFrequencyType )
  itkSuperclassTraitMacro( TotalAbsoluteFrequencyType )
  itkSuperclassTraitMacro( MeasurementVectorSizeType )
  itkSuperclassTraitMacro( InstanceIdentifier )

  typedef MeasurementVectorType                                ValueType;

  void SetImage(const TImage* image);

  const TImage* GetImage() const;

  InstanceIdentifier Size() const;

  virtual const MeasurementVectorType & GetMeasurementVector(InstanceIdentifier id) const;

  AbsoluteFrequencyType GetFrequency(InstanceIdentifier  id) const;

  TotalAbsoluteFrequencyType GetTotalFrequency() const;

  itkSetMacro( UsePixelContainer, bool );

  itkGetConstMacro( UsePixelContainer, bool );

  itkBooleanMacro( UsePixelContainer );

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

  class ConstIterator
    {
    friend class ImageToListSampleAdaptor;
    public:

    ConstIterator( const ImageToListSampleAdaptor * adaptor )
      {
      *this = adaptor->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
      {
      MeasurementVectorTraits::Assign( this->m_MeasurementVectorCache, m_Iter.Get());
      return this->m_MeasurementVectorCache;
      }

    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(
      ImageConstIteratorType iter,
      InstanceIdentifier iid)
      {
      m_Iter = iter;
      m_InstanceIdentifier = iid;
      }

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

  private:
    ImageConstIteratorType            m_Iter;
    mutable MeasurementVectorType     m_MeasurementVectorCache;
    InstanceIdentifier                m_InstanceIdentifier;
    };

  class Iterator : public ConstIterator
  {

  friend class ImageToListSampleAdaptor;

  public:

    Iterator(Self * adaptor):ConstIterator(adaptor)
      {
      }

    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 * adaptor);
    Iterator(  ImageConstIteratorType iter, InstanceIdentifier iid);
    Iterator(const ConstIterator & it);
    ConstIterator& operator=(const ConstIterator& it);
    Iterator(
      ImageIteratorType iter,
      InstanceIdentifier iid):ConstIterator( iter, iid )
      {
      }

  private:
  };


  Iterator Begin()
    {
    ImagePointer  nonConstImage = const_cast< ImageType* >(m_Image.GetPointer());
    ImageIteratorType imageIterator( nonConstImage, nonConstImage->GetLargestPossibleRegion());  
    Iterator iter(imageIterator.Begin(), 0);
    return iter;
    }

  Iterator End()
    {
    ImagePointer  nonConstImage = const_cast< ImageType* >(m_Image.GetPointer());
    ImageIteratorType imageIterator( nonConstImage, nonConstImage->GetLargestPossibleRegion());  
    Iterator iter(imageIterator.End(), m_Image->GetPixelContainer()->Size());
    return iter;
    }


  ConstIterator Begin() const
    {
    ImageConstIteratorType imageConstIterator( m_Image, m_Image->GetLargestPossibleRegion());  
    ConstIterator iter(imageConstIterator.Begin(), 0);
    return iter;
    }

  ConstIterator End() const
    {
    ImageConstIteratorType imageConstIterator( m_Image, m_Image->GetLargestPossibleRegion());  
    ConstIterator iter(imageConstIterator.End(), m_Image->GetPixelContainer()->Size());
    return iter;
    }


protected:
  ImageToListSampleAdaptor();
  virtual ~ImageToListSampleAdaptor() {}
  void PrintSelf(std::ostream& os, Indent indent) const;  

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

  ImageConstPointer                  m_Image;
  mutable MeasurementVectorType      m_MeasurementVectorInternal;
  bool                               m_UsePixelContainer;

  PixelContainerConstPointer m_PixelContainer;

}; // end of class ImageToListSampleAdaptor

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

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

#endif