external_force.hpp

Go to the documentation of this file.

#pragma once
 
#include "modifier.hpp"
#include "particle.hpp"
#include "utils_math.hpp"
#include <unordered_map>
#include <unordered_set>
 
namespace netdem {
 
class ExternalForce : public Modifier {
protected:
  std::unordered_set<int> particle_id_list;
 
  VecXT<Particle *> particle_list;
 
  VecXT<Vec3d> force_list;
 
  // List of particle moments to apply
  VecXT<Vec3d> moment_list;
 
public:
  ExternalForce();
 
  void SetForce(int id, double fx, double fy, double fz);
 
  void SetMoment(int id, double mx, double my, double mz);
 
  void SetForce(const VecXT<int> &id_list, double fx, double fy, double fz);
 
  void SetMoment(const VecXT<int> &id_list, double mx, double my, double mz);
 
  void AddForce(int id, double fx, double fy, double fz);
 
  void AddMoment(int id, double mx, double my, double mz);
 
  void AddForce(const VecXT<int> &id_list, double fx, double fy, double fz);
 
  void AddMoment(const VecXT<int> &id_list, double mx, double my, double mz);
 
  Vec3d GetForce(int id);
 
  Vec3d GetMoment(int id);
 
  void Clear();
 
  Modifier *Clone() const override;
 
  void SyncToAllProcessors();
 
  void Execute() override;
 
  void Update() override;
 
private:
  std::unordered_map<int, VecNd<6>> id_force_map;
};
 
} // namespace netdem