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"
 
#pragma omp declare reduction( + : arma::vec : omp_out += omp_in ) \
  initializer( omp_priv = omp_orig )
 
class PenningTrap {
private:
    double B_0; 
    double V_0; 
    double d; 
    std::vector<Particle> particles; 
 
public:
    PenningTrap(double B_0 = T, double V_0 = 25.*V/1000., double d = 500.);
 
    void add_particle(Particle particle);
 
    arma::vec external_E_field(arma::vec r);
 
    arma::vec external_B_field(arma::vec r);
 
    arma::vec force_on_particle(int i, int j);
 
    arma::vec total_force_external(int i);
 
    arma::vec total_force_particles(int i);
 
    arma::vec total_force(int i);
 
    void evolve_RK4(double dt);
 
    void evolve_forward_euler(double dt);
 
    arma::vec get_particle(int i);
 
    double get_d();
};
 
#endif