N0_3D.h

Go to the documentation of this file.

 
#ifndef N0_3D_H
#define N0_3D_H
#include "../ascot5.h"
#include "../linint/linint.h" /* for 3D interpolation routines */
 
typedef struct {
    int n_r;        
    int n_z;        
    int n_phi;      
    real r_min;     
    real r_max;     
    real z_min;     
    real z_max;     
    real phi_min;   
    real phi_max;   
    int n_species;               
    int anum[MAX_SPECIES];       
    int znum[MAX_SPECIES];       
    int maxwellian[MAX_SPECIES]; 
    int offload_array_length;    
} N0_3D_offload_data;
 
typedef struct {
    int n_species;            
    int anum[MAX_SPECIES];    
    int znum[MAX_SPECIES];    
    int maxwellian[MAX_SPECIES];    
    linint3D_data n0[MAX_SPECIES];  
    linint3D_data t0[MAX_SPECIES];  
} N0_3D_data;
 
int N0_3D_init_offload(N0_3D_offload_data* offload_data, real** offload_array);
void N0_3D_free_offload(N0_3D_offload_data* offload_data, real** offload_array);
 
void N0_3D_init(N0_3D_data* ndata, N0_3D_offload_data* offload_data,
                real* offload_array);
DECLARE_TARGET_SIMD_UNIFORM(ndata)
a5err N0_3D_eval_n0(real* n0, real r, real phi, real z, N0_3D_data* ndata);
DECLARE_TARGET_SIMD_UNIFORM(ndata)
a5err N0_3D_eval_t0(real* t0, real r, real phi, real z, N0_3D_data* ndata);
DECLARE_TARGET_SIMD_UNIFORM(ndata)
int N0_3D_get_n_species(N0_3D_data* ndata);
#endif