itkMultipleImageIterator.h

Go to the documentation of this file.

/*=========================================================================
*
*  Copyright Insight Software Consortium
*
*  Licensed under the Apache License, Version 2.0 (the "License");
*  you may not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
*         http://www.apache.org/licenses/LICENSE-2.0.txt
*
*  Unless required by applicable law or agreed to in writing, software
*  distributed under the License is distributed on an "AS IS" BASIS,
*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*  See the License for the specific language governing permissions and
*  limitations under the License.
*
*=========================================================================*/

#ifndef itkMultipleImageIterator_h
#define itkMultipleImageIterator_h
#include <vector>
#include <itkImageRegionIterator.h>


namespace itk {
template<typename TIterator>
class MultipleImageIterator {
public:
  using Self = MultipleImageIterator;
  using IteratorType = TIterator;
  using ImageType = typename IteratorType::ImageType;
  IteratorType& operator[](const int i) {return m_Iterators[i];}
  void AddIterator(const IteratorType& it) {m_Iterators.push_back(it);}
  Self& operator++() {
    for (auto it = m_Iterators.begin();
      it != m_Iterators.end(); ++it) {
      ++(*it);
    }
    return *this;
  }
  void GoToBegin() {
    for (auto it = m_Iterators.begin();
      it != m_Iterators.end(); ++it) {
      it->GoToBegin();
    }
  }
  bool IsAtEnd() {
#ifdef NDEBUG
    return m_Iterators[0].IsAtEnd();
#else
    assert(m_Iterators.size());
    bool retval = m_Iterators[0].IsAtEnd();
    for (unsigned int i=0; i<m_Iterators.size();++i)
      assert(m_Iterators[i].IsAtEnd() == retval);
    return retval;
#endif
  }
  unsigned int Size () const { return m_Iterators.size(); }
protected:
  std::vector<IteratorType> m_Iterators;

};
}
#endif //itkMultipleImageIterator_h