plasma.h

Go to the documentation of this file.

 
#ifndef PLASMA_H
#define PLASMA_H
 
#include "ascot5.h"
#include "error.h"
#include "plasma/plasma_1D.h"
#include "plasma/plasma_1Dt.h"
#include "plasma/plasma_1DS.h"
 
typedef enum plasma_type {
    plasma_type_1D,  
    plasma_type_1Dt, 
    plasma_type_1DS  
} plasma_type;
 
 
typedef struct {
    plasma_type type;         
    plasma_1D_offload_data plasma_1D;   
    plasma_1Dt_offload_data plasma_1Dt; 
    plasma_1DS_offload_data plasma_1DS; 
    int offload_array_length; 
} plasma_offload_data;
 
typedef struct {
    plasma_type type;           
    plasma_1D_data plasma_1D;   
    plasma_1Dt_data plasma_1Dt; 
    plasma_1DS_data plasma_1DS; 
} plasma_data;
 
int plasma_init_offload(plasma_offload_data* offload_data,
                        real** offload_array);
void plasma_free_offload(plasma_offload_data* offload_data,
                          real** offload_array);
 
int plasma_init(plasma_data* plasma_data, plasma_offload_data* offload_data,
                real* offload_array);
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_eval_temp(real* temp, real rho, real r, real phi, real z, real t,
                       int species, plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_eval_dens(real* dens, real rho, real r, real phi, real z, real t,
                       int species, plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
a5err plasma_eval_densandtemp(real* dens, real* temp, real rho,
                              real r, real phi, real z, real t,
                              plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
int plasma_get_n_species(plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
const real* plasma_get_species_mass(plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
const real* plasma_get_species_charge(plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
const int* plasma_get_species_znum(plasma_data* pls_data);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(pls_data)
const int* plasma_get_species_anum(plasma_data* pls_data);
DECLARE_TARGET_END
 
#endif