FlatTop.h

Go to the documentation of this file.

 
#ifndef IPPL_FLAT_TOP_H
#define IPPL_FLAT_TOP_H
 
#include <Kokkos_Random.hpp>
#include <cmath>
#include <memory>
#include "Distribution.h"
#include "Ippl.h"
#include "OPALTypes.h"
#include "SamplingBase.hpp"
#include "Utilities/Options.h"
 
using GeneratorPool = typename Kokkos::Random_XorShift64_Pool<>;
using Dist_t        = ippl::random::NormalDistribution<double, 3>;
 
class FlatTop : public SamplingBase {
public:
    FlatTop(std::shared_ptr<ParticleContainer_t> pc, std::shared_ptr<FieldContainer_t> fc,
            Distribution_t* opalDist);
 
    FlatTop(std::shared_ptr<ParticleContainer_t> pc, std::shared_ptr<FieldContainer_t> fc,
            bool emitting, double sigmaTFall, double sigmaTRise, Vector_t<double, 3> cutoff,
            double tPulseLengthFWHM, Vector_t<double, 3> sigmaR);
 
    FlatTop(std::shared_ptr<ParticleContainer_t> pc, bool emitting, double sigmaTFall,
            double sigmaTRise, Vector_t<double, 3> cutoff, double tPulseLengthFWHM,
            Vector_t<double, 3> sigmaR);
 
    void testNumEmitParticles(size_type nsteps, double dt) override;
 
    void testEmitParticles(size_type nsteps, double dt) override;
 
    void setWithDomainDecomp(bool withDomainDecomp) override;
 
    void setDistArea(double distArea) { distArea_m = distArea; }
 
    double getDistArea() { return distArea_m; }
 
    double getEmissionTime() { return emissionTime_m; }
 
    bool isEmissionDone(double t) const override { return (t - t0_m) >= emissionTime_m; }
 
private:
    using size_type = ippl::detail::size_type;
    GeneratorPool rand_pool_m;      
    double flattopTime_m;           
    double normalizedFlankArea_m;   
    double distArea_m;              
    double sigmaTFall_m;            
    double sigmaTRise_m;            
    Vector_t<double, 3> cutoffR_m;  
    double fallTime_m;              
    double riseTime_m;              
    bool emitting_m;                
    size_type totalN_m;             
    bool withDomainDecomp_m;        
    double emissionTime_m;          
    Vector_t<double, 3> nr_m;       
    Vector_t<double, 3> hr_m;       
    Vector_t<double, 3> sigmaR_m;
 
    static size_t determineRandInit();
 
    void setParameters(Distribution_t* opalDist);
 
public:
    void generateUniformDisk(size_type nlocal, size_t nNew, double dt);
 
    void setNr(Vector_t<double, 3> nr);
 
    void generateParticles(size_t& numberOfParticles, Vector_t<double, 3> nr) override;
 
    double FlatTopProfile(double t);
 
    size_t computeNlocalUniformly(size_t nglobal);
 
    double integrateTrapezoidal(double x1, double x2, double y1, double y2);
 
    void initDomainDecomp(double BoxIncr) override;
 
    size_type countEnteringParticlesPerRank(double t0, double tf);
 
    void allocateParticles(size_t numberOfParticles);
 
    void emitParticles(double t, double dt) override;
 
    void setInternalVariables(
            bool emitting, double sigmaTFall, double sigmaTRise, Vector_t<double, 3> cutoff,
            double tPulseLengthFWHM, Vector_t<double, 3> sigmaR);
};
 
#endif  // IPPL_FLAT_TOP_H