charge.hpp

Go to the documentation of this file.

#ifndef CHARGE_HPP
#define CHARGE_HPP
#include "space_settings.hpp"
#include "electric_field.hpp"

// Charge class
template<class Device>
class Charge {
    public:

    // Constructor
    Charge(bool use_current, int nnode, int nphi, int nrho, int npsi00, int nthreads, bool is_electrons);

    // Default constructor
    Charge(){}

    bool is_electrons;
    bool use_current; 

    // Host views
    View<double**,CLayout,HostType> density_h;
    View<double*,CLayout,HostType> density0_h;
    View<double*,CLayout,HostType> den00_1d_h;

    View<double**,CLayout,HostType> jpar_h;
    View<double*,CLayout,HostType> jpar0_h;
    
    View<double**,CLayout,HostType> den_f0_h;

    // ions only
    View<double***,CLayout,HostType> den_rho_ff_h;
    View<double***,CLayout,HostType> jpar_rho_ff_h;

    // Device views
    View<double****,CLayout,Device> density;   

    // EXPLICIT_EM
    View<double****,CLayout,Device> jpar;   

    void reset_to_zero();
};

struct Charges{
    Charge<DeviceType> electron;
    Charge<DeviceType> ion;

    Charges(const Grid<DeviceType>& grid, const Simulation<DeviceType>& sml){
#ifdef USE_ARRAY_REPLICATION
        int nthreads = sml.nthreads;
#else
        int nthreads = 1;
#endif
        int nphi = (sml.is_XGCa ? 1 : 2);
        electron = Charge<DeviceType>(sml.explicit_electromagnetic, grid.nnode, nphi, 0, grid.npsi00, nthreads, true);
        ion = Charge<DeviceType>(sml.explicit_electromagnetic, grid.nnode, nphi, grid.nrho, grid.npsi00, nthreads, false);
    }

    // Probably temporary: constructor that doesn't require Grid or PerturbedBField as inputs so that it can be constructed independently
    Charges(const Simulation<DeviceType>& sml, int nnode, int npsi00, int nrho){
#ifdef USE_ARRAY_REPLICATION
        int nthreads = sml.nthreads;
#else
        int nthreads = 1;
#endif
        int nphi = (sml.is_XGCa ? 1 : 2);
        electron = Charge<DeviceType>(sml.explicit_electromagnetic, nnode, nphi, 0, npsi00, nthreads, true);
        ion = Charge<DeviceType>(sml.explicit_electromagnetic, nnode, nphi, nrho, npsi00, nthreads, false);
    }
};

#include "charge.tpp"

#endif