SoAView.h

Go to the documentation of this file.

 
#pragma once
 
namespace autopas {
 
template <class SoAArraysType>
class SoA;
 
template <class SoAArraysType>
class SoAView {
 public:
  SoAView() : _soa(nullptr), _startIndex(0), _endIndex(0) {}
 
  SoAView(SoA<SoAArraysType> *soa, size_t startIndex, size_t endIndex)
      : _soa(soa), _startIndex(startIndex), _endIndex(endIndex) {
    if (not(soa->size() >= endIndex and endIndex >= startIndex)) /* @todo C++20 [[unlikely]] */ {
      utils::ExceptionHandler::exception("SoAView: Trying to view particles outside of the SoA.");
    }
  }
 
  explicit SoAView(SoA<SoAArraysType> *soa) : _soa(soa), _startIndex(0), _endIndex(soa->size()) {}
 
  SoAView(SoA<SoAArraysType> &soa) : _soa(&soa), _startIndex(0), _endIndex(soa.size()) {}
 
  template <size_t attribute>
  [[nodiscard]] auto begin() {
    return _soa->template begin<attribute>() + _startIndex;
  }
 
  template <size_t attribute>
  [[nodiscard]] auto begin() const {
    return _soa->template begin<attribute>() + _startIndex;
  }
 
  [[nodiscard]] size_t size() const { return _endIndex - _startIndex; }
 
 private:
  SoA<SoAArraysType> *_soa;
 
  size_t _startIndex;
 
  size_t _endIndex;
};
 
}  // namespace autopas