TargettedForceSource.h

Go to the documentation of this file.

#pragma once
 
#include <iostream>
#include <vector>
 
#include "particles/Particle.h"
 
class TargettedForceSource {
   protected:
    std::vector<size_t> target_indices;
 
   public:
    inline TargettedForceSource(const std::vector<size_t>& target_indices) : target_indices(target_indices){};
 
    virtual ~TargettedForceSource() = default;
 
    virtual void applyForce(std::vector<Particle>& particle_vector, double curr_simulation_time) const = 0;
 
    virtual explicit operator std::string() const = 0;
};
 
// overload the << operator for the ForceSource class to allow easier printing
inline std::ostream& operator<<(std::ostream& os, const TargettedForceSource& forceSource) {
    os << static_cast<std::string>(forceSource);
    return os;
}