B_field.h

Go to the documentation of this file.

 
#ifndef B_FIELD_H
#define B_FIELD_H
 
#include "ascot5.h"
#include "offload.h"
#include "error.h"
#include "Bfield/B_GS.h"
#include "Bfield/B_2DS.h"
#include "Bfield/B_3DS.h"
#include "Bfield/B_STS.h"
#include "Bfield/B_TC.h"
 
typedef enum B_field_type {
    B_field_type_GS,  
    B_field_type_2DS, 
    B_field_type_3DS, 
    B_field_type_STS, 
    B_field_type_TC   
} B_field_type;
 
typedef struct {
    B_field_type type; 
    B_GS_data BGS;     
    B_2DS_data B2DS;   
    B_3DS_data B3DS;   
    B_STS_data BSTS;   
    B_TC_data BTC;     
} B_field_data;
 
void B_field_free(B_field_data* data);
void B_field_offload(B_field_data* data);
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_psi(
    real* psi, real r, real phi, real z, real t, B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_psi_dpsi(
    real psi_dpsi[4], real r, real phi, real z, real t, B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_rho(real rho[2], real psi, B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_rho_drho(
    real rho_drho[4], real r, real phi, real z, B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_B(real B[3], real r, real phi, real z, real t,
                     B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_eval_B_dB(
    real B_dB[15], real r, real phi, real z, real t, B_field_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_field_get_axis_rz(real rz[2], B_field_data* Bdata, real phi);
DECLARE_TARGET_END
 
#endif