B_3DS.h

Go to the documentation of this file.

#ifndef B_3DS_H
#define B_3DS_H
#include "../offload.h"
#include "../ascot5.h"
#include "../error.h"
#include "../spline/interp.h"

typedef struct {
    real psi0;           
    real psi1;           
    real axis_r;         
    real axis_z;         
    interp2D_data psi;   
    interp3D_data B_r;   
    interp3D_data B_phi; 
    interp3D_data B_z;   
} B_3DS_data;
 
int B_3DS_init(B_3DS_data* data,
               int p_n_r, real p_r_min, real p_r_max,
               int p_n_z, real p_z_min, real p_z_max,
               int b_n_r, real b_r_min, real b_r_max,
               int b_n_phi, real b_phi_min, real b_phi_max,
               int b_n_z, real b_z_min, real b_z_max,
               real axis_r, real axis_z, real psi0, real psi1,
               real* psi, real* B_r, real* B_phi, real* B_z);
void B_3DS_free(B_3DS_data* data);
void B_3DS_offload(B_3DS_data* data);
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_eval_psi(real* psi, real r, real phi, real z, B_3DS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_eval_psi_dpsi(real psi_dpsi[4], real r, real phi, real z,
                          B_3DS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_eval_rho_drho(real rho_drho[4], real r, real phi, real z,
                          B_3DS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_eval_B(real B[3], real r, real phi, real z, B_3DS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_eval_B_dB(real B_dB[12], real r, real phi, real z,
                      B_3DS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_3DS_get_axis_rz(real rz[2], B_3DS_data* Bdata);
DECLARE_TARGET_END
#endif