DSSequentialTraversal.h

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "DSTraversalInterface.h"
#include "autopas/baseFunctors/CellFunctor.h"
#include "autopas/baseFunctors/CellFunctor3B.h"
#include "autopas/containers/TraversalInterface.h"
#include "autopas/containers/cellTraversals/CellTraversal.h"
#include "autopas/options/DataLayoutOption.h"
#include "autopas/utils/DataLayoutConverter.h"
#include "autopas/utils/checkFunctorType.h"
 
namespace autopas {
 
template <class ParticleCell, class Functor>
class DSSequentialTraversal : public CellTraversal<ParticleCell>,
                              public TraversalInterface,
                              public DSTraversalInterface {
 public:
  explicit DSSequentialTraversal(Functor &functor, double cutoff, DataLayoutOption dataLayout, bool useNewton3)
      : CellTraversal<ParticleCell>({2, 1, 1}),
        TraversalInterface(dataLayout, useNewton3),
        _cellFunctor(functor, cutoff /*should use cutoff here, if not used to build verlet-lists*/, dataLayout,
                     useNewton3),
        _dataLayoutConverter(functor, dataLayout) {}
 
  [[nodiscard]] TraversalOption getTraversalType() const override { return TraversalOption::ds_sequential; }
 
  [[nodiscard]] bool isApplicableToDomain() const override { return true; }
 
  void initTraversal() override {
    auto &cells = *(this->_cells);
    for (auto &cell : cells) {
      _dataLayoutConverter.loadDataLayout(cell);
    }
  }
 
  void endTraversal() override {
    auto &cells = *(this->_cells);
    for (auto &cell : cells) {
      _dataLayoutConverter.storeDataLayout(cell);
    }
  }
 
  void traverseParticles() override;
 
  void setAoSSortingThresholds(const SortingThresholdInfoInterface &aosSortingThreshold) override {
    _cellFunctor.setAoSSortingThresholds(aosSortingThreshold);
  }
 
  void setSoASortingThresholds(const SortingThresholdInfoInterface &soaSortingThreshold) override {
    _cellFunctor.setSoASortingThresholds(soaSortingThreshold);
  }
 
 private:
  // CellFunctor type for either Pairwise or Triwise Functors.
  using CellFunctorType = std::conditional_t<decltype(utils::isPairwiseFunctor<Functor>())::value,
                                             internal::CellFunctor<ParticleCell, Functor, /*bidirectional*/ false>,
                                             internal::CellFunctor3B<ParticleCell, Functor, /*bidirectional*/ false>>;
 
  CellFunctorType _cellFunctor;
 
  utils::DataLayoutConverter<Functor> _dataLayoutConverter;
};
 
template <class ParticleCell, class Functor>
void DSSequentialTraversal<ParticleCell, Functor>::traverseParticles() {
  using namespace autopas::utils::ArrayMath::literals;
 
  // cells[0] is the owned domain and cells[1-6] are the halo cells.
  auto &cells = *(this->_cells);
 
  // Process interactions between 3 owned particles.
  _cellFunctor.processCell(cells[0]);
 
  constexpr std::array<std::array<double, 3>, 7> haloDirections{
      {{0., 0., 0.}, {-1., 0., 0.}, {1., 0., 0.}, {0., -1., 0.}, {0., 1., 0.}, {0., 0., -1.}, {0., 0., 1.}}};
 
  //  Process pair interactions with all six halo cells.
  for (int i = 1; i < cells.size(); i++) {
    _cellFunctor.processCellPair(cells[0], cells[i], std::array<double, 3>(haloDirections[i]));
  }
 
  if constexpr (utils::isTriwiseFunctor<Functor>()) {
    // Process cell triplet interactions with 1 owned particle and two halo particles from different cells.
    for (int i = 1; i < cells.size(); i++) {
      for (int j = i + 1; j < cells.size(); j++) {
        // Sorting direction should be from owned to first halo cell.
        _cellFunctor.processCellTriple(cells[0], cells[i], cells[j], haloDirections[i]);
      }
    }
  }
}
 
}  // namespace autopas