FM2DMagnetoStatic.h

Go to the documentation of this file.

#ifndef OPALX_FIELDMAP2DMAGNETOSTATIC_HH
#define OPALX_FIELDMAP2DMAGNETOSTATIC_HH
 
#include "Fields/Fieldmap.h"
 
#include <Kokkos_Core.hpp>
#include <Kokkos_DualView.hpp>
 
class FM2DMagnetoStatic : public Fieldmap {
public:
    virtual bool getFieldstrength(
            const Vector_t<double, 3>& R, Vector_t<double, 3>& E,
            Vector_t<double, 3>& B) const override;
 
    virtual bool getFieldDerivative(
            const Vector_t<double, 3>& R, Vector_t<double, 3>& E, Vector_t<double, 3>& B,
            const DiffDirection& dir) const override;
 
    virtual void getFieldDimensions(double& zBegin, double& zEnd) const override;
 
    virtual void getFieldDimensions(
            double& xIni, double& xFinal, double& yIni, double& yFinal, double& zIni,
            double& zFinal) const override;
 
    virtual void swap() override;
 
    virtual void getInfo(Inform* msg) override;
 
    virtual double getFrequency() const override;
 
    virtual void setFrequency(double freq) override;
 
    bool isInside(const Vector_t<double, 3>& r) const override {
        return r(2) >= zbegin_m && r(2) < zend_m && sqrt(r(0) * r(0) + r(1) * r(1)) < rend_m;
    }
 
    template <class ViewType>
    KOKKOS_INLINE_FUNCTION static void computeField(
            const Vector_t<double, 3>& R, Vector_t<double, 3>& B, const ViewType& Bz,
            const ViewType& Br, double hr, double hz, double zbegin, int num_gridpr,
            int num_gridpz) {
        const double RR = sqrt(R(0) * R(0) + R(1) * R(1));
 
        const int indexr    = (int)floor(RR / hr);
        const double leverr = (RR / hr) - indexr;
 
        const int indexz    = (int)floor((R(2) - zbegin) / hz);
        const double leverz = (R(2) - zbegin) / hz - indexz;
 
        if ((indexz < 0) || (indexz + 2 > num_gridpz)) return;
 
        if (indexr + 2 > num_gridpr) return;
 
        int index1 = indexz + indexr * num_gridpz;
        int index2 = index1 + num_gridpz;
 
        double BfieldR = (1.0 - leverz) * (1.0 - leverr) * Br(index1)
                         + leverz * (1.0 - leverr) * Br(index1 + 1)
                         + (1.0 - leverz) * leverr * Br(index2) + leverz * leverr * Br(index2 + 1);
 
        double BfieldZ = (1.0 - leverz) * (1.0 - leverr) * Bz(index1)
                         + leverz * (1.0 - leverr) * Bz(index1 + 1)
                         + (1.0 - leverz) * leverr * Bz(index2) + leverz * leverr * Bz(index2 + 1);
 
        if (RR != 0) {
            B(0) += BfieldR * R(0) / RR;
            B(1) += BfieldR * R(1) / RR;
        }
        B(2) += BfieldZ;
        return;
    }
 
    void applyField(std::shared_ptr<ParticleContainer_t> pc, double scale = 1.0) override;
 
private:
    FM2DMagnetoStatic(std::string aFilename);
    ~FM2DMagnetoStatic();
 
    void readMap() override;
    void freeMap() override;
 
    Kokkos::DualView<double*> FieldstrengthBz_m;
    Kokkos::DualView<double*> FieldstrengthBr_m;
 
    double rbegin_m;
    double rend_m;
 
    double zbegin_m;
    double zend_m;
 
    double hz_m;
    double hr_m;
    int num_gridpr_m;
    int num_gridpz_m;
 
    bool swap_m;
    friend class Fieldmap;
};
 
#endif