C04BasedTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include "ColorBasedTraversal.h"
#include "autopas/utils/ArrayMath.h"
#include "autopas/utils/DataLayoutConverter.h"
 
namespace autopas {
 
template <class ParticleCell, class Functor, int collapseDepth = 3>
class C04BasedTraversal : public ColorBasedTraversal<ParticleCell, Functor, collapseDepth> {
 public:
  explicit C04BasedTraversal(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, collapseDepth>(dims, functor, interactionLength, cellLength,
                                                                  dataLayout, useNewton3) {}
 
 protected:
  template <typename LoopBody>
  inline void c04Traversal(LoopBody &&loopBody);
};
 
template <class ParticleCell, class Functor, int collapseDepth>
template <typename LoopBody>
inline void C04BasedTraversal<ParticleCell, Functor, collapseDepth>::c04Traversal(LoopBody &&loopBody) {
  using namespace autopas::utils::ArrayMath::literals;
 
  const auto end = this->_cellsPerDimension - this->_overlap;
  auto stride = this->_overlap + 1ul;
  stride[0] = 1;
  this->colorTraversal(std::forward<LoopBody>(loopBody), end, stride);
}
}  // namespace autopas