ParticleVector.h

Go to the documentation of this file.

 
#pragma once
 
#include <vector>
 
#include "autopas/utils/WrapOpenMP.h"
 
template <class Particle_T>
class ParticleVector {
 public:
  ParticleVector() = default;
 
  bool isDirty() { return _dirty; }
 
  void markAsClean() {
    _dirty = false;
    _dirtyIndex = _particleListImp.size();
  }
 
  void clearAllParticles() {
    _particleListImp.clear();
    _dirty = true;
    _dirtyIndex = 0;
  }
 
  void clearHaloParticles() {
    _particleListImp.erase(std::remove_if(_particleListImp.begin(), _particleListImp.end(),
                                          [](const auto &particle) { return particle.isHalo(); }),
                           _particleListImp.end());
    _dirty = true;
    _dirtyIndex = 0;
  }
 
  void deleteDummyParticles() {
    _particleListImp.erase(std::remove_if(_particleListImp.begin(), _particleListImp.end(),
                                          [](const auto &particle) { return particle.isDummy(); }),
                           _particleListImp.end());
    _dirty = true;
    _dirtyIndex = 0;
  }
 
  void push_back(const Particle_T &particle) {
    _particleListLock.lock();
    _dirty = true;
    if (_particleListImp.capacity() == _particleListImp.size()) {
      _dirtyIndex = 0;
    }
    _particleListImp.push_back(particle);
    _particleListLock.unlock();
  }
 
  int totalSize() { return _particleListImp.size(); }
 
  int dirtySize() { return totalSize() - _dirtyIndex; }
 
  bool needsRebuild() { return _dirtyIndex == 0; }
 
  auto beginDirty() { return _particleListImp.begin() + _dirtyIndex; }
  auto endDirty() { return _particleListImp.end(); }
 
  template <typename Lambda>
  void forEach(Lambda forEachLambda) {
    for (Particle_T &p : _particleListImp) {
      forEachLambda(p);
    }
  }
 
  template <typename Lambda, typename A>
  void reduce(Lambda reduceLambda, A &result) {
    for (Particle_T &p : _particleListImp) {
      reduceLambda(p, result);
    }
  }
 
 private:
  bool _dirty = false;
  size_t _dirtyIndex{0};
  autopas::AutoPasLock _particleListLock;
  std::vector<Particle_T> _particleListImp;
};