sources.hpp

Go to the documentation of this file.

#ifndef SOURCES_HPP
#define SOURCES_HPP

#include "step_trigger.hpp"
#include "monte_carlo_collisions.hpp"
#include "col_grid.hpp"

enum class CollisionMode{
    None = 0,
    MonteCarloNonConserving, // monte-carlo non-conserving
    NonLinearFokkerPlanckLandau, // fully non-linear multi-species Fokker-Planck-Landau [Hager JCP 315, 644-660 (2016)]
    PETScLandauAMR // PETSc multi-species Landau with adaptive mesh refinement (ongoing implementation)
};

struct Sources{
    private:

    StepTrigger step_trigger; // Triggers when f_source is called

    public:

    int f_source_period; 


    CollisionMode col_mode; 
    MonteCarloCollider<DeviceType> col_mc; 
    CollisionGrid<DeviceType> col_grid; 

    Sources(NLReader::NamelistReader& nlr, const MagneticField<DeviceType>& magnetic_field, const VelocityGrid& vgrid, const Plasma& plasma) {
        nlr.use_namelist("sml_param");
        f_source_period = nlr.get<int>("sml_f_source_period", 1);

        step_trigger = StepTrigger(f_source_period);

        // Collisions
        nlr.use_namelist("col_param");
        int col_mode_in = nlr.get<int>("col_mode", 0);
        if(col_mode_in==0) col_mode = CollisionMode::None;
        else if(col_mode_in==1) col_mode = CollisionMode::MonteCarloNonConserving;
        else if(col_mode_in==4) col_mode = CollisionMode::NonLinearFokkerPlanckLandau;
        else if(col_mode_in==5) col_mode = CollisionMode::PETScLandauAMR;
        else{
            if(col_mode_in==2){
                if(is_rank_zero()) printf("\ncol_mode==2 is currently not supported. If you are reviving it, dont forget to address this potential issue: https://github.com/PrincetonUniversity/XGC-Devel/issues/266");
            }
            exit_XGC("\nInvalid col_mode specified. Acceptable options are 0, 1, 4, or 5.\n");
        }

        if(col_mode==CollisionMode::NonLinearFokkerPlanckLandau){
#ifndef USE_FORTRAN_COLLISIONS
            col_grid = CollisionGrid<DeviceType>(nlr);
#endif
        }else if(col_mode==CollisionMode::MonteCarloNonConserving){
            col_mc = MonteCarloCollider<DeviceType>(nlr, magnetic_field, plasma.n_nonadiabatic_species);
        }
    }

    /* Returns whether the sources should be executed on this step
     * */
    bool execute(int step){
        return step_trigger.execute(step);
    }
};

#endif