particledatabase.hpp

Go to the documentation of this file.

 
/* Copyright (c) 2005-2015 Taneli Kalvas. All rights reserved.
 *
 * You can redistribute this software and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option)
 * any later version.
 * 
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this library (file "COPYING" included in the package);
 * if not, write to the Free Software Foundation, Inc., 51 Franklin
 * Street, Fifth Floor, Boston, MA 02110-1301 USA
 * 
 * If you have questions about your rights to use or distribute this
 * software, please contact Berkeley Lab's Technology Transfer
 * Department at TTD@lbl.gov. Other questions, comments and bug
 * reports should be sent directly to the author via email at
 * taneli.kalvas@jyu.fi.
 * 
 * NOTICE. This software was developed under partial funding from the
 * U.S.  Department of Energy.  As such, the U.S. Government has been
 * granted for itself and others acting on its behalf a paid-up,
 * nonexclusive, irrevocable, worldwide license in the Software to
 * reproduce, prepare derivative works, and perform publicly and
 * display publicly.  Beginning five (5) years after the date
 * permission to assert copyright is obtained from the U.S. Department
 * of Energy, and subject to any subsequent five (5) year renewals,
 * the U.S. Government is granted for itself and others acting on its
 * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
 * the Software to reproduce, prepare derivative works, distribute
 * copies to the public, perform publicly and display publicly, and to
 * permit others to do so.
 */
 
#ifndef PARTICLEDATABASE_HPP
#define PARTICLEDATABASE_HPP 1
 
 
#include "scalarfield.hpp"
#include "vectorfield.hpp"
#include "particles.hpp"
#include "trajectorydiagnostics.hpp"
#include "particlestatistics.hpp"
#include "constants.hpp"
#include "types.hpp"
 
 
class TrajectoryHandlerCallback {
public:
 
    virtual ~TrajectoryHandlerCallback() {}
 
    virtual void operator()( ParticleBase *particle, ParticlePBase *xcur, ParticlePBase *xend ) = 0;
};
 
 
class TrajectoryEndCallback {
public:
 
    virtual ~TrajectoryEndCallback() {}
 
    virtual void operator()( ParticleBase *particle, class ParticleDataBase *pdb ) = 0;
};
 
 
class TrajectorySurfaceCollisionCallback {
public:
 
    virtual ~TrajectorySurfaceCollisionCallback() {}
 
    virtual void operator()( ParticleBase *particle, ParticlePBase *x, uint32_t tri, 
                             double s, double t ) = 0;
};
 
 
class PPlasmaBfieldSuppression : public CallbackFunctorD_V {
 
    double                 _phi;    
    const MeshScalarField &_epot;   
public:
 
    PPlasmaBfieldSuppression( const MeshScalarField &epot, double phi ) 
        : _phi(phi), _epot(epot) {}
 
    ~PPlasmaBfieldSuppression() {}
 
    virtual double operator()( const Vec3D &x ) const {
        if( _epot( x ) > _phi )
            return( 0.0 );
        return( 1.0 );
    }
};
 
 
class NPlasmaBfieldSuppression : public CallbackFunctorD_V {
 
    double             _phi;    
    const MeshScalarField &_epot;   
public:
 
    NPlasmaBfieldSuppression( const MeshScalarField &epot, double phi ) 
        : _phi(phi), _epot(epot) {}
 
    ~NPlasmaBfieldSuppression() {}
    
    virtual double operator()( const Vec3D &x ) const {
        if( _epot( x ) < _phi )
            return( 0.0 );
        return( 1.0 );
    }
};
 
 
/* ******************************************************************************************* *
 * ParticleDataBase classes                                                                    *
 * ******************************************************************************************* */
 
 
class ParticleDataBase {
 
    class ParticleDataBaseImp *_imp;
    
protected:
 
/* ************************************** *
 * Constructors                           *
 * ************************************** */
 
    ParticleDataBase();
 
    ParticleDataBase( const ParticleDataBase &pdb );
 
    const ParticleDataBase &operator=( const ParticleDataBase &pdb );
 
    void set_implementation_pointer( class ParticleDataBaseImp *imp );
 
public:
 
/* ************************************** *
 * Destructor                             *
 * ************************************** */
 
    virtual ~ParticleDataBase();
 
/* ****************************************** *
 * Particle iteration settings and statictics *
 * ****************************************** */
 
    void set_thread_count( uint32_t threadcount ) {}
 
    void set_accuracy( double epsabs, double epsrel );
 
    void set_bfield_suppression( const CallbackFunctorD_V *functor );
 
    void set_trajectory_handler_callback( TrajectoryHandlerCallback *thand_cb );
 
    void set_trajectory_end_callback( TrajectoryEndCallback *tend_cb );
 
    void set_trajectory_surface_collision_callback( TrajectorySurfaceCollisionCallback *tsur_cb );
 
    void set_relativistic( bool enable );
 
    void set_surface_collision( bool surface_collision );
 
    void set_polyint( bool polyint );
    
    bool get_polyint( void ) const;
    
    void set_trajectory_interpolation( trajectory_interpolation_e intrp );
 
    trajectory_interpolation_e get_trajectory_interpolation( void ) const;
 
    void set_scharge_deposition( scharge_deposition_e type );
 
    scharge_deposition_e get_scharge_deposition( void ) const;
 
    void set_max_steps( uint32_t maxsteps );
 
    void set_max_time( double maxt );
 
    void set_save_all_points( bool save_points );
 
    void set_save_trajectories( uint32_t div );
 
    uint32_t get_save_trajectories( void ) const;
 
    void set_mirror( const bool mirror[6] );
 
    void get_mirror( bool mirror[6] ) const;
 
    int get_iteration_number( void ) const;
 
    double get_rhosum( void ) const;
 
    void set_rhosum( double rhosum );
 
    const ParticleStatistics &get_statistics( void ) const;
 
/* ************************************** *
 * Information and queries                *
 * ************************************** */
 
    geom_mode_e geom_mode() const;
 
    size_t size( void ) const;
 
    virtual ParticleBase &particle( uint32_t i ) = 0;
 
    virtual const ParticleBase &particle( uint32_t i ) const = 0;
 
    double traj_length( uint32_t i ) const;
    
    size_t traj_size( uint32_t i ) const;
    
    virtual const ParticlePBase &trajectory_point( uint32_t i, uint32_t j ) const = 0;
 
    void trajectory_point( double &t, Vec3D &loc, Vec3D &vel, uint32_t i, uint32_t j ) const;
 
    void trajectories_at_plane( TrajectoryDiagnosticData &tdata, 
                                coordinate_axis_e axis,
                                double val,
                                const std::vector<trajectory_diagnostic_e> &diagnostics ) const;
 
    void build_trajectory_density_field( MeshScalarField &tdens ) const;
 
/* ************************************** *
 * Particle and trajectory clearing       *
 * ************************************** */
 
    void clear( void );
 
    void clear_trajectories( void );
 
    void clear_trajectory( size_t a );
 
    void reset_trajectories( void );
 
    void reset_trajectory( size_t a );
 
/* ************************************** *
 * Particle definition                    *
 * ************************************** */
 
    void reserve( size_t size );
 
/* ************************************** *
 * Particle iterators                     *
 * ************************************** */
 
    void iterate_trajectories( MeshScalarField &scharge, const VectorField &efield, 
                               const VectorField &bfield );
 
    void step_particles( MeshScalarField &scharge, const VectorField &efield, 
                         const VectorField &bfield, double dt );
 
/* ************************************** *
 * Debugging, plotting and saving         *
 * ************************************** */
 
    virtual void save( const std::string &filename ) const = 0;
 
    virtual void save( std::ostream &os ) const = 0;
 
    virtual void debug_print( std::ostream &os ) const = 0;
};
 
 
class ParticleDataBase2D : public ParticleDataBase {
 
    class ParticleDataBase2DImp *_imp;
 
public:
 
/* ************************************** *
 * Constructors and destructor            *
 * ************************************** */
 
    ParticleDataBase2D( const Geometry &geom );
 
    ParticleDataBase2D( const ParticleDataBase2D &pdb );
 
    ParticleDataBase2D( std::istream &s, const Geometry &geom );
 
    ~ParticleDataBase2D();
 
    const ParticleDataBase2D &operator=( const ParticleDataBase2D &pdb );
    
/* ************************************** *
 * Information and queries                *
 * ************************************** */
 
    virtual Particle2D &particle( uint32_t i );
 
    virtual const Particle2D &particle( uint32_t i ) const;
    
    virtual const ParticleP2D &trajectory_point( uint32_t i, uint32_t j ) const;
 
    using ParticleDataBase::trajectory_point;
 
    void trajectories_at_plane( std::vector<Particle2D> &tdata,
                                coordinate_axis_e axis,
                                double val ) const;
 
    using ParticleDataBase::trajectories_at_plane;
 
/* ************************************** *
 * Particle definition                    *
 * ************************************** */
 
    void add_particle( double IQ, double q, double m, const ParticleP2D &x );
 
    void add_particle( const Particle2D &p );
 
/* ************************************** *
 * Particle beam definition               *
 * ************************************** */
 
    void add_2d_beam_with_energy( uint32_t N, double J, double q, double m, 
                                  double E, double Tp, double Tt, 
                                  double x1, double y1, double x2, double y2 );
 
    void add_2d_beam_with_velocity( uint32_t N, double J, double q, double m, 
                                    double v, double dvp, double dvt, 
                                    double x1, double y1, double x2, double y2 );
 
    void add_2d_KV_beam_with_emittance( uint32_t N, double I, double q, double m,
                                        double a, double b, double e,
                                        double Ex, double x0, double y0 );
 
    void add_2d_gaussian_beam_with_emittance( uint32_t N, double I, double q, double m,
                                              double a, double b, double e,
                                              double Ex, double x0, double y0 );
 
/* ************************************** *
 * Debugging, plotting and saving         *
 * ************************************** */
 
    virtual void save( const std::string &filename ) const;
 
    virtual void save( std::ostream &s ) const;
 
    virtual void debug_print( std::ostream &os ) const;
};
 
 
 
 
class ParticleDataBaseCyl : public ParticleDataBase {
 
    class ParticleDataBaseCylImp *_imp;
 
public:
 
/* ************************************** *
 * Constructors and destructor            *
 * ************************************** */
 
    ParticleDataBaseCyl( const Geometry &geom );
 
    ParticleDataBaseCyl( const ParticleDataBaseCyl &pdb );
 
    ParticleDataBaseCyl( std::istream &s, const Geometry &geom );
 
    ~ParticleDataBaseCyl();
 
    const ParticleDataBaseCyl &operator=( const ParticleDataBaseCyl &pdb );
 
/* ************************************** *
 * Information and queries                *
 * ************************************** */
 
    virtual ParticleCyl &particle( uint32_t i );
 
    virtual const ParticleCyl &particle( uint32_t i ) const;
    
    virtual const ParticlePCyl &trajectory_point( uint32_t i, uint32_t j ) const;
 
    using ParticleDataBase::trajectory_point;
 
    void trajectories_at_plane( std::vector<ParticleCyl> &tdata,
                                coordinate_axis_e axis,
                                double val ) const;
 
    using ParticleDataBase::trajectories_at_plane;
 
    void export_path_manager_data( const std::string &filename, 
                                   double ref_E, double ref_q, double ref_m, 
                                   double val, uint32_t Np ) const;
 
/* ************************************** *
 * Particle definition                    *
 * ************************************** */
 
    void add_particle( double IQ, double q, double m, const ParticlePCyl &x );
 
    void add_particle( const ParticleCyl &p );
 
 /* ************************************** *
 * Particle beam definition               *
 * ************************************** */
 
    void add_2d_beam_with_energy( uint32_t N, double J, double q, double m, 
                                  double E, double Tp, double Tt, 
                                  double x1, double y1, double x2, double y2 );
 
    void add_2d_beam_with_total_energy( uint32_t N, double J, double q, double m, 
                                        double Etot, const ScalarField &epot, 
                                        double Tp, double Tt, 
                                        double x1, double y1, double x2, double y2 );
 
    void add_2d_beam_with_velocity( uint32_t N, double J, double q, double m, 
                                    double v, double dvp, double dvt, 
                                    double x1, double y1, double x2, double y2 );
 
    void add_2d_full_gaussian_beam( uint32_t N, double I, double q, double m,
                                    double Ex, double Tp, double Tt, 
                                    double x0, double dr );
 
    void add_2d_gaussian_beam_with_emittance( uint32_t N, double I, double q, double m,
                                              double a, double b, double e,
                                              double Ex, double x0 );
 
/* ************************************** *
 * Debugging, plotting and saving         *
 * ************************************** */
 
    virtual void save( const std::string &filename ) const;
 
    virtual void save( std::ostream &s ) const;
 
    virtual void debug_print( std::ostream &os ) const;
};
 
 
 
class ParticleDataBase3D : public ParticleDataBase {
 
    class ParticleDataBase3DImp *_imp;
 
public:
 
/* ************************************** *
 * Constructors and destructor            *
 * ************************************** */
 
    ParticleDataBase3D( const Geometry &geom );
 
    ParticleDataBase3D( const ParticleDataBase3D &pdb );
 
    ParticleDataBase3D( std::istream &s, const Geometry &geom );
 
    ~ParticleDataBase3D();
 
    const ParticleDataBase3D &operator=( const ParticleDataBase3D &pdb );
 
/* ************************************** *
 * Information and queries                *
 * ************************************** */
 
    virtual Particle3D &particle( uint32_t i );
 
    virtual const Particle3D &particle( uint32_t i ) const;
    
    virtual const ParticleP3D &trajectory_point( uint32_t i, uint32_t j ) const;
 
    using ParticleDataBase::trajectory_point;
 
    void trajectories_at_plane( std::vector<Particle3D> &tdata,
                                coordinate_axis_e axis,
                                double val ) const;
 
    using ParticleDataBase::trajectories_at_plane;
 
/* ************************************** *
 * Particle definition                    *
 * ************************************** */
 
    void add_particle( double IQ, double q, double m, const ParticleP3D &x );
 
    void add_particle( const Particle3D &p );
 
/* ************************************** *
 * Particle beam definition               *
 * ************************************** */
 
 
    void add_cylindrical_beam_with_total_energy( uint32_t N, double J, double q, double m, 
                                                 double Etot, const ScalarField &epot, 
                                                 double Tp, double Tt, Vec3D c, 
                                                 Vec3D dir1, Vec3D dir2, double r );
 
    void add_cylindrical_beam_with_energy( uint32_t N, double J, double q, double m, 
                                           double E, double Tp, double Tt, Vec3D c, 
                                           Vec3D dir1, Vec3D dir2, double r );
 
    void add_cylindrical_beam_with_velocity( uint32_t N, double J, double q, double m, 
                                             double v, double dvp, double dvt, Vec3D c, 
                                             Vec3D dir1, Vec3D dir2, double r );
 
    void add_rectangular_beam_with_energy( uint32_t N, double J, double q, double m, 
                                           double E, double Tp, double Tt, Vec3D c, 
                                           Vec3D dir1, Vec3D dir2, double size1, double size2 );
 
    void add_rectangular_beam_with_velocity( uint32_t N, double J, double q, double m, 
                                             double v, double dvp, double dvt, Vec3D c, 
                                             Vec3D dir1, Vec3D dir2, double size1, double size2 );
 
 
    void add_3d_KV_beam_with_emittance( uint32_t N, double I, double q, double m,
                                        double E0,
                                        double a1, double b1, double e1,
                                        double a2, double b2, double e2,
                                        Vec3D c, Vec3D dir1, Vec3D dir2 );
 
    void add_3d_waterbag_beam_with_emittance( uint32_t N, double I, double q, double m,
                                              double E0,
                                              double a1, double b1, double e1,
                                              double a2, double b2, double e2,
                                              Vec3D c, Vec3D dir1, Vec3D dir2 );
    
    void add_3d_gaussian_beam_with_emittance( uint32_t N, double I, double q, double m,
                                              double E0, 
                                              double a1, double b1, double e1,
                                              double a2, double b2, double e2,
                                              Vec3D c, Vec3D dir1, Vec3D dir2 );
 
 
/* ************************************** *
 * Information and queries                *
 * ************************************** */
 
 
    void trajectories_at_free_plane( TrajectoryDiagnosticData &tdata, 
                                     const Vec3D &c, const Vec3D &o, const Vec3D &p,
                                     const std::vector<trajectory_diagnostic_e> &diagnostics ) const;
 
 
    void export_path_manager_data( const std::string &filename, 
                                   double ref_E, double ref_q, double ref_m, 
                                   const Vec3D &c, const Vec3D &o, const Vec3D &p ) const;
 
/* ************************************** *
 * Debugging, plotting and saving         *
 * ************************************** */
 
    virtual void save( const std::string &filename ) const;
 
    virtual void save( std::ostream &s ) const;
 
    virtual void debug_print( std::ostream &os ) const;
};
 
#endif