VLCNeighborListInterface.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/containers/linkedCells/LinkedCells.h"
#include "autopas/containers/verletListsCellBased/verletListsCells/VerletListsCellsHelpers.h"
#include "autopas/options/TraversalOption.h"
 
namespace autopas {
template <class Particle_T>
class VLCNeighborListInterface {
 public:
  virtual ~VLCNeighborListInterface() = default;
 
  virtual void buildAoSNeighborList(TraversalOption vlcTraversalOpt, LinkedCells<Particle_T> &linkedCells,
                                    bool useNewton3) = 0;
 
  virtual size_t getNumberOfPartners(const Particle_T *particle) const = 0;
 
  [[nodiscard]] virtual ContainerOption getContainerType() const = 0;
 
  virtual void generateSoAFromAoS(LinkedCells<Particle_T> &linkedCells) = 0;
 
  template <class TFunctor>
  auto *loadSoA(TFunctor *f) {
    _soa.clear();
 
    // First resize the SoA to the required number of elements to store. This avoids resizing successively the SoA in
    // SoALoader.
    auto &cells = _internalLinkedCells->getCells();
    std::vector<size_t> offsets(cells.size() + 1);
    std::inclusive_scan(
        cells.begin(), cells.end(), offsets.begin() + 1,
        [](const size_t &partialSum, const auto &cell) { return partialSum + cell.size(); }, 0);
    _soa.resizeArrays(offsets.back());
 
    AUTOPAS_OPENMP(parallel for)
    for (size_t i = 0; i < cells.size(); ++i) {
      f->SoALoader(cells[i], _soa, offsets[i], /*skipSoAResize*/ true);
    }
 
    return &_soa;
  }
 
  template <class TFunctor>
  void extractSoA(TFunctor *f) {
    size_t offset = 0;
    for (auto &cell : _internalLinkedCells->getCells()) {
      f->SoAExtractor(cell, _soa, offset);
      offset += cell.size();
    }
  }
 
  virtual void setUpTraversal(TraversalInterface *traversal) = 0;
 
  void setLinkedCellsPointer(LinkedCells<Particle_T> *linkedCells) { this->_internalLinkedCells = linkedCells; }
 
 protected:
  LinkedCells<Particle_T> *_internalLinkedCells;
 
  SoA<typename Particle_T::SoAArraysType> _soa;
};
 
}  // namespace autopas