B_field.h

Go to the documentation of this file.

 
#ifndef B_FIELD_H
#define B_FIELD_H
 
#include "offload_acc_omp.h"
#include "ascot5.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_offload_data BGS;    
    B_2DS_offload_data B2DS;  
    B_3DS_offload_data B3DS;  
    B_STS_offload_data BSTS;  
    B_TC_offload_data BTC;    
    int offload_array_length; 
} B_field_offload_data;
 
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;
 
int B_field_init_offload(B_field_offload_data* offload_data,
                         real** offload_array);
void B_field_free_offload(B_field_offload_data* offload_data,
                          real** offload_array);
 
int B_field_init(
    B_field_data* Bdata, B_field_offload_data* offload_data,
    real* offload_array);
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