LCSlicedTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
 
#include "LCTraversalInterface.h"
#include "autopas/containers/cellTraversals/SlicedLockBasedTraversal.h"
#include "autopas/containers/linkedCells/traversals/LCC08CellHandler.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
 
namespace autopas {
 
template <class ParticleCell_T, class Functor_T>
class LCSlicedTraversal : public SlicedLockBasedTraversal<ParticleCell_T, Functor_T>, public LCTraversalInterface {
 public:
  explicit LCSlicedTraversal(const std::array<unsigned long, 3> &dims, Functor_T &functor, double interactionLength,
                             const std::array<double, 3> &cellLength, DataLayoutOption dataLayout, bool useNewton3)
      : SlicedLockBasedTraversal<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; }
 
  void setSortingThreshold(size_t sortingThreshold) override { _cellHandler.setSortingThreshold(sortingThreshold); }
 
 private:
  LCC08CellHandler<ParticleCell_T, Functor_T> _cellHandler;
};
 
template <class ParticleCell_T, class Functor_T>
inline void LCSlicedTraversal<ParticleCell_T, Functor_T>::traverseParticles() {
  auto &cells = *(this->_cells);
  this->slicedTraversal([&](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