ParticleContainer.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <vector>
 
#include "particles/Particle.h"
#include "physics/pairwiseforces/PairwiseForceSource.h"
#include "physics/simpleforces/SimpleForceSource.h"
#include "physics/targettedforces/TargettedForceSource.h"
 
class ParticleContainer {
   public:
    virtual ~ParticleContainer() = default;
 
    virtual void addParticle(const Particle& p) = 0;
 
    virtual void addParticle(Particle&& p) = 0;
 
    virtual void prepareForceCalculation() = 0;
 
    virtual void applySimpleForces(const std::vector<std::shared_ptr<SimpleForceSource>>& simple_force_sources) = 0;
 
    virtual void applyPairwiseForces(const std::vector<std::shared_ptr<PairwiseForceSource>>& pairwise_force_sources) = 0;
 
    virtual void applyTargettedForces(const std::vector<std::shared_ptr<TargettedForceSource>>& targetted_force_sources,
                                      double curr_simulation_time) = 0;
 
    virtual void reserve(size_t n) = 0;
 
    [[nodiscard]] virtual size_t size() const = 0;
 
    virtual Particle& operator[](int i) = 0;
 
    [[nodiscard]] virtual std::vector<Particle>::iterator begin() = 0;
 
    [[nodiscard]] virtual std::vector<Particle>::iterator end() = 0;
 
    [[nodiscard]] virtual std::vector<Particle>::const_iterator begin() const = 0;
 
    [[nodiscard]] virtual std::vector<Particle>::const_iterator end() const = 0;
 
    [[nodiscard]] virtual const std::vector<Particle>& getParticles() const = 0;
};