Gaussian.h

Go to the documentation of this file.

#ifndef IPPL_GAUSSIAN_H
#define IPPL_GAUSSIAN_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 Gaussian : public SamplingBase {
public:
    IpplTimings::TimerRef samperTimer_m;
 
    Gaussian(
            std::shared_ptr<ParticleContainer_t> pc, std::shared_ptr<FieldContainer_t> fc,
            Distribution_t* opalDist);
 
    Gaussian(
            std::shared_ptr<ParticleContainer_t> pc, const Vector_t<double, 3>& sigmaR,
            const Vector_t<double, 3>& sigmaP, double avrgpz, const Vector_t<double, 3>& cutoffR,
            bool fix_meanR = true);
    void generateParticles(size_t& numberOfParticles, Vector_t<double, 3> nr) override;
 
    void emitParticles(double t, double dt) override;
 
    void setSigmaR(const Vector_t<double, 3>& sigmaR) { sigmaR_m = sigmaR; }
 
    void setSigmaP(const Vector_t<double, 3>& sigmaP) { sigmaP_m = sigmaP; }
 
    void setAvrgpz(double avrgpz) { avrgpz_m = avrgpz; }
 
    void setCutoffR(const Vector_t<double, 3>& cutoffR) { cutoffR_m = cutoffR; }
 
    void getParameters(
            Vector_t<double, 3>& sigmaR, Vector_t<double, 3>& sigmaP, double& avrgpz,
            Vector_t<double, 3>& cutoffR) const {
        sigmaR  = sigmaR_m;
        sigmaP  = sigmaP_m;
        avrgpz  = avrgpz_m;
        cutoffR = cutoffR_m;
    }
 
    void setFixMeanR(bool fixMeanR) { fixMeanR_m = fixMeanR; }
 
    void getFixMeanR(bool& fixMeanR) const { fixMeanR = fixMeanR_m; }
 
private:
    void initRandomPool();
 
    GeneratorPool randPool_m;
 
    Vector_t<double, 3> sigmaR_m;
    Vector_t<double, 3> sigmaP_m;
 
    double avrgpz_m;
 
    Vector_t<double, 3> cutoffR_m;
 
    bool fixMeanR_m = true;
};
 
#endif  // IPPL_GAUSSIAN_H