OTTraversalInterface.h

Go to the documentation of this file.

 
#pragma once
 
#include "autopas/baseFunctors/CellFunctor.h"
#include "autopas/containers/cellTraversals/CellTraversal.h"
#include "autopas/containers/octree/OctreeNodeWrapper.h"
#include "autopas/options/DataLayoutOption.h"
#include "autopas/utils/DataLayoutConverter.h"
 
namespace autopas {
 
template <typename ParticleCell>
class OTTraversalInterface : public TraversalInterface {
  using ParticleType = typename ParticleCell::ParticleType;
 
 public:
  OTTraversalInterface(double interactionLength, DataLayoutOption dataLayout, bool useNewton3)
      : TraversalInterface(dataLayout, useNewton3), _interactionLength(interactionLength) {}
 
  void setCells(std::vector<OctreeNodeWrapper<ParticleType>> *cells) { _cells = cells; }
 
 protected:
  template <typename PairwiseFunctor>
  void loadBuffers(utils::DataLayoutConverter<PairwiseFunctor> &dataLayoutConverter,
                   OctreeNodeWrapper<ParticleType> *wrapper, std::vector<OctreeLeafNode<ParticleType> *> &leaves) {
    wrapper->appendAllLeaves(leaves);
 
    for (OctreeLeafNode<ParticleType> *leaf : leaves) {
      dataLayoutConverter.loadDataLayout(*leaf);
    }
  }
 
  template <typename PairwiseFunctor>
  void unloadBuffers(utils::DataLayoutConverter<PairwiseFunctor> &dataLayoutConverter,
                     std::vector<OctreeLeafNode<ParticleType> *> &leaves) {
    for (OctreeLeafNode<ParticleType> *leaf : leaves) {
      dataLayoutConverter.storeDataLayout(*leaf);
    }
 
    // Remove the cached leaves
    leaves.clear();
  }
 
  OctreeNodeWrapper<ParticleType> *getOwned() { return dynamic_cast<OctreeNodeWrapper<ParticleType> *>(&(*_cells)[0]); }
 
  OctreeNodeWrapper<ParticleType> *getHalo() { return dynamic_cast<OctreeNodeWrapper<ParticleType> *>(&(*_cells)[1]); }
 
  std::vector<OctreeNodeWrapper<ParticleType>> *_cells;
 
  std::vector<OctreeLeafNode<ParticleType> *> _ownedLeaves;
 
  std::vector<OctreeLeafNode<ParticleType> *> _haloLeaves;
 
  double _interactionLength;
};
 
}  // namespace autopas