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);
    
    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;
    }
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(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;
 
    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