FlatTop.h

Go to the documentation of this file.

 
#ifndef IPPL_FLAT_TOP_H
#define IPPL_FLAT_TOP_H
 
#include "Distribution.h"
#include "SamplingBase.hpp"
#include <Kokkos_Random.hpp>
#include "Ippl.h"
#include "Utilities/Options.h"
#include "OPALTypes.h"
#include <memory>
#include <cmath>
 
using ParticleContainer_t = ParticleContainer<double, 3>;
using FieldContainer_t = FieldContainer<double, 3>;
using Distribution_t = Distribution;
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, std::shared_ptr<Distribution_t> &opalDist);

    void testNumEmitParticles(size_type nsteps, double dt) override;

    void testEmitParticles(size_type nsteps, double dt) override;
 
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; 

    void setWithDomainDecomp(bool withDomainDecomp) override;

    static size_t determineRandInit();

    void setParameters(const std::shared_ptr<Distribution_t> &opalDist);
 
public:
    void generateUniformDisk(size_type nlocal, size_t nNew);

    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;

    double countEnteringParticlesPerRank(double t0, double tf);

    void allocateParticles(size_t numberOfParticles);

    void emitParticles(double t, double dt) override;
};
 
#endif // IPPL_FLAT_TOP_H