DirectSumContainer.h

Go to the documentation of this file.

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