fgrid_coarse_graining.hpp

Go to the documentation of this file.

 
#ifndef FGRID_COARSE_GRAINING_HPP
#define FGRID_COARSE_GRAINING_HPP
 
#include "source.hpp"                 // base class + StepTrigger
#include "plasma.hpp"                 // Plasma / Species
#include "vgrid_distribution.hpp"     // VGridDistribution
#include "NamelistReader.hpp"         // (same helper used elsewhere)
#include "view_arithmetic.hpp"
 
class FGridCoarseGraining : public Source
{
    double alpha;
    bool   enabled;
 
public:
    // Default constructor
    FGridCoarseGraining(){}
 
    FGridCoarseGraining(NLReader::NamelistReader& nlr)
    {
        nlr.use_namelist("f0_param");
        enabled = nlr.get<bool>  ("f0_coarse_graining_on",    false);
        alpha   = nlr.get<double>("f0_coarse_graining_alpha", 1.0e-3);
 
        // call every f_source step
        step_trigger = StepTrigger(1);
    }
 
    void apply(Plasma& plasma,
               VGridDistribution<DeviceType>& df0g) const {
        plasma.for_all_nonadiabatic_species([&](Species<DeviceType>& species){
            const int isp   = species.nonadiabatic_idx;
            species.f0.f0g_send_to_device();
            add_ay_to_x(my_subview(df0g.f,isp), species.f0.f0g, alpha);
            add_ay_to_x(species.f0.f0g, species.f0.f0g, -alpha);
            species.f0.deallocate_device_views();
 
            //Kokkos::parallel_for("recycle_f0g", Kokkos::RangePolicy<ExSpace>(0,df0g.n_nodes()), KOKKOS_LAMBDA( const int inode ){
            //    for(int imu=0; imu<df0g.n_vr(); imu++){
            //        for(int ivp=0; ivp<df0g.n_vz(); ivp++){
            //            double val = alpha*species.f0.f0g(ivp, inode, imu);
            //            species.f0.f0g(ivp, inode, imu) -= val;
            //            df0g(isp, ivp, inode, imu)      += val;
            //        }
            //    }
            //});
            //Kokkos::fence();
        });
    }
};
 
#endif   // FGRID_COARSE_GRAINING_HPP