FunctionDomainMD.cpp

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 +
//----------------------------------------------------------------------
// Includes
//----------------------------------------------------------------------
#include "MantidAPI/FunctionDomainMD.h"
#include "MantidAPI/IMDIterator.h"
 
#include <stdexcept>
 
namespace Mantid::API {
 
FunctionDomainMD::FunctionDomainMD(const IMDWorkspace_const_sptr &ws, size_t start, size_t length)
    : m_iterator(ws->createIterator()), m_startIndex(start), m_currentIndex(0), m_justReset(true), m_workspace(ws) {
  size_t dataSize = m_iterator->getDataSize();
  m_size = length == 0 ? dataSize : length;
  if (start >= dataSize) {
    throw std::out_of_range("Start point out of range");
  }
  if (start + length > dataSize) {
    throw std::out_of_range("End point out of range");
  }
  if (start > 0) {
    m_iterator->jumpTo(start);
  }
}
 
FunctionDomainMD::~FunctionDomainMD() = default;
 
void FunctionDomainMD::reset() const {
  m_iterator->jumpTo(m_startIndex);
  m_currentIndex = 0;
  m_justReset = true;
}
 
const IMDIterator *FunctionDomainMD::getNextIterator() const {
  if (m_justReset) {
    m_justReset = false;
    return m_iterator.get();
  }
  ++m_currentIndex;
  if (!m_iterator->next() || m_currentIndex >= m_size) {
    m_currentIndex = m_size;
    return nullptr;
  }
  return m_iterator.get();
}
 
IMDWorkspace_const_sptr FunctionDomainMD::getWorkspace() const { return m_workspace; }
 
} // namespace Mantid::API