PenningTrap.hpp

Go to the documentation of this file.

 
#ifndef __PENNING_TRAP__
#define __PENNING_TRAP__
 
#include <armadillo>
#include <omp.h>
 
#include "Particle.hpp"
#include "constants.hpp"
#include "typedefs.hpp"
#include "utils.hpp"
 
#pragma omp declare reduction(+ : vec_3d : omp_out += omp_in)                  \
    initializer(omp_priv = omp_orig)
 
class PenningTrap {
private:
    double B_0;                        
    std::function<double(double)> V_0; 
    double d;                          
    double t;                          
    std::vector<Particle> particles;   
    sim_arr k_v; 
    sim_arr k_r; 
 
    vec_3d v_func(unsigned int i, unsigned int j, double dt);
 
    vec_3d r_func(unsigned int i, unsigned int j, double dt);
 
public:
    PenningTrap(
        double B_0 = T,
        std::function<double(double)> V_0 =
            [](double t) { return 25. * V / 1000.; },
        double d = 500., double t = 0.);
 
    PenningTrap(
        unsigned int i, double B_0 = T,
        std::function<double(double)> V_0 =
            [](double t) { return 25. * V / 1000.; },
        double d = 500., double t = 0.);
 
    PenningTrap(
        std::vector<Particle> particles, double B_0 = T,
        std::function<double(double)> V_0 =
            [](double t) { return 25. * V / 1000.; },
        double d = 500., double t = 0.);
 
    void add_particle(Particle particle);
 
    vec_3d external_E_field(vec_3d r);
 
    vec_3d external_B_field(vec_3d r);
 
    vec_3d force_on_particle(unsigned int i, unsigned int j);
 
    vec_3d total_force_external(unsigned int i);
 
    vec_3d total_force_particles(unsigned int i);
 
    vec_3d total_force(unsigned int i);
 
    void evolve_RK4(double dt, bool particle_interaction = true);
 
    void evolve_forward_euler(double dt, bool particle_interaction = true);
 
    simulation_t simulate(double time, unsigned int steps,
                          std::string method = "rk4",
                          bool particle_interaction = true);
 
    void write_simulation_to_dir(std::string path, double time,
                                 unsigned int steps, std::string method = "rk4",
                                 bool particle_interaction = true);
 
    double fraction_of_particles_left(double time, unsigned int steps,
                                      std::string method = "rk4",
                                      bool particle_interaction = true);
 
    vec_3d get_r(int i);
    double get_t();
};
 
#endif