plasma_2D.h

Go to the documentation of this file.

#ifndef PLASMA_2D_H
#define PLASMA_2D_H
#include "../ascot5.h"
#include "../offload.h"
#include "../error.h"
#include "../linint/linint.h"

typedef struct {
    int n_species;       
    real* mass;          
    real* charge;        
    int* anum;           
    int* znum;           
    linint2D_data* temp; 
    linint2D_data* dens; 
    linint2D_data* vtor; 
} plasma_2D_data;
 
int plasma_2D_init(plasma_2D_data* data, int nr, int nz, int nion,
                   real r_min, real r_max, real z_min, real z_max,
                   int* anum, int* znum, real* mass, real* charge,
                   real* Te, real* Ti, real* ne, real* ni, real* vtor);
void plasma_2D_free(plasma_2D_data* data);
void plasma_2D_offload(plasma_2D_data* data);
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_2D_eval_temp(real* dens, real r, real z, int species,
                          plasma_2D_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_2D_eval_dens(real* temp, real r, real z, int species,
                          plasma_2D_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_2D_eval_densandtemp(real* dens, real* temp, real r, real z,
                                 plasma_2D_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_2D_eval_flow(real* vflow, real r, real z,
                          plasma_2D_data* pls_data);
DECLARE_TARGET_END
 
#endif