C18BasedTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/containers/cellTraversals/ColorBasedTraversal.h"
#include "autopas/utils/ThreeDimensionalMapping.h"
 
namespace autopas {
 
template <class ParticleCell, class Functor>
class C18BasedTraversal : public ColorBasedTraversal<ParticleCell, Functor> {
 public:
  explicit C18BasedTraversal(const std::array<unsigned long, 3> &dims, Functor *functor, const double interactionLength,
                             const std::array<double, 3> &cellLength, DataLayoutOption dataLayout, bool useNewton3)
      : ColorBasedTraversal<ParticleCell, Functor>(dims, functor, interactionLength, cellLength, dataLayout,
                                                   useNewton3) {}
 
 protected:
  template <bool allCells, typename LoopBody>
  inline void c18Traversal(LoopBody &&loopBody);
};
 
template <class ParticleCell, class Functor>
template <bool allCells, typename LoopBody>
inline void C18BasedTraversal<ParticleCell, Functor>::c18Traversal(LoopBody &&loopBody) {
  const std::array<unsigned long, 3> stride = {
      2ul * this->_overlap[0] + 1ul,
      2ul * this->_overlap[1] + 1ul,
      this->_overlap[2] + 1ul,
  };
  auto end(this->_cellsPerDimension);
  if (not allCells) {
    end[2] -= this->_overlap[2];
  }
  this->colorTraversal(std::forward<LoopBody>(loopBody), end, stride);
}
 
}  // namespace autopas