Decay.h

Go to the documentation of this file.

#ifndef OPAL_GLOBALPROCESS_DECAY_H
#define OPAL_GLOBALPROCESS_DECAY_H
 
#include "Processes/GlobalProcesses/GlobalProcess.h"
 
#include "Ippl.h"
 
#include <Kokkos_Core.hpp>
#include <Kokkos_Random.hpp>
 
#include <cstddef>
#include <memory>
 
class Decay : public GlobalProcess {
public:
    Decay(double restLifetimeSeconds, std::size_t containerIndex, double parentMassGeV,
          int parentChargeSign);
 
    ~Decay() override = default;
 
    void setDaughterContainer(
            std::shared_ptr<ParticleContainer<double, 3>> daughterPC, double daughterMassGeV);
 
    size_t apply(
            ParticleContainer<double, 3>& pc, double dt, long long globalTrackStep,
            size_t containerIdx) override;
 
    virtual void createDaughterParticles(
            std::size_t localDestroyNum, std::size_t oldDaughterLocal,
            const Kokkos::View<ippl::Vector<double, 3>*>& parentR,
            const Kokkos::View<ippl::Vector<double, 3>*>& parentP,
            const Kokkos::View<double*>& parentDt,
            const Kokkos::View<ippl::Vector<float, 3>*>& parentPol) = 0;
 
    struct DecayedParentViews {
        Kokkos::View<ippl::Vector<double, 3>*> R;
        Kokkos::View<ippl::Vector<double, 3>*> P;
        Kokkos::View<double*> dt;
        Kokkos::View<ippl::Vector<float, 3>*> Pol;
    };
 
    ippl::detail::size_type markDecayedParticles(
            ippl::detail::size_type nLocal, double dt, Kokkos::View<ippl::Vector<double, 3>*> Pview,
            Kokkos::View<bool*> decayed);
 
    DecayedParentViews collectDecayedParents(
            ippl::detail::size_type nLocal, ippl::detail::size_type localDestroyNum,
            Kokkos::View<bool*> decayed, Kokkos::View<ippl::Vector<double, 3>*> Rview,
            Kokkos::View<ippl::Vector<double, 3>*> Pview, Kokkos::View<double*> dtView,
            Kokkos::View<ippl::Vector<float, 3>*> PolView);
 
protected:
    double tau0_m;
 
    Kokkos::Random_XorShift64_Pool<> randPool_m;
 
    std::shared_ptr<ParticleContainer<double, 3>> daughterPC_m;
 
    short allowedDaughterSpecies_m = -1;
 
    double daughterMassGeV_m = 0.0;
 
    double parentMassGeV_m;
 
    int parentChargeSign_m;
};
 
#endif