LCSlicedBalancedTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include <algorithm>
 
#include "LCTraversalInterface.h"
#include "autopas/containers/cellTraversals/SlicedBalancedBasedTraversal.h"
#include "autopas/containers/linkedCells/traversals/LCC08CellHandler.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
 
namespace autopas {
 
template <class ParticleCell, class PairwiseFunctor>
class LCSlicedBalancedTraversal : public SlicedBalancedBasedTraversal<ParticleCell, PairwiseFunctor>,
                                  public LCTraversalInterface {
 public:
  explicit LCSlicedBalancedTraversal(const std::array<unsigned long, 3> &dims, PairwiseFunctor &pairwiseFunctor,
                                     double interactionLength, const std::array<double, 3> &cellLength,
                                     DataLayoutOption dataLayout, bool useNewton3)
      : SlicedBalancedBasedTraversal<ParticleCell, PairwiseFunctor>(dims, pairwiseFunctor, interactionLength,
                                                                    cellLength, dataLayout, useNewton3, true),
        _cellHandler(pairwiseFunctor, this->_cellsPerDimension, interactionLength, cellLength, this->_overlap,
                     dataLayout, useNewton3) {}
 
  void traverseParticles() override;
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::lc_sliced_balanced; }
 
  void setAoSSortingThresholds(const SortingThresholdInfoInterface &aosSortingThreshold) override {
    _cellHandler.setAoSSortingThresholds(aosSortingThreshold);
  }
 
  void setSoASortingThresholds(const SortingThresholdInfoInterface &soaSortingThreshold) override {
    _cellHandler.setSoASortingThresholds(soaSortingThreshold);
  }
 
 private:
  LCC08CellHandler<ParticleCell, PairwiseFunctor> _cellHandler;
};
 
template <class ParticleCell, class PairwiseFunctor>
inline void LCSlicedBalancedTraversal<ParticleCell, PairwiseFunctor>::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