NeighborIdentificationFunctor.h

Go to the documentation of this file.

 
#pragma once
 
#include <string>
#include <unordered_map>
#include <vector>
 
#include "autopas/baseFunctors/InteractionListGeneratorFunctor.h"
 
namespace autopas {
 
template <class Particle_T>
class AoSNeighborListPolicy {
 public:
  explicit AoSNeighborListPolicy(std::unordered_map<Particle_T *, std::vector<Particle_T *>> &neighborListAoS)
      : _neighborListsAoS(neighborListAoS) {}
 
  template <class ParticleIterator_T>
  void initializeNeighborList(ParticleIterator_T particlesBegin) {
    _neighborListsAoS.clear();
    for (auto iter = particlesBegin; iter.isValid(); ++iter) {
      _neighborListsAoS[&(*iter)];
    }
  }
 
  void add(Particle_T *i, Particle_T *j) { _neighborListsAoS[i].push_back(j); }
 
 private:
  std::unordered_map<Particle_T *, std::vector<Particle_T *>> &_neighborListsAoS;
};
 
template <class Particle_T>
class NeighborIdentificationFunctor
    : public InteractionListGeneratorFunctor<Particle_T, AoSNeighborListPolicy<Particle_T>> {
 public:
  NeighborIdentificationFunctor(AoSNeighborListPolicy<Particle_T> &policy, double interactionLength,
                                const bool gatherNewton3Lists)
      : InteractionListGeneratorFunctor<Particle_T, AoSNeighborListPolicy<Particle_T>>(policy, interactionLength,
                                                                                       gatherNewton3Lists),
        _policy(policy) {}
 
  std::string getName() override { return "NeighborIdentificationFunctor"; }
 
  template <class ParticleIterator_T>
  void initializeNeighborList(ParticleIterator_T particlesBegin) {
    _policy.initializeNeighborList(particlesBegin);
  }
 
  bool isRelevantForTuning() override { return true; }
 
 private:
  AoSNeighborListPolicy<Particle_T> &_policy;
};
 
}  // namespace autopas