LCSlicedC02Traversal.h

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
 
#include "LCTraversalInterface.h"
#include "autopas/containers/cellTraversals/SlicedC02BasedTraversal.h"
#include "autopas/containers/linkedCells/traversals/LCC08CellHandler.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
#include "autopas/utils/WrapOpenMP.h"
 
namespace autopas {
 
template <class ParticleCell_T, class Functor_T>
class LCSlicedC02Traversal : public SlicedC02BasedTraversal<ParticleCell_T, Functor_T>, public LCTraversalInterface {
 public:
  explicit LCSlicedC02Traversal(const std::array<unsigned long, 3> &dims, Functor_T &functor, double interactionLength,
                                const std::array<double, 3> &cellLength, DataLayoutOption dataLayout, bool useNewton3)
      : SlicedC02BasedTraversal<ParticleCell_T, Functor_T>(dims, functor, interactionLength, cellLength, dataLayout,
                                                           useNewton3, true),
        _cellHandler(functor, this->_cellsPerDimension, interactionLength, cellLength, this->_overlap, dataLayout,
                     useNewton3) {}
 
  void traverseParticles() override;
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::lc_sliced_c02; }
 
  void setAoSSortingThreshold(size_t aosSortingThreshold) override {
    _cellHandler.setAoSSortingThreshold(aosSortingThreshold);
  }
  void setSoASortingThreshold(size_t soaSortingThreshold) override {
    _cellHandler.setSoASortingThreshold(soaSortingThreshold);
  }
 
 private:
  LCC08CellHandler<ParticleCell_T, Functor_T> _cellHandler;
};
 
template <class ParticleCell_T, class Functor_T>
inline void LCSlicedC02Traversal<ParticleCell_T, Functor_T>::traverseParticles() {
  auto &cells = *(this->_cells);
  this->cSlicedTraversal([&](unsigned long x, unsigned long y, unsigned long z) {
    auto id = utils::ThreeDimensionalMapping::threeToOneD(x, y, z, this->_cellsPerDimension);
    _cellHandler.processBaseCell(cells, id);
  });
}
 
}  // namespace autopas