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 "field.hpp"

extern "C" void get_3db_rampup_fac(const double *rampup_fac);
extern "C" void get_ptb_3db_arrays(Field<VarType::Vector,PhiInterpType::None>** ptb_3db_bfield_re_vac, Field<VarType::Vector,PhiInterpType::None>** ptb_3db_bfield_im_vac);

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

    // Field variables
    bool rampup_vac;      
    int num_ntor;         
    int rampup_time;      
    int start_time;       
    int mode;             
    bool full_spec_on;    

    double rampup_fac;    

    bool active; 

    // Field arrays
    Kokkos::View<int*,Kokkos::LayoutRight,Device> ntor;             
    Kokkos::View<Field<VarType::Vector,PhiInterpType::None>**,Kokkos::LayoutRight,Device> bfield_im_vac;
    Kokkos::View<Field<VarType::Vector,PhiInterpType::None>**,Kokkos::LayoutRight,Device> bfield_re_vac;

    // Constructor
    PerturbedBField(NLReader::NamelistReader& nlr, const DomainDecomposition<DeviceType>& pol_decomp, const Grid<DeviceType>& grid);

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

    void initialize(NLReader::NamelistReader& nlr, const DomainDecomposition<DeviceType>& pol_decomp, const Grid<DeviceType>& grid);

    // Update ramp_up
    void update_rampup_fac_from_fortran() const{
        get_3db_rampup_fac(&rampup_fac);
    }

    void copy_bfield_from_fortran() const{
#ifndef NO_FORTRAN_MODULES
        Field<VarType::Vector,PhiInterpType::None>* ptb_3db_bfield_re_vac;
        Field<VarType::Vector,PhiInterpType::None>* ptb_3db_bfield_im_vac;
        get_ptb_3db_arrays(&ptb_3db_bfield_re_vac, &ptb_3db_bfield_im_vac);

        array_deep_copy(bfield_re_vac, ptb_3db_bfield_re_vac);
        array_deep_copy(bfield_im_vac, ptb_3db_bfield_im_vac);
#endif
    }

    void copy_to_device_if_on(int gstep){
        if(is_triggered(gstep)){
            update_rampup_fac_from_fortran();
            copy_bfield_from_fortran();
            active = true;
        }else{
            active = false;
        }
    }

    KOKKOS_INLINE_FUNCTION bool is_triggered(int gstep) const;

    // Get perturbed b-field
    template<Order OT>
    KOKKOS_INLINE_FUNCTION void get_delta_b(const Grid<Device> &grid, const Simd<double>& phi, const SimdGridWeights<OT, PhiInterpType::None>& grid_wts, SimdVector &tdb) const;

};

#include "perturbed_B_field.tpp"
#endif