Loading...
Searching...
No Matches
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
{
friend class PenningTrapTest;
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);
};
#endif
Particle.hpp
A class that holds the properties of a particle.
Particle
A class that holds attributes of a particle.
Definition: Particle.hpp:23
PenningTrapTest
Test class for the Penning trap.
Definition: test_suite.cpp:24
PenningTrap
A class that simulates a Penning trap.
Definition: PenningTrap.hpp:32
PenningTrap::particles
std::vector< Particle > particles
The particles in the Penning trap.
Definition: PenningTrap.hpp:42
PenningTrap::B_0
double B_0
Magnetic field strength.
Definition: PenningTrap.hpp:37
PenningTrap::total_force_external
vec3 total_force_external(uint i)
Calculate the total external force on a particle.
Definition: PenningTrap.cpp:82
PenningTrap::k_r
sim_arr k_r
A 2D vector containing all where is the index of a particle.
Definition: PenningTrap.hpp:52
PenningTrap::total_force_particles
vec3 total_force_particles(uint i)
Calculate the total force on a particle p_i from other particles.
Definition: PenningTrap.cpp:95
PenningTrap::external_B_field
vec3 external_B_field(vec3 r)
Calculate B at point r.
Definition: PenningTrap.cpp:66
PenningTrap::evolve_RK4
void evolve_RK4(double dt, bool particle_interaction=true)
Go forward one timestep using the RK4 method.
Definition: PenningTrap.cpp:163
PenningTrap::v_func
vec3 v_func(uint i, uint j, double dt)
Helper for evolve_RK4 when calculating values.
Definition: PenningTrap.cpp:20
PenningTrap::external_E_field
vec3 external_E_field(vec3 r)
Calculate E at point r.
Definition: PenningTrap.cpp:58
PenningTrap::d
double d
Characteristic dimension.
Definition: PenningTrap.hpp:40
PenningTrap::add_particle
void add_particle(Particle particle)
Add a particle to the system.
Definition: PenningTrap.cpp:158
PenningTrap::V_0
double V_0
Applied potential.
Definition: PenningTrap.hpp:38
PenningTrap::simulate
simulation_t simulate(double time, uint steps, std::string method="rk4", bool particle_interaction=true)
Simulate the particle system inside the Penning trap over a certain amount of time.
Definition: PenningTrap.cpp:228
PenningTrap::force_on_particle
vec3 force_on_particle(uint i, uint j)
Calculate the force between 2 particles.
Definition: PenningTrap.cpp:71
PenningTrap::r_func
vec3 r_func(uint i, uint j, double dt)
Helper for evolve_RK4 when calculating values.
Definition: PenningTrap.cpp:39
PenningTrap::t
double t
Current time.
Definition: PenningTrap.hpp:41
PenningTrap::total_force
vec3 total_force(uint i)
calculate the total force on a particle p_i.
Definition: PenningTrap.cpp:107
PenningTrap::set_pertubation
void set_pertubation(double f, double omega_V)
Time dependent perturbation to V_0.
Definition: PenningTrap.cpp:141
PenningTrap::reinitialize
void reinitialize(double f, double omega_V, double t=0.)
Give all particles new positions and velocities, and change t and V_0.
Definition: PenningTrap.cpp:148
PenningTrap::evolve_forward_euler
void evolve_forward_euler(double dt, bool particle_interaction=true)
Go forward one timestep using the forward Euler method.
Definition: PenningTrap.cpp:201
PenningTrap::fraction_of_particles_left
double fraction_of_particles_left(double time, uint steps, std::string method="rk4", bool particle_interaction=true)
Simulate and calculate what fraction of particles are still left inside the Penning trap after the si...
Definition: PenningTrap.cpp:297
PenningTrap::write_simulation_to_dir
void write_simulation_to_dir(std::string path, double time, uint steps, std::string method="rk4", bool particle_interaction=true)
Simulate and write the displacement of all particles to files.
Definition: PenningTrap.cpp:259
PenningTrap::perturbation
std::function< double(double)> perturbation
Time-dependent perturbation.
Definition: PenningTrap.hpp:39
PenningTrap::k_v
sim_arr k_v
A 2D vector containing all where is the index of a particle.
Definition: PenningTrap.hpp:47
constants.hpp
Library of constants.
T
#define T
1 Tesla. unit:
Definition: constants.hpp:21
V
#define V
1 Volt. unit:
Definition: constants.hpp:25
simulation
Typedef for PenningTrap::simulation return value.
Definition: typedefs.hpp:40
typedefs.hpp
Useful typedefs for cleaner code.
vec3
arma::vec::fixed< 3 > vec3
Typedef for a fixed 3d arma vector.
Definition: typedefs.hpp:23
sim_arr
std::vector< sim_cols > sim_arr
Typedef for the result of the simulate method.
Definition: typedefs.hpp:36
utils.hpp
Function prototypes and macros that are useful.