solvers.hpp

Go to the documentation of this file.

#ifndef SOLVERS_HPP
#define SOLVERS_HPP
 
#include "plasma.hpp"
#include "electric_field.hpp"
#include "perturbed_B_field.hpp"
#include "charge.hpp"
#include "sml.hpp"
#include "solver_settings.hpp"
#include "step_trigger.hpp"
#include "simple00_solver.hpp"
#include "bias_potential.hpp"
#include "solver.hpp"
 
class Solvers{
    bool initialized;
 
    bool poisson_0m_full_geo;
    bool helmholtz_spectral;    
    bool use_pade; 
 
    // Simple 00
    Simple00Solver simple00;
 
    // Bias potential
    BiasPotential bias_potential;
 
 
    void init_solvers(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, Plasma& plasma);
 
    public:
 
    Solver axisym_poisson;
    Solver turb_poisson;
    Solver ampere;
    Solver ampere_cv;
 
    Solvers() : initialized(false) {}
 
    Solvers(NLReader::NamelistReader& nlr, const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, ElectricField<DeviceType>& electric_field, Plasma& plasma);
 
    void update(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, Plasma& plasma);
 
 
    // Poisson axisym
 
    View<double*,CLayout,DeviceType> get_initial_guess_axisym(const Simulation<DeviceType>& sml, const Grid<DeviceType>& grid, const View<double*,CLayout,DeviceType>& cden00_1d) const;
 
    View<double*,CLayout,HostType> solve_for_E(const Simulation<DeviceType>& sml, const DomainDecomposition<DeviceType>& pol_decomp, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, Plasma& plasma, Smoothing& smoothing, const Charges& charges, SolverSetting solver_setting) const;
 
    void solve_poisson_axisym(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, Smoothing& smoothing, Plasma& plasma, const Charges& charges, SolverSetting solver_setting, ElectricField<DeviceType>& electric_field) const;
 
    // Poisson turb
    void solve_poisson_turb(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, Smoothing& smoothing, Plasma& plasma, const Charges& charges, int ipc, SolverSetting solver_setting, const ElectricField<DeviceType>& electric_field, const PerturbedBField<DeviceType>& perturbed_B_field) const;
 
    void solve(const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, const MagneticField<DeviceType>& magnetic_field, Smoothing& smoothing, const View<double*,CLayout,HostType>& rhs, const View<double*,CLayout,HostType>& dpot_double) const;
 
 
    // Ampere
    void solve_ampere(const Simulation<DeviceType>& sml, const MagneticField<DeviceType>& magnetic_field, const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, Plasma& plasma, const Charges& charges, Smoothing& smoothing, int ipc, bool is_cv_solve, ElectricField<DeviceType> &electric_field, const PerturbedBField<DeviceType>& perturbed_B_field) const;
 
    void solve(const Grid<DeviceType>& grid, const DomainDecomposition<DeviceType>& pol_decomp, const MagneticField<DeviceType>& magnetic_field, Smoothing& smoothing, const View<double*,CLayout,HostType>& rhs, const View<double*,CLayout,HostType>& rhs2, const View<double*,CLayout,HostType>& dpot_double, bool is_cv_solve) const;
 
    void control_variate(int ipc, Simulation<DeviceType> &sml, const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, ElectricField<DeviceType> &electric_field,
        Charges& charges, Plasma& plasma,
        const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp,
        Smoothing& smoothing, const PerturbedBField<DeviceType>& perturbed_B_field) const;
 
    // General
    void solve_algorithm(int ipc, Simulation<DeviceType> &sml, const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, ElectricField<DeviceType> &electric_field,
        Charges& charges, Plasma& plasma,
        const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp,
        Smoothing& smoothing, const PerturbedBField<DeviceType>& perturbed_B_field) const;
 
    void solve_for_fields(int ipc, Simulation<DeviceType> &sml, const Grid<DeviceType> &grid,
        const MagneticField<DeviceType> &magnetic_field, ElectricField<DeviceType> &electric_field,
        Charges& charges, Plasma& plasma,
        const VelocityGrid& vgrid, const DomainDecomposition<DeviceType>& pol_decomp,
        Smoothing& smoothing, const PerturbedBField<DeviceType>& perturbed_B_field) const;
};
 
#endif