Gaussian.h

Go to the documentation of this file.

#ifndef IPPL_GAUSSIAN_H
#define IPPL_GAUSSIAN_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 Gaussian : public SamplingBase {
public:
    IpplTimings::TimerRef samperTimer_m;

    Gaussian(std::shared_ptr<ParticleContainer_t> &pc, 
             std::shared_ptr<FieldContainer_t> &fc, 
             std::shared_ptr<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 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