B_GS.h

Go to the documentation of this file.

 
#ifndef B_GS_H
#define B_GS_H
#include "../offload_acc_omp.h"
#include "../ascot5.h"
#include "../error.h"
 
typedef struct {
    real R0;                  
    real z0;                  
    real raxis;               
    real zaxis;               
    real B_phi0;              
    real psi0;                
    real psi1;                
    real psi_mult;            
    real psi_coeff[13];       
    int Nripple;              
    real a0;                  
    real alpha0;              
    real delta0;              
    int offload_array_length; 
} B_GS_offload_data;
 
typedef struct {
    real R0;                  
    real z0;                  
    real raxis;               
    real zaxis;               
    real B_phi0;              
    real psi0;                
    real psi1;                
    real psi_mult;            
    real* psi_coeff;          
    int Nripple;              
    real a0;                  
    real alpha0;              
    real delta0;              
} B_GS_data;
 
int B_GS_init_offload(B_GS_offload_data* offload_data, real** offload_array);
void B_GS_free_offload(B_GS_offload_data* offload_data, real** offload_array);
 
void B_GS_init(B_GS_data* Bdata, B_GS_offload_data* offload_data,
               real* offload_array);
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_eval_B(real B[3], real r, real phi, real z, B_GS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_eval_psi(real* psi, real r, real phi, real z, B_GS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_eval_psi_dpsi(real psi_dpsi[4], real r, real phi, real z,
                         B_GS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_eval_rho_drho(real rho_drho[4], real r, real phi, real z,
                         B_GS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_eval_B_dB(real B_dB[12], real r, real phi, real z, B_GS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_GS_get_axis_rz(real rz[2], B_GS_data* Bdata);
DECLARE_TARGET_END
 
#endif