asigma_loc.h

Go to the documentation of this file.

 
#ifndef ASIGMALOCAL_H
#define ASIGMALOCAL_H
 
#include "../ascot5.h"
#include "../error.h"
#include "../spline/interp.h"
 
typedef struct {
    int N_reac;               
    int z_1[MAX_ATOMIC];      
    int a_1[MAX_ATOMIC];      
    int z_2[MAX_ATOMIC];      
    int a_2[MAX_ATOMIC];      
    int N_E[MAX_ATOMIC];      
    int N_n[MAX_ATOMIC];      
    int N_T[MAX_ATOMIC];      
    int reac_type[MAX_ATOMIC];
    int offload_array_length; 
} asigma_loc_offload_data;
 
typedef struct {
    int N_reac;                      
    int z_1[MAX_ATOMIC];             
    int a_1[MAX_ATOMIC];             
    int z_2[MAX_ATOMIC];             
    int a_2[MAX_ATOMIC];             
    int reac_type[MAX_ATOMIC];       
    interp1D_data sigma[MAX_ATOMIC]; 
    interp2D_data sigmav[MAX_ATOMIC];
    interp3D_data BMSsigmav[MAX_ATOMIC];
} asigma_loc_data;
 
int asigma_loc_init_offload(asigma_loc_offload_data* offload_data,
                            real** offload_array);
void asigma_loc_free_offload(asigma_loc_offload_data* offload_data,
                             real** offload_array);
 
#pragma omp declare target
void asigma_loc_init(
    asigma_loc_data* asigma_data,
    asigma_loc_offload_data* offload_data, real* offload_array);
DECLARE_TARGET_SIMD_UNIFORM(asigma_data, reac_type, z_2, a_2,\
    extrapolate)
a5err asigma_loc_eval_sigma(
    real* sigma, int z_1, int a_1, int z_2, int a_2, real E_coll_per_amu,
    int reac_type, int extrapolate, asigma_loc_data* asigma_data);
DECLARE_TARGET_SIMD_UNIFORM(asigma_data, reac_type, z_2, a_2,\
    extrapolate)
a5err asigma_loc_eval_sigmav(
    real* sigmav, int z_1, int a_1, real m_1, int z_2, int a_2,
    real E, real T_e, real T_0, real n_i, int reac_type, int extrapolate,
    asigma_loc_data* asigma_data);
#pragma omp declare simd uniform(asigmadata, znum, anum, nspec, extrapolate)
a5err asigma_loc_eval_cx(
    real* ratecoeff, int z_1, int a_1, real E, real mass, int nspec,
    const int* znum, const int* anum, real T_0, real* n_0, int extrapolate,
    asigma_loc_data* asigmadata);
#pragma omp declare simd uniform(asigma_data, znum, anum, nion, extrapolate)
a5err asigma_loc_eval_bms(
    real* sigmav, int z_1, int a_1, real E, real mass, int nion,
    const int* znum, const int* anum, real T_e, real* n_i, int extrapolate,
    asigma_loc_data* asigma_data);
#pragma omp end declare target
 
#endif