VLCCellPairC08Traversal.h

Go to the documentation of this file.

 
#pragma once
#include "autopas/containers/cellTraversals/C08BasedTraversal.h"
#include "autopas/containers/verletListsCellBased/verletListsCells/neighborLists/VLCCellPairNeighborList.h"
#include "autopas/containers/verletListsCellBased/verletListsCells/traversals/VLCCellPairC08CellHandler.h"
#include "autopas/options/DataLayoutOption.h"
 
namespace autopas {
 
template <class ParticleCell, class PairwiseFunctor>
 
class VLCCellPairC08Traversal : public C08BasedTraversal<ParticleCell, PairwiseFunctor>,
                                public VLCCellPairTraversalInterface<typename ParticleCell::ParticleType> {
 public:
  explicit VLCCellPairC08Traversal(const std::array<unsigned long, 3> &dims, PairwiseFunctor *pairwiseFunctor,
                                   double interactionLength, const std::array<double, 3> &cellLength,
                                   DataLayoutOption dataLayout, bool useNewton3)
      : C08BasedTraversal<ParticleCell, PairwiseFunctor>(dims, pairwiseFunctor, interactionLength, cellLength,
                                                         dataLayout, useNewton3),
        _functor(pairwiseFunctor),
        _cellHandler(dims, interactionLength, cellLength) {}
 
  void traverseParticles() override;
 
  [[nodiscard]] bool isApplicable() const override {
    return (this->_dataLayout == DataLayoutOption::aos or this->_dataLayout == DataLayoutOption::soa);
  }
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::vlp_c08; }
 
  void setSortingThreshold(size_t sortingThreshold) override {}
 
 private:
  PairwiseFunctor *_functor;
  VLCCellPairC08CellHandler<ParticleCell, PairwiseFunctor> _cellHandler;
  SoA<typename ParticleCell::ParticleType::SoAArraysType> *_soa;
};
 
template <class ParticleCell, class PairwiseFunctor>
inline void VLCCellPairC08Traversal<ParticleCell, PairwiseFunctor>::traverseParticles() {
  if (this->_dataLayout == DataLayoutOption::soa) {
    _soa = this->_cellPairVerletList->loadSoA(_functor);
  }
 
  this->c08Traversal([&](unsigned long x, unsigned long y, unsigned long z) {
    const auto baseIndex = utils::ThreeDimensionalMapping::threeToOneD(x, y, z, this->_cellsPerDimension);
    _cellHandler.processCellListsC08(*(this->_cellPairVerletList), baseIndex, _functor, this->_dataLayout, _soa,
                                     this->_useNewton3);
  });
 
  if (this->_dataLayout == DataLayoutOption::soa) {
    this->_cellPairVerletList->extractSoA(_functor);
    _soa = nullptr;
  }
}
 
}  // namespace autopas