perturbed_B_field.hpp

Go to the documentation of this file.

#ifndef PTB_3DB_HPP
#define PTB_3DB_HPP
#include "NamelistReader.hpp"
#include "grid.hpp"

#include "push_controls.hpp"

struct PertVec {
    double r;
    double z;
    double phi;
};

// Perturbed magnetic field class
template<class Device>
class PerturbedBField {
    public:

    // Field variables
    bool rampup_vac;      
    int num_ntor;         
    int rampup_interval;  
    int rampup_time;      
    int start_time;       
    int mode;             
    int jpar_avg_period;  

    // Field arrays
    Kokkos::View<int*,Kokkos::LayoutRight,Device> ntor;             
    Kokkos::View<PertVec**,Kokkos::LayoutRight,Device> bfield_im_vac;
    Kokkos::View<PertVec**,Kokkos::LayoutRight,Device> bfield_re_vac;
    Kokkos::View<PertVec**,Kokkos::LayoutRight,Device> bfield_im;    
    Kokkos::View<PertVec**,Kokkos::LayoutRight,Device> bfield_re;    

    // Constructor
    PerturbedBField(NLReader::NamelistReader& nlr, bool rampup_vac_in, int num_ntor_in,
                    int rampup_interval_in, int rampup_time_in, int start_time_in,
                    int nnode_in, int *ntor_in);

    // Null constructor if ptb_3db is off
    PerturbedBField(){}

    // Get perturbed b-field
    KOKKOS_INLINE_FUNCTION void get_delta_b(const PushControls &push_controls, const Grid<Device> &grid, const Simd<double>& fld_phi, const Simd<int>& itr, const SimdGridVec &p, SimdVector &tdb) const;

};

#include "perturbed_B_field.tpp"
#endif