collisioninfo_radial.hpp

// SPDX-License-Identifier: MIT
 
#pragma once
#include <paraconf.h>
 
#include "ddc_alias_inline_functions.hpp"
#include "ddc_aliases.hpp"
#include "paraconfpp.hpp"
#include "species_info.hpp"
 
template <class GridR>
class CollisionInfoRadial
{
private:
    using IdxR = Idx<GridR>;
    using IdxRangeR = IdxRange<GridR>;
 
public:
    using DFieldMemR = DFieldMem<IdxRangeR>;
    using DFieldR = DField<IdxRangeR>;
    using DConstFieldR = DConstField<IdxRangeR>;
    using radial_chunk_type = DConstFieldR;
 
private:
    double m_nustar0_rpeak;
 
    // read in the YAML input file
    std::int8_t m_collisions_interspecies;
 
    // [TODO] Values that are temporarily need for the interface with koliop
    // but that will be deleted as soon as no more required
    double const m_rpeak;
    double const m_q_rpeak;
    double const m_R0;
 
    DConstFieldR m_rg; //VG ?
    DConstFieldR m_safety_factor; //VG?
 
    DFieldMemR m_coeff_AD;
 
public:
    void compute_coeff_AD(IdxRangeR idxrange_r)
    {
        double const rpeak = m_rpeak;
        double const q_rpeak = m_q_rpeak;
        double const R0 = m_R0;
        double const nustar0_rpeak = m_nustar0_rpeak;
 
        // Compute coeff_AD_rpeak
        DFieldR coeff_AD = get_field(m_coeff_AD);
        double const eps_rpeak = rpeak / R0;
        double const coeff_AD_rpeak
                = std::sqrt(2.) * eps_rpeak * std::sqrt(eps_rpeak) * nustar0_rpeak / (q_rpeak * R0);
 
        // [TODO] coeff_AD should be a scalar and not a radial profile
        //  but to change that koliop interface must be changed
        //  so coeff_AD(r) is fixed equal to coeff_AD_rpeak
        ddc::parallel_for_each(
                Kokkos::DefaultExecutionSpace(),
                idxrange_r,
                KOKKOS_LAMBDA(Idx<GridR> idx_r) { coeff_AD(idx_r) = coeff_AD_rpeak; });
    }
 
public:
    CollisionInfoRadial(
            double const rpeak,
            double const q_rpeak,
            double const R0,
            double const nustar0_rpeak,
            std::int8_t const collisions_interspecies,
            IdxRangeR idxrange_r)
        : m_rpeak(rpeak)
        , m_q_rpeak(q_rpeak)
        , m_R0(R0)
        , m_nustar0_rpeak(nustar0_rpeak)
        , m_collisions_interspecies(collisions_interspecies)
        , m_coeff_AD(idxrange_r)
    {
        compute_coeff_AD(idxrange_r);
    };
 
    ~CollisionInfoRadial() = default;
 
 
    std::int8_t collisions_interspecies() const
    {
        return m_collisions_interspecies;
    }
 
 
    DConstFieldR coeff_AD() const
    {
        return get_const_field(m_coeff_AD);
    }
};