mitkDataCollectionImageIterator.h

Go to the documentation of this file.

/*============================================================================
 
The Medical Imaging Interaction Toolkit (MITK)
 
Copyright (c) German Cancer Research Center (DKFZ)
All rights reserved.
 
Use of this source code is governed by a 3-clause BSD license that can be
found in the LICENSE file.
 
============================================================================*/
 
 
#ifndef mitkDataCollectionImageIterator_h
#define mitkDataCollectionImageIterator_h
 
#include <mitkImageCast.h>
#include <mitkDataCollection.h>
#include <itkImageRegionIterator.h>
 
namespace mitk
{
  template <typename TDataType, int ImageDimension>
  class DataCollectionImageIterator
  {
  public:
    typedef itk::Image<TDataType, ImageDimension> ImageType;
    typedef typename ImageType::Pointer ImagePointerType;
    typedef itk::ImageRegionIterator<ImageType> ImageIterator;
 
    DataCollectionImageIterator(DataCollection::Pointer collection, std::string imageName);
 
    void ToBegin();
 
    bool IsAtEnd();
 
    void operator++();
 
    // TODO refactor implementation
    void operator++(int);
 
    TDataType GetVoxel();
 
    void SetVoxel(TDataType value);
 
    size_t GetIndex();
    size_t GetImageIndex () { return m_ImageIndex;}
 
    std::string GetFilePrefix();
 
    void NextObject();
 
    ImageIterator GetImageIterator();
 
  private:
    DataCollectionImageIterator<TDataType, ImageDimension>* GetNextDataCollectionIterator(size_t start);
 
    // DATA
    DataCollection::Pointer m_Collection;
    std::string m_ImageName;
 
    bool m_IsAtEnd;
    bool m_IteratingImages;
    size_t m_CurrentIndex;
    size_t m_ImageIndex;
 
    size_t m_CurrentElement;
    DataCollectionImageIterator<TDataType, ImageDimension>* m_CurrentCollectionIterator;
    ImageIterator m_CurrentIterator;
  };
} // end namespace
 
#include <Iterators/mitkDataCollectionImageIterator.cxx>
#endif