ParticleContainer.h

Go to the documentation of this file.

 
#pragma once
#include "objects/Particle.h"
#include <array>
#include <string>
#include <utility>
#include <vector>
 
class ParticleContainer {
    using ContainerType = std::vector<Particle>;
 
  private:
    ContainerType m_particles;
 
    int m_specialForceLimit;
 
    /* iterator definitions */
  public:
    ContainerType::iterator begin();
 
    ContainerType::iterator end();
 
    ContainerType::const_iterator begin() const;
 
    ContainerType::const_iterator end() const;
 
    class PairIterator {
      private:
        ContainerType::iterator outer_it;
        ContainerType::iterator inner_it;
        ContainerType::iterator container_begin;
        ContainerType::iterator container_end;
 
      public:
        PairIterator(ContainerType::iterator outer, ContainerType::iterator inner, ContainerType::iterator begin,
                     ContainerType::iterator end);
 
        std::pair<Particle &, Particle &> operator*() const;
 
        PairIterator &operator++();
 
        bool operator==(const PairIterator &other) const;
 
        bool operator!=(const PairIterator &other) const;
    };
 
    Particle &operator[](size_t index);
 
    const Particle &operator[](size_t index) const;
 
    PairIterator beginPairs();
 
    PairIterator endPairs();
 
    /* container functions */
    ParticleContainer();
 
    explicit ParticleContainer(size_t num_particles);
 
    virtual ~ParticleContainer();
 
    void addParticle(const Particle &particle);
 
    void addParticle(const std::array<double, 3> &x, const std::array<double, 3> &v, double m, int type = TYPE_DEFAULT,
                     double eps = EPSILON_DEFAULT, double sigma = SIGMA_DEFAULT, double k = K_DEFAULT,
                     double r_0 = R0_DEFAULT, double fzup = FZUP_DEFAULT);
 
    void addParticle(const std::array<double, 3> &x, const std::array<double, 3> &v, const std::array<double, 3> &f,
                     const std::array<double, 3> &old_f, double m, int type, double eps, double sigma, double k,
                     double r_0, double fzup, int cellIndex);
 
    void reserve(size_t capacity);
 
    Particle &get(size_t index);
 
    const Particle &get(size_t index) const;
 
    int getSpecialForceLimit() const;
 
    void setSpecialForceLimit(int limit);
 
    void decrementSpecialForceLimit();
 
    size_t size() const;
 
    size_t activeSize() const;
 
    size_t nonWallSize() const;
 
    bool isEmpty() const;
 
    ContainerType &getParticles();
 
    bool operator==(const ParticleContainer &other) const;
 
    bool operator!=(const ParticleContainer &other) const;
 
    std::string toString() const;
};