mitkDataCollectionSingleImageIterator.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 mitkDataCollectionSingleImageIterator_H
#define mitkDataCollectionSingleImageIterator_H


#include <mitkDataCollection.h>
#include <itkImageRegionIterator.h>

namespace mitk
{
  template <typename TDataType, int ImageDimension>
  class DataCollectionSingleImageIterator
  {
  public:
    typedef itk::Image<TDataType, ImageDimension> ImageType;
    typedef itk::ImageRegionIterator<ImageType> ImageIterator;

    DataCollectionSingleImageIterator(mitk::DataCollection * collection, std::string imageName);

    void ToBegin();

    bool IsAtEnd();

    void operator++();

    // TODO refactor implementation
    void operator++(int);

    size_t GetIndex();

    std::string GetFilePrefix();

    void NextObject();

    ImageType * GetImage();

    void AddImage(ImageType * image, std::string name);

  private:
    // Variables
    DataCollection * m_Collection;
    std::string m_ImageName;
    bool m_IsAtEnd;
    bool m_IteratingImages;
    size_t m_CurrentIndex;
    size_t m_CurrentElement;
    ImageType * m_Image;
    DataCollectionSingleImageIterator<TDataType, ImageDimension>* m_CurrentSingleCollectionIterator;

    // Methods
    DataCollectionSingleImageIterator<TDataType, ImageDimension>* GetNextDataCollectionIterator(size_t start);

  };
} // end namespace

#include <Iterators/mitkDataCollectionSingleImageIterator.cxx>
#endif //mitkDataCollectionSingleImageIterator_H