mitkDataCollectionVectorImageIterator.h

Go to the documentation of this file.

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

 The Medical Imaging Interaction Toolkit (MITK)

 Copyright (c) German Cancer Research Center,
 Division of Medical and Biological Informatics.
 All rights reserved.

 This software is distributed WITHOUT ANY WARRANTY; without
 even the implied warranty of MERCHANTABILITY or FITNESS FOR
 A PARTICULAR PURPOSE.

 See LICENSE.txt or http://www.mitk.org for details.

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

#ifndef DATACOLLECTIONFEATUREVECTORIMAGEITERATOR_H_
#define DATACOLLECTIONFEATUREVECTORIMAGEITERATOR_H_
#include <mitkDataCollection.h>
#include <mitkDataCollectionSingleImageIterator.h>

#include "itkImage.h"

namespace mitk
{

  /*
   * Iterate over a data collection and return a vector image for all selected
   * modalities.
   * TODO SW: this class needs a module test urgently.
   */
  template <typename TDataType, int ImageDimension>
  class DataCollectionVectorImageIterator
  {

  public:
    typedef itk::VectorImage<TDataType, ImageDimension> VectorImageType;
    typedef typename VectorImageType::Pointer VectorImagePointerType;
    typedef itk::ImageRegionIterator<VectorImageType> VectorImageIteratorType;

    DataCollectionVectorImageIterator(DataCollection::Pointer collection, std::vector<std::string> featureNames);

    void ToBegin();

    bool IsAtEnd();

    DataCollectionVectorImageIterator& operator++();

    DataCollectionVectorImageIterator operator++(int);

    VectorImagePointerType GetVectorImage();

    void
    AddImage(typename itk::Image<TDataType, ImageDimension>::Pointer image, std::string name);

    typedef typename mitk::DataCollectionSingleImageIterator<TDataType, ImageDimension> SingleImageIterator;
    typedef typename std::vector< SingleImageIterator > DataCollectionSingleImageIterators;

  private:

    DataCollection::Pointer m_Collection;
    DataCollectionSingleImageIterators m_FeatureIterators;
  };

} /* namespace mitk */


#include <Iterators/mitkDataCollectionVectorImageIterator.cxx>

#endif /* DATACOLLECTIONFEATUREVECTORIMAGEITERATOR_H_ */