particle_motion_control.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 ParticleMotionControl : public Modifier {
private:
  enum Type { fixed, fix_rotation, linear_spin, linear_vel, sin_vel, num_motions };
 
  std::unordered_set<int> particle_id_list;
 
  VecXT<Particle *> particle_list;
 
  VecXT<Type> motion_type_list;
 
  VecXT<VecNd<9>> params_list;
 
public:
  ParticleMotionControl();
 
  void SetFixed(int id);
 
  void SetFixRotation(int id);
 
 
  void SetLinearSpin(int id, double spin_x,double spin_y, double spin_z);
 
  void SetLinearVelocity(int id, double xa, double xb, double ya, double yb,
                         double za, double zb);
 
  void SetSinVelocity(int id, double x_amplit, double x_freq, double x_phase,
                      double y_amplit, double y_freq, double y_phase,
                      double z_amplit, double z_freq, double z_phase);
 
  void SyncToAllProcessors();
 
  void Clear();
 
  Modifier *Clone() const override;
 
  void Execute() override;
 
  void Update() override;
 
private:
  std::unordered_map<int, VecNd<10>> id_params_map;
};
 
} // namespace netdem