SpectrumInfoPythonIterator.h

Go to the documentation of this file.

// Mantid Repository : https://github.com/mantidproject/mantid
//
// Copyright © 2018 ISIS Rutherford Appleton Laboratory UKRI,
//   NScD Oak Ridge National Laboratory, European Spallation Source,
//   Institut Laue - Langevin & CSNS, Institute of High Energy Physics, CAS
// SPDX - License - Identifier: GPL - 3.0 +
#pragma once
 
#include "MantidAPI/SpectrumInfo.h"
#include "MantidAPI/SpectrumInfoItem.h"
#include "MantidAPI/SpectrumInfoIterator.h"
 
#include <boost/python/iterator.hpp>
 
using Mantid::API::SpectrumInfo;
using Mantid::API::SpectrumInfoItem;
using Mantid::API::SpectrumInfoIterator;
using namespace boost::python;
 
namespace Mantid {
namespace PythonInterface {
 
class SpectrumInfoPythonIterator {
public:
  explicit SpectrumInfoPythonIterator(SpectrumInfo &spectrumInfo)
      : m_begin(spectrumInfo.begin()), m_end(spectrumInfo.end()), m_firstOrDone(true) {}
 
  const SpectrumInfoItem<SpectrumInfo> &next() {
    if (!m_firstOrDone)
      ++m_begin;
    else
      m_firstOrDone = false;
    if (m_begin == m_end) {
      m_firstOrDone = true;
      objects::stop_iteration_error();
    }
    return *m_begin;
  }
 
private:
  SpectrumInfoIterator<SpectrumInfo> m_begin;
  SpectrumInfoIterator<SpectrumInfo> m_end;
  bool m_firstOrDone;
};
 
} // namespace PythonInterface
} // namespace Mantid