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(+ : vec3 : omp_out += omp_in)                    \
    initializer(omp_priv = omp_orig)
 
class PenningTrap
{
private:
    double B_0;                                 
    double V_0;                                 
    std::function<double(double)> perturbation; 
    double d;                                   
    double t;                                   
    std::vector<Particle> particles; 
    sim_arr k_v; 
    sim_arr k_r; 
 
    vec3 v_func(uint i, uint j, double dt);
 
    vec3 r_func(uint i, uint j, double dt);
 
    vec3 external_E_field(vec3 r);
 
    vec3 external_B_field(vec3 r);
 
    vec3 force_on_particle(uint i, uint j);
 
    vec3 total_force_external(uint i);
 
    vec3 total_force_particles(uint i);
 
    vec3 total_force(uint i);
 
public:
    PenningTrap(double B_0 = T, double V_0 = (25. * V) / 1000., double d = 500.,
                double t = 0.);
 
    PenningTrap(uint i, double B_0 = T, double V_0 = (25. * V) / 1000.,
                double d = 500., double t = 0.);
 
    PenningTrap(std::vector<Particle> particles, double B_0 = T,
                double V_0 = (25. * V) / 1000., double d = 500., double t = 0.);
 
    void set_pertubation(double f, double omega_V);
 
    void reinitialize(double f, double omega_V, double t = 0.);
 
    void add_particle(Particle particle);
 
    void evolve_RK4(double dt, bool particle_interaction = true);
 
    void evolve_forward_euler(double dt, bool particle_interaction = true);
 
    simulation_t simulate(double time, uint steps, std::string method = "rk4",
                          bool particle_interaction = true);
 
    void write_simulation_to_dir(std::string path, double time, uint steps,
                                 std::string method = "rk4",
                                 bool particle_interaction = true);
 
    double fraction_of_particles_left(double time, uint steps,
                                      std::string method = "rk4",
                                      bool particle_interaction = true);
 
    friend class PenningTrapTest;
};
 
#endif