LCC04CombinedSoATraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include "LCC04SoACellHandler.h"
#include "LCTraversalInterface.h"
#include "autopas/containers/cellTraversals/C04BasedTraversal.h"
#include "autopas/utils/WrapOpenMP.h"
 
namespace autopas {
 
template <class ParticleCell, class PairwiseFunctor>
class LCC04CombinedSoATraversal : public C04BasedTraversal<ParticleCell, PairwiseFunctor, 2>,
                                  public LCTraversalInterface {
 public:
  explicit LCC04CombinedSoATraversal(const std::array<unsigned long, 3> &dims, PairwiseFunctor *pairwiseFunctor,
                                     const double interactionLength, const std::array<double, 3> &cellLength,
                                     DataLayoutOption dataLayout, bool useNewton3)
      : C04BasedTraversal<ParticleCell, PairwiseFunctor, 2>(dims, pairwiseFunctor, interactionLength, cellLength,
                                                            dataLayout, useNewton3),
        _cellHandler(pairwiseFunctor, this->_cellsPerDimension, interactionLength, cellLength, dataLayout, useNewton3,
                     this->_overlap) {}
 
  void traverseParticles() override;
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::lc_c04_combined_SoA; }
 
  [[nodiscard]] bool isApplicable() const override {
    return this->_dataLayout == DataLayoutOption::soa and
           (this->_overlap[0] == 1 and this->_overlap[1] == 1 and this->_overlap[2] == 1);
  }
 
  void setSortingThreshold(size_t sortingThreshold) override {}
 
 private:
  LCC04SoACellHandler<ParticleCell, PairwiseFunctor> _cellHandler;
};
 
template <class ParticleCell, class PairwiseFunctor>
inline void LCC04CombinedSoATraversal<ParticleCell, PairwiseFunctor>::traverseParticles() {
  _cellHandler.resizeBuffers();
  auto &cells = *(this->_cells);
  this->c04Traversal(
      [&](unsigned long x, unsigned long y, unsigned long z) { _cellHandler.processBaseCell(cells, x, y, z); });
}
 
}  // namespace autopas