PushSelector.H

Go to the documentation of this file.

/* Copyright 2020 Andrew Myers
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_PARTICLES_PUSHER_PUSHSELECTOR_H_
#define WARPX_PARTICLES_PUSHER_PUSHSELECTOR_H_
 
// Import low-level single-particle kernels
#include "Particles/Pusher/UpdateMomentumBoris.H"
#include "Particles/Pusher/UpdateMomentumBorisWithRadiationReaction.H"
#include "Particles/Pusher/UpdateMomentumHigueraCary.H"
#include "Particles/Pusher/UpdateMomentumVay.H"
#include "Particles/Pusher/UpdatePosition.H"
#include "Particles/WarpXParticleContainer.H"
#include "Utils/WarpXAlgorithmSelection.H"
 
#include <AMReX_REAL.H>
 
#include <limits>
 
template <int do_sync>
AMREX_GPU_DEVICE AMREX_FORCE_INLINE
void doParticlePush(const GetParticlePosition& GetPosition,
                    const SetParticlePosition& SetPosition,
                    const CopyParticleAttribs& copyAttribs,
                    const long i,
                    amrex::ParticleReal& ux,
                    amrex::ParticleReal& uy,
                    amrex::ParticleReal& uz,
                    const amrex::ParticleReal Ex,
                    const amrex::ParticleReal Ey,
                    const amrex::ParticleReal Ez,
                    const amrex::ParticleReal Bx,
                    const amrex::ParticleReal By,
                    const amrex::ParticleReal Bz,
                    const int ion_lev,
                    const amrex::ParticleReal m,
                    const amrex::ParticleReal a_q,
                    const int pusher_algo,
                    const int do_crr,
                    const int do_copy,
#ifdef WARPX_QED
                    const amrex::Real t_chi_max,
#endif
                    const amrex::Real dt)
{
    amrex::ParticleReal qp = a_q;
    if (ion_lev) { qp *= ion_lev; }
 
    if (do_copy) copyAttribs(i);
    if (do_crr) {
#ifdef WARPX_QED
        amrex::ignore_unused(t_chi_max);
        if constexpr (do_sync) {
            auto chi = QedUtils::chi_ele_pos(m*ux, m*uy, m*uz,
                                            Ex, Ey, Ez,
                                            Bx, By, Bz);
            if (chi < t_chi_max) {
                UpdateMomentumBorisWithRadiationReaction(ux, uy, uz,
                                                         Ex, Ey, Ez, Bx,
                                                         By, Bz, qp, m, dt);
            }
            else {
                UpdateMomentumBoris( ux, uy, uz,
                                     Ex, Ey, Ez, Bx,
                                     By, Bz, qp, m, dt);
            }
            amrex::ParticleReal x, y, z;
            GetPosition(i, x, y, z);
            UpdatePosition(x, y, z, ux, uy, uz, dt );
            SetPosition(i, x, y, z);
        } else
#endif
        {
 
            UpdateMomentumBorisWithRadiationReaction(ux, uy, uz,
                                                     Ex, Ey, Ez, Bx,
                                                     By, Bz, qp, m, dt);
            amrex::ParticleReal x, y, z;
            GetPosition(i, x, y, z);
            UpdatePosition(x, y, z, ux, uy, uz, dt );
            SetPosition(i, x, y, z);
        }
    } else if (pusher_algo == ParticlePusherAlgo::Boris) {
        UpdateMomentumBoris( ux, uy, uz,
                             Ex, Ey, Ez, Bx,
                             By, Bz, qp, m, dt);
        amrex::ParticleReal x, y, z;
        GetPosition(i, x, y, z);
        UpdatePosition(x, y, z, ux, uy, uz, dt );
        SetPosition(i, x, y, z);
    } else if (pusher_algo == ParticlePusherAlgo::Vay) {
        UpdateMomentumVay( ux, uy, uz,
                           Ex, Ey, Ez, Bx,
                           By, Bz, qp, m, dt);
        amrex::ParticleReal x, y, z;
        GetPosition(i, x, y, z);
        UpdatePosition(x, y, z, ux, uy, uz, dt );
        SetPosition(i, x, y, z);
    } else if (pusher_algo == ParticlePusherAlgo::HigueraCary) {
        UpdateMomentumHigueraCary( ux, uy, uz,
                                   Ex, Ey, Ez, Bx,
                                   By, Bz, qp, m, dt);
        amrex::ParticleReal x, y, z;
        GetPosition(i, x, y, z);
        UpdatePosition(x, y, z, ux, uy, uz, dt );
        SetPosition(i, x, y, z);
    } //else {
//        amrex::Abort("Unknown particle pusher");
//    }
}
 
#endif // WARPX_PARTICLES_PUSHER_SELECTOR_H_