RotCounter.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 +
#include "MantidGeometry/Math/RotCounter.h"
#include <algorithm>
#include <fstream>
#include <iterator>
#include <vector>
 
namespace Mantid::Geometry {
std::ostream &operator<<(std::ostream &OX, const RotaryCounter &A)
{
  A.write(OX);
  return OX;
}
 
RotaryCounter::RotaryCounter(const int S, const int N)
    : Rmax(N), RC(S)
{
  for (int i = 0; i < S; i++)
    RC[i] = i;
}
 
bool RotaryCounter::operator==(const RotaryCounter &A) const
{
  if (RC.size() != A.RC.size())
    return false;
 
  for (size_t i = 0; i < RC.size(); i++) {
    if (RC[i] != A.RC[i]) {
      return false;
    }
  }
  return true;
}
 
bool RotaryCounter::operator>(const RotaryCounter &A) const
{
  const size_t ourSize = RC.size();
  const size_t theirSize = A.RC.size();
  const size_t maxI = (theirSize > ourSize) ? ourSize : theirSize;
  for (size_t i = 0; i < maxI; i++) {
    if (RC[i] != A.RC[i]) {
      return RC[i] > A.RC[i];
    }
  }
  if (theirSize != ourSize) {
    return ourSize > theirSize;
  }
  return false;
}
 
bool RotaryCounter::operator<(const RotaryCounter &A) const
{
  const size_t ourSize = RC.size();
  const size_t theirSize = A.RC.size();
  const size_t maxI = (theirSize > ourSize) ? ourSize : theirSize;
  for (size_t i = 0; i < maxI; i++) {
    if (RC[i] != A.RC[i]) {
      return RC[i] < A.RC[i];
    }
  }
  if (theirSize != ourSize)
    return ourSize < theirSize;
  return false;
}
 
int RotaryCounter::operator++(int a)
{
  (void)a; // Avoid compiler warning
  return this->operator++();
}
 
int RotaryCounter::operator++()
{
  int Npart = Rmax - 1;
  int I;
  for (I = static_cast<int>(RC.size()) - 1; I >= 0 && RC[I] == Npart; I--, Npart--)
    ;
 
  if (I < 0) {
    for (int i = 0; i < static_cast<int>(RC.size()); i++)
      RC[i] = i;
    return 1;
  }
  RC[I]++;
  for (I++; I < static_cast<int>(RC.size()); I++)
    RC[I] = RC[I - 1] + 1;
  return 0;
}
 
int RotaryCounter::operator--(int a)
{
  (void)a; // Avoid compiler warning
  return this->operator--();
}
 
int RotaryCounter::operator--()
{
  const auto Size(static_cast<int>(RC.size()));
  int I;
  for (I = Size - 1; I > 0 && RC[I] == RC[I - 1] + 1; I--)
    ;
  // Loop case
  if (!I && !RC[0]) {
    // In case of loop go to
    for (int i = 0; i < Size; i++) {
      RC[i] = Rmax + i - Size;
    }
    return 1;
  }
 
  RC[I]--;
  for (I++; I < Size; I++) {
    RC[I] = Rmax + I - Size;
  }
  return 0;
}
 
void RotaryCounter::write(std::ostream &OX) const
{
  OX << " ";
  copy(RC.begin(), RC.end() - 1, std::ostream_iterator<int>(OX, ":"));
  OX << RC.back() << " ";
}
} // namespace Mantid::Geometry